1 files changed, 146 insertions, 0 deletions

diff --git a/mqtt-controller/main.go b/mqtt-controller/main.go
new file mode 100644
index 0000000..29d5d28
--- /dev/null
+++ b/mqtt-controller/main.go
@@ -0,0 +1,146 @@
+package main
+
+import (
+        "encoding/json"
+        "log"
+        "os"
+        "os/signal"
+        "sync"
+        "syscall"
+        "time"
+
+        "github.com/brutella/hc"
+        "github.com/brutella/hc/accessory"
+        "github.com/brutella/hc/util"
+)
+
+type AppConfig struct {
+        Broker     string
+        BridgeName string
+        Pin        string
+}
+
+func loadAppConfig(s util.Storage) (*AppConfig, error) {
+        d, err := s.Get("bridgeConfig")
+        if err != nil {
+                return nil, err
+        }
+
+        c := &AppConfig{}
+        if err = json.Unmarshal(d, c); err != nil {
+                return nil, err
+        }
+
+        return c, nil
+}
+
+func main() {
+        wg := &sync.WaitGroup{}
+        done := make(chan interface{})
+
+        c := make(chan os.Signal)
+        signal.Notify(c, os.Interrupt)
+        signal.Notify(c, os.Kill)
+        signal.Notify(c, syscall.SIGTERM)
+
+        storage, err := util.NewFileStorage("db")
+        if err != nil {
+                panic(err)
+        }
+
+        cfg, err := loadAppConfig(storage)
+        if err != nil {
+                panic(err)
+        }
+
+        broker, err := NewMQTTBroker(cfg.Broker)
+        if err != nil {
+                panic(err)
+        }
+
+        if err := broker.Subscribe("shellies/announce", nil); err != nil {
+                panic(err)
+        }
+
+        if err := broker.Publish("shellies/command", "announce"); err != nil {
+                panic(err)
+        }
+
+        ac := make(chan *UnifiedSwitch, 10)
+
+        go func() {
+                wg.Add(1)
+                defer wg.Done()
+
+                log.Println("Starting homekit thread")
+
+                reg := []*accessory.Accessory{}
+
+                // Collect devices until the discovery process quiesces, then start the
+                // transport
+                quiescence := 10 * time.Second
+                t := time.NewTimer(quiescence)
+
+        C:
+                for {
+                        select {
+                        case a := <-ac:
+                                log.Println("Homekit: new device")
+                                reg = append(reg, a.HomeKitSwitch.Accessory)
+                                // TODO: Extend timer deadline
+                        case <-t.C:
+                                log.Println("Homekit: timeout over")
+                                t.Stop()
+                                break C
+                        case <-done:
+                                log.Println("Homekit: cancelled")
+                                return
+                        }
+                }
+
+                log.Println("MQTT quiesces, starting HomeKit")
+
+                br := accessory.NewBridge(accessory.Info{Name: cfg.BridgeName})
+
+                transport, err := hc.NewIPTransport(hc.Config{Pin: cfg.Pin}, br.Accessory, reg...)
+                if err != nil {
+                        log.Fatalf("%s", err)
+                }
+                go transport.Start()
+                log.Printf("HomeKit transport started")
+
+                select {
+                case <-done:
+                        log.Printf("Shutting down transport")
+                        if transport != nil {
+                                <-transport.Stop()
+                        }
+                }
+                log.Printf("Transport shut down")
+        }()
+
+        for {
+                select {
+                case d := <-broker.Devices:
+                        if d == nil {
+                                continue
+                        }
+
+                        us, err := NewUnifiedSwitch(broker, d)
+                        if err != nil {
+                                log.Printf("Error connecting device: %e", err)
+                                continue
+                        }
+
+                        go us.MessageLoop(done, wg)
+
+                        ac <- us
+                        log.Printf("New Device: %s\n", d.Identity())
+                case <-c:
+                        close(done)
+                        broker.Shutdown()
+                        wg.Wait()
+                        return
+                }
+        }
+}