1 files changed, 124 insertions, 0 deletions

diff --git a/mqtt-controller/unified_switch.go b/mqtt-controller/unified_switch.go
new file mode 100644
index 0000000..fb62c7a
--- /dev/null
+++ b/mqtt-controller/unified_switch.go
@@ -0,0 +1,124 @@
+package main
+
+import (
+        "hash/crc64"
+        "log"
+        "sync"
+
+        "github.com/brutella/hc/accessory"
+)
+
+type UnifiedSwitch struct {
+        HomeKitSwitch *accessory.Switch
+        MqttDevice    Device
+        Commands      <-chan Command
+        metrics       chan Metric
+        events        chan Event
+        infos         chan interface{}
+}
+
+func NewUnifiedSwitch(b *MQTTBroker, d Device) (*UnifiedSwitch, error) {
+        us := &UnifiedSwitch{
+                MqttDevice: d,
+                metrics:    make(chan Metric, 100),
+                events:     make(chan Event, 100),
+                infos:      make(chan interface{}, 100),
+        }
+
+        cmd, err := d.Connect(b, us.events, us.metrics, us.infos)
+        if err != nil {
+                log.Printf("Error connecting device: %e", err)
+                return nil, err
+        }
+
+        us.Commands = cmd
+        us.makeHomekitSwitch()
+
+        return us, nil
+}
+
+func (s *UnifiedSwitch) MessageLoop(done <-chan interface{}, wg *sync.WaitGroup) {
+        wg.Add(1)
+        defer wg.Done()
+
+        log.Printf("Starting switch %s", s.MqttDevice.Identity())
+
+C:
+        for {
+                select {
+                case e := <-s.events:
+                        direction := "output"
+                        if e.Direction == DirectionInput {
+                                direction = "input"
+                        }
+
+                        s.HomeKitSwitch.Switch.On.SetValue(e.Active)
+
+                        log.Printf("New event for '%s'.%d %s: %t", e.Device.Identity(), e.Index, direction, e.Active)
+                case m := <-s.metrics:
+                        /*
+                                switch t := m.Value.(type) {
+                                case float64:
+                                        log.Printf("New metric '%s' for '%s': %#v", m.Name, m.Device.Identity(), t)
+                                case int64:
+                                        log.Printf("New metric '%s' for '%s': %#v", m.Name, m.Device.Identity(), t)
+                                case bool:
+                                        log.Printf("New metric '%s' for '%s': %#v", m.Name, m.Device.Identity(), t)
+                                default:
+                                        log.Printf("Unknown metric: %#v", m.Value)
+                                }
+                        */
+                        _ = m
+                case i := <-s.infos:
+                        switch it := i.(type) {
+                        case ShellyStatus:
+                                // TODO: Handle multiple relays
+                                log.Printf("Info relays: %#v", it.Relays)
+                                s.HomeKitSwitch.Switch.On.SetValue(it.Relays[0].IsOn)
+                        default:
+                                // Just discard them for now
+                        }
+                case <-done:
+                        log.Printf("Shutting down switch %s", s.MqttDevice.Identity())
+                        break C
+                }
+        }
+
+        log.Printf("Switch %s shut down", s.MqttDevice.Identity())
+}
+
+func (s *UnifiedSwitch) HomeKitID() uint64 {
+        h64 := crc64.New(crc64.MakeTable(crc64.ISO))
+        h64.Write([]byte(s.MqttDevice.Identity()))
+        return h64.Sum64()
+}
+
+func (s *UnifiedSwitch) makeHomekitSwitch() {
+        // TODO: Get name from local DB?
+        switch dt := s.MqttDevice.(type) {
+        case *ShellyDevice:
+                s.HomeKitSwitch = accessory.NewSwitch(accessory.Info{
+                        ID:               s.HomeKitID(),
+                        Name:             s.MqttDevice.Identity(),
+                        Manufacturer:     dt.Manufacturer,
+                        Model:            dt.Model,
+                        FirmwareRevision: dt.FirmwareVersion,
+                })
+        default:
+                s.HomeKitSwitch = accessory.NewSwitch(accessory.Info{
+                        ID:   s.HomeKitID(),
+                        Name: s.MqttDevice.Identity(),
+                })
+        }
+        s.HomeKitSwitch.Switch.On.SetValue(false)
+        s.HomeKitSwitch.Switch.On.OnValueRemoteUpdate(s.HomeKitSetValue)
+}
+
+func (s *UnifiedSwitch) HomeKitSetValue(state bool) {
+        log.Printf("Client updated state: %t", state)
+        if state {
+                s.MqttDevice.OutputOn(0)
+        } else {
+                s.MqttDevice.OutputOff(0)
+        }
+}