Package fitbit: Merge the "Data" and "ShortData" sleep stages.

[kraftakt.git] / gfit / gfit.go

diff --git a/gfit/gfit.go b/gfit/gfit.go
index ec88b19..36e5d5f 100644 (file)
--- a/gfit/gfit.go
+++ b/gfit/gfit.go
@@ -204,7 +204,7 @@ func (c *Client) DatasetPatch(ctx context.Context, dataSourceID string, points [
 }
 
 func (c *Client) SetDistance(ctx context.Context, meters float64, startOfDay time.Time) error {
-       return c.updateCumulative(ctx,
+       return c.updateIncremental(ctx,
                &fitness.DataSource{
                        Application: Application(ctx),
                        DataType: &fitness.DataType{
@@ -226,7 +226,7 @@ func (c *Client) SetDistance(ctx context.Context, meters float64, startOfDay tim
 }
 
 func (c *Client) SetSteps(ctx context.Context, totalSteps int, startOfDay time.Time) error {
-       return c.updateCumulative(ctx,
+       return c.updateIncremental(ctx,
                &fitness.DataSource{
                        Application: Application(ctx),
                        DataType: &fitness.DataType{
@@ -248,7 +248,7 @@ func (c *Client) SetSteps(ctx context.Context, totalSteps int, startOfDay time.T
 }
 
 func (c *Client) SetCalories(ctx context.Context, totalCalories float64, startOfDay time.Time) error {
-       return c.updateCumulative(ctx,
+       return c.updateIncremental(ctx,
                &fitness.DataSource{
                        Application: Application(ctx),
                        DataType: &fitness.DataType{
@@ -272,11 +272,11 @@ func (c *Client) SetCalories(ctx context.Context, totalCalories float64, startOf
 type Activity struct {
        Start time.Time
        End   time.Time
-       Type  int64
+       Type  string
 }
 
 func (a Activity) String() string {
-       return fmt.Sprintf("%s-%s %d", a.Start.Format("15:04:05"), a.End.Format("15:04:05"), a.Type)
+       return fmt.Sprintf("%s-%s %q", a.Start.Format("15:04:05"), a.End.Format("15:04:05"), a.Type)
 }
 
 func (c *Client) SetActivities(ctx context.Context, activities []Activity, startOfDay time.Time) error {
@@ -313,9 +313,10 @@ Next:
        for _, a := range activities {
                startTimeNanos := a.Start.UnixNano()
                endTimeNanos := a.End.UnixNano()
+               activityType := ParseFitbitActivity(a.Type)
 
                for _, p := range dataset.Point {
-                       if p.StartTimeNanos == startTimeNanos && p.EndTimeNanos == endTimeNanos && p.Value[0].IntVal == a.Type {
+                       if p.StartTimeNanos == startTimeNanos && p.EndTimeNanos == endTimeNanos && p.Value[0].IntVal == activityType {
                                log.Debugf(ctx, "activity %s already stored in Google Fit", a)
                                continue Next
                        }
@@ -327,7 +328,7 @@ Next:
                        StartTimeNanos: startTimeNanos,
                        EndTimeNanos:   endTimeNanos,
                        Value: []*fitness.Value{
-                               &fitness.Value{IntVal: a.Type},
+                               &fitness.Value{IntVal: activityType},
                        },
                })
        }
@@ -339,7 +340,7 @@ Next:
        return c.DatasetPatch(ctx, dataStreamID, dataPoints)
 }
 
-func (c *Client) updateCumulative(ctx context.Context, dataSource *fitness.DataSource, rawValue *fitness.Value, startOfDay time.Time) error {
+func (c *Client) updateIncremental(ctx context.Context, dataSource *fitness.DataSource, rawValue *fitness.Value, startOfDay time.Time) error {
        switch f := dataSource.DataType.Field[0].Format; f {
        case "integer":
                if rawValue.IntVal == 0 {
@@ -360,30 +361,30 @@ func (c *Client) updateCumulative(ctx context.Context, dataSource *fitness.DataS
        dataSource.DataStreamId = dataSourceID
 
        endOfDay := startOfDay.Add(24 * time.Hour).Add(-1 * time.Nanosecond)
-       currValue, startTime, err := c.readCumulative(ctx, dataSource, startOfDay, endOfDay)
+       storedValue, startTime, err := c.readIncremental(ctx, dataSource, startOfDay, endOfDay)
        if err != nil {
                return err
        }
 
        var diffValue fitness.Value
        if dataSource.DataType.Field[0].Format == "integer" {
-               if rawValue.IntVal == currValue.IntVal {
+               if storedValue.IntVal > rawValue.IntVal {
+                       log.Warningf(ctx, "stored value (%d) is larger than new value (%d)", storedValue.IntVal, rawValue.IntVal)
                        return nil
                }
-               diffValue.IntVal = rawValue.IntVal - currValue.IntVal
-               if diffValue.IntVal < 0 {
-                       log.Warningf(ctx, "stored value (%d) is larger than new value (%d); assuming count was reset", currValue.IntVal, rawValue.IntVal)
-                       diffValue.IntVal = rawValue.IntVal
+               if rawValue.IntVal == storedValue.IntVal {
+                       return nil
                }
+               diffValue.IntVal = rawValue.IntVal - storedValue.IntVal
        } else { // if dataSource.DataType.Field[0].Format == "floatPoint"
-               if rawValue.FpVal == currValue.FpVal {
+               if storedValue.FpVal > rawValue.FpVal {
+                       log.Warningf(ctx, "stored value (%g) is larger than new value (%g)", storedValue.FpVal, rawValue.FpVal)
                        return nil
                }
-               diffValue.FpVal = rawValue.FpVal - currValue.FpVal
-               if diffValue.FpVal < 0 {
-                       log.Warningf(ctx, "stored value (%g) is larger than new value (%g); assuming count was reset", currValue.FpVal, rawValue.FpVal)
-                       diffValue.FpVal = rawValue.FpVal
+               if rawValue.FpVal == storedValue.FpVal {
+                       return nil
                }
+               diffValue.FpVal = rawValue.FpVal - storedValue.FpVal
        }
 
        endTime := endOfDay
@@ -402,7 +403,7 @@ func (c *Client) updateCumulative(ctx context.Context, dataSource *fitness.DataS
        })
 }
 
-func (c *Client) readCumulative(ctx context.Context, dataSource *fitness.DataSource, startTime, endTime time.Time) (*fitness.Value, time.Time, error) {
+func (c *Client) readIncremental(ctx context.Context, dataSource *fitness.DataSource, startTime, endTime time.Time) (*fitness.Value, time.Time, error) {
        dataset, err := c.DatasetGet(ctx, dataSource.DataStreamId, startTime, endTime)
        if err != nil {
                return nil, time.Time{}, err