new-server/app/api/devices.go

package api

import (
	"context"
	"git.aiterp.net/lucifer/new-server/app/config"
	"git.aiterp.net/lucifer/new-server/app/services/publisher"
	"git.aiterp.net/lucifer/new-server/models"
	"github.com/gin-gonic/gin"
	"log"
	"strconv"
	"time"
)

func fetchDevices(ctx context.Context, fetchStr string) ([]models.Device, error) {
	kind, value := models.ParseFetchString(fetchStr)
	return config.DeviceRepository().FetchByReference(ctx, kind, value)
}

func Devices(r gin.IRoutes) {
	r.GET("", handler(func(c *gin.Context) (interface{}, error) {
		devices, err := config.DeviceRepository().FetchByReference(ctxOf(c), models.RKAll, "")
		if err != nil {
			return nil, err
		}

		return withSceneState(devices), nil
	}))

	r.GET("/:fetch", handler(func(c *gin.Context) (interface{}, error) {
		devices, err := fetchDevices(ctxOf(c), c.Param("fetch"))
		if err != nil {
			return nil, err
		}

		return withSceneState(devices), nil
	}))

	r.PUT("", handler(func(c *gin.Context) (interface{}, error) {
		var body []struct {
			Fetch    string                `json:"fetch"`
			SetState models.NewDeviceState `json:"setState"`
		}
		err := parseBody(c, &body)
		if err != nil {
			return nil, err
		}

		set := make(map[int]bool)
		changed := make([]models.Device, 0, 64)
		for _, job := range body {
			devices, err := fetchDevices(ctxOf(c), job.Fetch)
			if err != nil {
				return nil, err
			}
			if len(devices) == 0 {
				return []models.Device{}, nil
			}

			for i := range devices {
				if set[devices[i].ID] {
					continue
				}

				err := devices[i].SetState(job.SetState)
				if err != nil {
					return nil, err
				}

				set[devices[i].ID] = true
				changed = append(changed, devices[i])
			}
		}

		config.PublishChannel <- changed

		go func() {
			for _, device := range changed {
				err := config.DeviceRepository().Save(context.Background(), &device, models.SMState)
				if err != nil {
					log.Println("Failed to save device for state:", err)
					continue
				}
			}
		}()

		return withSceneState(changed), nil
	}))

	r.PUT("/:fetch", handler(func(c *gin.Context) (interface{}, error) {
		update := models.DeviceUpdate{}
		err := parseBody(c, &update)
		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 {
			devices[i].ApplyUpdate(update)

			err := config.DeviceRepository().Save(context.Background(), &devices[i], models.SMProperties)
			if err != nil {
				log.Println("Failed to save device for state:", err)
				continue
			}
		}

		return withSceneState(devices), nil
	}))

	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

		err = config.DeviceRepository().SaveMany(ctxOf(c), models.SMState, devices)
		if err != nil {
			log.Println("Failed to save devices states")
		}

		return withSceneState(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
					}
				}
				if index == -1 {
					continue
				}

				device.Tags = append(device.Tags[:index], device.Tags[index+1:]...)
			}

			err = config.DeviceRepository().Save(ctxOf(c), device, models.SMTags)
			if err != nil {
				return nil, err
			}
		}

		return withSceneState(devices), nil
	}))

	r.PUT("/:fetch/scene", handler(func(c *gin.Context) (interface{}, error) {
		var body models.DeviceSceneAssignment
		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
		}

		_, err = config.SceneRepository().Find(ctxOf(c), body.SceneID)
		if err != nil {
			return nil, err
		}
		if body.DurationMS < 0 {
			body.DurationMS = 0
		}
		body.StartTime = time.Now()

		pushMode := c.Query("push") == "true"
		for i := range devices {
			if pushMode {
				devices[i].SceneAssignments = append(devices[i].SceneAssignments, body)
			} else {
				devices[i].SceneAssignments = []models.DeviceSceneAssignment{body}
			}
		}
		config.PublishChannel <- devices

		err = config.DeviceRepository().SaveMany(ctxOf(c), 0, devices)
		if err != nil {
			return nil, err
		}

		return withSceneState(devices), nil
	}))

	r.DELETE("/:fetch/scene", handler(func(c *gin.Context) (interface{}, error) {
		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 {
			devices[i].SceneAssignments = []models.DeviceSceneAssignment{}
		}
		config.PublishChannel <- devices

		err = config.DeviceRepository().SaveMany(ctxOf(c), 0, devices)
		if err != nil {
			return nil, err
		}

		return withSceneState(devices), nil
	}))

	r.DELETE("/:fetch", handler(func(c *gin.Context) (interface{}, error) {
		id, err := strconv.Atoi(c.Param("fetch"))
		if err != nil {
			return nil, err
		}

		device, err := config.DeviceRepository().Find(ctxOf(c), id)
		if err != nil {
			return nil, err
		}

		if c.Query("forgetmenot") != "true" {
			bridge, err := config.BridgeRepository().Find(ctxOf(c), device.BridgeID)
			if err != nil {
				return nil, err
			}

			driver, err := config.DriverProvider().Provide(bridge.Driver)
			if err != nil {
				return nil, err
			}

			forgettableDriver, ok := driver.(models.ForgettableDriver)
			if !ok {
				return nil, models.ErrCannotForget
			}
			err = forgettableDriver.ForgetDevice(ctxOf(c), bridge, *device)
			if err != nil {
				return nil, err
			}
		}

		err = config.DeviceRepository().Delete(ctxOf(c), device)
		if err != nil {
			return nil, err
		}

		return device, nil
	}))
}

func withSceneState(devices []models.Device) []models.Device {
	res := make([]models.Device, 0, len(devices))
	for _, device := range devices {
		device.SceneState = publisher.Global().SceneState(device.ID)
		res = append(res, device)
	}

	return res
}