feat: Added GPX export api

bahattincinic · May 21, 2024 · e775adb · e775adb
1 parent e7ea6b5
commit e775adb
Show file tree

Hide file tree

Showing 11 changed files with 218 additions and 77 deletions.
diff --git a/api/activity.go b/api/activity.go
@@ -1,8 +1,11 @@
 package api
 
 import (
+	"encoding/xml"
+	"fmt"
 	"net/http"
 
+	"github.com/bahattincinic/fitwave/strava"
 	"github.com/labstack/echo/v4"
 )
 
@@ -53,3 +56,40 @@ func (a *API) getActivity(c echo.Context) error {
 
 	return c.JSON(http.StatusOK, act)
 }
+
+// getActivity godoc
+//
+//	@Summary	Export Activity GPX
+//	@Tags		activity
+//	@Accept		json
+//	@Param		id				path	string	true	"Activity ID"
+//	@Param		Authorization	header	string	true	"Strava Access Token"
+//	@Success	200
+//	@Router		/activities/{id}/gpx [get]
+func (a *API) exportActivityGPS(c echo.Context) error {
+	user := c.Get(userContextKey).(*strava.User)
+
+	act, err := a.db.GetActivity(c.Param("id"))
+	if err != nil {
+		return err
+	}
+
+	if act == nil {
+		return echo.NewHTTPError(http.StatusNotFound, "activity Not Found")
+	}
+
+	gpx, err := a.st.ExportGPX(user, act.Id)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusInternalServerError, err)
+	}
+
+	if gpx == "" {
+		return echo.NewHTTPError(http.StatusNotFound, "GPX Not Found")
+	}
+
+	c.Response().Header().Set(echo.HeaderContentDisposition,
+		fmt.Sprintf("attachment; filename=\"activity_%d.gpx\"", act.Id))
+	c.Response().Header().Set(echo.HeaderContentType,
+		"application/gpx+xml")
+	return c.Blob(http.StatusOK, "application/gpx+xml", []byte(xml.Header+gpx))
+}
diff --git a/api/docs/docs.go b/api/docs/docs.go
@@ -97,6 +97,38 @@ const docTemplate = `{
                 }
             }
         },
+        "/activities/{id}/gpx": {
+            "get": {
+                "consumes": [
+                    "application/json"
+                ],
+                "tags": [
+                    "activity"
+                ],
+                "summary": "Export Activity GPX",
+                "parameters": [
+                    {
+                        "type": "string",
+                        "description": "Activity ID",
+                        "name": "id",
+                        "in": "path",
+                        "required": true
+                    },
+                    {
+                        "type": "string",
+                        "description": "Strava Access Token",
+                        "name": "Authorization",
+                        "in": "header",
+                        "required": true
+                    }
+                ],
+                "responses": {
+                    "200": {
+                        "description": "OK"
+                    }
+                }
+            }
+        },
         "/athletes": {
             "get": {
                 "consumes": [
@@ -960,10 +992,10 @@ const docTemplate = `{
                 "athlete": {
                     "$ref": "#/definitions/models.Athlete"
                 },
-                "athleteID": {
+                "athlete_count": {
                     "type": "integer"
                 },
-                "athlete_count": {
+                "athlete_id": {
                     "type": "integer"
                 },
                 "average_cadence": {
@@ -1005,8 +1037,10 @@ const docTemplate = `{
                 "flagged": {
                     "type": "boolean"
                 },
+                "gear": {
+                    "$ref": "#/definitions/models.Gear"
+                },
                 "gear_id": {
-                    "description": "bike or pair of shoes",
                     "type": "string"
                 },
                 "has_kudos": {

diff --git a/api/docs/swagger.json b/api/docs/swagger.json
@@ -86,6 +86,38 @@
                 }
             }
         },
+        "/activities/{id}/gpx": {
+            "get": {
+                "consumes": [
+                    "application/json"
+                ],
+                "tags": [
+                    "activity"
+                ],
+                "summary": "Export Activity GPX",
+                "parameters": [
+                    {
+                        "type": "string",
+                        "description": "Activity ID",
+                        "name": "id",
+                        "in": "path",
+                        "required": true
+                    },
+                    {
+                        "type": "string",
+                        "description": "Strava Access Token",
+                        "name": "Authorization",
+                        "in": "header",
+                        "required": true
+                    }
+                ],
+                "responses": {
+                    "200": {
+                        "description": "OK"
+                    }
+                }
+            }
+        },
         "/athletes": {
             "get": {
                 "consumes": [
@@ -949,10 +981,10 @@
                 "athlete": {
                     "$ref": "#/definitions/models.Athlete"
                 },
-                "athleteID": {
+                "athlete_count": {
                     "type": "integer"
                 },
-                "athlete_count": {
+                "athlete_id": {
                     "type": "integer"
                 },
                 "average_cadence": {
@@ -994,8 +1026,10 @@
                 "flagged": {
                     "type": "boolean"
                 },
+                "gear": {
+                    "$ref": "#/definitions/models.Gear"
+                },
                 "gear_id": {
-                    "description": "bike or pair of shoes",
                     "type": "string"
                 },
                 "has_kudos": {

diff --git a/api/docs/swagger.yaml b/api/docs/swagger.yaml
@@ -53,7 +53,7 @@ definitions:
         $ref: '#/definitions/models.Athlete'
       athlete_count:
         type: integer
-      athleteID:
+      athlete_id:
         type: integer
       average_cadence:
         type: number
@@ -81,8 +81,9 @@ definitions:
         type: string
       flagged:
         type: boolean
+      gear:
+        $ref: '#/definitions/models.Gear'
       gear_id:
-        description: bike or pair of shoes
         type: string
       has_kudos:
         type: boolean
@@ -448,6 +449,27 @@ paths:
       summary: Get Activity
       tags:
       - activity
+  /activities/{id}/gpx:
+    get:
+      consumes:
+      - application/json
+      parameters:
+      - description: Activity ID
+        in: path
+        name: id
+        required: true
+        type: string
+      - description: Strava Access Token
+        in: header
+        name: Authorization
+        required: true
+        type: string
+      responses:
+        "200":
+          description: OK
+      summary: Export Activity GPX
+      tags:
+      - activity
   /athletes:
     get:
       consumes:

diff --git a/api/handlers.go b/api/handlers.go
@@ -26,6 +26,7 @@ func (a *API) setupHandlers() {
 		act := a.ec.Group("/activities")
 		act.GET("", a.listActivities)
 		act.GET("/:id", a.getActivity)
+		act.GET("/:id/gpx", a.exportActivityGPS, requireAuth)
 	}
 
 	{

diff --git a/database/activity.go b/database/activity.go
@@ -52,6 +52,7 @@ func (d *Database) ListActivities(offset, limit int) (int64, []models.Activity,
 		Offset(offset).
 		Order("id desc").
 		Preload("Athlete").
+		Preload("Gear").
 		Find(&activities).
 		Count(&count).
 		Error
@@ -65,7 +66,14 @@ func (d *Database) ListActivities(offset, limit int) (int64, []models.Activity,
 
 func (d *Database) GetActivity(id string) (*models.Activity, error) {
 	var act models.Activity
-	if err := d.db.Preload("Athlete").First(&act, id).Error; err != nil {
+
+	err := d.db.
+		Preload("Athlete").
+		Preload("Gear").
+		First(&act, id).
+		Error
+
+	if err != nil {
 		if err == gorm.ErrRecordNotFound {
 			return nil, nil
 		}

diff --git a/database/athlete.go b/database/athlete.go
@@ -8,18 +8,15 @@ import (
 	"gorm.io/gorm/clause"
 )
 
-func (d *Database) UpsertAthletes(tx *gorm.DB, athletes []models.Athlete) error {
-	for _, row := range athletes {
-		currentRow := row
-		err := tx.Clauses(clause.OnConflict{
-			Columns:   []clause.Column{{Name: "id"}},
-			UpdateAll: true,
-		}).Create(&currentRow).Error
-		if err != nil {
-			d.log.Error("could not upsert athletes",
-				zap.Any("athlete", currentRow))
-			return err
-		}
+func (d *Database) UpsertAthlete(tx *gorm.DB, athlete *models.Athlete) error {
+	err := tx.Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "id"}},
+		UpdateAll: true,
+	}).Create(&athlete).Error
+	if err != nil {
+		d.log.Error("could not upsert athletes",
+			zap.Any("athlete", athlete))
+		return err
 	}
 	return nil
 }