rename the project to catbus-wakeonlan

1 files changed, 81 insertions, 0 deletions

diff --git a/cmd/catbus-actuator-wakeonlan/main.go b/cmd/catbus-actuator-wakeonlan/main.go
new file mode 100644
index 0000000..ada08a0
--- /dev/null
+++ b/cmd/catbus-actuator-wakeonlan/main.go
@@ -0,0 +1,81 @@
+// SPDX-FileCopyrightText: 2020 Ethel Morgan
+//
+// SPDX-License-Identifier: MIT
+
+package main
+
+import (
+	"context"
+	"flag"
+	"log"
+
+	"go.eth.moe/catbus-wakeonlan/config"
+	"go.eth.moe/catbus-wakeonlan/logger"
+	"go.eth.moe/catbus-wakeonlan/mqtt"
+	"go.eth.moe/catbus-wakeonlan/wakeonlan"
+)
+
+var (
+	configPath = flag.String("config-path", "", "path to config")
+)
+
+func main() {
+	flag.Parse()
+
+	if *configPath == "" {
+		log.Fatal("must set -config-path")
+	}
+
+	log, _ := logger.FromContext(context.Background())
+
+	config, err := config.ParseFile(*configPath)
+	if err != nil {
+		log.AddField("config-path", *configPath)
+		log.WithError(err).Fatal("could not parse config file")
+	}
+
+	log.AddField("broker-uri", config.Broker)
+
+	brokerOptions := mqtt.NewClientOptions()
+	brokerOptions.AddBroker(config.Broker)
+	brokerOptions.SetAutoReconnect(true)
+	brokerOptions.SetConnectionLostHandler(func(_ mqtt.Client, err error) {
+		log := log
+		if err != nil {
+			log = log.WithError(err)
+		}
+		log.Error("disconnected from MQTT broker")
+	})
+	brokerOptions.SetOnConnectHandler(func(broker mqtt.Client) {
+		log.Info("connected to MQTT broker")
+
+		for topic := range config.MACsByTopic {
+			token := broker.Subscribe(topic, mqtt.AtLeastOnce, func(_ mqtt.Client, msg mqtt.Message) {
+				mac, ok := config.MACsByTopic[msg.Topic()]
+				if !ok {
+					return
+				}
+
+				log.AddField("mac", mac)
+				log.AddField("topic", topic)
+				if err := wakeonlan.Wake(mac); err != nil {
+					log.WithError(err).Error("could not send wake-on-lan packet")
+					return
+				}
+				log.Info("sent wake-on-lan packet")
+			})
+			if err := token.Error(); err != nil {
+				log := log.WithError(err)
+				log.AddField("topic", topic)
+				log.Error("could not subscribe to MQTT topic")
+			}
+		}
+	})
+
+	broker := mqtt.NewClient(brokerOptions)
+	if token := broker.Connect(); token.Error() != nil {
+		log.WithError(token.Error()).Fatal("could not connect to MQTT broker")
+	}
+
+	select {}
+}