import (
"context"
"fmt"
- "math"
"net/http"
"strings"
"time"
- "github.com/octo/gfitsync/app"
- "github.com/octo/gfitsync/fitbit"
+ "github.com/octo/kraftakt/app"
+ "github.com/octo/kraftakt/fitbit"
+ "github.com/octo/retry"
"golang.org/x/oauth2"
oauth2google "golang.org/x/oauth2/google"
fitness "google.golang.org/api/fitness/v1"
csrfToken = "@CSRFTOKEN@"
userID = "me"
- dataTypeNameCalories = "com.google.calories.expended"
- dataTypeNameSteps = "com.google.step_count.delta"
- dataTypeNameHeartrate = "com.google.heart_rate.summary"
+ dataTypeNameCalories = "com.google.calories.expended"
+ dataTypeNameDistance = "com.google.distance.delta"
+ dataTypeNameSteps = "com.google.step_count.delta"
+ dataTypeNameHeartrate = "com.google.heart_rate.summary"
+ dataTypeNameActivitySegment = "com.google.activity.segment"
)
-var oauthConfig = &oauth2.Config{
- ClientID: "@GOOGLE_CLIENT_ID@",
- ClientSecret: "@GOOGLE_CLIENT_SECRET@",
- Endpoint: oauth2google.Endpoint,
- RedirectURL: "https://kraftakt.octo.it/google/grant",
- Scopes: []string{
- fitness.FitnessActivityWriteScope,
- fitness.FitnessBodyWriteScope,
- },
+func oauthConfig() *oauth2.Config {
+ return &oauth2.Config{
+ ClientID: app.Config.GoogleClientID,
+ ClientSecret: app.Config.GoogleClientSecret,
+ Endpoint: oauth2google.Endpoint,
+ RedirectURL: "https://kraftakt.octo.it/google/grant",
+ Scopes: []string{
+ fitness.FitnessActivityWriteScope,
+ fitness.FitnessBodyWriteScope,
+ fitness.FitnessLocationWriteScope,
+ },
+ }
}
func Application(ctx context.Context) *fitness.Application {
return &fitness.Application{
- Name: "Fitbit to Google Fit sync",
+ Name: "Kraftakt",
Version: appengine.VersionID(ctx),
DetailsUrl: "", // optional
}
}
func AuthURL() string {
- return oauthConfig.AuthCodeURL(csrfToken, oauth2.AccessTypeOffline)
+ return oauthConfig().AuthCodeURL(csrfToken, oauth2.AccessTypeOffline)
}
func ParseToken(ctx context.Context, r *http.Request, u *app.User) error {
return fmt.Errorf("invalid state parameter: %q", state)
}
- tok, err := oauthConfig.Exchange(ctx, r.FormValue("code"))
+ tok, err := oauthConfig().Exchange(ctx, r.FormValue("code"))
if err != nil {
return err
}
type Client struct {
*fitness.Service
+ appUser *app.User
}
func NewClient(ctx context.Context, u *app.User) (*Client, error) {
- c, err := u.OAuthClient(ctx, "Google", oauthConfig)
+ c, err := u.OAuthClient(ctx, "Google", oauthConfig())
if err != nil {
return nil, err
}
return &Client{
Service: service,
+ appUser: u,
}, nil
}
+func (c *Client) DeleteToken(ctx context.Context) error {
+ return c.appUser.DeleteToken(ctx, "Google")
+}
+
func DataStreamID(dataSource *fitness.DataSource) string {
fields := []string{
dataSource.Type,
dataSource.DataType.Name,
- "@PROJECT_NUMBER@", // FIXME
+ app.Config.ProjectNumber,
}
if dev := dataSource.Device; dev != nil {
}
}
+ if dataSource.DataStreamName != "" {
+ fields = append(fields, dataSource.DataStreamName)
+ }
+
return strings.Join(fields, ":")
}
+func wrapGoogleError(err error) error {
+ if err == nil {
+ return nil
+ }
+
+ if gerr, ok := err.(*googleapi.Error); ok && gerr.Code >= 400 && gerr.Code < 500 {
+ return retry.Abort(err)
+ }
+
+ return err
+}
+
func (c *Client) DataSourceCreate(ctx context.Context, dataSource *fitness.DataSource) (string, error) {
- res, err := c.Service.Users.DataSources.Create(userID, dataSource).Context(ctx).Do()
+ var dataStreamID string
+ err := retry.Do(ctx, func(ctx context.Context) error {
+ res, err := c.Service.Users.DataSources.Create(userID, dataSource).Context(ctx).Do()
+ if err != nil {
+ return wrapGoogleError(err)
+ }
+ dataStreamID = res.DataStreamId
+ return nil
+ })
if err != nil {
if gerr, ok := err.(*googleapi.Error); ok && gerr.Code == http.StatusConflict {
if dataSource.DataStreamId != "" {
}
return DataStreamID(dataSource), nil
}
- log.Errorf(ctx, "c.Service.Users.DataSources.Create() = (%+v, %v)", res, err)
+ log.Errorf(ctx, "c.Service.Users.DataSources.Create(%q) = %v", DataStreamID(dataSource), err)
return "", err
}
- return res.DataStreamId, nil
+ return dataStreamID, nil
+}
+
+func (c *Client) DatasetGet(ctx context.Context, dataStreamID string, startTime, endTime time.Time) (*fitness.Dataset, error) {
+ datasetID := fmt.Sprintf("%d-%d", startTime.UnixNano(), endTime.UnixNano())
+
+ var dataset *fitness.Dataset
+ err := retry.Do(ctx, func(ctx context.Context) error {
+ res, err := c.Service.Users.DataSources.Datasets.Get(userID, dataStreamID, datasetID).Context(ctx).Do()
+ if err != nil {
+ return wrapGoogleError(err)
+ }
+ dataset = res
+ return nil
+ })
+ if err != nil {
+ log.Errorf(ctx, "c.Service.Users.DataSources.Datasets.Get(%q, %q) = %v", dataStreamID, datasetID, err)
+ return nil, err
+ }
+ return dataset, nil
}
-func (c *Client) DataSetPatch(ctx context.Context, dataSourceID string, points []*fitness.DataPoint) error {
+func (c *Client) DatasetPatch(ctx context.Context, dataSourceID string, points []*fitness.DataPoint) error {
startTimeNanos, endTimeNanos := int64(-1), int64(-1)
for _, p := range points {
if startTimeNanos == -1 || startTimeNanos > p.StartTimeNanos {
Point: points,
}
- _, err := c.Service.Users.DataSources.Datasets.Patch(userID, dataSourceID, datasetID, dataset).Context(ctx).Do()
+ err := retry.Do(ctx, func(ctx context.Context) error {
+ _, err := c.Service.Users.DataSources.Datasets.Patch(userID, dataSourceID, datasetID, dataset).Context(ctx).Do()
+ return wrapGoogleError(err)
+ })
if err != nil {
log.Errorf(ctx, "c.Service.Users.DataSources.Datasets.Patch() = %v", err)
return err
return nil
}
+func (c *Client) SetDistance(ctx context.Context, meters float64, startOfDay time.Time) error {
+ return c.updateCumulative(ctx,
+ &fitness.DataSource{
+ Application: Application(ctx),
+ DataType: &fitness.DataType{
+ Field: []*fitness.DataTypeField{
+ &fitness.DataTypeField{
+ Name: "distance",
+ Format: "floatPoint",
+ },
+ },
+ Name: dataTypeNameDistance,
+ },
+ Name: "Distance covered",
+ Type: "raw",
+ },
+ &fitness.Value{
+ FpVal: meters,
+ },
+ startOfDay)
+}
+
func (c *Client) SetSteps(ctx context.Context, totalSteps int, startOfDay time.Time) error {
return c.updateCumulative(ctx,
&fitness.DataSource{
startOfDay)
}
+type Activity struct {
+ Start time.Time
+ End time.Time
+ Type int64
+}
+
+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)
+}
+
+func (c *Client) SetActivities(ctx context.Context, activities []Activity, startOfDay time.Time) error {
+ if len(activities) == 0 {
+ return nil
+ }
+
+ dataStreamID, err := c.DataSourceCreate(ctx, &fitness.DataSource{
+ Application: Application(ctx),
+ DataType: &fitness.DataType{
+ Field: []*fitness.DataTypeField{
+ &fitness.DataTypeField{
+ Name: "activity",
+ Format: "integer",
+ },
+ },
+ Name: dataTypeNameActivitySegment,
+ },
+ Type: "raw",
+ })
+ if err != nil {
+ return err
+ }
+
+ endOfDay := startOfDay.Add(24 * time.Hour).Add(-1 * time.Nanosecond)
+
+ dataset, err := c.DatasetGet(ctx, dataStreamID, startOfDay, endOfDay)
+ if err != nil {
+ return err
+ }
+
+ var dataPoints []*fitness.DataPoint
+Next:
+ for _, a := range activities {
+ startTimeNanos := a.Start.UnixNano()
+ endTimeNanos := a.End.UnixNano()
+
+ for _, p := range dataset.Point {
+ if p.StartTimeNanos == startTimeNanos && p.EndTimeNanos == endTimeNanos && p.Value[0].IntVal == a.Type {
+ log.Debugf(ctx, "activity %s already stored in Google Fit", a)
+ continue Next
+ }
+ }
+
+ log.Debugf(ctx, "activity %s will be added to Google Fit", a)
+ dataPoints = append(dataPoints, &fitness.DataPoint{
+ DataTypeName: dataTypeNameActivitySegment,
+ StartTimeNanos: startTimeNanos,
+ EndTimeNanos: endTimeNanos,
+ Value: []*fitness.Value{
+ &fitness.Value{IntVal: a.Type},
+ },
+ })
+ }
+
+ if len(dataPoints) == 0 {
+ return nil
+ }
+
+ return c.DatasetPatch(ctx, dataStreamID, dataPoints)
+}
+
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":
endOfDay := startOfDay.Add(24 * time.Hour).Add(-1 * time.Nanosecond)
currValue, startTime, err := c.readCumulative(ctx, dataSource, startOfDay, endOfDay)
+ if err != nil {
+ return err
+ }
var diffValue fitness.Value
if dataSource.DataType.Field[0].Format == "integer" {
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)
+ log.Debugf(ctx, "add cumulative data %s until %v: %+v", dataSource.DataStreamId, endTime, diffValue)
- return c.DataSetPatch(ctx, dataSource.DataStreamId, []*fitness.DataPoint{
+ return c.DatasetPatch(ctx, dataSource.DataStreamId, []*fitness.DataPoint{
&fitness.DataPoint{
DataTypeName: dataSource.DataType.Name,
StartTimeNanos: startTime.UnixNano(),
}
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()
+ dataset, err := c.DatasetGet(ctx, dataSource.DataStreamId, startTime, endTime)
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 {
+ if len(dataset.Point) == 0 {
+ log.Debugf(ctx, "read cumulative data %s until %v: []", dataSource.DataStreamId, endTime)
return &fitness.Value{}, startTime, nil
}
var sum fitness.Value
maxEndTime := startTime
- for _, p := range res.Point {
+ for _, p := range dataset.Point {
switch f := dataSource.DataType.Field[0].Format; f {
case "integer":
sum.IntVal += p.Value[0].IntVal
}
func (c *Client) heartRate(ctx context.Context, dataSource *fitness.DataSource, startTime, endTime time.Time) (heartRateDurations, 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()
+ dataset, err := c.DatasetGet(ctx, dataSource.DataStreamId, startTime, endTime)
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 {
+ if len(dataset.Point) == 0 {
return nil, startTime, nil
}
var results heartRateDurations
maxEndTime := startTime
- for _, p := range res.Point {
+ for _, p := range dataset.Point {
max := int(p.Value[1].FpVal)
min := int(p.Value[2].FpVal)
duration := time.Unix(0, p.EndTimeNanos).Sub(time.Unix(0, p.StartTimeNanos))
return results, maxEndTime, nil
}
-func (c *Client) SetHeartRate(ctx context.Context, totalDurations []fitbit.HeartRateZone, startOfDay time.Time) error {
+func (c *Client) SetHeartRate(ctx context.Context, totalDurations []fitbit.HeartRateZone, restingHeartRate int, startOfDay time.Time) error {
dataSource := &fitness.DataSource{
Application: Application(ctx),
DataType: &fitness.DataType{
break
}
+ average := float64(d.Min+d.Max) / 2.0
+ if d.Min <= restingHeartRate && restingHeartRate <= d.Max {
+ average = float64(restingHeartRate)
+ }
+
dataPoints = append(dataPoints, &fitness.DataPoint{
DataTypeName: dataSource.DataType.Name,
StartTimeNanos: startTime.UnixNano(),
EndTimeNanos: endTime.UnixNano(),
Value: []*fitness.Value{
&fitness.Value{
- FpVal: math.NaN(),
+ FpVal: average,
},
&fitness.Value{
FpVal: float64(d.Max),
if len(dataPoints) == 0 {
return nil
}
- return c.DataSetPatch(ctx, dataSource.DataStreamId, dataPoints)
+ return c.DatasetPatch(ctx, dataSource.DataStreamId, dataPoints)
}