update

pkg/mqtt/mqtt.go

@@ -0,0 +1,102 @@
+package mqtt
+
+import (
+	"app/pkg/datastore"
+	"app/pkg/telldus"
+	"fmt"
+
+	mqtt "github.com/eclipse/paho.mqtt.golang"
+)
+
+// Client handles all MQTT operations
+type Client struct {
+	client        mqtt.Client
+	store         *datastore.DataStore
+	subscriptions []string
+}
+
+// Config holds MQTT connection configuration
+type Config struct {
+	BrokerURL string
+	Username  string
+	Password  string
+}
+
+// New creates a new MQTT client
+func New(cfg Config, store *datastore.DataStore) (*Client, error) {
+	opts := mqtt.NewClientOptions().AddBroker(cfg.BrokerURL)
+	opts.SetUsername(cfg.Username)
+	opts.SetPassword(cfg.Password)
+
+	client := mqtt.NewClient(opts)
+	if token := client.Connect(); token.Wait() && token.Error() != nil {
+		return nil, token.Error()
+	}
+
+	return &Client{
+		client: client,
+		store:  store,
+	}, nil
+}
+
+// Close disconnects the MQTT client
+func (c *Client) Close() {
+	c.client.Disconnect(250)
+}
+
+// PublishDeviceState publishes a device state change
+func (c *Client) PublishDeviceState(deviceID int, state string) {
+	topic := fmt.Sprintf("telldus/device/%d/state", deviceID)
+	c.client.Publish(topic, 0, false, state)
+}
+
+// PublishSensorValue publishes a sensor value
+func (c *Client) PublishSensorValue(protocol, model string, id int, dataType int, value string) {
+	var topic string
+	switch dataType {
+	case telldus.DataTypeTemperature:
+		topic = fmt.Sprintf("telldus/sensor/%s/%s/%d/temperature", protocol, model, id)
+	case telldus.DataTypeHumidity:
+		topic = fmt.Sprintf("telldus/sensor/%s/%s/%d/humidity", protocol, model, id)
+	default:
+		return
+	}
+	c.client.Publish(topic, 0, false, value)
+}
+
+// UnsubscribeFromDeviceCommands unsubscribes from all tracked command topics
+func (c *Client) UnsubscribeFromDeviceCommands() {
+	if len(c.subscriptions) > 0 {
+		if token := c.client.Unsubscribe(c.subscriptions...); token.Wait() && token.Error() != nil {
+			// Log error but don't fail
+		}
+		c.subscriptions = nil
+	}
+}
+
+// SubscribeToDeviceCommands subscribes to command topics for all devices
+func (c *Client) SubscribeToDeviceCommands() error {
+	// Unsubscribe from existing subscriptions first
+	c.UnsubscribeFromDeviceCommands()
+
+	numDevices := telldus.GetNumberOfDevices()
+	for i := 0; i < numDevices; i++ {
+		deviceID := telldus.GetDeviceId(i)
+		topic := fmt.Sprintf("telldus/device/%d/set", deviceID)
+
+		// Capture deviceID in closure
+		id := deviceID
+		c.client.Subscribe(topic, 0, func(client mqtt.Client, msg mqtt.Message) {
+			payload := string(msg.Payload())
+			if payload == "ON" {
+				telldus.TurnOn(id)
+			} else if payload == "OFF" {
+				telldus.TurnOff(id)
+			}
+		})
+
+		// Track subscription
+		c.subscriptions = append(c.subscriptions, topic)
+	}
+	return nil
+}