87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package mqtt
|
|
|
|
import "gobot.io/x/gobot/v2"
|
|
|
|
const (
|
|
// Data event when data is available for Driver
|
|
Data = "data"
|
|
|
|
// Error event when error occurs in Driver
|
|
Error = "error"
|
|
)
|
|
|
|
// Driver for mqtt
|
|
type Driver struct {
|
|
name string
|
|
topic string
|
|
connection gobot.Connection
|
|
gobot.Eventer
|
|
gobot.Commander
|
|
}
|
|
|
|
// NewDriver returns a new Gobot MQTT Driver
|
|
func NewDriver(a *Adaptor, topic string) *Driver {
|
|
m := &Driver{
|
|
name: gobot.DefaultName("MQTT"),
|
|
topic: topic,
|
|
connection: a,
|
|
Eventer: gobot.NewEventer(),
|
|
Commander: gobot.NewCommander(),
|
|
}
|
|
|
|
m.AddEvent(Data)
|
|
m.AddEvent(Error)
|
|
|
|
return m
|
|
}
|
|
|
|
// Name returns name for the Driver
|
|
func (m *Driver) Name() string { return m.name }
|
|
|
|
// Name sets name for the Driver
|
|
func (m *Driver) SetName(name string) { m.name = name }
|
|
|
|
// Connection returns Connections used by the Driver
|
|
func (m *Driver) Connection() gobot.Connection {
|
|
return m.connection
|
|
}
|
|
|
|
func (m *Driver) adaptor() *Adaptor {
|
|
//nolint:forcetypeassert // ok here
|
|
return m.Connection().(*Adaptor)
|
|
}
|
|
|
|
// Start starts the Driver
|
|
func (m *Driver) Start() error {
|
|
return nil
|
|
}
|
|
|
|
// Halt halts the Driver
|
|
func (m *Driver) Halt() error {
|
|
return nil
|
|
}
|
|
|
|
// Topic returns the current topic for the Driver
|
|
func (m *Driver) Topic() string { return m.topic }
|
|
|
|
// SetTopic sets the current topic for the Driver
|
|
func (m *Driver) SetTopic(topic string) { m.topic = topic }
|
|
|
|
// Publish a message to the current device topic
|
|
func (m *Driver) Publish(data interface{}) bool {
|
|
//nolint:forcetypeassert // ok here
|
|
message := data.([]byte)
|
|
return m.adaptor().Publish(m.topic, message)
|
|
}
|
|
|
|
// On subscribes to data updates for the current device topic,
|
|
// and then calls the message handler function when data is received
|
|
func (m *Driver) On(n string, f func(msg interface{})) error {
|
|
// TODO: also be able to subscribe to Error updates
|
|
f1 := func(msg Message) {
|
|
f(msg)
|
|
}
|
|
m.adaptor().On(m.topic, f1)
|
|
return nil
|
|
}
|