1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
package mqtt
import (
"errors"
"github.com/charmbracelet/log"
mqtt "github.com/eclipse/paho.mqtt.golang"
"strings"
"sync"
"time"
)
const MQTTProtoTopic = "/2/e/"
type Client struct {
server string
username string
password string
topicRoot string
clientID string
client mqtt.Client
sync.RWMutex
channelHandlers map[string][]HandlerFunc
}
type HandlerFunc func(message Message)
var DefaultClient = Client{
server: "tcp://mqtt.meshtastic.org:1883",
username: "meshdev",
password: "large4cats",
topicRoot: "msh", //TODO: this will need to change
channelHandlers: make(map[string][]HandlerFunc),
}
func NewClient(url, username, password, rootTopic string) *Client {
return &Client{
server: url,
username: username,
password: password,
topicRoot: rootTopic,
channelHandlers: make(map[string][]HandlerFunc),
}
}
func (c *Client) TopicRoot() string {
return c.topicRoot
}
func (c *Client) Connect() error {
var alphabet = []rune("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789")
c.clientID = randomString(23, alphabet)
mqtt.DEBUG = log.StandardLog(log.StandardLogOptions{ForceLevel: log.DebugLevel})
mqtt.ERROR = log.StandardLog(log.StandardLogOptions{ForceLevel: log.ErrorLevel})
opts := mqtt.NewClientOptions().
AddBroker(c.server).
SetUsername(c.username).
SetOrderMatters(false).
SetPassword(c.password).
SetClientID(c.clientID).
SetCleanSession(false)
opts.SetKeepAlive(30 * time.Second)
opts.SetResumeSubs(true)
//opts.SetDefaultPublishHandler(f)
opts.SetPingTimeout(5 * time.Second)
opts.SetAutoReconnect(true)
opts.SetMaxReconnectInterval(1 * time.Minute)
opts.SetConnectionLostHandler(func(c mqtt.Client, err error) {
log.Error("mqtt connection lost", "err", err)
})
opts.SetReconnectingHandler(func(c mqtt.Client, options *mqtt.ClientOptions) {
log.Info("mqtt reconnecting")
})
opts.SetOnConnectHandler(func(client mqtt.Client) {
log.Info("connected to", "server", c.server)
})
c.client = mqtt.NewClient(opts)
if token := c.client.Connect(); token.Wait() && token.Error() != nil {
return token.Error()
}
return nil
}
// MQTT Message
type Message struct {
Topic string
Payload []byte
Retained bool
}
// Publish a message to the broker
func (c *Client) Publish(m *Message) error {
tok := c.client.Publish(m.Topic, 0, m.Retained, m.Payload)
if !tok.WaitTimeout(10 * time.Second) {
tok.Wait()
return errors.New("timeout on mqtt publish")
}
if tok.Error() != nil {
return tok.Error()
}
return nil
}
// Register a handler for messages on the specified channel
func (c *Client) Handle(channel string, h HandlerFunc) {
c.Lock()
defer c.Unlock()
topic := c.GetFullTopicForChannel(channel)
c.channelHandlers[channel] = append(c.channelHandlers[channel], h)
c.client.Subscribe(topic+"/+", 0, c.handleBrokerMessage)
}
func (c *Client) GetFullTopicForChannel(channel string) string {
return c.topicRoot + MQTTProtoTopic + channel
}
func (c *Client) GetChannelFromTopic(topic string) string {
trimmed := strings.TrimPrefix(topic, c.topicRoot+MQTTProtoTopic)
sepIndex := strings.Index(trimmed, "/")
if sepIndex > 0 {
return trimmed[:sepIndex]
}
return trimmed
}
func (c *Client) handleBrokerMessage(client mqtt.Client, message mqtt.Message) {
msg := Message{
Topic: message.Topic(),
Payload: message.Payload(),
Retained: message.Retained(),
}
c.RLock()
defer c.RUnlock()
channel := c.GetChannelFromTopic(msg.Topic)
chans := c.channelHandlers[channel]
if len(chans) == 0 {
log.Error("no handlers found", "topic", channel)
}
for _, ch := range chans {
go ch(msg)
}
}
|
