X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=fitbit%2Ffitbit.go;h=8a07fb7e70df55038f4dae1a84b4b1e3c2d22ad6;hb=bbefec247f38ced868d35a15ce40efc53e63a764;hp=04c50cd8f5d476b67f6a8994b96a3c17c47776be;hpb=b218c26b9fe377f02f22915e86736e94d4d3b803;p=kraftakt.git diff --git a/fitbit/fitbit.go b/fitbit/fitbit.go index 04c50cd..8a07fb7 100644 --- a/fitbit/fitbit.go +++ b/fitbit/fitbit.go @@ -9,34 +9,41 @@ import ( "fmt" "io/ioutil" "net/http" + "strings" "time" - "github.com/octo/gfitsync/app" + "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://kraftakt.octo.it/fitbit/grant", - Scopes: []string{"activity", "heartrate", "profile"}, +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", + }, + } } -const csrfToken = "@CSRFTOKEN@" - -func AuthURL() string { - return oauth2Config.AuthCodeURL(csrfToken, oauth2.AccessTypeOffline) +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 != csrfToken { + if state := r.FormValue("state"); state != u.Sign("Fitbit") { return fmt.Errorf("invalid state parameter: %q", state) } - tok, err := oauth2Config.Exchange(ctx, r.FormValue("code")) + tok, err := oauthConfig().Exchange(ctx, r.FormValue("code")) if err != nil { return err } @@ -51,7 +58,7 @@ func CheckSignature(ctx context.Context, payload []byte, rawSig string) bool { return false } - mac := hmac.New(sha1.New, []byte(oauth2Config.ClientSecret+"&")) + mac := hmac.New(sha1.New, []byte(oauthConfig().ClientSecret+"&")) mac.Write(payload) signatureWant := mac.Sum(nil) @@ -125,6 +132,11 @@ type Subscription struct { 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 { fitbitUserID string appUser *app.User @@ -136,9 +148,9 @@ func NewClient(ctx context.Context, fitbitUserID string, u *app.User) (*Client, fitbitUserID = "-" } - c, err := u.OAuthClient(ctx, "Fitbit", oauth2Config) + c, err := u.OAuthClient(ctx, "Fitbit", oauthConfig()) if err != nil { - return nil, err + return nil, fmt.Errorf("OAuthClient(%q) = %v", "Fitbit", err) } return &Client{ @@ -169,29 +181,151 @@ func (c *Client) ActivitySummary(ctx context.Context, date string) (*ActivitySum 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 { - subscriberID, err := c.appUser.ID(ctx) + 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, userID, collection, subscriptionID string) error { + if userID == "" { + userID = c.fitbitUserID + } url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/%s/apiSubscriptions/%s.json", - c.fitbitUserID, collection, subscriberID) - res, err := c.client.Post(url, "", nil) + userID, collection, subscriptionID) + 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 { + if res.StatusCode >= 400 && res.StatusCode != http.StatusNotFound { data, _ := ioutil.ReadAll(res.Body) - log.Errorf(ctx, "creating subscription failed: status %d %q", res.StatusCode, data) - return fmt.Errorf("creating subscription failed") + 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 { + var errs appengine.MultiError + + for _, collection := range []string{"activities", "sleep"} { + subs, err := c.ListSubscriptions(ctx, collection) + if err != nil { + errs = append(errs, err) + continue + } + + for _, sub := range subs { + if err := c.unsubscribe(ctx, sub.OwnerID, sub.CollectionType, sub.SubscriptionID); err != nil { + errs = append(errs, err) + } + } + } + if len(errs) != 0 { + return errs } return nil } +func (c *Client) ListSubscriptions(ctx context.Context, collection string) ([]Subscription, error) { + url := fmt.Sprintf("https://api.fitbit.com/1/user/%s/%s/apiSubscriptions.json", c.fitbitUserID, collection) + 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 == http.StatusNotFound { + log.Infof(ctx, "get %q subscription: not found", collection) + return nil, nil + } + + data, err := ioutil.ReadAll(res.Body) + if err != nil { + return nil, err + } + log.Debugf(ctx, "GET %s -> %s", url, data) + + if res.StatusCode >= 400 { + return nil, fmt.Errorf("Get(%q) = %d", url, res.StatusCode) + } + + var parsed struct { + Subscriptions []Subscription `json:"apiSubscriptions"` + } + if err := json.Unmarshal(data, &parsed); err != nil { + return nil, err + } + + var errs appengine.MultiError + var ret []Subscription + for _, sub := range parsed.Subscriptions { + if sub.CollectionType != collection { + errs = append(errs, fmt.Errorf("unexpected collection type: got %q, want %q", sub.CollectionType, collection)) + continue + } + if sub.SubscriptionID == "" { + errs = append(errs, fmt.Errorf("missing subscription ID: %+v", sub)) + continue + } + if sub.OwnerID == "" { + sub.OwnerID = c.fitbitUserID + } + ret = append(ret, sub) + } + + if len(ret) == 0 && len(errs) != 0 { + return nil, errs + } + + for _, err := range errs { + log.Warningf(ctx, err) + } + + return ret, nil +} + +func (c *Client) DeleteToken(ctx context.Context) error { + return c.appUser.DeleteToken(ctx, "Fitbit") +} + type Profile struct { Name string Timezone *time.Location