Merge pull request #243 from nemunaire/f/sync-3.4

Add routes for sync v3.4
ddvk · Jul 1, 2023 · 9209bc3 · 9209bc3
2 parents 5784582 + 16d3e65
commit 9209bc3
Show file tree

Hide file tree

Showing 3 changed files with 56 additions and 0 deletions.
diff --git a/internal/app/handlers.go b/internal/app/handlers.go
@@ -1,6 +1,7 @@
 package app
 
 import (
+	"bytes"
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
@@ -702,6 +703,52 @@ func (app *App) blobStorageUpload(c *gin.Context) {
 	c.JSON(http.StatusOK, response)
 }
 
+func (app *App) syncUpdateRootV3(c *gin.Context) {
+	var rootv3 messages.SyncRootV3
+	err := json.NewDecoder(c.Request.Body).Decode(&rootv3)
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusBadRequest)
+		return
+	}
+
+	uid := c.GetString(userIDKey)
+	newgeneration, err := app.blobStorer.StoreBlob(uid, "root", bytes.NewBufferString(rootv3.Hash), rootv3.Generation)
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	c.JSON(http.StatusOK, messages.SyncRootV3{
+		Generation: newgeneration,
+		Hash:       rootv3.Hash,
+	})
+}
+
+func (app *App) syncGetRootV3(c *gin.Context) {
+	uid := c.GetString(userIDKey)
+
+	reader, generation, _, err := app.blobStorer.LoadBlob(uid, "root")
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	roothash, err := io.ReadAll(reader)
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	c.JSON(http.StatusOK, messages.SyncRootV3{
+		Generation: generation,
+		Hash:       string(roothash),
+	})
+}
+
 func (app *App) integrationsGetMetadata(c *gin.Context) {
 	var metadata messages.IntegrationMetadata
 	metadata.Thumbnail = ""

diff --git a/internal/app/routes.go b/internal/app/routes.go
@@ -127,5 +127,8 @@ func (app *App) registerRoutes(router *gin.Engine) {
 		authRoutes.POST("/sync/v2/signed-urls/downloads", app.blobStorageDownload)
 		authRoutes.POST("/sync/v2/signed-urls/uploads", app.blobStorageUpload)
 		authRoutes.POST("/sync/v2/sync-complete", app.syncCompleteV2)
+
+		authRoutes.GET("/sync/v3/root", app.syncGetRootV3)
+		authRoutes.PUT("/sync/v3/root", app.syncUpdateRootV3)
 	}
 }
diff --git a/internal/messages/messages.go b/internal/messages/messages.go
@@ -136,6 +136,12 @@ type SyncCompletedRequestV2 struct {
 	Generation int64 `json:"generation"`
 }
 
+// SyncRootV3
+type SyncRootV3 struct {
+	Generation int64  `json:"generation"`
+	Hash       string `json:"hash"`
+}
+
 // IntegrationsResponse integrations
 type IntegrationsResponse struct {
 	Integrations []Integration `json:"integrations"`