streamdeck-remote-server/main.go

package main

import (
	"github.com/atotto/clipboard"
	"github.com/getlantern/systray"
	"io/ioutil"
	"log"
	"math/rand"
	"meow.tf/streamdeck-remote-server/icon"
	"net/http"
	"os"
	"time"
)

func init() {
	rand.Seed(time.Now().UnixNano())
}

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

func onReady() {
	systray.SetIcon(icon.Data)
	systray.SetTitle("StreamDeck Remote")
	systray.SetTooltip("StreamDeck Remote Server (Running)")

	mCopy := systray.AddMenuItem("Copy Token", "Copy the token to the clipboard")
	mQuit := systray.AddMenuItem("Quit", "Close StreamDeck Remote Server")

	go listenServer()

	go func() {
		for {
			select {
			case <-mCopy.ClickedCh:
				clipboard.WriteAll(token)
			case <-mQuit.ClickedCh:
				systray.Quit()
			}
		}
	}()
}

func onExit() {
	// clean up here
}

var (
	token string
)

func loadToken() {
	if _, err := os.Stat("token.txt"); os.IsNotExist(err) {
		token = RandStringRunes(32)

		f, err := os.Create("token.txt")

		if err != nil {
			log.Fatalln("Unable to save token:", err)
		}

		defer f.Close()

		f.WriteString(token)
	} else {
		f, err := os.Open("token.txt")

		if err != nil {
			log.Fatalln("Unable to open token file:", err)
		}

		defer f.Close()

		b, err := ioutil.ReadAll(f)

		if err != nil {
			f.Close()
			log.Fatalln("Unable to read token file:", err)
		}

		token = string(b)
	}
}

func listenServer() {
	loadToken()

	if _, err := os.Stat("server.crt"); os.IsNotExist(err) {
		err = generateCert("server.crt", "server.key")

		if err != nil {
			log.Fatalln("Unable to generate certificates:", err)
		}
	}

	mux := http.NewServeMux()

	mux.HandleFunc("/url/open", authenticationHandler(requirePost(urlOpenHandler)))
	mux.HandleFunc("/cmd/open", authenticationHandler(requirePost(openHandler)))

	http.ListenAndServeTLS(":4443", "server.crt", "server.key", mux)
}