Compare commits
2 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
83cbcdaa42 | ||
![]() |
2839b615c3 |
@@ -13,3 +13,11 @@ func GetIDFromAuth(c *gin.Context) (string, error) {
|
|||||||
}
|
}
|
||||||
return "", errors.New("authorization token missing")
|
return "", errors.New("authorization token missing")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetIDFromQuery(c *gin.Context) (string, error) {
|
||||||
|
auth, exists := c.GetQuery("id")
|
||||||
|
if !exists {
|
||||||
|
return "", errors.New("id missing")
|
||||||
|
}
|
||||||
|
return auth, nil
|
||||||
|
}
|
||||||
|
66
client/client.go
Normal file
66
client/client.go
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/coder/websocket"
|
||||||
|
"github.com/coder/websocket/wsjson"
|
||||||
|
json_data "github.com/tecamino/tecamino-json_data"
|
||||||
|
json_dataModels "github.com/tecamino/tecamino-json_data/models"
|
||||||
|
"github.com/tecamino/tecamino-logger/logging"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Client struct {
|
||||||
|
conn *websocket.Conn
|
||||||
|
ctx context.Context
|
||||||
|
cancel context.CancelFunc
|
||||||
|
log *logging.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initialize new websocket server
|
||||||
|
func NewClient(log *logging.Logger) *Client {
|
||||||
|
c := Client{
|
||||||
|
log: log,
|
||||||
|
}
|
||||||
|
c.ctx, c.cancel = context.WithCancel(context.Background())
|
||||||
|
return &c
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) Connect(ip, id string, port uint) (err error) {
|
||||||
|
c.conn, _, err = websocket.Dial(c.ctx, fmt.Sprintf("ws://%s:%d/ws?id=%s", ip, port, id), nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) Disconnect() {
|
||||||
|
c.cancel()
|
||||||
|
}
|
||||||
|
func (c *Client) Subscribe(id string) error {
|
||||||
|
|
||||||
|
req := json_data.NewRequest()
|
||||||
|
req.AddDriverSubscription(".*", id, 0, true, false, false)
|
||||||
|
if err := wsjson.Write(c.ctx, c.conn, req); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) ReadJsonData() (response json_dataModels.Response, err error) {
|
||||||
|
|
||||||
|
err = wsjson.Read(c.ctx, c.conn, &response)
|
||||||
|
if err != nil {
|
||||||
|
code := websocket.CloseStatus(err)
|
||||||
|
|
||||||
|
switch code {
|
||||||
|
case websocket.StatusNormalClosure,
|
||||||
|
websocket.StatusGoingAway,
|
||||||
|
websocket.StatusNoStatusRcvd:
|
||||||
|
c.log.Info("webSocket.readJsonData", fmt.Sprintf("WebSocket closed: %v (code: %v)\n", err, code))
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
c.log.Error("webSocket.readJsonData", fmt.Sprintf("WebSocket read error: %v (code: %v)\n", err, code))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
@@ -2,8 +2,8 @@ package driver
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"artNet/cfg"
|
"artNet/cfg"
|
||||||
|
"artNet/client"
|
||||||
"artNet/models"
|
"artNet/models"
|
||||||
serverModels "artNet/server/models"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"path"
|
"path"
|
||||||
|
|
||||||
@@ -14,8 +14,9 @@ type ArtNetDriver struct {
|
|||||||
Name string `yaml:"driver" json:"driver"`
|
Name string `yaml:"driver" json:"driver"`
|
||||||
Buses map[string]*models.Bus `yaml:"buses,omitempty" json:"buses,omitempty"`
|
Buses map[string]*models.Bus `yaml:"buses,omitempty" json:"buses,omitempty"`
|
||||||
cfgHandler *cfg.Cfg `yaml:"-" json:"-"`
|
cfgHandler *cfg.Cfg `yaml:"-" json:"-"`
|
||||||
Connections serverModels.Clients `yaml:"-"`
|
Conn *client.Client `yaml:"-" json:"-"`
|
||||||
Log *logging.Logger `yaml:"-"`
|
Subscriptions models.Subscriptions `yaml:"-" json:"-"`
|
||||||
|
Log *logging.Logger `yaml:"-" json:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// initialize new Art-Net driver
|
// initialize new Art-Net driver
|
||||||
@@ -42,7 +43,6 @@ func NewDriver(cfgDir, name string, debug bool) (*ArtNetDriver, error) {
|
|||||||
Name: name,
|
Name: name,
|
||||||
Buses: make(map[string]*models.Bus),
|
Buses: make(map[string]*models.Bus),
|
||||||
cfgHandler: cfg.NewCfgHandler(cfgDir, name),
|
cfgHandler: cfg.NewCfgHandler(cfgDir, name),
|
||||||
Connections: serverModels.NewClients(),
|
|
||||||
Log: logger,
|
Log: logger,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -51,6 +51,8 @@ func NewDriver(cfgDir, name string, debug bool) (*ArtNetDriver, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
d.Conn = client.NewClient(logger)
|
||||||
|
|
||||||
return &d, nil
|
return &d, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -70,16 +72,43 @@ func (d *ArtNetDriver) LoadCfg() error {
|
|||||||
// adds new Art-Net bus/interface to driver port 0 = 6454 (default art-net)
|
// 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 {
|
func (d *ArtNetDriver) NewBus(name, ip string, port int) *models.Bus {
|
||||||
b := models.NewBus(name, ip, port)
|
b := models.NewBus(name, ip, port)
|
||||||
b.Start(d.Log)
|
|
||||||
d.Buses[name] = b
|
d.Buses[name] = b
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *ArtNetDriver) SetValue(set models.Set) error {
|
func (d *ArtNetDriver) SetValue(bus string, address uint, value uint8) error {
|
||||||
if _, ok := d.Buses[set.Bus]; !ok {
|
if _, ok := d.Buses[bus]; !ok {
|
||||||
return fmt.Errorf("no bus '%s' found", set.Bus)
|
return fmt.Errorf("no bus '%s' found", bus)
|
||||||
}
|
}
|
||||||
d.Buses[set.Bus].Data.SetValue(set.Address, set.Value)
|
d.Buses[bus].Data.SetValue(address, value)
|
||||||
|
|
||||||
return d.Buses[set.Bus].SendData()
|
return d.Buses[bus].SendData()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) Connect(ip, id string, port uint) error {
|
||||||
|
if err := d.Conn.Connect(ip, id, port); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer d.Conn.Disconnect()
|
||||||
|
|
||||||
|
if err := d.Conn.Subscribe(id); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
respond, err := d.Conn.ReadJsonData()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
d.Subscribe(respond.Subscribe...)
|
||||||
|
|
||||||
|
for _, pub := range respond.Publish {
|
||||||
|
if sub, ok := d.Subscriptions[pub.Uuid]; ok {
|
||||||
|
if err := d.SetValue(sub.Bus, sub.Address, uint8(pub.Value.(float64))); err != nil {
|
||||||
|
d.Log.Info("artNet.Connect", err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
144
driver/bus.go
144
driver/bus.go
@@ -8,6 +8,8 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
json_data "github.com/tecamino/tecamino-json_data"
|
||||||
|
json_dataModels "github.com/tecamino/tecamino-json_data/models"
|
||||||
)
|
)
|
||||||
|
|
||||||
// sends a list of all buses in the driver
|
// sends a list of all buses in the driver
|
||||||
@@ -25,124 +27,132 @@ func (d *ArtNetDriver) GetAllBuses(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (d *ArtNetDriver) CreateBus(c *gin.Context) {
|
func (d *ArtNetDriver) CreateBus(c *gin.Context) {
|
||||||
_, err := auth.GetIDFromAuth(c)
|
_, err := auth.GetIDFromQuery(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
r := json_data.NewResponse()
|
||||||
|
r.SendError(err.Error())
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var payload models.Bus
|
busPayload := models.Bus{}
|
||||||
|
if err := busPayload.ParsePayload(c); err != nil {
|
||||||
if err := c.BindJSON(&payload); err != nil {
|
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if payload.Name == "" {
|
if addr := net.ParseIP(busPayload.Ip); addr == nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
r := json_data.NewResponse()
|
||||||
return
|
r.SendError("wrong ip '" + busPayload.Ip + "'")
|
||||||
} else if addr := net.ParseIP(payload.Ip); addr == nil {
|
c.JSON(http.StatusBadRequest, r)
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "wrong ip '" + payload.Ip + "'"})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, ok := d.Buses[payload.Name]; ok {
|
if _, ok := d.Buses[busPayload.Name]; ok {
|
||||||
c.JSON(http.StatusOK, gin.H{"message": "bus " + payload.Name + " exists already"})
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("bus " + busPayload.Name + " exists already")
|
||||||
|
c.JSON(http.StatusOK, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
bus := d.NewBus(payload.Name, payload.Ip, payload.GetPort())
|
bus := d.NewBus(busPayload.Name, busPayload.Ip, busPayload.GetPort())
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{
|
r := json_data.NewResponse()
|
||||||
"message": fmt.Sprintf("bus '%s' successfully created with ip: %s and on port: %d", bus.Name, bus.Ip, bus.GetPort()),
|
r.SendMessage(fmt.Sprintf("bus '%s' successfully created with ip: %s and on port: %d", bus.Name, bus.Ip, bus.GetPort()))
|
||||||
})
|
c.JSON(http.StatusOK, r)
|
||||||
d.cfgHandler.SaveCfg(*d)
|
d.cfgHandler.SaveCfg(*d)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *ArtNetDriver) RemoveBus(c *gin.Context) {
|
func (d *ArtNetDriver) RemoveBus(c *gin.Context) {
|
||||||
_, err := auth.GetIDFromAuth(c)
|
_, err := auth.GetIDFromQuery(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("id: " + err.Error())
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var payload models.Bus
|
busPayload := models.Bus{}
|
||||||
|
if err := busPayload.ParsePayload(c); err != nil {
|
||||||
if err := c.BindJSON(&payload); err != nil {
|
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if payload.Name == "" {
|
if _, ok := d.Buses[busPayload.Name]; !ok {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
r := json_dataModels.NewResponse()
|
||||||
return
|
r.SendMessage("bus " + busPayload.Name + " not found")
|
||||||
}
|
c.JSON(http.StatusOK, r)
|
||||||
|
|
||||||
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
|
return
|
||||||
} else {
|
} else {
|
||||||
delete(d.Buses, payload.Name)
|
delete(d.Buses, busPayload.Name)
|
||||||
}
|
}
|
||||||
|
r := json_dataModels.NewResponse()
|
||||||
c.JSON(http.StatusOK, models.JsonResponse{
|
r.SendMessage(fmt.Sprintf("bus '%s' successfully removed", busPayload.Name))
|
||||||
Message: fmt.Sprintf("bus '%s' successfully removed", payload.Name),
|
c.JSON(http.StatusOK, r)
|
||||||
})
|
|
||||||
d.cfgHandler.SaveCfg(*d)
|
d.cfgHandler.SaveCfg(*d)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *ArtNetDriver) Start(c *gin.Context) {
|
func (d *ArtNetDriver) Start(c *gin.Context) {
|
||||||
_, err := auth.GetIDFromAuth(c)
|
_, err := auth.GetIDFromQuery(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("id: " + err.Error())
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var payload models.Bus
|
busPayload := models.Bus{}
|
||||||
|
if busPayload.ParsePayload(c); err != nil {
|
||||||
if err := c.BindJSON(&payload); err != nil {
|
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if payload.Name == "" {
|
d.Buses[busPayload.Name].Start(d.Log)
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
d.Buses[payload.Name].Start(d.Log)
|
r := json_dataModels.NewResponse()
|
||||||
|
r.SendMessage(fmt.Sprintf("bus '%s' running", busPayload.Name))
|
||||||
c.JSON(http.StatusOK, models.JsonResponse{
|
c.JSON(http.StatusOK, r)
|
||||||
Message: fmt.Sprintf("bus '%s' running", payload.Name),
|
|
||||||
})
|
|
||||||
d.cfgHandler.SaveCfg(*d)
|
d.cfgHandler.SaveCfg(*d)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *ArtNetDriver) Stop(c *gin.Context) {
|
func (d *ArtNetDriver) Stop(c *gin.Context) {
|
||||||
_, err := auth.GetIDFromAuth(c)
|
_, err := auth.GetIDFromQuery(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("id: " + err.Error())
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var payload models.Bus
|
busPayload := models.Bus{}
|
||||||
|
if err := busPayload.ParsePayload(c); err != nil {
|
||||||
if err := c.BindJSON(&payload); err != nil {
|
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if payload.Name == "" {
|
d.Buses[busPayload.Name].Stop()
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
d.Buses[payload.Name].Stop()
|
r := json_dataModels.NewResponse()
|
||||||
|
r.SendMessage(fmt.Sprintf("bus '%s' stopped", busPayload.Name))
|
||||||
c.JSON(http.StatusOK, models.JsonResponse{
|
c.JSON(http.StatusOK, r)
|
||||||
Message: fmt.Sprintf("bus '%s' stopped", payload.Name),
|
|
||||||
})
|
|
||||||
d.cfgHandler.SaveCfg(*d)
|
d.cfgHandler.SaveCfg(*d)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) Status(c *gin.Context) {
|
||||||
|
_, err := auth.GetIDFromQuery(c)
|
||||||
|
if err != nil {
|
||||||
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("id: " + err.Error())
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
busPayload := models.Bus{}
|
||||||
|
if err := busPayload.ParsePayload(c); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
r := json_dataModels.NewResponse()
|
||||||
|
state := "stopped"
|
||||||
|
if d.Buses[busPayload.Name].Status() {
|
||||||
|
state = "running"
|
||||||
|
}
|
||||||
|
r.SendMessage(fmt.Sprintf("bus '%s' %s", busPayload.Name, state))
|
||||||
|
c.JSON(http.StatusOK, r)
|
||||||
|
}
|
||||||
|
20
driver/subscribe.go
Normal file
20
driver/subscribe.go
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
package driver
|
||||||
|
|
||||||
|
import (
|
||||||
|
"artNet/models"
|
||||||
|
|
||||||
|
json_dataModels "github.com/tecamino/tecamino-json_data/models"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) Subscribe(subs ...json_dataModels.Subscribe) {
|
||||||
|
if d.Subscriptions == nil {
|
||||||
|
d.Subscriptions = models.NewSubscriptions()
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, sub := range subs {
|
||||||
|
if drv, ok := (*sub.Drivers)[sub.Driver]; ok {
|
||||||
|
d.Subscriptions.AddSubscription(sub.Uuid, drv)
|
||||||
|
d.SetValue(drv.Bus, drv.Address, uint8(sub.Value.(float64)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@@ -1,63 +1,49 @@
|
|||||||
package driver
|
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 (
|
const (
|
||||||
OnCreate = "onCreate"
|
OnCreate = "onCreate"
|
||||||
OnChange = "onChange"
|
OnChange = "onChange"
|
||||||
OnDelete = "onDelete"
|
OnDelete = "onDelete"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (d *ArtNetDriver) Websocket(c *gin.Context) {
|
//func (d *ArtNetDriver) Websocket(c *gin.Context) {
|
||||||
id, err := auth.GetIDFromAuth(c)
|
// id, err := auth.GetIDFromQuery(c)
|
||||||
if err != nil {
|
// if err != nil {
|
||||||
d.Log.Error("artNet.webSocket.Websocket", "error GetIDFromAuth: "+err.Error())
|
// d.Log.Error("artNet.webSocket.Websocket", "error GetIDFromAuth: "+err.Error())
|
||||||
log.Println("error id:", err)
|
// return
|
||||||
return
|
// }
|
||||||
}
|
// d.Log.Debug("artNet.webSocket.Websocket", "authorization id token: "+id)
|
||||||
d.Log.Debug("artNet.webSocket.Websocket", "authorization id token: "+id)
|
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(c.Request.Context(), 10*time.Minute)
|
// ctx, cancel := context.WithTimeout(c.Request.Context(), 10*time.Minute)
|
||||||
defer cancel()
|
// defer cancel()
|
||||||
conn, err := d.Connections.ConnectRecievingWsConnection(id, c)
|
// conn, err := d.Connections.ConnectRecievingWsConnection(id, c)
|
||||||
if err != nil {
|
// if err != nil {
|
||||||
d.Log.Error("artNet.webSocket.Websocket", "error connecting recieving websocket conection: "+err.Error())
|
// d.Log.Error("artNet.webSocket.Websocket", "error connecting recieving websocket conection: "+err.Error())
|
||||||
return
|
// return
|
||||||
}
|
// }
|
||||||
defer d.Connections.DisconnectRecievingWsConnection(id, websocket.StatusInternalError, "Internal error")
|
// defer d.Connections.DisconnectRecievingWsConnection(id, websocket.StatusInternalError, "Internal error")
|
||||||
|
|
||||||
var request models.JsonData
|
// var request any
|
||||||
//Read loop
|
// //Read loop
|
||||||
for {
|
// 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
|
|
||||||
}
|
|
||||||
|
|
||||||
|
// 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
|
||||||
|
// }
|
||||||
|
// fmt.Println(request)
|
||||||
// Set
|
// Set
|
||||||
if request.Set != nil {
|
// if request.Set != nil {
|
||||||
for _, set := range *request.Set {
|
// for _, set := range *request.Set {
|
||||||
if err = d.SetValue(set); err != nil {
|
// if err = d.SetValue(set); err != nil {
|
||||||
d.Log.Error("artNet.webSocket.Websocket", "set value error"+err.Error())
|
// d.Log.Error("artNet.webSocket.Websocket", "set value error"+err.Error())
|
||||||
log.Println(err)
|
// log.Println(err)
|
||||||
continue
|
// continue
|
||||||
}
|
// }
|
||||||
time.Sleep(23 * time.Millisecond)
|
// time.Sleep(23 * time.Millisecond)
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
|
2
go.mod
2
go.mod
@@ -7,7 +7,9 @@ toolchain go1.23.8
|
|||||||
require (
|
require (
|
||||||
github.com/coder/websocket v1.8.13
|
github.com/coder/websocket v1.8.13
|
||||||
github.com/gin-gonic/gin v1.10.0
|
github.com/gin-gonic/gin v1.10.0
|
||||||
|
github.com/google/uuid v1.6.0
|
||||||
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e
|
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e
|
||||||
|
github.com/tecamino/tecamino-json_data v0.0.11
|
||||||
github.com/tecamino/tecamino-logger v0.2.0
|
github.com/tecamino/tecamino-logger v0.2.0
|
||||||
gopkg.in/yaml.v3 v3.0.1
|
gopkg.in/yaml.v3 v3.0.1
|
||||||
)
|
)
|
||||||
|
4
go.sum
4
go.sum
@@ -30,6 +30,8 @@ github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MG
|
|||||||
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
||||||
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||||
@@ -63,6 +65,8 @@ github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsT
|
|||||||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e h1:nt2877sKfojlHCTOBXbpWjBkuWKritFaGIfgQwbQUls=
|
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e h1:nt2877sKfojlHCTOBXbpWjBkuWKritFaGIfgQwbQUls=
|
||||||
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e/go.mod h1:B4+Kq1u5FlULTjFSM707Q6e/cOHFv0z/6QRoxubDIQ8=
|
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e/go.mod h1:B4+Kq1u5FlULTjFSM707Q6e/cOHFv0z/6QRoxubDIQ8=
|
||||||
|
github.com/tecamino/tecamino-json_data v0.0.11 h1:WVcF0tj+ElM9hRM1PccbSnY8DHJnLPauqzY0o0ib9O0=
|
||||||
|
github.com/tecamino/tecamino-json_data v0.0.11/go.mod h1:LLlyD7Wwqplb2BP4PeO86EokEcTRidlW5MwgPd1T2JY=
|
||||||
github.com/tecamino/tecamino-logger v0.2.0 h1:NPH/Gg9qRhmVoW8b39i1eXu/LEftHc74nyISpcRG+XU=
|
github.com/tecamino/tecamino-logger v0.2.0 h1:NPH/Gg9qRhmVoW8b39i1eXu/LEftHc74nyISpcRG+XU=
|
||||||
github.com/tecamino/tecamino-logger v0.2.0/go.mod h1:0M1E9Uei/qw3e3WA1x3lBo1eP3H5oeYE7GjYrMahnj8=
|
github.com/tecamino/tecamino-logger v0.2.0/go.mod h1:0M1E9Uei/qw3e3WA1x3lBo1eP3H5oeYE7GjYrMahnj8=
|
||||||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||||
|
24
main.go
24
main.go
@@ -6,6 +6,7 @@ import (
|
|||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
@@ -14,9 +15,11 @@ const DriverName string = "ArtNetDriver"
|
|||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
// cli flags
|
// cli flags
|
||||||
port := flag.Uint("port-http", 8110, "http server listening port")
|
|
||||||
cfgDir := flag.String("cfg", "./cfg", "config directory")
|
cfgDir := flag.String("cfg", "./cfg", "config directory")
|
||||||
workingDir := flag.String("workingDirectory", ".", "current working directory")
|
workingDir := flag.String("workingDirectory", ".", "current working directory")
|
||||||
|
serverIp := flag.String("serverIp", "127.0.0.1", "ip address of server")
|
||||||
|
serverPort := flag.Uint("serverPort", 8100, "port of server")
|
||||||
|
wsPort := flag.Uint("port", 8200, "websocket port")
|
||||||
debug := flag.Bool("debug", false, "debug logging")
|
debug := flag.Bool("debug", false, "debug logging")
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
@@ -38,21 +41,28 @@ func main() {
|
|||||||
|
|
||||||
//set routes
|
//set routes
|
||||||
artNetDriver.Log.Debug("main", "setting routes")
|
artNetDriver.Log.Debug("main", "setting routes")
|
||||||
s.Routes.GET("/ws", artNetDriver.Websocket)
|
|
||||||
s.Routes.GET("/buses/all", artNetDriver.GetAllBuses)
|
s.Routes.GET("/buses/all", artNetDriver.GetAllBuses)
|
||||||
s.Routes.POST("/buses/create", artNetDriver.CreateBus)
|
s.Routes.POST("/buses/create", artNetDriver.CreateBus)
|
||||||
s.Routes.POST("/buses/remove", artNetDriver.RemoveBus)
|
s.Routes.POST("/buses/remove", artNetDriver.RemoveBus)
|
||||||
s.Routes.POST("/buses/start", artNetDriver.Start)
|
s.Routes.POST("/buses/start", artNetDriver.Start)
|
||||||
|
s.Routes.POST("/buses/status", artNetDriver.Status)
|
||||||
s.Routes.POST("/buses/stop", artNetDriver.Stop)
|
s.Routes.POST("/buses/stop", artNetDriver.Stop)
|
||||||
|
|
||||||
s.Routes.GET("/", func(c *gin.Context) {
|
s.Routes.GET("/", func(c *gin.Context) {
|
||||||
c.String(200, "ArtNet Driver WebSocket Server is running!")
|
c.String(200, "ArtNet Driver WebSocket Server is running!")
|
||||||
})
|
})
|
||||||
|
|
||||||
artNetDriver.Log.Info("main", fmt.Sprintf("listen on %d", *port))
|
go func() {
|
||||||
// start http server
|
if err := s.ServeHttp(*wsPort); err != nil {
|
||||||
if err := s.ServeHttp(*port); err != nil {
|
artNetDriver.Log.Error("main", err.Error())
|
||||||
artNetDriver.Log.Error("main", "error http server "+err.Error())
|
|
||||||
panic(err)
|
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
artNetDriver.Log.Info("main", fmt.Sprintf("connect to server ws://%s:%d with id %s", *serverIp, serverPort, DriverName))
|
||||||
|
// connect to server
|
||||||
|
for {
|
||||||
|
artNetDriver.Log.Error("main", artNetDriver.Connect(*serverIp, DriverName, *serverPort))
|
||||||
|
time.Sleep(10 * time.Second)
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@@ -2,11 +2,15 @@ package models
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/tatsushid/go-fastping"
|
"github.com/tatsushid/go-fastping"
|
||||||
|
json_data "github.com/tecamino/tecamino-json_data"
|
||||||
"github.com/tecamino/tecamino-logger/logging"
|
"github.com/tecamino/tecamino-logger/logging"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -135,6 +139,11 @@ func (b *Bus) Stop() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// status bus
|
||||||
|
func (b *Bus) Status() bool {
|
||||||
|
return b.Watchdog != nil
|
||||||
|
}
|
||||||
|
|
||||||
// send dmx data
|
// send dmx data
|
||||||
func (b *Bus) SendData() error {
|
func (b *Bus) SendData() error {
|
||||||
if !b.Reachable {
|
if !b.Reachable {
|
||||||
@@ -156,6 +165,24 @@ func (b *Bus) SendData() error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Bus) ParsePayload(c *gin.Context) error {
|
||||||
|
|
||||||
|
if err := c.BindJSON(b); err != nil {
|
||||||
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("json: " + err.Error())
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if b.Name == "" {
|
||||||
|
r := json_data.NewResponse()
|
||||||
|
r.SendError("bus name missing")
|
||||||
|
c.JSON(http.StatusBadRequest, r)
|
||||||
|
return errors.New("bus name missing")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func isUDPReachable(ip string) (recieved bool, err error) {
|
func isUDPReachable(ip string) (recieved bool, err error) {
|
||||||
p := fastping.NewPinger()
|
p := fastping.NewPinger()
|
||||||
ra, err := net.ResolveIPAddr("ip4:icmp", ip)
|
ra, err := net.ResolveIPAddr("ip4:icmp", ip)
|
||||||
|
@@ -1,23 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
type JsonData struct {
|
|
||||||
Set *[]Set `json:"set,omitempty"`
|
|
||||||
Create *[]Bus `json:"create,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewRequest() *JsonData {
|
|
||||||
return &JsonData{}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *JsonData) AddSet(bus string, address uint, value uint8) {
|
|
||||||
if r.Set == nil {
|
|
||||||
r.Set = &[]Set{}
|
|
||||||
}
|
|
||||||
|
|
||||||
*r.Set = append(*r.Set, Set{
|
|
||||||
Bus: bus,
|
|
||||||
Address: address,
|
|
||||||
Value: value,
|
|
||||||
})
|
|
||||||
}
|
|
@@ -1,7 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
type JsonResponse struct {
|
|
||||||
Error bool `json:"error,omitempty"`
|
|
||||||
Message string `json:"message,omitempty"`
|
|
||||||
Data string `json:"data,omitempty"`
|
|
||||||
}
|
|
25
models/subscriptions.go
Normal file
25
models/subscriptions.go
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/google/uuid"
|
||||||
|
json_dataModels "github.com/tecamino/tecamino-json_data/models"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Subscriptions map[uuid.UUID]Subscription
|
||||||
|
|
||||||
|
type Subscription struct {
|
||||||
|
Bus string
|
||||||
|
Address uint
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSubscriptions() Subscriptions {
|
||||||
|
return make(Subscriptions)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Subscriptions) AddSubscription(uid uuid.UUID, drv *json_dataModels.Driver) {
|
||||||
|
sub := Subscription{
|
||||||
|
Bus: drv.Bus,
|
||||||
|
Address: drv.Address,
|
||||||
|
}
|
||||||
|
(*s)[uid] = sub
|
||||||
|
}
|
@@ -1,51 +1,44 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
import (
|
// type Clients map[string]Client
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"github.com/coder/websocket"
|
// type Client struct {
|
||||||
"github.com/gin-gonic/gin"
|
// Connected *bool `json:"connected"`
|
||||||
)
|
// SndConn *websocket.Conn `json:"-"` //sending connection
|
||||||
|
// RvcConn *websocket.Conn `json:"-"` // revieving connection
|
||||||
|
// }
|
||||||
|
|
||||||
type Clients map[string]Client
|
// func NewClients() Clients {
|
||||||
|
// return make(Clients)
|
||||||
|
// }
|
||||||
|
|
||||||
type Client struct {
|
// // Connect a recieving websocket connection
|
||||||
Connected *bool `json:"connected"`
|
// func (c *Clients) ConnectRecievingWsConnection(id string, ctx *gin.Context) (*websocket.Conn, error) {
|
||||||
SndConn *websocket.Conn `json:"-"` //sending connection
|
// conn, err := websocket.Accept(ctx.Writer, ctx.Request, &websocket.AcceptOptions{
|
||||||
RvcConn *websocket.Conn `json:"-"` // revieving connection
|
// OriginPatterns: []string{"*"},
|
||||||
}
|
// })
|
||||||
|
|
||||||
func NewClients() Clients {
|
// if err != nil {
|
||||||
return make(Clients)
|
// return nil, fmt.Errorf("error accept websocket client: %s", err)
|
||||||
}
|
// }
|
||||||
|
|
||||||
// Connect a recieving websocket connection
|
// b := true
|
||||||
func (c *Clients) ConnectRecievingWsConnection(id string, ctx *gin.Context) (*websocket.Conn, error) {
|
// (*c)[id] = Client{
|
||||||
conn, err := websocket.Accept(ctx.Writer, ctx.Request, &websocket.AcceptOptions{
|
// Connected: &b,
|
||||||
OriginPatterns: []string{"*"},
|
// RvcConn: conn,
|
||||||
})
|
// }
|
||||||
|
// return conn, nil
|
||||||
|
// }
|
||||||
|
|
||||||
if err != nil {
|
// func (c *Clients) RemoveClient(id string) {
|
||||||
return nil, fmt.Errorf("error accept websocket client: %s", err)
|
// delete(*c, id)
|
||||||
}
|
// }
|
||||||
|
|
||||||
b := true
|
// func (c *Clients) GetClientPointer(id string) *bool {
|
||||||
(*c)[id] = Client{
|
// return (*c)[id].Connected
|
||||||
Connected: &b,
|
// }
|
||||||
RvcConn: conn,
|
|
||||||
}
|
|
||||||
return conn, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Clients) RemoveClient(id string) {
|
// func (c *Clients) DisconnectRecievingWsConnection(id string, code websocket.StatusCode, reason string) {
|
||||||
delete(*c, id)
|
// *(*c)[id].Connected = false
|
||||||
}
|
// (*c)[id].RvcConn.Close(code, reason)
|
||||||
|
// }
|
||||||
func (c *Clients) GetClientPointer(id string) *bool {
|
|
||||||
return (*c)[id].Connected
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Clients) DisconnectRecievingWsConnection(id string, code websocket.StatusCode, reason string) {
|
|
||||||
*(*c)[id].Connected = false
|
|
||||||
(*c)[id].RvcConn.Close(code, reason)
|
|
||||||
}
|
|
||||||
|
@@ -1,9 +1,8 @@
|
|||||||
package server
|
package server
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"artNet/cert"
|
"artNet/cert"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
@@ -12,6 +11,17 @@ type Server struct {
|
|||||||
Routes *gin.Engine
|
Routes *gin.Engine
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Send struct {
|
||||||
|
Set []Set `json:"subscribe"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Set struct {
|
||||||
|
Path string `json:"path"`
|
||||||
|
OnChange bool `json:"onChange"`
|
||||||
|
Depth int `json:"depth"`
|
||||||
|
Driver string `json:"driver"`
|
||||||
|
}
|
||||||
|
|
||||||
// Initialize new websocket server
|
// Initialize new websocket server
|
||||||
func NewServer() *Server {
|
func NewServer() *Server {
|
||||||
return &Server{
|
return &Server{
|
||||||
|
Reference in New Issue
Block a user