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

diff --git a/fitbit/fitbit.go b/fitbit/fitbit.go
index 6f15b7c..52b89d7 100644
--- a/fitbit/fitbit.go
+++ b/fitbit/fitbit.go
@@ -15,6 +15,7 @@ import (
 	"github.com/octo/kraftakt/app"
 	"golang.org/x/oauth2"
 	oauth2fitbit "golang.org/x/oauth2/fitbit"
+	"google.golang.org/appengine"
 	"google.golang.org/appengine/log"
 )
 
@@ -131,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
@@ -144,7 +150,7 @@ func NewClient(ctx context.Context, fitbitUserID string, u *app.User) (*Client,
 
 	c, err := u.OAuthClient(ctx, "Fitbit", oauthConfig())
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("OAuthClient(%q) = %v", "Fitbit", err)
 	}
 
 	return &Client{
@@ -231,6 +237,59 @@ func (c *Client) Unsubscribe(ctx context.Context, collection string) error {
 	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")
 }