Compare commits
13 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
dc64d08e9d | ||
![]() |
a92995d766 | ||
![]() |
348a6f7622 | ||
![]() |
83cbcdaa42 | ||
![]() |
2839b615c3 | ||
![]() |
f5e66af3d8 | ||
![]() |
274e6acf0e | ||
![]() |
c21d9b0dd0 | ||
![]() |
7b201f6e63 | ||
![]() |
b777716aa7 | ||
![]() |
2273a6140f | ||
![]() |
11c8340be5 | ||
![]() |
56188a6ad0 |
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
*.cfg
|
||||
*.log
|
47
README.md
47
README.md
@@ -1,2 +1,45 @@
|
||||
# ArtNet Driver
|
||||
Tecamino artNet driver DMX-Communication
|
||||
# 🎛️ ArtNet Driver
|
||||
|
||||
**Tecamino ArtNet Driver** — A lightweight driver for DMX communication using the ArtNet protocol over TCP/IP.
|
||||
|
||||
This tool allows you to send DMX values to an ArtNet-compatible device, enabling remote lighting control and stage automation.
|
||||
|
||||
---
|
||||
|
||||
## 🚀 Features
|
||||
|
||||
- 🔌 Sends DMX data over the ArtNet protocol (UDP)
|
||||
- 📁 Configurable working and config directories
|
||||
- 🐞 Optional debug logging for development and troubleshooting
|
||||
- ⚙️ Easy to run with CLI flags
|
||||
|
||||
---
|
||||
|
||||
## 🧰 Command Line Arguments
|
||||
|
||||
| Flag | Default | Description |
|
||||
|-------------------|-------------|--------------------------------------|
|
||||
| `--port` | `8110` | Port on which the server listens |
|
||||
| `--cfg` | `./cfg` | Path to the configuration directory |
|
||||
| `--workingDirectory` | `"."` | Working directory for runtime files |
|
||||
| `--debug` | `false` | Enable verbose debug logging |
|
||||
|
||||
---
|
||||
|
||||
## 🏁 Getting Started
|
||||
|
||||
### 🔧 Run the Driver
|
||||
|
||||
```bash
|
||||
go run main.go --port=8110 --cfg=./cfg --workingDirectory=. --debug=true
|
||||
|
||||
```
|
||||
### 🔧 Build the App
|
||||
|
||||
To compile the Go code into a runnable binary:
|
||||
|
||||
#### On Linux or macOS:
|
||||
|
||||
```bash
|
||||
go build -o artnet-driver
|
||||
```
|
23
auth/auth.go
Normal file
23
auth/auth.go
Normal file
@@ -0,0 +1,23 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func GetIDFromAuth(c *gin.Context) (string, error) {
|
||||
auth := c.GetHeader("Authorization")
|
||||
if len(auth) > 7 && auth[:7] == "Bearer " {
|
||||
return auth[7:], nil
|
||||
}
|
||||
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
|
||||
}
|
62
cfg/cfg.go
Normal file
62
cfg/cfg.go
Normal file
@@ -0,0 +1,62 @@
|
||||
package cfg
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
type Cfg struct {
|
||||
Name string
|
||||
Dir string
|
||||
}
|
||||
|
||||
// Initializes new config handler
|
||||
// dir is config directory from working dir on
|
||||
// name is filename, file extention is automatically added '.cfg'
|
||||
// file is in yaml format
|
||||
// if file exists it will read its content in to the driver
|
||||
func NewCfgHandler(dir, name string) *Cfg {
|
||||
return &Cfg{
|
||||
Dir: dir,
|
||||
Name: name,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Cfg) SaveCfg(data any) error {
|
||||
if _, err := os.Stat(c.Dir); err != nil {
|
||||
if err := os.MkdirAll(c.Dir, 0666); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
f, err := os.Create(path.Join(c.Dir, c.Name) + ".cfg")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
encoder := yaml.NewEncoder(f)
|
||||
encoder.SetIndent(2)
|
||||
|
||||
if err := encoder.Encode(data); err != nil {
|
||||
return err
|
||||
}
|
||||
encoder.Close()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Cfg) LoadCfg(data any) error {
|
||||
p := path.Join(c.Dir, c.Name) + ".cfg"
|
||||
if _, err := os.Stat(p); err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
b, err := os.ReadFile(p)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return yaml.Unmarshal(b, data)
|
||||
}
|
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
|
||||
}
|
118
driver/artNet.go
118
driver/artNet.go
@@ -1,29 +1,111 @@
|
||||
package driver
|
||||
|
||||
import "artNet/models"
|
||||
import (
|
||||
"artNet/cfg"
|
||||
"artNet/client"
|
||||
"artNet/models"
|
||||
"fmt"
|
||||
"path"
|
||||
|
||||
"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:"-"`
|
||||
Conn *client.Client `yaml:"-" json:"-"`
|
||||
Subscriptions models.Subscriptions `yaml:"-" json:"-"`
|
||||
Log *logging.Logger `yaml:"-" json:"-"`
|
||||
}
|
||||
|
||||
// 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(path.Join(cfgDir, name+".log"), &logging.Config{
|
||||
MaxSize: 1,
|
||||
MaxBackup: 3,
|
||||
MaxAge: 28,
|
||||
Debug: debug,
|
||||
TerminalOut: true,
|
||||
})
|
||||
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),
|
||||
Log: logger,
|
||||
}
|
||||
|
||||
if err := d.LoadCfg(); err != nil {
|
||||
logger.Error("artNet.NewDriver.LoadCfg", "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(bus string, address uint, value uint8) error {
|
||||
if _, ok := d.Buses[bus]; !ok {
|
||||
return fmt.Errorf("no bus '%s' found", bus)
|
||||
}
|
||||
d.Buses[bus].Data.SetValue(address, value)
|
||||
return d.Buses[bus].SendData()
|
||||
}
|
||||
|
||||
func (d *ArtNetDriver) Connect(ip, id string, port uint) error {
|
||||
d.Conn = client.NewClient(d.Log)
|
||||
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())
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
165
driver/bus.go
Normal file
165
driver/bus.go
Normal file
@@ -0,0 +1,165 @@
|
||||
package driver
|
||||
|
||||
import (
|
||||
"artNet/auth"
|
||||
"artNet/models"
|
||||
"fmt"
|
||||
"net"
|
||||
"net/http"
|
||||
|
||||
"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
|
||||
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.GetIDFromQuery(c)
|
||||
if err != nil {
|
||||
r := json_data.NewResponse()
|
||||
r.SetError()
|
||||
r.SetMessage(err.Error())
|
||||
c.JSON(http.StatusBadRequest, r)
|
||||
return
|
||||
}
|
||||
|
||||
busPayload := models.Bus{}
|
||||
if err := busPayload.ParsePayload(c); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if addr := net.ParseIP(busPayload.Ip); addr == nil {
|
||||
r := json_data.NewResponse()
|
||||
r.SetError()
|
||||
r.SetMessage("wrong ip '" + busPayload.Ip + "'")
|
||||
c.JSON(http.StatusBadRequest, r)
|
||||
return
|
||||
}
|
||||
|
||||
if _, ok := d.Buses[busPayload.Name]; ok {
|
||||
r := json_data.NewResponse()
|
||||
r.SetError()
|
||||
r.SetMessage("bus " + busPayload.Name + " exists already")
|
||||
c.JSON(http.StatusOK, r)
|
||||
return
|
||||
}
|
||||
|
||||
bus := d.NewBus(busPayload.Name, busPayload.Ip, busPayload.GetPort())
|
||||
|
||||
r := json_data.NewResponse()
|
||||
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)
|
||||
}
|
||||
|
||||
func (d *ArtNetDriver) RemoveBus(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.StatusOK, r)
|
||||
return
|
||||
} else {
|
||||
delete(d.Buses, busPayload.Name)
|
||||
}
|
||||
r := json_dataModels.NewResponse()
|
||||
r.SetMessage(fmt.Sprintf("bus '%s' successfully removed", busPayload.Name))
|
||||
c.JSON(http.StatusOK, r)
|
||||
d.cfgHandler.SaveCfg(*d)
|
||||
}
|
||||
|
||||
func (d *ArtNetDriver) Start(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 busPayload.ParsePayload(c); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
d.Buses[busPayload.Name].Start(d.Log)
|
||||
|
||||
r := json_dataModels.NewResponse()
|
||||
r.SetMessage(fmt.Sprintf("bus '%s' running", busPayload.Name))
|
||||
c.JSON(http.StatusOK, r)
|
||||
d.cfgHandler.SaveCfg(*d)
|
||||
}
|
||||
|
||||
func (d *ArtNetDriver) Stop(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
|
||||
}
|
||||
|
||||
d.Buses[busPayload.Name].Stop()
|
||||
|
||||
r := json_dataModels.NewResponse()
|
||||
r.SetMessage(fmt.Sprintf("bus '%s' stopped", busPayload.Name))
|
||||
c.JSON(http.StatusOK, r)
|
||||
d.cfgHandler.SaveCfg(*d)
|
||||
}
|
||||
|
||||
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
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.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)))
|
||||
}
|
||||
}
|
||||
}
|
7
go.mod
7
go.mod
@@ -7,8 +7,11 @@ toolchain go1.23.8
|
||||
require (
|
||||
github.com/coder/websocket v1.8.13
|
||||
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/tecamino/tecamino-logger v0.1.1
|
||||
github.com/tecamino/tecamino-json_data v0.0.12
|
||||
github.com/tecamino/tecamino-logger v0.2.0
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
)
|
||||
|
||||
require (
|
||||
@@ -39,5 +42,5 @@ require (
|
||||
golang.org/x/sys v0.20.0 // indirect
|
||||
golang.org/x/text v0.15.0 // indirect
|
||||
google.golang.org/protobuf v1.34.1 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
|
||||
)
|
||||
|
10
go.sum
10
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/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||
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/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||
@@ -63,8 +65,10 @@ 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/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/tecamino/tecamino-logger v0.1.1 h1:bK0SEQpbjui42OsnM6JCDThaGMBoqwVj/Op6ESqDwV4=
|
||||
github.com/tecamino/tecamino-logger v0.1.1/go.mod h1:sGysmiFGIdr4vLJRAI+fJgsa7EoRRuxvRrKW7GnGQkw=
|
||||
github.com/tecamino/tecamino-json_data v0.0.12 h1:S4Y+WcfQNrin7P73ZI+4eJWh62IwJVhriRsPGGM8N34=
|
||||
github.com/tecamino/tecamino-json_data v0.0.12/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/go.mod h1:0M1E9Uei/qw3e3WA1x3lBo1eP3H5oeYE7GjYrMahnj8=
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||
github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
|
||||
@@ -94,6 +98,8 @@ google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFW
|
||||
google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
|
||||
gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
103
main.go
103
main.go
@@ -3,71 +3,66 @@ package main
|
||||
import (
|
||||
"artNet/driver"
|
||||
"artNet/server"
|
||||
"flag"
|
||||
"fmt"
|
||||
"math"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
const DriverName string = "ArtNetDriver"
|
||||
|
||||
func main() {
|
||||
// cli flags
|
||||
cfgDir := flag.String("cfg", "./cfg", "config 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")
|
||||
flag.Parse()
|
||||
|
||||
bus := driver.NewDriver().NewInterface("2.0.0.1", 0)
|
||||
d, _ := bus.AddDevice(47, 9)
|
||||
|
||||
var i uint8
|
||||
|
||||
s := server.NewServer()
|
||||
|
||||
s.ServeHttp(8120)
|
||||
|
||||
for {
|
||||
if i == math.MaxUint8 {
|
||||
i = 0
|
||||
}
|
||||
err := d.SetChannelValue(0, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(1, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(2, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(3, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(4, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(5, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(6, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(7, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = d.SetChannelValue(8, i)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
//change working directory only if value is given
|
||||
if *workingDir != "." && *workingDir != "" {
|
||||
os.Chdir(*workingDir)
|
||||
}
|
||||
|
||||
if err := bus.SendData(); err != nil {
|
||||
fmt.Println(12, err)
|
||||
//initialize new ArtNet driver
|
||||
artNetDriver, err := driver.NewDriver(*cfgDir, DriverName, *debug)
|
||||
if err != nil {
|
||||
artNetDriver.Log.Error("main", "error initialize new artnet driver "+err.Error())
|
||||
panic(err)
|
||||
}
|
||||
time.Sleep(100 * time.Microsecond)
|
||||
i += 1
|
||||
i = 0
|
||||
|
||||
//initialize new server
|
||||
artNetDriver.Log.Debug("main", "initialize new server instance")
|
||||
s := server.NewServer()
|
||||
|
||||
//set routes
|
||||
artNetDriver.Log.Debug("main", "setting routes")
|
||||
s.Routes.GET("/buses/all", artNetDriver.GetAllBuses)
|
||||
s.Routes.POST("/buses/create", artNetDriver.CreateBus)
|
||||
s.Routes.POST("/buses/remove", artNetDriver.RemoveBus)
|
||||
s.Routes.POST("/buses/start", artNetDriver.Start)
|
||||
s.Routes.POST("/buses/status", artNetDriver.Status)
|
||||
s.Routes.POST("/buses/stop", artNetDriver.Stop)
|
||||
|
||||
s.Routes.GET("/", func(c *gin.Context) {
|
||||
c.String(200, "ArtNet Driver WebSocket Server is running!")
|
||||
})
|
||||
|
||||
go func() {
|
||||
if err := s.ServeHttp(*wsPort); err != nil {
|
||||
artNetDriver.Log.Error("main", err.Error())
|
||||
}
|
||||
}()
|
||||
|
||||
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)
|
||||
}
|
||||
|
||||
}
|
||||
|
141
models/bus.go
141
models/bus.go
@@ -1,11 +1,17 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/tatsushid/go-fastping"
|
||||
json_data "github.com/tecamino/tecamino-json_data"
|
||||
"github.com/tecamino/tecamino-logger/logging"
|
||||
)
|
||||
|
||||
// Art-Net constants
|
||||
@@ -15,43 +21,47 @@ const (
|
||||
|
||||
// Art-Net Interface
|
||||
type Bus struct {
|
||||
ip string `yaml:"ip"`
|
||||
port int `yaml:"port"`
|
||||
Devices []*Device `yaml:"devices"`
|
||||
Data *DMX
|
||||
Name string `yaml:"name" json:"name"`
|
||||
Ip string `yaml:"ip" json:"ip"`
|
||||
Port *int `yaml:"port" json:"port,omitempty"`
|
||||
Data *DMX `yaml:"-" json:"-"`
|
||||
Watchdog context.CancelFunc `yaml:"-" json:"-"`
|
||||
Reachable bool `yaml:"-" json:"-"`
|
||||
}
|
||||
|
||||
// adds new Art-Net interface to driver port 0 = 6454 (default art-net)
|
||||
func NewBus(ip string, port int) *Bus {
|
||||
func NewBus(name, ip string, port int) *Bus {
|
||||
if port == 0 {
|
||||
port = artPort
|
||||
}
|
||||
|
||||
i := Bus{
|
||||
ip: ip,
|
||||
port: port,
|
||||
Name: name,
|
||||
Ip: ip,
|
||||
Port: &port,
|
||||
Data: NewDMXUniverse(),
|
||||
}
|
||||
return &i
|
||||
}
|
||||
|
||||
// adds new dmx device to interface
|
||||
func (i *Bus) AddDevice(address uint, channels uint) (*Device, error) {
|
||||
d := NewDevice(address, channels, i.Data)
|
||||
i.Devices = append(i.Devices, d)
|
||||
return d, nil
|
||||
// get bus port from pointer
|
||||
func (b *Bus) GetPort() int {
|
||||
if b.Port == nil {
|
||||
return artPort
|
||||
}
|
||||
return *b.Port
|
||||
}
|
||||
|
||||
// start polling dmx data in milliseconds 0 = aprox. 44Hertz
|
||||
func (i *Bus) Poll(interval time.Duration) error {
|
||||
func (b *Bus) Poll(interval time.Duration) error {
|
||||
if interval == 0 {
|
||||
interval = 23
|
||||
}
|
||||
|
||||
// Send packet over UDP
|
||||
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
||||
IP: net.ParseIP(i.ip),
|
||||
Port: i.port,
|
||||
IP: net.ParseIP(b.Ip),
|
||||
Port: *b.Port,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
@@ -63,7 +73,7 @@ func (i *Bus) Poll(interval time.Duration) error {
|
||||
for {
|
||||
go func() {
|
||||
for {
|
||||
if reached, _ := isUDPReachable(i.ip); !reached {
|
||||
if reached, _ := isUDPReachable(b.Ip); !reached {
|
||||
if errCount > 20 {
|
||||
break
|
||||
} else {
|
||||
@@ -77,7 +87,7 @@ func (i *Bus) Poll(interval time.Duration) error {
|
||||
}
|
||||
}()
|
||||
|
||||
_, err = conn.Write(NewArtNetPackage(i.Data))
|
||||
_, err = conn.Write(NewArtNetPackage(b.Data))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -89,12 +99,60 @@ func (i *Bus) Poll(interval time.Duration) error {
|
||||
}
|
||||
}
|
||||
|
||||
// start polling dmx data in milliseconds 0 = aprox. 44Hertz
|
||||
func (i *Bus) SendData() error {
|
||||
// start bus
|
||||
func (b *Bus) Start(log *logging.Logger) {
|
||||
var ctx context.Context
|
||||
ctx, b.Watchdog = context.WithCancel(context.Background())
|
||||
|
||||
go func() {
|
||||
var interval time.Duration = 10 * time.Second
|
||||
log.Info("bus.Start", fmt.Sprintf("device:%s ip:%s watchdog stopped", b.Name, b.Ip))
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
log.Info("bus.Start", fmt.Sprintf("device:%s ip:%s watchdog stopped", b.Name, b.Ip))
|
||||
b.Reachable = false
|
||||
return
|
||||
default:
|
||||
b.Reachable = false
|
||||
if reached, err := isUDPReachable(b.Ip); err != nil {
|
||||
log.Error("bus.Start", err)
|
||||
interval = 5 * time.Second
|
||||
} else if !reached {
|
||||
log.Error("bus.Start", fmt.Sprintf("device:%s ip:%s not reached", b.Name, b.Ip))
|
||||
interval = 5 * time.Second
|
||||
} else {
|
||||
b.Reachable = true
|
||||
log.Info("bus.Start", fmt.Sprintf("device:%s ip:%s watchdog running", b.Name, b.Ip))
|
||||
interval = 30 * time.Second
|
||||
}
|
||||
time.Sleep(interval)
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// stop bus
|
||||
func (b *Bus) Stop() {
|
||||
if b.Watchdog != nil {
|
||||
b.Watchdog()
|
||||
}
|
||||
}
|
||||
|
||||
// status bus
|
||||
func (b *Bus) Status() bool {
|
||||
return b.Watchdog != nil
|
||||
}
|
||||
|
||||
// send dmx data
|
||||
func (b *Bus) SendData() error {
|
||||
if !b.Reachable {
|
||||
return nil
|
||||
}
|
||||
// Send packet over UDP
|
||||
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
||||
IP: net.ParseIP(i.ip),
|
||||
Port: i.port,
|
||||
IP: net.ParseIP(b.Ip),
|
||||
Port: *b.Port,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
@@ -102,28 +160,30 @@ func (i *Bus) SendData() error {
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
errChan := make(chan error)
|
||||
go func() {
|
||||
if reached, _ := isUDPReachable(i.ip); !reached {
|
||||
errChan <- fmt.Errorf("device not reachable")
|
||||
return
|
||||
}
|
||||
errChan <- nil
|
||||
}()
|
||||
|
||||
err = <-errChan
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = conn.Write(NewArtNetPackage(i.Data))
|
||||
_, err = conn.Write(NewArtNetPackage(b.Data))
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
const (
|
||||
protocolICMP = 1
|
||||
)
|
||||
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) {
|
||||
p := fastping.NewPinger()
|
||||
@@ -132,10 +192,11 @@ func isUDPReachable(ip string) (recieved bool, err error) {
|
||||
return
|
||||
}
|
||||
p.AddIPAddr(ra)
|
||||
|
||||
p.OnRecv = func(addr *net.IPAddr, rtt time.Duration) {
|
||||
recieved = true
|
||||
return
|
||||
}
|
||||
|
||||
p.OnIdle = func() {}
|
||||
|
||||
err = p.Run()
|
||||
|
@@ -1,25 +0,0 @@
|
||||
package models
|
||||
|
||||
import "fmt"
|
||||
|
||||
type Device struct {
|
||||
startAddress uint
|
||||
length uint
|
||||
channels *DMX
|
||||
}
|
||||
|
||||
func NewDevice(startAddress uint, channels uint, dmx *DMX) *Device {
|
||||
return &Device{
|
||||
startAddress: startAddress,
|
||||
length: channels,
|
||||
channels: dmx,
|
||||
}
|
||||
}
|
||||
|
||||
func (d *Device) SetChannelValue(channel uint, value uint8) error {
|
||||
if d.length == channel {
|
||||
return fmt.Errorf("channel out of range %d", channel)
|
||||
}
|
||||
d.channels.SetValue(d.startAddress+channel, value)
|
||||
return nil
|
||||
}
|
@@ -1,5 +0,0 @@
|
||||
package models
|
||||
|
||||
type Get struct {
|
||||
Bus uint `json:"bus"`
|
||||
}
|
@@ -1,22 +0,0 @@
|
||||
package models
|
||||
|
||||
type JsonData struct {
|
||||
Set *[]Set `json:"set,omitempty"`
|
||||
}
|
||||
|
||||
func NewRequest() *JsonData {
|
||||
return &JsonData{}
|
||||
|
||||
}
|
||||
|
||||
func (r *JsonData) AddSet(bus, 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 +1,7 @@
|
||||
package models
|
||||
|
||||
type Set struct {
|
||||
Bus uint `json:"bus"`
|
||||
Bus string `json:"bus"`
|
||||
Address uint `json:"address"`
|
||||
Value uint8 `json:"value"`
|
||||
}
|
||||
|
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,18 +0,0 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func (s *Server) AllBuses(c *gin.Context) {
|
||||
var data any
|
||||
if s.Driver.Bus == nil {
|
||||
data = "no buses avaiable"
|
||||
} else {
|
||||
data = s.Driver.Bus
|
||||
}
|
||||
|
||||
c.JSON(200, gin.H{
|
||||
"buses": data,
|
||||
})
|
||||
}
|
@@ -1,25 +0,0 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"artNet/models"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func (s *Server) JsonRequest(c *gin.Context) {
|
||||
var payload models.JsonData
|
||||
|
||||
if err := c.BindJSON(&payload); err != nil {
|
||||
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(200, gin.H{
|
||||
"name": payload,
|
||||
})
|
||||
return
|
||||
|
||||
}
|
44
server/models/clients.go
Normal file
44
server/models/clients.go
Normal file
@@ -0,0 +1,44 @@
|
||||
package models
|
||||
|
||||
// type Clients map[string]Client
|
||||
|
||||
// type Client struct {
|
||||
// Connected *bool `json:"connected"`
|
||||
// SndConn *websocket.Conn `json:"-"` //sending connection
|
||||
// RvcConn *websocket.Conn `json:"-"` // revieving connection
|
||||
// }
|
||||
|
||||
// func NewClients() Clients {
|
||||
// return make(Clients)
|
||||
// }
|
||||
|
||||
// // Connect a recieving websocket connection
|
||||
// func (c *Clients) ConnectRecievingWsConnection(id string, ctx *gin.Context) (*websocket.Conn, error) {
|
||||
// conn, err := websocket.Accept(ctx.Writer, ctx.Request, &websocket.AcceptOptions{
|
||||
// OriginPatterns: []string{"*"},
|
||||
// })
|
||||
|
||||
// if err != nil {
|
||||
// return nil, fmt.Errorf("error accept websocket client: %s", err)
|
||||
// }
|
||||
|
||||
// b := true
|
||||
// (*c)[id] = Client{
|
||||
// Connected: &b,
|
||||
// RvcConn: conn,
|
||||
// }
|
||||
// return conn, nil
|
||||
// }
|
||||
|
||||
// func (c *Clients) RemoveClient(id string) {
|
||||
// delete(*c, id)
|
||||
// }
|
||||
|
||||
// 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,7 +0,0 @@
|
||||
package models
|
||||
|
||||
type Response struct {
|
||||
StatusCode int `json:"statusCode"`
|
||||
Message string `json:"message,omitempty"`
|
||||
Data string `json:"dat,omitempty"`
|
||||
}
|
@@ -1,26 +0,0 @@
|
||||
package models
|
||||
|
||||
import "github.com/coder/websocket"
|
||||
|
||||
type Subscribers map[*websocket.Conn]*bool
|
||||
|
||||
func InitSubsrcibers() Subscribers {
|
||||
return make(Subscribers)
|
||||
}
|
||||
|
||||
func (s *Subscribers) Connect(conn *websocket.Conn) {
|
||||
b := true
|
||||
(*s)[conn] = &b
|
||||
}
|
||||
|
||||
func (s *Subscribers) DeleteSubsrcibers(conn *websocket.Conn) {
|
||||
delete(*s, conn)
|
||||
}
|
||||
|
||||
func (s *Subscribers) GetPointer(conn *websocket.Conn) *bool {
|
||||
return (*s)[conn]
|
||||
}
|
||||
|
||||
func (s *Subscribers) Disconnect(conn *websocket.Conn) {
|
||||
*(*s)[conn] = false
|
||||
}
|
@@ -1,16 +0,0 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func (s *Server) AddRoutes() {
|
||||
|
||||
s.engine.GET("/ws", s.handleWebSocket)
|
||||
|
||||
s.engine.GET("/allBuses", s.AllBuses)
|
||||
|
||||
s.engine.GET("/", func(c *gin.Context) {
|
||||
c.String(200, "WebSocket Broadcast Server is running!")
|
||||
})
|
||||
}
|
@@ -1,48 +1,38 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"artNet/cert"
|
||||
"artNet/driver"
|
||||
serverModels "artNet/server/models"
|
||||
"fmt"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/tecamino/tecamino-logger/logging"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
engine *gin.Engine
|
||||
sync.RWMutex
|
||||
Subscribers serverModels.Subscribers
|
||||
Logger *logging.Logger
|
||||
Driver *driver.ArtNetDriver
|
||||
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
|
||||
func NewServer() *Server {
|
||||
s := Server{
|
||||
engine: gin.Default(),
|
||||
Subscribers: serverModels.InitSubsrcibers(),
|
||||
return &Server{
|
||||
Routes: gin.Default(),
|
||||
}
|
||||
s.AddRoutes()
|
||||
|
||||
s.Driver = driver.NewDriver()
|
||||
|
||||
return &s
|
||||
}
|
||||
|
||||
func (s *Server) ServeHttp(port uint) error {
|
||||
if err := s.engine.Run(fmt.Sprintf(":%d", port)); err != nil {
|
||||
return fmt.Errorf("failed to run http server: %v", err)
|
||||
}
|
||||
return nil
|
||||
return s.Routes.Run(fmt.Sprintf(":%d", port))
|
||||
}
|
||||
|
||||
func (s *Server) ServeHttps(port uint, cert cert.Cert) error {
|
||||
cert.GenerateSelfSignedCert()
|
||||
if err := s.engine.RunTLS(fmt.Sprintf(":%d", port), cert.CertFile, cert.KeyFile); err != nil {
|
||||
return fmt.Errorf("failed to run https server: %v", err)
|
||||
}
|
||||
return nil
|
||||
return s.Routes.RunTLS(fmt.Sprintf(":%d", port), cert.CertFile, cert.KeyFile)
|
||||
}
|
||||
|
@@ -1,13 +0,0 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"artNet/models"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
func (s *Server) Set(get models.Get) error {
|
||||
if len(s.Driver.Bus) <= int(get.Bus) {
|
||||
return fmt.Errorf("no bus number '%d' found", get.Bus)
|
||||
}
|
||||
return nil
|
||||
}
|
@@ -1,65 +0,0 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"artNet/models"
|
||||
|
||||
"github.com/coder/websocket"
|
||||
"github.com/coder/websocket/wsjson"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
const (
|
||||
OnCreate = "onCreate"
|
||||
OnChange = "onChange"
|
||||
OnDelete = "onDelete"
|
||||
)
|
||||
|
||||
func (s *Server) handleWebSocket(c *gin.Context) {
|
||||
conn, err := websocket.Accept(c.Writer, c.Request, &websocket.AcceptOptions{
|
||||
OriginPatterns: []string{"*"},
|
||||
})
|
||||
if err != nil {
|
||||
log.Println("Upgrade error:", err)
|
||||
return
|
||||
}
|
||||
|
||||
defer conn.Close(websocket.StatusInternalError, "Internal error")
|
||||
|
||||
ctx, cancel := context.WithTimeout(c.Request.Context(), 10*time.Minute)
|
||||
defer cancel()
|
||||
|
||||
s.Subscribers.Connect(conn)
|
||||
|
||||
//Read loop
|
||||
for {
|
||||
var request models.JsonData
|
||||
err := wsjson.Read(ctx, conn, &request)
|
||||
if err != nil {
|
||||
fmt.Println(s.Subscribers[conn])
|
||||
log.Println("WebSocket read error:", err)
|
||||
s.Subscribers.Disconnect(conn)
|
||||
fmt.Println(s.Subscribers[conn])
|
||||
break
|
||||
}
|
||||
|
||||
// Set
|
||||
// if request.Set != nil {
|
||||
// for _, set := range *request.Set {
|
||||
// err = s.Unsubscribe(unsub, conn)
|
||||
// if err != nil {
|
||||
// fmt.Println(err)
|
||||
// break
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user