Also subscribe to the "sleep" Fitbit collection.

[kraftakt.git] / gfitsync.go

diff --git a/gfitsync.go b/gfitsync.go
index 3bd416f..7f28bb0 100644 (file)
--- a/gfitsync.go
+++ b/gfitsync.go
@@ -130,6 +130,10 @@ func fitbitGrantHandler(ctx context.Context, w http.ResponseWriter, r *http.Requ
                return fmt.Errorf("c.Subscribe() = %v", err)
        }
 
+       if err := c.Subscribe(ctx, "sleep"); err != nil {
+               return fmt.Errorf("c.Subscribe() = %v", err)
+       }
+
        redirectURL := r.URL
        redirectURL.Path = "/"
        redirectURL.RawQuery = ""
@@ -292,6 +296,8 @@ func handleNotification(ctx context.Context, s *fitbit.Subscription) error {
 
        wg.Add(1)
        go func() {
+               defer wg.Done()
+
                var distanceMeters float64
                for _, d := range summary.Summary.Distances {
                        if d.Activity != "total" {
@@ -302,8 +308,8 @@ func handleNotification(ctx context.Context, s *fitbit.Subscription) error {
                }
                if err := gfitClient.SetDistance(ctx, distanceMeters, tm); err != nil {
                        errs = append(errs, fmt.Errorf("gfitClient.SetDistance(%d) = %v", distanceMeters, err))
+                       return
                }
-               wg.Done()
        }()
 
        wg.Add(1)
@@ -314,6 +320,35 @@ func handleNotification(ctx context.Context, s *fitbit.Subscription) error {
                wg.Done()
        }()
 
+       wg.Add(1)
+       go func() {
+               defer wg.Done()
+
+               var activities []gfit.Activity
+               for _, a := range summary.Activities {
+                       if !a.HasStartTime {
+                               continue
+                       }
+
+                       startTime, err := time.ParseInLocation("2006-01-02T15:04:05", a.StartDate+"T"+a.StartTime, profile.Timezone)
+                       if err != nil {
+                               errs = append(errs, fmt.Errorf("gfitClient.SetActivities() = %v", err))
+                               return
+                       }
+                       endTime := startTime.Add(time.Duration(a.Duration) * time.Millisecond)
+
+                       activities = append(activities, gfit.Activity{
+                               Start: startTime,
+                               End:   endTime,
+                               Type:  gfit.ParseFitbitActivity(a.Name),
+                       })
+               }
+               if err := gfitClient.SetActivities(ctx, activities, tm); err != nil {
+                       errs = append(errs, fmt.Errorf("gfitClient.SetActivities() = %v", err))
+                       return
+               }
+       }()
+
        wg.Wait()
 
        if len(errs) != 0 {