feat: tray icon, connect to home assistant, misc functions

internal/app/tray.go

@@ -0,0 +1,87 @@
+package app
+
+import (
+	"fmt"
+	"ha-tray/internal"
+	"time"
+
+	"github.com/getlantern/systray"
+)
+
+type IconReference string
+
+const (
+	IconOpen    IconReference = "open"
+	IconClosed  IconReference = "closed"
+	IconUnknown IconReference = "unknown"
+)
+
+// Path returns the path to the icon file
+func (i IconReference) Path() string {
+	switch i {
+	case IconOpen:
+		return "resources/open.ico"
+	case IconClosed:
+		return "resources/closed.ico"
+	default:
+		return "resources/unknown.ico"
+	}
+}
+
+type Tray struct {
+	active      bool
+	currentIcon IconReference
+}
+
+func (t *Tray) SetIcon(icon IconReference) error {
+	if !t.active {
+		return fmt.Errorf("tray is not active")
+	}
+
+	iconBytes, err := internal.Icons.ReadFile(icon.Path())
+	if err != nil {
+		return fmt.Errorf("failed to read icon: %w", err)
+	}
+	systray.SetIcon(iconBytes)
+	t.currentIcon = icon
+
+	return nil
+}
+
+func (t *Tray) Start(title string) error {
+	if t.active {
+		return fmt.Errorf("tray is already active")
+	}
+
+	readyTimeout := make(chan struct{}, 1)
+	go systray.Run(func() {
+		systray.SetTitle(title)
+		systray.SetTooltip(title)
+
+		readyTimeout <- struct{}{}
+		close(readyTimeout)
+	}, func() {
+		t.active = false
+	})
+
+	select {
+	case <-readyTimeout:
+		fmt.Println("tray started")
+		t.active = true
+		return nil
+	case <-time.After(5 * time.Second):
+		close(readyTimeout)
+		return fmt.Errorf("tray did not start in time")
+	}
+}
+
+func (t *Tray) Stop() error {
+	if !t.active {
+		return fmt.Errorf("tray is not active")
+	}
+
+	systray.Quit()
+	t.active = false
+
+	return nil
+}