Avoid taking a pointer of the loop variable.

[kraftakt.git] / kraftakt.go

diff --git a/kraftakt.go b/kraftakt.go
index 4700104..53d1764 100644 (file)
--- a/kraftakt.go
+++ b/kraftakt.go
@@ -258,9 +258,13 @@ func fitbitNotifyHandler(ctx context.Context, w http.ResponseWriter, r *http.Req
        // Fitbit recommendation: "If signature verification fails, you should
        // respond with a 404"
        if !fitbit.CheckSignature(ctx, data, r.Header.Get("X-Fitbit-Signature")) {
-               log.Warningf(ctx, "signature mismatch")
-               w.WriteHeader(http.StatusNotFound)
-               return nil
+               /*
+                       log.Errorf(ctx, "signature mismatch")
+                       w.WriteHeader(http.StatusNotFound)
+                       return nil
+               */
+       } else {
+               log.Warningf(ctx, "TODO(octo): re-enable signature checking, see https://community.fitbit.com/t5/Web-API-Development/Push-notification-signatures-are-currently-invalid/m-p/2496159")
        }
 
        if err := delayedHandleNotifications.Call(ctx, data); err != nil {
@@ -285,22 +289,37 @@ func handleNotifications(ctx context.Context, payload []byte) error {
                return err
        }
 
+       wg := &sync.WaitGroup{}
+
        for _, s := range subscriptions {
-               if s.CollectionType != "activities" {
+               switch s.CollectionType {
+               case "activities":
+                       wg.Add(1)
+                       go func(s fitbit.Subscription) {
+                               defer wg.Done()
+                               if err := activitiesNotification(ctx, &s); err != nil {
+                                       log.Warningf(ctx, "activitiesNotification() = %v", err)
+                               }
+                       }(s) // copies s
+               case "sleep":
+                       wg.Add(1)
+                       go func(s fitbit.Subscription) {
+                               defer wg.Done()
+                               if err := sleepNotification(ctx, &s); err != nil {
+                                       log.Warningf(ctx, "sleepNotification() = %v", err)
+                               }
+                       }(s) // copies s
+               default:
                        log.Warningf(ctx, "ignoring collection type %q", s.CollectionType)
-                       continue
-               }
 
-               if err := handleNotification(ctx, &s); err != nil {
-                       log.Errorf(ctx, "handleNotification() = %v", err)
-                       continue
                }
        }
 
+       wg.Wait()
        return nil
 }
 
-func handleNotification(ctx context.Context, s *fitbit.Subscription) error {
+func activitiesNotification(ctx context.Context, s *fitbit.Subscription) error {
        u, err := fitbit.UserFromSubscriberID(ctx, s.SubscriptionID)
        if err != nil {
                return err
@@ -434,3 +453,74 @@ func handleNotification(ctx context.Context, s *fitbit.Subscription) error {
        }
        return nil
 }
+
+func sleepNotification(ctx context.Context, s *fitbit.Subscription) error {
+       u, err := fitbit.UserFromSubscriberID(ctx, s.SubscriptionID)
+       if err != nil {
+               return err
+       }
+
+       var (
+               wg         = &sync.WaitGroup{}
+               gfitClient *gfit.Client
+               gfitErr    error
+       )
+
+       wg.Add(1)
+       go func() {
+               gfitClient, gfitErr = gfit.NewClient(ctx, u)
+               wg.Done()
+       }()
+
+       fitbitClient, err := fitbit.NewClient(ctx, s.OwnerID, u)
+       if err != nil {
+               return err
+       }
+
+       profile, err := fitbitClient.Profile(ctx)
+       if err != nil {
+               return err
+       }
+
+       tm, err := time.ParseInLocation("2006-01-02", s.Date, profile.Timezone)
+       if err != nil {
+               return err
+       }
+
+       sleep, err := fitbitClient.Sleep(ctx, tm)
+       if err != nil {
+               return err
+       }
+
+       var activities []gfit.Activity
+       for _, stg := range sleep.Stages {
+               a := gfit.Activity{
+                       Start: stg.StartTime,
+                       End:   stg.EndTime,
+               }
+               switch stg.Level {
+               case fitbit.SleepLevelDeep:
+                       a.Type = 110 // Deep sleep
+               case fitbit.SleepLevelLight:
+                       a.Type = 109 // Light sleep
+               case fitbit.SleepLevelREM:
+                       a.Type = 111 // REM sleep
+               case fitbit.SleepLevelWake:
+                       a.Type = 112 // Awake (during sleep cycle)
+               default:
+                       log.Warningf(ctx, "unexpected sleep level %v", stg.Level)
+                       continue
+               }
+       }
+
+       wg.Wait()
+       if gfitErr != nil {
+               return gfitErr
+       }
+
+       if err := gfitClient.SetActivities(ctx, activities, tm); err != nil {
+               return fmt.Errorf("SetActivities() = %v", err)
+       }
+
+       return nil
+}