Compare commits
8 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
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
|
# 🎛️ ArtNet Driver
|
||||||
Tecamino artNet driver DMX-Communication
|
|
||||||
|
**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
|
||||||
|
```
|
15
auth/auth.go
Normal file
15
auth/auth.go
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
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")
|
||||||
|
}
|
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)
|
||||||
|
}
|
@@ -1,29 +1,85 @@
|
|||||||
package driver
|
package driver
|
||||||
|
|
||||||
import "artNet/models"
|
import (
|
||||||
|
"artNet/cfg"
|
||||||
|
"artNet/models"
|
||||||
|
serverModels "artNet/server/models"
|
||||||
|
"fmt"
|
||||||
|
"path"
|
||||||
|
|
||||||
|
"github.com/tecamino/tecamino-logger/logging"
|
||||||
|
)
|
||||||
|
|
||||||
type ArtNetDriver struct {
|
type ArtNetDriver struct {
|
||||||
Bus []*models.Bus
|
Name string `yaml:"driver" json:"driver"`
|
||||||
|
Buses map[string]*models.Bus `yaml:"buses,omitempty" json:"buses,omitempty"`
|
||||||
|
cfgHandler *cfg.Cfg `yaml:"-" json:"-"`
|
||||||
|
Connections serverModels.Clients `yaml:"-"`
|
||||||
|
Log *logging.Logger `yaml:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// initialize new Art-Net driver
|
// initialize new Art-Net driver
|
||||||
func NewDriver() *ArtNetDriver {
|
// cfgDir config directory
|
||||||
return &ArtNetDriver{}
|
// name name of driver
|
||||||
|
func NewDriver(cfgDir, name string, debug bool) (*ArtNetDriver, error) {
|
||||||
|
if cfgDir == "" {
|
||||||
|
cfgDir = "./cfg"
|
||||||
}
|
}
|
||||||
|
|
||||||
// adds new Art-Net interface to driver port 0 = 6454 (default art-net)
|
logger, err := logging.NewLogger(path.Join(cfgDir, name+".log"), &logging.Config{
|
||||||
func (d *ArtNetDriver) NewInterface(ip string, port int) *models.Bus {
|
MaxSize: 1,
|
||||||
i := models.NewBus(ip, port)
|
MaxBackup: 3,
|
||||||
d.Bus = append(d.Bus, i)
|
MaxAge: 28,
|
||||||
return i
|
Debug: debug,
|
||||||
|
TerminalOut: true,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// dmxData[46] = byte(255) // Channel 1: Red
|
logger.Debug("artNet.NewDriver", "initialize "+name+" driver")
|
||||||
// dmxData[47] = byte(255) // Channel 2: Green
|
d := ArtNetDriver{
|
||||||
// dmxData[48] = byte(255) // Channel 3: Blue
|
Name: name,
|
||||||
// dmxData[49] = byte(255) // Channel 4: White
|
Buses: make(map[string]*models.Bus),
|
||||||
// dmxData[50] = byte(255) // Channel 5: Amber
|
cfgHandler: cfg.NewCfgHandler(cfgDir, name),
|
||||||
// dmxData[51] = byte(255) // Channel 6: UV Lila
|
Connections: serverModels.NewClients(),
|
||||||
// dmxData[52] = byte(255) // Channel 7: 0-9 10-255 strobo
|
Log: logger,
|
||||||
// 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
|
|
||||||
|
if err := d.LoadCfg(); err != nil {
|
||||||
|
logger.Error("artNet.NewDriver.LoadCfg", "error load driver configuration: "+err.Error())
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &d, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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)
|
||||||
|
b.Start(d.Log)
|
||||||
|
d.Buses[name] = b
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) SetValue(set models.Set) error {
|
||||||
|
if _, ok := d.Buses[set.Bus]; !ok {
|
||||||
|
return fmt.Errorf("no bus '%s' found", set.Bus)
|
||||||
|
}
|
||||||
|
d.Buses[set.Bus].Data.SetValue(set.Address, set.Value)
|
||||||
|
|
||||||
|
return d.Buses[set.Bus].SendData()
|
||||||
|
}
|
||||||
|
148
driver/bus.go
Normal file
148
driver/bus.go
Normal file
@@ -0,0 +1,148 @@
|
|||||||
|
package driver
|
||||||
|
|
||||||
|
import (
|
||||||
|
"artNet/auth"
|
||||||
|
"artNet/models"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
// sends a list of all buses in the driver
|
||||||
|
func (d *ArtNetDriver) GetAllBuses(c *gin.Context) {
|
||||||
|
var data any
|
||||||
|
if len(d.Buses) == 0 {
|
||||||
|
data = "no buses avaiable"
|
||||||
|
} else {
|
||||||
|
data = *d
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"buses": data,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) CreateBus(c *gin.Context) {
|
||||||
|
_, err := auth.GetIDFromAuth(c)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var payload models.Bus
|
||||||
|
|
||||||
|
if err := c.BindJSON(&payload); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if payload.Name == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
||||||
|
return
|
||||||
|
} else if addr := net.ParseIP(payload.Ip); addr == nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "wrong ip '" + payload.Ip + "'"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, ok := d.Buses[payload.Name]; ok {
|
||||||
|
c.JSON(http.StatusOK, gin.H{"message": "bus " + payload.Name + " exists already"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
bus := d.NewBus(payload.Name, payload.Ip, payload.GetPort())
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, gin.H{
|
||||||
|
"message": fmt.Sprintf("bus '%s' successfully created with ip: %s and on port: %d", bus.Name, bus.Ip, bus.GetPort()),
|
||||||
|
})
|
||||||
|
d.cfgHandler.SaveCfg(*d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) RemoveBus(c *gin.Context) {
|
||||||
|
_, err := auth.GetIDFromAuth(c)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var payload models.Bus
|
||||||
|
|
||||||
|
if err := c.BindJSON(&payload); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if payload.Name == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, ok := d.Buses[payload.Name]; !ok {
|
||||||
|
c.JSON(http.StatusOK, gin.H{"message": "bus " + payload.Name + " not found"})
|
||||||
|
c.JSON(http.StatusOK, models.JsonResponse{
|
||||||
|
Message: "bus " + payload.Name + " not found"})
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
delete(d.Buses, payload.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, models.JsonResponse{
|
||||||
|
Message: fmt.Sprintf("bus '%s' successfully removed", payload.Name),
|
||||||
|
})
|
||||||
|
d.cfgHandler.SaveCfg(*d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) Start(c *gin.Context) {
|
||||||
|
_, err := auth.GetIDFromAuth(c)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var payload models.Bus
|
||||||
|
|
||||||
|
if err := c.BindJSON(&payload); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if payload.Name == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
d.Buses[payload.Name].Start(d.Log)
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, models.JsonResponse{
|
||||||
|
Message: fmt.Sprintf("bus '%s' running", payload.Name),
|
||||||
|
})
|
||||||
|
d.cfgHandler.SaveCfg(*d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) Stop(c *gin.Context) {
|
||||||
|
_, err := auth.GetIDFromAuth(c)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "id: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var payload models.Bus
|
||||||
|
|
||||||
|
if err := c.BindJSON(&payload); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "json: " + err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if payload.Name == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "bus name missing"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
d.Buses[payload.Name].Stop()
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, models.JsonResponse{
|
||||||
|
Message: fmt.Sprintf("bus '%s' stopped", payload.Name),
|
||||||
|
})
|
||||||
|
d.cfgHandler.SaveCfg(*d)
|
||||||
|
}
|
63
driver/webSocket.go
Normal file
63
driver/webSocket.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package driver
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"artNet/auth"
|
||||||
|
"artNet/models"
|
||||||
|
|
||||||
|
"github.com/coder/websocket"
|
||||||
|
"github.com/coder/websocket/wsjson"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
OnCreate = "onCreate"
|
||||||
|
OnChange = "onChange"
|
||||||
|
OnDelete = "onDelete"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d *ArtNetDriver) Websocket(c *gin.Context) {
|
||||||
|
id, err := auth.GetIDFromAuth(c)
|
||||||
|
if err != nil {
|
||||||
|
d.Log.Error("artNet.webSocket.Websocket", "error GetIDFromAuth: "+err.Error())
|
||||||
|
log.Println("error id:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
d.Log.Debug("artNet.webSocket.Websocket", "authorization id token: "+id)
|
||||||
|
|
||||||
|
ctx, cancel := context.WithTimeout(c.Request.Context(), 10*time.Minute)
|
||||||
|
defer cancel()
|
||||||
|
conn, err := d.Connections.ConnectRecievingWsConnection(id, c)
|
||||||
|
if err != nil {
|
||||||
|
d.Log.Error("artNet.webSocket.Websocket", "error connecting recieving websocket conection: "+err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer d.Connections.DisconnectRecievingWsConnection(id, websocket.StatusInternalError, "Internal error")
|
||||||
|
|
||||||
|
var request models.JsonData
|
||||||
|
//Read loop
|
||||||
|
for {
|
||||||
|
|
||||||
|
err := wsjson.Read(ctx, conn, &request)
|
||||||
|
if err != nil {
|
||||||
|
d.Log.Error("artNet.webSocket.Websocket", "read error:"+err.Error())
|
||||||
|
log.Println("WebSocket read error:", err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set
|
||||||
|
if request.Set != nil {
|
||||||
|
for _, set := range *request.Set {
|
||||||
|
if err = d.SetValue(set); err != nil {
|
||||||
|
d.Log.Error("artNet.webSocket.Websocket", "set value error"+err.Error())
|
||||||
|
log.Println(err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
time.Sleep(23 * time.Millisecond)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
5
go.mod
5
go.mod
@@ -8,7 +8,8 @@ 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/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e
|
github.com/tatsushid/go-fastping v0.0.0-20160109021039-d7bb493dee3e
|
||||||
github.com/tecamino/tecamino-logger v0.1.1
|
github.com/tecamino/tecamino-logger v0.2.0
|
||||||
|
gopkg.in/yaml.v3 v3.0.1
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
@@ -39,5 +40,5 @@ require (
|
|||||||
golang.org/x/sys v0.20.0 // indirect
|
golang.org/x/sys v0.20.0 // indirect
|
||||||
golang.org/x/text v0.15.0 // indirect
|
golang.org/x/text v0.15.0 // indirect
|
||||||
google.golang.org/protobuf v1.34.1 // 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
|
||||||
)
|
)
|
||||||
|
6
go.sum
6
go.sum
@@ -63,8 +63,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-logger v0.1.1 h1:bK0SEQpbjui42OsnM6JCDThaGMBoqwVj/Op6ESqDwV4=
|
github.com/tecamino/tecamino-logger v0.2.0 h1:NPH/Gg9qRhmVoW8b39i1eXu/LEftHc74nyISpcRG+XU=
|
||||||
github.com/tecamino/tecamino-logger v0.1.1/go.mod h1:sGysmiFGIdr4vLJRAI+fJgsa7EoRRuxvRrKW7GnGQkw=
|
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=
|
||||||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
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=
|
github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
|
||||||
@@ -94,6 +94,8 @@ google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFW
|
|||||||
google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
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 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
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.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 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
97
main.go
97
main.go
@@ -3,71 +3,56 @@ package main
|
|||||||
import (
|
import (
|
||||||
"artNet/driver"
|
"artNet/driver"
|
||||||
"artNet/server"
|
"artNet/server"
|
||||||
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"math"
|
"os"
|
||||||
"time"
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const DriverName string = "ArtNetDriver"
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
// cli flags
|
||||||
|
port := flag.Uint("port-http", 8110, "http server listening port")
|
||||||
|
cfgDir := flag.String("cfg", "./cfg", "config directory")
|
||||||
|
workingDir := flag.String("workingDirectory", ".", "current working directory")
|
||||||
|
debug := flag.Bool("debug", false, "debug logging")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
bus := driver.NewDriver().NewInterface("2.0.0.1", 0)
|
//change working directory only if value is given
|
||||||
d, _ := bus.AddDevice(47, 9)
|
if *workingDir != "." && *workingDir != "" {
|
||||||
|
os.Chdir(*workingDir)
|
||||||
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)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := bus.SendData(); err != nil {
|
//initialize new ArtNet driver
|
||||||
fmt.Println(12, err)
|
artNetDriver, err := driver.NewDriver(*cfgDir, DriverName, *debug)
|
||||||
|
if err != nil {
|
||||||
|
artNetDriver.Log.Error("main", "error initialize new artnet driver "+err.Error())
|
||||||
panic(err)
|
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("/ws", artNetDriver.Websocket)
|
||||||
|
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/stop", artNetDriver.Stop)
|
||||||
|
|
||||||
|
s.Routes.GET("/", func(c *gin.Context) {
|
||||||
|
c.String(200, "ArtNet Driver WebSocket Server is running!")
|
||||||
|
})
|
||||||
|
|
||||||
|
artNetDriver.Log.Info("main", fmt.Sprintf("listen on %d", *port))
|
||||||
|
// start http server
|
||||||
|
if err := s.ServeHttp(*port); err != nil {
|
||||||
|
artNetDriver.Log.Error("main", "error http server "+err.Error())
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
114
models/bus.go
114
models/bus.go
@@ -1,11 +1,13 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/tatsushid/go-fastping"
|
"github.com/tatsushid/go-fastping"
|
||||||
|
"github.com/tecamino/tecamino-logger/logging"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Art-Net constants
|
// Art-Net constants
|
||||||
@@ -15,43 +17,47 @@ const (
|
|||||||
|
|
||||||
// Art-Net Interface
|
// Art-Net Interface
|
||||||
type Bus struct {
|
type Bus struct {
|
||||||
ip string `yaml:"ip"`
|
Name string `yaml:"name" json:"name"`
|
||||||
port int `yaml:"port"`
|
Ip string `yaml:"ip" json:"ip"`
|
||||||
Devices []*Device `yaml:"devices"`
|
Port *int `yaml:"port" json:"port,omitempty"`
|
||||||
Data *DMX
|
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)
|
// 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 {
|
if port == 0 {
|
||||||
port = artPort
|
port = artPort
|
||||||
}
|
}
|
||||||
|
|
||||||
i := Bus{
|
i := Bus{
|
||||||
ip: ip,
|
Name: name,
|
||||||
port: port,
|
Ip: ip,
|
||||||
|
Port: &port,
|
||||||
Data: NewDMXUniverse(),
|
Data: NewDMXUniverse(),
|
||||||
}
|
}
|
||||||
return &i
|
return &i
|
||||||
}
|
}
|
||||||
|
|
||||||
// adds new dmx device to interface
|
// get bus port from pointer
|
||||||
func (i *Bus) AddDevice(address uint, channels uint) (*Device, error) {
|
func (b *Bus) GetPort() int {
|
||||||
d := NewDevice(address, channels, i.Data)
|
if b.Port == nil {
|
||||||
i.Devices = append(i.Devices, d)
|
return artPort
|
||||||
return d, nil
|
}
|
||||||
|
return *b.Port
|
||||||
}
|
}
|
||||||
|
|
||||||
// start polling dmx data in milliseconds 0 = aprox. 44Hertz
|
// 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 {
|
if interval == 0 {
|
||||||
interval = 23
|
interval = 23
|
||||||
}
|
}
|
||||||
|
|
||||||
// Send packet over UDP
|
// Send packet over UDP
|
||||||
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
||||||
IP: net.ParseIP(i.ip),
|
IP: net.ParseIP(b.Ip),
|
||||||
Port: i.port,
|
Port: *b.Port,
|
||||||
})
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -63,7 +69,7 @@ func (i *Bus) Poll(interval time.Duration) error {
|
|||||||
for {
|
for {
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
if reached, _ := isUDPReachable(i.ip); !reached {
|
if reached, _ := isUDPReachable(b.Ip); !reached {
|
||||||
if errCount > 20 {
|
if errCount > 20 {
|
||||||
break
|
break
|
||||||
} else {
|
} else {
|
||||||
@@ -77,7 +83,7 @@ func (i *Bus) Poll(interval time.Duration) error {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
_, err = conn.Write(NewArtNetPackage(i.Data))
|
_, err = conn.Write(NewArtNetPackage(b.Data))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -89,12 +95,55 @@ func (i *Bus) Poll(interval time.Duration) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// start polling dmx data in milliseconds 0 = aprox. 44Hertz
|
// start bus
|
||||||
func (i *Bus) SendData() error {
|
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()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// send dmx data
|
||||||
|
func (b *Bus) SendData() error {
|
||||||
|
if !b.Reachable {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
// Send packet over UDP
|
// Send packet over UDP
|
||||||
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
||||||
IP: net.ParseIP(i.ip),
|
IP: net.ParseIP(b.Ip),
|
||||||
Port: i.port,
|
Port: *b.Port,
|
||||||
})
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -102,29 +151,11 @@ func (i *Bus) SendData() error {
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
errChan := make(chan error)
|
_, err = conn.Write(NewArtNetPackage(b.Data))
|
||||||
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))
|
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
const (
|
|
||||||
protocolICMP = 1
|
|
||||||
)
|
|
||||||
|
|
||||||
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)
|
||||||
@@ -132,10 +163,11 @@ func isUDPReachable(ip string) (recieved bool, err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
p.AddIPAddr(ra)
|
p.AddIPAddr(ra)
|
||||||
|
|
||||||
p.OnRecv = func(addr *net.IPAddr, rtt time.Duration) {
|
p.OnRecv = func(addr *net.IPAddr, rtt time.Duration) {
|
||||||
recieved = true
|
recieved = true
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
p.OnIdle = func() {}
|
p.OnIdle = func() {}
|
||||||
|
|
||||||
err = p.Run()
|
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"`
|
|
||||||
}
|
|
@@ -2,6 +2,7 @@ package models
|
|||||||
|
|
||||||
type JsonData struct {
|
type JsonData struct {
|
||||||
Set *[]Set `json:"set,omitempty"`
|
Set *[]Set `json:"set,omitempty"`
|
||||||
|
Create *[]Bus `json:"create,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewRequest() *JsonData {
|
func NewRequest() *JsonData {
|
||||||
@@ -9,7 +10,7 @@ func NewRequest() *JsonData {
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *JsonData) AddSet(bus, address uint, value uint8) {
|
func (r *JsonData) AddSet(bus string, address uint, value uint8) {
|
||||||
if r.Set == nil {
|
if r.Set == nil {
|
||||||
r.Set = &[]Set{}
|
r.Set = &[]Set{}
|
||||||
}
|
}
|
||||||
|
7
models/jsonResponse.go
Normal file
7
models/jsonResponse.go
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
package models
|
||||||
|
|
||||||
|
type JsonResponse struct {
|
||||||
|
Error bool `json:"error,omitempty"`
|
||||||
|
Message string `json:"message,omitempty"`
|
||||||
|
Data string `json:"data,omitempty"`
|
||||||
|
}
|
@@ -1,7 +1,7 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
type Set struct {
|
type Set struct {
|
||||||
Bus uint `json:"bus"`
|
Bus string `json:"bus"`
|
||||||
Address uint `json:"address"`
|
Address uint `json:"address"`
|
||||||
Value uint8 `json:"value"`
|
Value uint8 `json:"value"`
|
||||||
}
|
}
|
||||||
|
@@ -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
|
|
||||||
|
|
||||||
}
|
|
51
server/models/clients.go
Normal file
51
server/models/clients.go
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/coder/websocket"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
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!")
|
|
||||||
})
|
|
||||||
}
|
|
@@ -2,47 +2,27 @@ package server
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"sync"
|
|
||||||
|
|
||||||
"artNet/cert"
|
"artNet/cert"
|
||||||
"artNet/driver"
|
|
||||||
serverModels "artNet/server/models"
|
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/tecamino/tecamino-logger/logging"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Server struct {
|
type Server struct {
|
||||||
engine *gin.Engine
|
Routes *gin.Engine
|
||||||
sync.RWMutex
|
|
||||||
Subscribers serverModels.Subscribers
|
|
||||||
Logger *logging.Logger
|
|
||||||
Driver *driver.ArtNetDriver
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Initialize new websocket server
|
||||||
func NewServer() *Server {
|
func NewServer() *Server {
|
||||||
s := Server{
|
return &Server{
|
||||||
engine: gin.Default(),
|
Routes: gin.Default(),
|
||||||
Subscribers: serverModels.InitSubsrcibers(),
|
|
||||||
}
|
}
|
||||||
s.AddRoutes()
|
|
||||||
|
|
||||||
s.Driver = driver.NewDriver()
|
|
||||||
|
|
||||||
return &s
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) ServeHttp(port uint) error {
|
func (s *Server) ServeHttp(port uint) error {
|
||||||
if err := s.engine.Run(fmt.Sprintf(":%d", port)); err != nil {
|
return s.Routes.Run(fmt.Sprintf(":%d", port))
|
||||||
return fmt.Errorf("failed to run http server: %v", err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) ServeHttps(port uint, cert cert.Cert) error {
|
func (s *Server) ServeHttps(port uint, cert cert.Cert) error {
|
||||||
cert.GenerateSelfSignedCert()
|
return s.Routes.RunTLS(fmt.Sprintf(":%d", port), cert.CertFile, cert.KeyFile)
|
||||||
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
|
|
||||||
}
|
}
|
||||||
|
@@ -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