7 Commits

Author SHA1 Message Date
Adrian Zürcher
c85590e5a5 updated websocket as client broker 2025-06-19 18:32:47 +02:00
Adrian Zürcher
b5d9d96c52 add resubscribe (in progress)
fix crash when no bus found in map
2025-05-28 22:00:04 +02:00
Adrian Zürcher
dc64d08e9d new son_data model 2025-05-05 07:32:26 +02:00
Adrian Zürcher
a92995d766 removie unused websocket 2025-04-29 17:16:36 +02:00
Adrian Zürcher
348a6f7622 move client context to correct place 2025-04-29 17:15:20 +02:00
Adrian Zürcher
83cbcdaa42 updated 2025-04-29 09:16:27 +02:00
Adrian Zürcher
2839b615c3 implement new json_data model 2025-04-29 08:34:51 +02:00
17 changed files with 599 additions and 243 deletions

48
.github/workflows/build.yml vendored Normal file
View File

@@ -0,0 +1,48 @@
name: Build Go Binaries
on:
push:
branches: [ main ]
pull_request:
jobs:
build:
runs-on: ubuntu-latest
strategy:
matrix:
goos: [linux, windows]
goarch: [amd64, arm, arm64]
exclude:
- goos: windows
goarch: arm
- goos: windows
goarch: arm64
steps:
- uses: actions/checkout@v3
- name: Set up Go
uses: actions/setup-go@v4
with:
go-version: '1.24.0'
- name: Set up Git credentials for private modules
run: |
git config --global url."https://${{ secrets.GH_PAT }}@github.com/".insteadOf "https://github.com/"
echo "GOPRIVATE=github.com/tecamino/*" >> $GITHUB_ENV
- name: Build binary
run: |
mkdir -p build
if [ "${{ matrix.goos }}" == "windows" ]; then
GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }} go build -o build/tecamino-driver-artNet-${{ matrix.goos }}-${{ matrix.goarch }}.exe main.go
else
GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }} go build -o build/tecamino-driver-artNet-${{ matrix.goos }}-${{ matrix.goarch }} main.go
fi
- name: Upload artifacts
uses: actions/upload-artifact@v4
with:
name: binaries-${{ matrix.goos }}-${{ matrix.goarch }}
path: build/

3
.gitignore vendored
View File

@@ -1,2 +1,3 @@
*.cfg *.cfg
*.log *.log
artNetDriver-arm64

View File

@@ -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
}

View File

@@ -3,19 +3,21 @@ package driver
import ( import (
"artNet/cfg" "artNet/cfg"
"artNet/models" "artNet/models"
serverModels "artNet/server/models" ws "artNet/websocket"
"fmt" "fmt"
"path" "path"
"time"
json_data "github.com/tecamino/tecamino-json_data"
"github.com/tecamino/tecamino-logger/logging" "github.com/tecamino/tecamino-logger/logging"
) )
type ArtNetDriver struct { 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:"-"` Subscriptions models.Subscriptions `yaml:"-" json:"-"`
Log *logging.Logger `yaml:"-"` Log *logging.Logger `yaml:"-" json:"-"`
} }
// initialize new Art-Net driver // initialize new Art-Net driver
@@ -39,18 +41,16 @@ func NewDriver(cfgDir, name string, debug bool) (*ArtNetDriver, error) {
logger.Debug("artNet.NewDriver", "initialize "+name+" driver") logger.Debug("artNet.NewDriver", "initialize "+name+" driver")
d := ArtNetDriver{ d := ArtNetDriver{
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,
} }
if err := d.LoadCfg(); err != nil { if err := d.LoadCfg(); err != nil {
logger.Error("artNet.NewDriver.LoadCfg", "error load driver configuration: "+err.Error()) logger.Error("artNet.NewDriver.LoadCfg", "error load driver configuration: "+err.Error())
return nil, err return nil, err
} }
return &d, nil return &d, nil
} }
@@ -61,8 +61,8 @@ func (d *ArtNetDriver) LoadCfg() error {
return err return err
} }
for _, b := range d.Buses { for name, b := range d.Buses {
d.NewBus(b.Name, b.Ip, *b.Port) d.NewBus(name, b.Ip, *b.Port)
} }
return nil return nil
} }
@@ -70,16 +70,87 @@ 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) if d.Buses[bus].Data == nil {
return fmt.Errorf("no dmx data on bus '%s' found", bus)
return d.Buses[set.Bus].SendData() }
d.Buses[bus].Data.SetValue(address, value)
return d.Buses[bus].SendData()
}
// connect to websocket server and listen to subscriptions
// ip: ip address of server
// id: id of driver
// port: port of server
func (d *ArtNetDriver) Connect(ip, id string, port uint) error {
var err error
client, err := ws.NewClient(ip, id, port)
if err != nil {
return err
}
client.OnError = func(err error) {
d.Log.Error("websocket connection", err)
}
client.OnMessage = func(data []byte) {
//fmt.Println(100, string(data))
fmt.Println(100, string(data))
}
client.Connect(5)
req := json_data.NewRequest()
req.AddDriverSubscription(".*", id, 0, true, false, false)
if err := client.SendData(req); err != nil {
d.Log.Error("websocket send data", err)
}
for {
time.Sleep(1)
}
return nil
// d.Conn = websocket.NewClient()
// 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
// }
// Subscribe to websocket server
// 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
// }
// 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())
// }
// }
// }
// }
} }

View File

@@ -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,170 @@ 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.SetError()
r.SetMessage(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.SetError()
} else if addr := net.ParseIP(payload.Ip); addr == nil { r.SetMessage("wrong ip '" + busPayload.Ip + "'")
c.JSON(http.StatusBadRequest, gin.H{"error": "wrong ip '" + payload.Ip + "'"}) c.JSON(http.StatusBadRequest, r)
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.SetError()
r.SetMessage("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.SetMessage(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.SetError()
r.SetMessage("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.SetMessage("bus " + busPayload.Name + " not found")
} c.JSON(http.StatusBadRequest, 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.SetMessage(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.SetError()
r.SetMessage("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.SetMessage(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)
} }
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.SetError()
r.SetMessage("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.SetMessage(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) func (d *ArtNetDriver) Resubscribe(c *gin.Context) {
_, err := auth.GetIDFromQuery(c)
if err != nil {
r := json_data.NewResponse()
r.SetError()
r.SetMessage("id: " + err.Error())
c.JSON(http.StatusBadRequest, r)
return
}
busPayload := models.Bus{}
if err := busPayload.ParsePayload(c); err != nil {
return
}
if _, ok := d.Buses[busPayload.Name]; !ok {
r := json_dataModels.NewResponse()
r.SetMessage("bus " + busPayload.Name + " not found")
c.JSON(http.StatusBadRequest, r)
return
}
if busPayload.Resubscribe == nil {
r := json_dataModels.NewResponse()
r.SetMessage("no resubscriptions in request")
c.JSON(http.StatusBadRequest, r)
return
}
d.Subscribe(*busPayload.Resubscribe...)
}
func (d *ArtNetDriver) Status(c *gin.Context) {
_, err := auth.GetIDFromQuery(c)
if err != nil {
r := json_data.NewResponse()
r.SetError()
r.SetMessage("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.SetMessage(fmt.Sprintf("bus '%s' %s", busPayload.Name, state))
c.JSON(http.StatusOK, r)
} }

20
driver/subscribe.go Normal file
View 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.Subscription) {
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)))
}
}
}

View File

@@ -1,63 +0,0 @@
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)
}
}
}
}

4
go.mod
View File

@@ -5,9 +5,11 @@ go 1.23.0
toolchain go1.23.8 toolchain go1.23.8
require ( require (
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/gorilla/websocket v1.5.3
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.13
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
) )

8
go.sum
View File

@@ -6,8 +6,6 @@ github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/
github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg= github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
github.com/coder/websocket v1.8.13 h1:f3QZdXy7uGVz+4uCJy2nTZyM0yTBj8yANEHhqlXZ9FE=
github.com/coder/websocket v1.8.13/go.mod h1:LNVeNrXQZfe5qhS9ALED3uA+l5pPqvwXg3CKoDBB2gs=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@@ -30,6 +28,10 @@ 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/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
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.13 h1:hugbmCgXXh0F7YQAEbbJYHkSdq1caejD7SajDiMs42I=
github.com/tecamino/tecamino-json_data v0.0.13/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=

38
main.go
View File

@@ -6,6 +6,7 @@ import (
"flag" "flag"
"fmt" "fmt"
"os" "os"
"time"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
@@ -14,10 +15,13 @@ 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")
start := flag.Bool("start", false, "starts all buses on startup")
flag.Parse() flag.Parse()
//change working directory only if value is given //change working directory only if value is given
@@ -38,21 +42,41 @@ 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.POST("/buses/resubscribe", artNetDriver.Resubscribe)
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)
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))
if *start {
for busName, bus := range artNetDriver.Buses {
artNetDriver.Log.Info("main", fmt.Sprintf("starting bus %s", busName))
// start bus and listen to subscriptions
bus.Start(artNetDriver.Log)
}
} }
// connect to server
for {
if err := artNetDriver.Connect(*serverIp, DriverName, *serverPort); err != nil {
artNetDriver.Log.Error("main", err)
}
fmt.Println(555)
time.Sleep(10 * time.Second)
}
} }

View File

@@ -2,11 +2,16 @@ 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"
json_dataModels "github.com/tecamino/tecamino-json_data/models"
"github.com/tecamino/tecamino-logger/logging" "github.com/tecamino/tecamino-logger/logging"
) )
@@ -17,12 +22,13 @@ const (
// Art-Net Interface // Art-Net Interface
type Bus struct { type Bus struct {
Name string `yaml:"name" json:"name"` Name string `yaml:"name" json:"name"`
Ip string `yaml:"ip" json:"ip"` Ip string `yaml:"ip" json:"ip"`
Port *int `yaml:"port" json:"port,omitempty"` Port *int `yaml:"port" json:"port,omitempty"`
Data *DMX `yaml:"-" json:"-"` Data *DMX `yaml:"-" json:"-"`
Watchdog context.CancelFunc `yaml:"-" json:"-"` Resubscribe *[]json_dataModels.Subscription `yaml:"-" json:"resubscribe"`
Reachable bool `yaml:"-" json:"-"` Watchdog context.CancelFunc `yaml:"-" json:"-"`
Reachable bool `yaml:"-" json:"-"`
} }
// adds new Art-Net interface to driver port 0 = 6454 (default art-net) // adds new Art-Net interface to driver port 0 = 6454 (default art-net)
@@ -135,6 +141,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 +167,26 @@ 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.SetError()
r.SetMessage("json: " + err.Error())
c.JSON(http.StatusBadRequest, r)
return err
}
if b.Name == "" {
r := json_data.NewResponse()
r.SetError()
r.SetMessage("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)

View File

@@ -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,
})
}

View File

@@ -1,7 +0,0 @@
package models
type JsonResponse struct {
Error bool `json:"error,omitempty"`
Message string `json:"message,omitempty"`
Data string `json:"data,omitempty"`
}

24
models/subscriptions.go Normal file
View File

@@ -0,0 +1,24 @@
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) {
(*s)[uid] = Subscription{
Bus: drv.Bus,
Address: drv.Address,
}
}

View File

@@ -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)
}

View File

@@ -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{

165
websocket/client.go Normal file
View File

@@ -0,0 +1,165 @@
package websocket
import (
"encoding/json"
"fmt"
"log"
"sync"
"time"
"github.com/gorilla/websocket"
json_dataModels "github.com/tecamino/tecamino-json_data/models"
)
type Client struct {
conn *websocket.Conn
writeMu sync.Mutex
OnMessage func(data []byte)
OnOpen func()
OnClose func(code int, reason string)
OnError func(err error)
OnPing func()
OnPong func()
timeout time.Duration
}
// Connect to websocket server
// ip: ip address of server
func NewClient(ip, id string, port uint) (*Client, error) {
url := fmt.Sprintf("ws://%s:%d/ws?id=%s", ip, port, id)
c := &Client{}
dialer := websocket.DefaultDialer
conn, resp, err := dialer.Dial(url, nil)
if err != nil {
if c.OnError != nil {
c.OnError(err)
}
return nil, fmt.Errorf("dial error %v (status %v)", err, resp)
}
c.conn = conn
// Setup control handlers
c.conn.SetPingHandler(func(appData string) error {
if c.OnPing != nil {
c.OnPing()
}
return c.conn.WriteMessage(websocket.PongMessage, nil)
})
c.conn.SetPongHandler(func(appData string) error {
c.conn.SetReadDeadline(time.Now().Add(c.timeout))
if c.OnPong != nil {
c.OnPong()
}
return nil
})
c.conn.SetCloseHandler(func(code int, text string) error {
if c.OnClose != nil {
c.OnClose(code, text)
}
return nil
})
if c.OnOpen != nil {
c.OnOpen()
}
return c, nil
}
func (c *Client) Connect(timeout uint) {
if timeout > 0 {
fmt.Println(1234, timeout)
c.timeout = time.Duration(timeout) * time.Second
}
go c.pingLoop()
c.conn.SetReadDeadline(time.Now().Add(c.timeout))
go func() {
for {
msgType, msg, err := c.conn.ReadMessage()
if err != nil {
if websocket.IsCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
log.Println("WebSocket closed:", err)
}
if c.OnError != nil {
c.OnError(fmt.Errorf("read error: %w", err))
}
return
}
switch msgType {
case websocket.TextMessage, websocket.BinaryMessage:
if c.OnMessage != nil {
c.OnMessage(msg)
}
default:
log.Printf("Unhandled message type: %d", msgType)
}
}
}()
}
func (c *Client) pingLoop() {
interval := c.timeout / 2
if interval <= 0 {
interval = 5 * time.Second
}
ticker := time.NewTicker(interval)
defer ticker.Stop()
for range ticker.C {
if err := c.Write(websocket.PingMessage, nil); err != nil {
if c.OnError != nil {
c.OnError(fmt.Errorf("ping error: %w", err))
}
return
}
if c.OnPing != nil {
c.OnPing()
}
}
}
func (c *Client) Write(msgType int, data []byte) error {
c.writeMu.Lock()
defer c.writeMu.Unlock()
c.conn.SetWriteDeadline(time.Now().Add(c.timeout))
if err := c.conn.WriteMessage(msgType, data); err != nil {
if c.OnError != nil {
c.OnError(err)
}
return err
}
return nil
}
// Close connection to websocket server
func (c *Client) Close(code int, reason string) {
if c.conn != nil {
if c.OnClose != nil {
c.OnClose(code, reason)
}
c.conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(code, reason))
c.conn.Close()
}
}
func (c *Client) SendData(data any) error {
c.conn.SetWriteDeadline(time.Now().Add(c.timeout))
if err := c.conn.WriteJSON(data); err != nil {
if c.OnError != nil {
c.OnError(err)
}
return err
}
return nil
}
func (c *Client) ReadJsonData(data []byte) (json_dataModels.Response, error) {
var resp json_dataModels.Response
err := json.Unmarshal(data, &resp)
return resp, err
}