small fixed, device repo and some device api

app/api/bridges.go

@@ -1,6 +1,7 @@
 package api
 
 import (
+	"fmt"
 	"git.aiterp.net/lucifer/new-server/app/config"
 	"git.aiterp.net/lucifer/new-server/models"
 	"github.com/gin-gonic/gin"
@@ -18,9 +19,11 @@ func Bridges(r gin.IRoutes) {
 
 	r.POST("", handler(func(c *gin.Context) (interface{}, error) {
 		var body struct {
-			Driver  models.DriverKind `json:"driver"`
-			Address string            `json:"address"`
-			DryRun  bool              `json:"dryRun"`
+			Driver    models.DriverKind `json:"driver"`
+			Address   string            `json:"address"`
+			Token     string            `json:"token"`
+			ForceSave bool              `json:"forceSave"`
+			DryRun    bool              `json:"dryRun"`
 		}
 		err := parseBody(c, &body)
 		if err != nil {
@@ -32,6 +35,22 @@ func Bridges(r gin.IRoutes) {
 			return nil, err
 		}
 
+		if body.ForceSave {
+			bridge := models.Bridge{
+				Name:    fmt.Sprintf("Manually saved %s bridge (%s)", body.Driver, body.Address),
+				Driver:  body.Driver,
+				Address: body.Address,
+				Token:   body.Token,
+			}
+
+			err := config.BridgeRepository().Save(ctxOf(c), &bridge)
+			if err != nil {
+				return nil, err
+			}
+
+			return []models.Bridge{bridge}, nil
+		}
+
 		bridges, err := driver.SearchBridge(ctxOf(c), body.Address, body.DryRun)
 		if err != nil {
 			return nil, err

app/api/devices.go

@@ -0,0 +1,123 @@
+package api
+
+import (
+	"context"
+	"git.aiterp.net/lucifer/new-server/app/config"
+	"git.aiterp.net/lucifer/new-server/models"
+	"github.com/gin-gonic/gin"
+	"log"
+	"strings"
+)
+
+func fetchDevices(ctx context.Context, fetchStr string) ([]models.Device, error) {
+	if strings.HasPrefix(fetchStr, "tag:") {
+		return config.DeviceRepository().FetchByReference(ctx, models.RKTag, fetchStr[4:])
+	} else if strings.HasPrefix(fetchStr, "bridge:") {
+		return config.DeviceRepository().FetchByReference(ctx, models.RKBridgeID, fetchStr[7:])
+	} else if fetchStr == "all" {
+		return config.DeviceRepository().FetchByReference(ctx, models.RKAll, "")
+	} else {
+		return config.DeviceRepository().FetchByReference(ctx, models.RKDeviceID, fetchStr)
+	}
+}
+
+func Devices(r gin.IRoutes) {
+	r.GET("", handler(func(c *gin.Context) (interface{}, error) {
+		return config.DeviceRepository().FetchByReference(ctxOf(c), models.RKAll, "")
+	}))
+
+	r.GET("/:fetch", handler(func(c *gin.Context) (interface{}, error) {
+		return fetchDevices(ctxOf(c), c.Param("fetch"))
+	}))
+
+	r.PUT("/:fetch/state", handler(func(c *gin.Context) (interface{}, error) {
+		state := models.NewDeviceState{}
+		err := parseBody(c, &state)
+		if err != nil {
+			return nil, err
+		}
+
+		devices, err := fetchDevices(ctxOf(c), c.Param("fetch"))
+		if err != nil {
+			return nil, err
+		}
+		if len(devices) == 0 {
+			return []models.Device{}, nil
+		}
+
+		for i := range devices {
+			err := devices[i].SetState(state)
+			if err != nil {
+				return nil, err
+			}
+		}
+
+		config.PublishChannel <- devices
+
+		go func() {
+			for _, device := range devices {
+				err := config.DeviceRepository().Save(context.Background(), &device)
+				if err != nil {
+					log.Println("Failed to save device for state:", err)
+					continue
+				}
+			}
+		}()
+
+		return devices, nil
+	}))
+
+	r.PUT("/:fetch/tags", handler(func(c *gin.Context) (interface{}, error) {
+		var body struct {
+			Add    []string `json:"add"`
+			Remove []string `json:"remove"`
+		}
+		err := parseBody(c, &body)
+		if err != nil {
+			return nil, err
+		}
+
+		devices, err := fetchDevices(ctxOf(c), c.Param("fetch"))
+		if err != nil {
+			return nil, err
+		}
+		if len(devices) == 0 {
+			return []models.Device{}, nil
+		}
+
+		for i := range devices {
+			device := &devices[i]
+
+			for _, tag := range body.Add {
+				found := false
+				for _, tag2 := range device.Tags {
+					if tag == tag2 {
+						found = true
+						break
+					}
+				}
+
+				if !found {
+					device.Tags = append(device.Tags, tag)
+				}
+			}
+			for _, tag := range body.Remove {
+				index := -1
+				for i, tag2 := range device.Tags {
+					if tag == tag2 {
+						index = i
+					}
+				}
+
+				device.Tags = append(device.Tags[:index], device.Tags[index+1:]...)
+			}
+
+			err = config.DeviceRepository().Save(ctxOf(c), device)
+			if err != nil {
+				return nil, err
+			}
+		}
+
+		return devices, nil
+	}))
+}