Refactor federation + can sync track between instances

api/alarm.go

@@ -62,7 +62,7 @@ func declareAlarmRoutes(cfg *config.Config, router *gin.RouterGroup) {
 			}
 
 			for k, srv := range settings.Federation {
-				err = player.FederatedStop(srv)
+				err = srv.WakeStop()
 				if err != nil {
 					log.Printf("Unable to do federated wakeup on %s: %s", k, err.Error())
 				} else {

api/federation.go

@@ -1,12 +1,15 @@
 package api
 
 import (
+	"bytes"
+	"fmt"
 	"net/http"
 	"time"
 
 	"github.com/gin-gonic/gin"
 
 	"git.nemunai.re/nemunaire/reveil/config"
+	"git.nemunai.re/nemunaire/reveil/model"
 	"git.nemunai.re/nemunaire/reveil/player"
 )
 
@@ -47,4 +50,127 @@ func declareFederationRoutes(cfg *config.Config, router *gin.RouterGroup) {
 
 		c.JSON(http.StatusOK, true)
 	})
+
+	router.GET("/federation", func(c *gin.Context) {
+		settings, err := reveil.ReadSettings(cfg.SettingsFile)
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
+			return
+		}
+
+		c.JSON(http.StatusOK, settings.Federation)
+	})
+
+	federationsRoutes := router.Group("/federation/:fid")
+	federationsRoutes.Use(func(c *gin.Context) {
+		settings, err := reveil.ReadSettings(cfg.SettingsFile)
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
+			return
+		}
+
+		f, ok := settings.Federation[string(c.Param("fid"))]
+		if !ok {
+			c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"errmsg": "Action not found"})
+			return
+		}
+
+		c.Set("federation", &f)
+		c.Next()
+	})
+	federationsRoutes.GET("", func(c *gin.Context) {
+		c.JSON(http.StatusOK, c.MustGet("federation"))
+	})
+	federationsRoutes.POST("sync", func(c *gin.Context) {
+		srv := c.MustGet("federation").(*reveil.FederationServer)
+
+		// Retrieve music list on remote
+		remoteMusics, err := srv.GetMusics()
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to retrieve remote tracks lists: %s", err.Error())})
+			return
+		}
+
+		// Retrieve local music list
+		localMusics, err := reveil.LoadTracks(cfg)
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to retrieve local tracks: %s", err.Error())})
+			return
+		}
+
+		// Compute diff
+		var newMusics []reveil.Track
+		var oldMusics []reveil.Track
+		var musicsToEnable []reveil.Track
+
+		for _, rTrack := range remoteMusics {
+			found := false
+			for _, lTrack := range localMusics {
+				if bytes.Compare(lTrack.Id, rTrack.Id) == 0 || lTrack.Name == rTrack.Name {
+					if lTrack.Enabled != rTrack.Enabled {
+						if lTrack.Enabled {
+							musicsToEnable = append(musicsToEnable, rTrack)
+						} else {
+							oldMusics = append(oldMusics, *lTrack)
+						}
+					}
+
+					found = true
+					break
+				}
+			}
+
+			if !found && rTrack.Enabled {
+				oldMusics = append(oldMusics, rTrack)
+			}
+		}
+
+		for _, lTrack := range localMusics {
+			found := false
+			for _, rTrack := range remoteMusics {
+				if bytes.Compare(lTrack.Id, rTrack.Id) == 0 || lTrack.Name == rTrack.Name {
+					found = true
+					break
+				}
+			}
+
+			if !found && lTrack.Enabled {
+				newMusics = append(newMusics, *lTrack)
+			}
+		}
+
+		// Disable unexistant musics on local
+		for _, t := range oldMusics {
+			t.Enabled = false
+			err = srv.UpdateTrack(&t)
+			if err != nil {
+				c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("An error occurs when disabling remote tracks (unexistant on local): %s: %s", t.Id.ToString(), err.Error())})
+				return
+
+			}
+		}
+
+		// Enable existant musics on remote
+		for _, t := range musicsToEnable {
+			t.Enabled = true
+			err = srv.UpdateTrack(&t)
+			if err != nil {
+				c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
+				return
+
+			}
+		}
+
+		// Send new musics
+		for _, t := range newMusics {
+			err = srv.SendTrack(&t)
+			if err != nil {
+				c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
+				return
+
+			}
+		}
+
+		c.JSON(http.StatusOK, true)
+	})
 }