insert routes to main and driver

driver/artNet.go

@@ -1,29 +1,77 @@
 package driver
 
-import "artNet/models"
+import (
+	"artNet/cfg"
+	"artNet/models"
+	serverModels "artNet/server/models"
+	"fmt"
+
+	"github.com/tecamino/tecamino-logger/logging"
+)
 
 type ArtNetDriver struct {
-	Bus []*models.Bus
+	Name        string                 `yaml:"driver" json:"driver"`
+	Buses       map[string]*models.Bus `yaml:"buses,omitempty" json:"buses,omitempty"`
+	cfgHandler  *cfg.Cfg               `yaml:"-" json:"-"`
+	Connections serverModels.Clients   `yaml:"-"`
+	Log         *logging.Logger        `yaml:"-"`
 }
 
 // initialize new Art-Net driver
-func NewDriver() *ArtNetDriver {
-	return &ArtNetDriver{}
+// cfgDir config directory
+// name name of driver
+func NewDriver(cfgDir, name string, debug bool) (*ArtNetDriver, error) {
+	if cfgDir == "" {
+		cfgDir = "./cfg"
+	}
+
+	logger, err := logging.NewLogger(name, debug)
+	if err != nil {
+		panic(err)
+	}
+
+	logger.Debug("artNet.NewDriver", "initialize "+name+" driver")
+	d := ArtNetDriver{
+		Name:        name,
+		Buses:       make(map[string]*models.Bus),
+		cfgHandler:  cfg.NewCfgHandler(cfgDir, name),
+		Connections: serverModels.NewClients(),
+		Log:         logger,
+	}
+
+	if err := d.LoadCfg(); err != nil {
+		logger.Error("artNet.NewDriver", "error load driver configuration: "+err.Error())
+		return nil, err
+	}
+
+	return &d, nil
 }
 
-// adds new Art-Net interface to driver port 0 = 6454 (default art-net)
-func (d *ArtNetDriver) NewInterface(ip string, port int) *models.Bus {
-	i := models.NewBus(ip, port)
-	d.Bus = append(d.Bus, i)
-	return i
+// loads driver config
+func (d *ArtNetDriver) LoadCfg() error {
+	d.Log.Debug("artNet.LoadCfg", "load driver configuration")
+	if err := d.cfgHandler.LoadCfg(d); err != nil {
+		return err
+	}
+
+	for _, b := range d.Buses {
+		d.NewBus(b.Name, b.Ip, *b.Port)
+	}
+	return nil
 }
 
-// dmxData[46] = byte(255) // Channel 1: Red
-// dmxData[47] = byte(255) // Channel 2: Green
-// dmxData[48] = byte(255) // Channel 3: Blue
-// dmxData[49] = byte(255) // Channel 4: White
-// dmxData[50] = byte(255) // Channel 5: Amber
-// dmxData[51] = byte(255) // Channel 6: UV Lila
-// dmxData[52] = byte(255) // Channel 7: 0-9  10-255 strobo
-// dmxData[53] = byte(255) // Channel 8: 0-5 stop 6-127 static position 128-255 Motorgeschwindigkeit
-// dmxData[54] = byte(255) // Channel 9: 0-50 coincitence 51 -100 two color 101-150 all color
+// adds new Art-Net bus/interface to driver port 0 = 6454 (default art-net)
+func (d *ArtNetDriver) NewBus(name, ip string, port int) *models.Bus {
+	b := models.NewBus(name, ip, port)
+	d.Buses[name] = b
+	return b
+}
+
+func (d *ArtNetDriver) SetValue(set models.Set) error {
+	if _, ok := d.Buses[set.Bus]; !ok {
+		return fmt.Errorf("no bus '%s' found", set.Bus)
+	}
+	d.Buses[set.Bus].Data.SetValue(set.Address, set.Value)
+
+	return d.Buses[set.Bus].SendData()
+}

driver/bus.go

@@ -0,0 +1,94 @@
+package driver
+
+import (
+	"artNet/auth"
+	"artNet/models"
+	"fmt"
+	"net"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+// sends a list of all buses in the driver
+func (d *ArtNetDriver) GetAllBuses(c *gin.Context) {
+	var data any
+	if len(d.Buses) == 0 {
+		data = "no buses avaiable"
+	} else {
+		data = *d
+	}
+
+	c.JSON(200, gin.H{
+		"buses": data,
+	})
+}
+
+func (d *ArtNetDriver) CreateBus(c *gin.Context) {
+	_, err := auth.GetIDFromAuth(c)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
+		return
+	}
+
+	var payload models.Bus
+
+	if err := c.BindJSON(&payload); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
+		return
+	}
+
+	if payload.Name == "" {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
+		return
+	} else if addr := net.ParseIP(payload.Ip); addr == nil {
+
+		c.JSON(http.StatusBadRequest, gin.H{"error": "wrong ip '" + payload.Ip + "'"})
+		return
+	}
+
+	if _, ok := d.Buses[payload.Name]; ok {
+		c.JSON(http.StatusOK, gin.H{"message": "bus " + payload.Name + " exists already"})
+		return
+	}
+
+	bus := d.NewBus(payload.Name, payload.Ip, payload.GetPort())
+	c.JSON(http.StatusOK, gin.H{
+		"message": fmt.Sprintf("bus '%s' successfully created with ip: %s and on port: %d", bus.Name, bus.Ip, bus.GetPort()),
+	})
+	d.cfgHandler.SaveCfg(*d)
+}
+
+func (d *ArtNetDriver) RemoveBus(c *gin.Context) {
+	_, err := auth.GetIDFromAuth(c)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
+		return
+	}
+
+	var payload models.Bus
+
+	if err := c.BindJSON(&payload); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
+		return
+	}
+
+	if payload.Name == "" {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
+		return
+	}
+
+	if _, ok := d.Buses[payload.Name]; !ok {
+		c.JSON(http.StatusOK, gin.H{"message": "bus " + payload.Name + " not found"})
+		c.JSON(http.StatusOK, models.JsonResponse{
+			Message: "bus " + payload.Name + " not found"})
+		return
+	} else {
+		delete(d.Buses, payload.Name)
+	}
+
+	c.JSON(http.StatusOK, models.JsonResponse{
+		Message: fmt.Sprintf("bus '%s' successfully removed", payload.Name),
+	})
+	d.cfgHandler.SaveCfg(*d)
+}

driver/webSocket.go

@@ -0,0 +1,63 @@
+package driver
+
+import (
+	"context"
+	"log"
+	"time"
+
+	"artNet/auth"
+	"artNet/models"
+
+	"github.com/coder/websocket"
+	"github.com/coder/websocket/wsjson"
+	"github.com/gin-gonic/gin"
+)
+
+const (
+	OnCreate = "onCreate"
+	OnChange = "onChange"
+	OnDelete = "onDelete"
+)
+
+func (d *ArtNetDriver) Websocket(c *gin.Context) {
+	id, err := auth.GetIDFromAuth(c)
+	if err != nil {
+		d.Log.Error("artNet.webSocket.Websocket", "error GetIDFromAuth: "+err.Error())
+		log.Println("error id:", err)
+		return
+	}
+	d.Log.Debug("artNet.webSocket.Websocket", "authorization id token: "+id)
+
+	ctx, cancel := context.WithTimeout(c.Request.Context(), 10*time.Minute)
+	defer cancel()
+	conn, err := d.Connections.ConnectRecievingWsConnection(id, c)
+	if err != nil {
+		d.Log.Error("artNet.webSocket.Websocket", "error connecting recieving websocket conection: "+err.Error())
+		return
+	}
+	defer d.Connections.DisconnectRecievingWsConnection(id, websocket.StatusInternalError, "Internal error")
+
+	var request models.JsonData
+	//Read loop
+	for {
+
+		err := wsjson.Read(ctx, conn, &request)
+		if err != nil {
+			d.Log.Error("artNet.webSocket.Websocket", "read error:"+err.Error())
+			log.Println("WebSocket read error:", err)
+			break
+		}
+
+		// Set
+		if request.Set != nil {
+			for _, set := range *request.Set {
+				if err = d.SetValue(set); err != nil {
+					d.Log.Error("artNet.webSocket.Websocket", "set value error"+err.Error())
+					log.Println(err)
+					continue
+				}
+				time.Sleep(23 * time.Millisecond)
+			}
+		}
+	}
+}