jfa-go/tray.go

//go:build tray
// +build tray

package main

import (
	"log"
	"os"
	"os/signal"
	"runtime"
	"syscall"
	"time"

	"github.com/lutischan-ferenc/systray"
)

var TRAY = true

func BuildTagsTray() {
	buildTags = append(buildTags, "tray")
}

func RunTray() {
	systray.Run(onReady, onExit)
}

func QuitTray() {
	systray.Quit()
}

func onExit() {
	if RUNNING {
		QUIT = true
		RESTART <- true
		// Safety Sleep (Ensure shutdown tasks get done)
		time.Sleep(time.Second)
	}
	os.Remove(SOCK)
}

func onReady() {
	var icon []byte
	var err error = nil
	if runtime.GOOS == "windows" {
		icon, err = localFS.ReadFile("web/favicon.ico")
	} else {
		icon, err = localFS.ReadFile("web/favicon-256x256.png")
	}
	if err != nil {
		log.Fatalf("Failed to load favicon: %v", err)
	}
	systray.SetIcon(icon)
	systray.SetTitle("jfa-go")
	mStart := systray.AddMenuItem("Start", "Start jfa-go")
	mStop := systray.AddMenuItem("Stop", "Stop jfa-go")
	mRestart := systray.AddMenuItem("Restart", "Restart jfa-go")
	mOpenLogs := systray.AddMenuItem("Open logs", "Open jfa-go log file.")
	as := NewAutostart("jfa-go", "A user management system for Jellyfin", "Run on login", "Run jfa-go on user login.")
	mQuit := systray.AddMenuItem("Quit", "Quit jfa-go")

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
	go func() {
		<-c
		systray.Quit()
		os.Exit(1)
	}()

	defer func() {
		systray.Quit()
	}()

	RESTART = make(chan bool, 1)
	TRAYRESTART = make(chan bool, 1)
	go start(false, true)
	mStart.Disable()
	mStop.Enable()
	mRestart.Enable()
	as.Register()
	trayRestart := func() {
		if RUNNING {
			RESTART <- true
			mStop.Disable()
			mStart.Enable()
			mRestart.Disable()
			for {
				if !RUNNING {
					break
				}
			}
			go start(false, false)
			mStart.Disable()
			mStop.Enable()
			mRestart.Enable()
		}
	}

	mStart.Click(func() {
		if !RUNNING {
			go start(false, false)
			mStart.Disable()
			mStop.Enable()
			mRestart.Enable()
		}
	})
	mStop.Click(func() {
		if RUNNING {
			RESTART <- true
			mStop.Disable()
			mStart.Enable()
			mRestart.Disable()
		}
	})
	mRestart.Click(func() {
		trayRestart()
	})
	mOpenLogs.Click(func() {
		log.Printf("Opening %s\n", logPath)
		OpenFile(logPath)
	})
	mQuit.Click(func() {
		systray.Quit()
	})
	for {
		select {
		case <-TRAYRESTART:
			trayRestart()
		}
	}
}