X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=fitbit%2Ffitbit.go;h=52b89d7aa340269647889068f496b3fe6a005efb;hb=ac25c9764310be649dc18d0a02a1c127f0d73565;hp=6c7a61de7d012f38c5866dca64831d0af3229ec9;hpb=2a96ce53ec33a7fbd6d5054671ed2dfcc0ff379e;p=kraftakt.git

diff --git a/fitbit/fitbit.go b/fitbit/fitbit.go
index 6c7a61d..52b89d7 100644
--- a/fitbit/fitbit.go
+++ b/fitbit/fitbit.go
@@ -2,36 +2,85 @@ package fitbit
 
 import (
 	"context"
+	"crypto/hmac"
+	"crypto/sha1"
+	"encoding/base64"
 	"encoding/json"
 	"fmt"
+	"io/ioutil"
 	"net/http"
+	"strings"
 	"time"
 
+	"github.com/octo/kraftakt/app"
 	"golang.org/x/oauth2"
 	oauth2fitbit "golang.org/x/oauth2/fitbit"
+	"google.golang.org/appengine"
+	"google.golang.org/appengine/log"
 )
 
-var oauth2Config = &oauth2.Config{
-	ClientID:     "@FITBIT_CLIENT_ID@",
-	ClientSecret: "@FITBIT_CLIENT_SECRET@",
-	Endpoint:     oauth2fitbit.Endpoint,
-	RedirectURL:  "https://fitbit-gfit-sync.appspot.com/fitbit/grant",
-	Scopes:       []string{"activity"},
+func oauthConfig() *oauth2.Config {
+	return &oauth2.Config{
+		ClientID:     app.Config.FitbitClientID,
+		ClientSecret: app.Config.FitbitClientSecret,
+		Endpoint:     oauth2fitbit.Endpoint,
+		RedirectURL:  "https://kraftakt.octo.it/fitbit/grant",
+		Scopes: []string{
+			"activity",
+			"heartrate",
+			"profile",
+			"sleep",
+		},
+	}
+}
+
+func AuthURL(ctx context.Context, u *app.User) string {
+	return oauthConfig().AuthCodeURL(u.Sign("Fitbit"), oauth2.AccessTypeOffline)
+}
+
+func ParseToken(ctx context.Context, r *http.Request, u *app.User) error {
+	if state := r.FormValue("state"); state != u.Sign("Fitbit") {
+		return fmt.Errorf("invalid state parameter: %q", state)
+	}
+
+	tok, err := oauthConfig().Exchange(ctx, r.FormValue("code"))
+	if err != nil {
+		return err
+	}
+
+	return u.SetToken(ctx, "Fitbit", tok)
+}
+
+func CheckSignature(ctx context.Context, payload []byte, rawSig string) bool {
+	signatureGot, err := base64.StdEncoding.DecodeString(rawSig)
+	if err != nil {
+		log.Errorf(ctx, "base64.StdEncoding.DecodeString(%q) = %v", rawSig, err)
+		return false
+	}
+
+	mac := hmac.New(sha1.New, []byte(oauthConfig().ClientSecret+"&"))
+	mac.Write(payload)
+	signatureWant := mac.Sum(nil)
+
+	return hmac.Equal(signatureGot, signatureWant)
 }
 
 type Activity struct {
-	ActivityID       int     `json:"activityId"`
-	ActivityParentID int     `json:"activityParentId"`
-	Calories         int     `json:"calories"`
-	Description      string  `json:"description"`
-	Distance         float64 `json:"distance"`
-	Duration         int     `json:"duration"`
-	HasStartTime     bool    `json:"hasStartTime"`
-	IsFavorite       bool    `json:"isFavorite"`
-	LogID            int     `json:"logId"`
-	Name             string  `json:"name"`
-	StartTime        string  `json:"startTime"`
-	Steps            int     `json:"steps"`
+	ActivityID         int       `json:"activityId"`
+	ActivityParentID   int       `json:"activityParentId"`
+	ActivityParentName string    `json:"activityParentName"`
+	Calories           int       `json:"calories"`
+	Description        string    `json:"description"`
+	Distance           float64   `json:"distance"`
+	Duration           int       `json:"duration"`
+	HasStartTime       bool      `json:"hasStartTime"`
+	IsFavorite         bool      `json:"isFavorite"`
+	LastModified       time.Time `json:"lastModified"`
+	LogID              int       `json:"logId"`
+	Name               string    `json:"name"`
+	StartTime          string    `json:"startTime"`
+	StartDate          string    `json:"startDate"`
+	Steps              int       `json:"steps"`
 }
 
 type Distance struct {
@@ -39,6 +88,14 @@ type Distance struct {
 	Distance float64 `json:"distance"`
 }
 
+type HeartRateZone struct {
+	Name        string  `json:"name"`
+	Min         int     `json:"min"`
+	Max         int     `json:"max"`
+	Minutes     int     `json:"minutes"`
+	CaloriesOut float64 `json:"caloriesOut"`
+}
+
 type ActivitySummary struct {
 	Activities []Activity `json:"activities"`
 	Goals      struct {
@@ -48,36 +105,64 @@ type ActivitySummary struct {
 		Steps       int     `json:"steps"`
 	} `json:"goals"`
 	Summary struct {
-		ActivityCalories     int        `json:"activityCalories"`
-		CaloriesBMR          int        `json:"caloriesBMR"`
-		CaloriesOut          int        `json:"caloriesOut"`
-		MarginalCalories     int        `json:"marginalCalories"`
-		Distances            []Distance `json:"distances"`
-		Elevation            float64    `json:"elevation"`
-		Floors               int        `json:"floors"`
-		Steps                int        `json:"steps"`
-		SedentaryMinutes     int        `json:"sedentaryMinutes"`
-		LightlyActiveMinutes int        `json:"lightlyActiveMinutes"`
-		FairlyActiveMinutes  int        `json:"fairlyActiveMinutes"`
-		VeryActiveMinutes    int        `json:"veryActiveMinutes"`
+		ActiveScore          int             `json:"activeScore"`
+		ActivityCalories     int             `json:"activityCalories"`
+		CaloriesBMR          int             `json:"caloriesBMR"`
+		CaloriesOut          float64         `json:"caloriesOut"`
+		Distances            []Distance      `json:"distances"`
+		Elevation            float64         `json:"elevation"`
+		Floors               int             `json:"floors"`
+		HeartRateZones       []HeartRateZone `json:"heartRateZones"`
+		CustomHeartRateZones []HeartRateZone `json:"customHeartRateZones"`
+		MarginalCalories     int             `json:"marginalCalories"`
+		RestingHeartRate     int             `json:"restingHeartRate"`
+		Steps                int             `json:"steps"`
+		SedentaryMinutes     int             `json:"sedentaryMinutes"`
+		LightlyActiveMinutes int             `json:"lightlyActiveMinutes"`
+		FairlyActiveMinutes  int             `json:"fairlyActiveMinutes"`
+		VeryActiveMinutes    int             `json:"veryActiveMinutes"`
 	} `json:"summary"`
 }
 
+type Subscription struct {
+	CollectionType string `json:"collectionType"`
+	Date           string `json:"date"`
+	OwnerID        string `json:"ownerId"`
+	OwnerType      string `json:"ownerType"`
+	SubscriptionID string `json:"subscriptionId"`
+}
+
+func (s Subscription) String() string {
+	return fmt.Sprintf("https://api.fitbit.com/1/%s/%s/%s/apiSubscriptions/%s.json",
+		s.OwnerType, s.OwnerID, s.CollectionType, s.SubscriptionID)
+}
+
 type Client struct {
-	userID string
-	client *http.Client
+	fitbitUserID string
+	appUser      *app.User
+	client       *http.Client
 }
 
-func NewClient(ctx context.Context, userID string, tok *oauth2.Token) *Client {
-	return &Client{
-		userID: userID,
-		client: oauth2Config.Client(ctx, tok),
+func NewClient(ctx context.Context, fitbitUserID string, u *app.User) (*Client, error) {
+	if fitbitUserID == "" {
+		fitbitUserID = "-"
 	}
+
+	c, err := u.OAuthClient(ctx, "Fitbit", oauthConfig())
+	if err != nil {
+		return nil, fmt.Errorf("OAuthClient(%q) = %v", "Fitbit", err)
+	}
+
+	return &Client{
+		fitbitUserID: fitbitUserID,
+		appUser:      u,
+		client:       c,
+	}, nil
 }
 
-func (c *Client) ActivitySummary(t time.Time) (*ActivitySummary, error) {
-	url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/activity/date/%s.json",
-		c.userID, t.Format("2006-01-02"))
+func (c *Client) ActivitySummary(ctx context.Context, date string) (*ActivitySummary, error) {
+	url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/activities/date/%s.json",
+		c.fitbitUserID, date)
 
 	res, err := c.client.Get(url)
 	if err != nil {
@@ -85,10 +170,166 @@ func (c *Client) ActivitySummary(t time.Time) (*ActivitySummary, error) {
 	}
 	defer res.Body.Close()
 
+	data, _ := ioutil.ReadAll(res.Body)
+	log.Debugf(ctx, "GET %s -> %s", url, data)
+
 	var summary ActivitySummary
-	if err := json.NewDecoder(res.Body).Decode(&summary); err != nil {
+	if err := json.Unmarshal(data, &summary); err != nil {
 		return nil, err
 	}
 
 	return &summary, nil
 }
+
+func (c *Client) subscriberID(collection string) string {
+	return fmt.Sprintf("%s:%s", c.appUser.ID, collection)
+}
+
+func UserFromSubscriberID(ctx context.Context, subscriberID string) (*app.User, error) {
+	uid := strings.Split(subscriberID, ":")[0]
+	return app.UserByID(ctx, uid)
+}
+
+func (c *Client) Subscribe(ctx context.Context, collection string) error {
+	url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/%s/apiSubscriptions/%s.json",
+		c.fitbitUserID, collection, c.subscriberID(collection))
+	res, err := c.client.Post(url, "", nil)
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode >= 400 && res.StatusCode != http.StatusConflict {
+		data, _ := ioutil.ReadAll(res.Body)
+		log.Errorf(ctx, "creating %q subscription failed: status %d %q", collection, res.StatusCode, data)
+		return fmt.Errorf("creating %q subscription failed", collection)
+	}
+	if res.StatusCode == http.StatusConflict {
+		log.Infof(ctx, "creating %q subscription: already exists", collection)
+	}
+
+	return nil
+}
+
+func (c *Client) Unsubscribe(ctx context.Context, collection string) error {
+	url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/%s/apiSubscriptions/%s.json",
+		c.fitbitUserID, collection, c.subscriberID(collection))
+	req, err := http.NewRequest(http.MethodDelete, url, nil)
+	if err != nil {
+		return err
+	}
+
+	res, err := c.client.Do(req.WithContext(ctx))
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode >= 400 && res.StatusCode != http.StatusNotFound {
+		data, _ := ioutil.ReadAll(res.Body)
+		log.Errorf(ctx, "deleting %q subscription failed: status %d %q", collection, res.StatusCode, data)
+		return fmt.Errorf("deleting %q subscription failed", collection)
+	}
+	if res.StatusCode == http.StatusNotFound {
+		log.Infof(ctx, "deleting %q subscription: not found", collection)
+	}
+
+	return nil
+}
+
+func (c *Client) UnsubscribeAll(ctx context.Context) error {
+	subs, err := c.ListSubscriptions(ctx)
+	if err != nil {
+		return err
+	}
+
+	var errs appengine.MultiError
+	for _, s := range subs {
+		if s.OwnerType != "user" {
+			log.Infof(ctx, "unexpected OwnerType: %q", s.OwnerType)
+			continue
+		}
+		if err := c.Unsubscribe(ctx, s.CollectionType); err != nil {
+			errs = append(errs, err)
+		}
+	}
+	if len(errs) != 0 {
+		return errs
+	}
+
+	return nil
+}
+
+func (c *Client) ListSubscriptions(ctx context.Context) ([]Subscription, error) {
+	url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/apiSubscriptions.json", c.fitbitUserID)
+	res, err := c.client.Get(url)
+	if err != nil {
+		return nil, fmt.Errorf("Get(%q) = %v", url, err)
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode >= 400 && res.StatusCode != http.StatusNotFound {
+		data, _ := ioutil.ReadAll(res.Body)
+		log.Errorf(ctx, "listing subscriptions failed: status %d %q", res.StatusCode, data)
+		return nil, fmt.Errorf("listing subscriptions failed")
+	}
+	if res.StatusCode == http.StatusNotFound {
+		log.Infof(ctx, "listing subscriptions: not found")
+		return nil, nil
+	}
+
+	var subscriptions []Subscription
+	if err := json.NewDecoder(res.Body).Decode(&subscriptions); err != nil {
+		return nil, err
+	}
+
+	for i, s := range subscriptions {
+		log.Debugf(ctx, "ListSubscriptions() = %d: %s", i, s)
+	}
+
+	return subscriptions, nil
+}
+
+func (c *Client) DeleteToken(ctx context.Context) error {
+	return c.appUser.DeleteToken(ctx, "Fitbit")
+}
+
+type Profile struct {
+	Name     string
+	Timezone *time.Location
+}
+
+func (c *Client) Profile(ctx context.Context) (*Profile, error) {
+	res, err := c.client.Get("https://api.fitbit.com/1/user/-/profile.json")
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode >= 400 {
+		data, _ := ioutil.ReadAll(res.Body)
+		log.Errorf(ctx, "reading profile failed: %s", data)
+		return nil, fmt.Errorf("HTTP %d error", res.StatusCode)
+	}
+
+	var data struct {
+		User struct {
+			FullName            string
+			OffsetFromUTCMillis int
+			Timezone            string
+		}
+	}
+	if err := json.NewDecoder(res.Body).Decode(&data); err != nil {
+		return nil, err
+	}
+
+	loc, err := time.LoadLocation(data.User.Timezone)
+	if err != nil {
+		loc = time.FixedZone("Fitbit preference", data.User.OffsetFromUTCMillis/1000)
+	}
+
+	return &Profile{
+		Name:     data.User.FullName,
+		Timezone: loc,
+	}, nil
+}