const (
csrfToken = "@CSRFTOKEN@"
userID = "me"
+
+ dataTypeNameCalories = "com.google.calories.expended"
+ dataTypeNameSteps = "com.google.step_count.delta"
)
var oauthConfig = &oauth2.Config{
return nil
}
-func (c *Client) SetSteps(ctx context.Context, steps int, date time.Time) error {
- const dataTypeName = "com.google.step_count.delta"
-
- dataSourceID, err := c.DataSourceCreate(ctx, &fitness.DataSource{
- Application: Application(ctx),
- DataStreamId: "", // COMPUTED
- DataStreamName: "", // "daily summary"?
- DataType: &fitness.DataType{
- Field: []*fitness.DataTypeField{
- &fitness.DataTypeField{
- Format: "integer",
- Name: "steps",
+func (c *Client) SetSteps(ctx context.Context, totalSteps int, startOfDay time.Time) error {
+ return c.updateCumulative(ctx,
+ &fitness.DataSource{
+ Application: Application(ctx),
+ DataType: &fitness.DataType{
+ Field: []*fitness.DataTypeField{
+ &fitness.DataTypeField{
+ Name: "steps",
+ Format: "integer",
+ },
},
+ Name: dataTypeNameSteps,
},
- Name: dataTypeName,
+ Name: "Step Count",
+ Type: "raw",
},
- Name: "Step Count",
- Type: "raw",
- })
+ &fitness.Value{
+ IntVal: int64(totalSteps),
+ },
+ startOfDay)
+}
+
+func (c *Client) SetCalories(ctx context.Context, totalCalories float64, startOfDay time.Time) error {
+ return c.updateCumulative(ctx,
+ &fitness.DataSource{
+ Application: Application(ctx),
+ DataType: &fitness.DataType{
+ Field: []*fitness.DataTypeField{
+ &fitness.DataTypeField{
+ Name: "calories",
+ Format: "floatPoint",
+ },
+ },
+ Name: dataTypeNameCalories,
+ },
+ Name: "Calories expended",
+ Type: "raw",
+ },
+ &fitness.Value{
+ FpVal: totalCalories,
+ },
+ startOfDay)
+}
+
+func (c *Client) updateCumulative(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 {
+ return nil
+ }
+ case "floatPoint":
+ if rawValue.FpVal == 0 {
+ return nil
+ }
+ default:
+ return fmt.Errorf("unexpected data type field format %q", f)
+ }
+
+ dataSourceID, err := c.DataSourceCreate(ctx, dataSource)
if err != nil {
return err
}
+ dataSource.DataStreamId = dataSourceID
- return c.DataSetPatch(ctx, dataSourceID, []*fitness.DataPoint{
+ endOfDay := startOfDay.Add(24 * time.Hour).Add(-1 * time.Nanosecond)
+ currValue, startTime, err := c.readCumulative(ctx, dataSource, startOfDay, endOfDay)
+
+ var diffValue fitness.Value
+ if dataSource.DataType.Field[0].Format == "integer" {
+ if rawValue.IntVal == currValue.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
+ }
+ } else { // if dataSource.DataType.Field[0].Format == "floatPoint"
+ if rawValue.FpVal == currValue.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
+ }
+ }
+
+ endTime := endOfDay
+ if now := time.Now().In(startOfDay.Location()); now.Before(endOfDay) {
+ endTime = now
+ }
+ log.Debugf(ctx, "adding cumulative data point: %v-%v %+v", startTime, endTime, diffValue)
+
+ return c.DataSetPatch(ctx, dataSource.DataStreamId, []*fitness.DataPoint{
&fitness.DataPoint{
- ComputationTimeMillis: time.Now().UnixNano() / 1000000,
- DataTypeName: dataTypeName,
- StartTimeNanos: date.UnixNano(),
- EndTimeNanos: date.Add(24 * time.Hour).Add(-1 * time.Nanosecond).UnixNano(),
- Value: []*fitness.Value{
- &fitness.Value{
- IntVal: int64(steps),
- },
- },
+ DataTypeName: dataSource.DataType.Name,
+ StartTimeNanos: startTime.UnixNano(),
+ EndTimeNanos: endTime.UnixNano(),
+ Value: []*fitness.Value{&diffValue},
},
})
}
+
+func (c *Client) readCumulative(ctx context.Context, dataSource *fitness.DataSource, startTime, endTime time.Time) (*fitness.Value, time.Time, error) {
+ datasetID := fmt.Sprintf("%d-%d", startTime.UnixNano(), endTime.UnixNano())
+
+ res, err := c.Service.Users.DataSources.Datasets.Get(userID, dataSource.DataStreamId, datasetID).Context(ctx).Do()
+ if err != nil {
+ log.Errorf(ctx, "c.Service.Users.DataSources.Datasets.Get(%q, %q) = %v", dataSource.DataStreamId, datasetID, err)
+ return nil, time.Time{}, err
+ }
+
+ if len(res.Point) == 0 {
+ return &fitness.Value{}, startTime, nil
+ }
+
+ var sum fitness.Value
+ maxEndTime := startTime
+ for _, p := range res.Point {
+ switch f := dataSource.DataType.Field[0].Format; f {
+ case "integer":
+ sum.IntVal += p.Value[0].IntVal
+ case "floatPoint":
+ sum.FpVal += p.Value[0].FpVal
+ default:
+ return nil, time.Time{}, fmt.Errorf("unexpected data type field format %q", f)
+ }
+
+ pointEndTime := time.Unix(0, p.EndTimeNanos).In(startTime.Location())
+ if maxEndTime.Before(pointEndTime) {
+ maxEndTime = pointEndTime
+ }
+ }
+
+ log.Debugf(ctx, "read cumulative data %s until %v: %+v", dataSource.DataStreamId, maxEndTime, sum)
+ return &sum, maxEndTime, nil
+}
projects / kraftakt.git / blobdiff