Code cleanup

fbx/session.go

@@ -0,0 +1,153 @@
+package fbx
+
+import (
+	"crypto/hmac"
+	"crypto/sha1"
+	"encoding/hex"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/trazfr/freebox-exporter/log"
+)
+
+// FreeboxSession represents all the variables used in a session
+type FreeboxSession struct {
+	client             *FreeboxHttpClient
+	getSessionTokenURL string
+	getChallengeURL    string
+
+	appToken     string
+	sessionToken string
+	challenge    string
+}
+
+func GetAppToken(client *FreeboxHttpClient, apiVersion *FreeboxAPIVersion) (string, error) {
+	reqStruct := getFreeboxAuthorize()
+	postResponse := struct {
+		AppToken string `json:"app_token"`
+		TrackID  int64  `json:"track_id"`
+	}{}
+
+	url, err := apiVersion.GetURL("login/authorize/")
+	if err != nil {
+		return "", err
+	}
+
+	if err := client.Post(url, reqStruct, &postResponse); err != nil {
+		return "", err
+	}
+
+	counter := 0
+	for {
+		counter++
+		status := struct {
+			Status string `json:"status"`
+		}{}
+
+		url, err := apiVersion.GetURL("login/authorize/%d", postResponse.TrackID)
+		if err != nil {
+			return "", err
+		}
+		client.Get(url, &status)
+
+		switch status.Status {
+		case "pending":
+			log.Warning.Println(counter, "Please accept the login on the Freebox Server")
+			time.Sleep(10 * time.Second)
+		case "granted":
+			return postResponse.AppToken, nil
+		default:
+			return "", fmt.Errorf("Access is %s", status.Status)
+		}
+	}
+}
+
+func NewFreeboxSession(appToken string, client *FreeboxHttpClient, apiVersion *FreeboxAPIVersion) (*FreeboxSession, error) {
+	getChallengeURL, err := apiVersion.GetURL("login/")
+	if err != nil {
+		return nil, err
+	}
+
+	getSessionTokenURL, err := apiVersion.GetURL("login/session/")
+	if err != nil {
+		return nil, err
+	}
+
+	result := &FreeboxSession{
+		client:             client,
+		getSessionTokenURL: getSessionTokenURL,
+		getChallengeURL:    getChallengeURL,
+
+		appToken: appToken,
+	}
+	if err := result.Refresh(); err != nil {
+		return nil, err
+	}
+	return result, nil
+}
+
+func (f *FreeboxSession) IsValid() bool {
+	return f.sessionToken != "" && f.challenge != ""
+}
+
+func (f *FreeboxSession) AddHeader(req *http.Request) {
+	if f != nil && f.sessionToken != "" {
+		req.Header.Set("X-Fbx-App-Auth", f.sessionToken)
+	}
+}
+
+func (f *FreeboxSession) Refresh() error {
+	challenge, err := f.getChallenge()
+	if err != nil {
+		return err
+	}
+	sessionToken, err := f.getSessionToken(challenge)
+	if err != nil {
+		return err
+	}
+	f.challenge = challenge
+	f.sessionToken = sessionToken
+	return nil
+}
+
+func (f *FreeboxSession) getChallenge() (string, error) {
+	log.Debug.Println("GET challenge:", f.getChallengeURL)
+	resStruct := struct {
+		Challenge string `json:"challenge"`
+	}{}
+
+	if err := f.client.Get(f.getChallengeURL, &resStruct); err != nil {
+		return "", err
+	}
+
+	log.Debug.Println("Challenge:", resStruct.Challenge)
+	return resStruct.Challenge, nil
+}
+
+func (f *FreeboxSession) getSessionToken(challenge string) (string, error) {
+	log.Debug.Println("GET SessionToken:", f.getSessionTokenURL)
+	freeboxAuthorize := getFreeboxAuthorize()
+
+	hash := hmac.New(sha1.New, []byte(f.appToken))
+	hash.Write([]byte(challenge))
+	password := hex.EncodeToString(hash.Sum(nil))
+
+	reqStruct := struct {
+		AppID    string `json:"app_id"`
+		Password string `json:"password"`
+	}{
+		AppID:    freeboxAuthorize.AppID,
+		Password: password,
+	}
+	resStruct := struct {
+		SessionToken string `json:"session_token"`
+	}{}
+
+	if err := f.client.Post(f.getSessionTokenURL, &reqStruct, &resStruct); err != nil {
+		return "", err
+	}
+
+	log.Debug.Println("SessionToken:", resStruct.SessionToken)
+	return resStruct.SessionToken, nil
+}