X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=gfitsync.go;h=9eadde9efb66f919d278328f0daf06a0168840f7;hb=f1acb1c2891ade2daa46e6e0f5fbafc1d7d6dd05;hp=f12e7f411c871d158a626b783b7a2dc470db110c;hpb=ae4b69da3bf45a53f82205f6000f7cb56d0d6d26;p=kraftakt.git

diff --git a/gfitsync.go b/gfitsync.go
index f12e7f4..9eadde9 100644
--- a/gfitsync.go
+++ b/gfitsync.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io/ioutil"
 	"net/http"
+	"sync"
 	"time"
 
 	"github.com/octo/gfitsync/app"
@@ -125,8 +126,11 @@ func fitbitGrantHandler(ctx context.Context, w http.ResponseWriter, r *http.Requ
 		return err
 	}
 
-	if err := c.Subscribe(ctx, "activities"); err != nil {
-		return fmt.Errorf("c.Subscribe() = %v", err)
+	for _, collection := range []string{"activities", "sleep"} {
+		if err := c.Subscribe(ctx, collection); err != nil {
+			return fmt.Errorf("c.Subscribe(%q) = %v", collection, err)
+		}
+		log.Infof(ctx, "Successfully subscribed to %q", collection)
 	}
 
 	redirectURL := r.URL
@@ -205,6 +209,7 @@ func handleNotifications(ctx context.Context, payload []byte) error {
 
 	for _, s := range subscriptions {
 		if s.CollectionType != "activities" {
+			log.Warningf(ctx, "ignoring collection type %q", s.CollectionType)
 			continue
 		}
 
@@ -228,36 +233,126 @@ func handleNotification(ctx context.Context, s *fitbit.Subscription) error {
 		return err
 	}
 
-	profile, err := fitbitClient.Profile(ctx)
-	if err != nil {
-		return err
+	var (
+		wg      = &sync.WaitGroup{}
+		errs    appengine.MultiError
+		summary *fitbit.ActivitySummary
+		profile *fitbit.Profile
+	)
+
+	wg.Add(1)
+	go func() {
+		var err error
+		summary, err = fitbitClient.ActivitySummary(ctx, s.Date)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("fitbitClient.ActivitySummary(%q) = %v", s.Date, err))
+		}
+		wg.Done()
+	}()
+
+	wg.Add(1)
+	go func() {
+		var err error
+		profile, err = fitbitClient.Profile(ctx)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("fitbitClient.Profile(%q) = %v", s.Date, err))
+		}
+		wg.Done()
+	}()
+
+	wg.Wait()
+	if len(errs) != 0 {
+		return errs
 	}
-	log.Debugf(ctx, "profile = %+v", profile)
 
 	tm, err := time.ParseInLocation("2006-01-02", s.Date, profile.Timezone)
 	if err != nil {
 		return err
 	}
 
-	summary, err := fitbitClient.ActivitySummary(tm)
-	if err != nil {
-		return err
-	}
 	log.Debugf(ctx, "%s (%s) took %d steps on %s",
-		profile.Name, u.Email, summary.Summary.Steps, s.Date)
+		profile.Name, u.Email, summary.Summary.Steps, tm)
 
 	gfitClient, err := gfit.NewClient(ctx, u)
 	if err != nil {
 		return err
 	}
 
-	if err := gfitClient.SetSteps(ctx, summary.Summary.Steps, tm); err != nil {
-		return fmt.Errorf("gfitClient.SetSteps(%d) = %v", summary.Summary.Steps, err)
-	}
+	wg.Add(1)
+	go func() {
+		if err := gfitClient.SetSteps(ctx, summary.Summary.Steps, tm); err != nil {
+			errs = append(errs, fmt.Errorf("gfitClient.SetSteps(%d) = %v", summary.Summary.Steps, err))
+		}
+		wg.Done()
+	}()
 
-	if err := gfitClient.SetCalories(ctx, summary.Summary.CaloriesOut, tm); err != nil {
-		return fmt.Errorf("gfitClient.SetCalories(%d) = %v", summary.Summary.CaloriesOut, err)
-	}
+	wg.Add(1)
+	go func() {
+		if err := gfitClient.SetCalories(ctx, summary.Summary.CaloriesOut, tm); err != nil {
+			errs = append(errs, fmt.Errorf("gfitClient.SetCalories(%d) = %v", summary.Summary.CaloriesOut, err))
+		}
+		wg.Done()
+	}()
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+
+		var distanceMeters float64
+		for _, d := range summary.Summary.Distances {
+			if d.Activity != "total" {
+				continue
+			}
+			distanceMeters = 1000.0 * d.Distance
+			break
+		}
+		if err := gfitClient.SetDistance(ctx, distanceMeters, tm); err != nil {
+			errs = append(errs, fmt.Errorf("gfitClient.SetDistance(%d) = %v", distanceMeters, err))
+			return
+		}
+	}()
 
+	wg.Add(1)
+	go func() {
+		if err := gfitClient.SetHeartRate(ctx, summary.Summary.HeartRateZones, summary.Summary.RestingHeartRate, tm); err != nil {
+			errs = append(errs, fmt.Errorf("gfitClient.SetHeartRate() = %v", err))
+		}
+		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", 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 {
+		return errs
+	}
 	return nil
 }