main.go

package main

import (
	"github.com/atotto/clipboard"
	"github.com/getlantern/systray"
	"github.com/pkg/browser"

	icon "github.com/hoppscotch/proxyscotch/icons"
	"github.com/hoppscotch/proxyscotch/inputbox"
	"github.com/hoppscotch/proxyscotch/libproxy"
	"github.com/hoppscotch/proxyscotch/notifier"
)

var (
	VersionName string
	VersionCode string
)

var (
	mStatus          *systray.MenuItem
	mCopyAccessToken *systray.MenuItem
)

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

func onReady() {
	systray.SetIcon(icon.Data)
	systray.SetTooltip("Proxyscotch v" + VersionName + " (" + VersionCode + ") - created by NBTX")

	/** Set up menu items. **/

	// Status
	mStatus = systray.AddMenuItem("Starting...", "")
	mStatus.Disable()
	mCopyAccessToken = systray.AddMenuItem("Copy Access Token...", "")
	mCopyAccessToken.Disable()

	systray.AddSeparator()

	// Open Hoppscotch Interface
	mOpenHoppscotch := systray.AddMenuItem("Open Hoppscotch", "")

	systray.AddSeparator()

	// View Help
	mViewHelp := systray.AddMenuItem("Help...", "")
	// Set Proxy Authentication Token
	mSetAccessToken := systray.AddMenuItem("Set Access Token...", "")
	// Check for Updates
	mUpdateCheck := systray.AddMenuItem("Check for Updates...", "")

	systray.AddSeparator()

	// Quit Proxy
	mQuit := systray.AddMenuItem("Quit Proxyscotch", "")

	/** Start proxy server. **/
	go runHoppscotchProxy()

	/** Wait for menu input. **/
	for {
		select {
		case <-mOpenHoppscotch.ClickedCh:
			_ = browser.OpenURL("https://hoppscotch.io/")

		case <-mCopyAccessToken.ClickedCh:
			_ = clipboard.WriteAll(libproxy.GetAccessToken())
			_ = notifier.Notify("Proxyscotch", "Proxy Access Token copied...", "The Proxy Access Token has been copied to your clipboard.", notifier.GetIcon())

		case <-mViewHelp.ClickedCh:
			_ = browser.OpenURL("https://github.com/hoppscotch/proxyscotch/wiki")

		case <-mSetAccessToken.ClickedCh:
			newAccessToken, success := inputbox.InputBox("Proxyscotch", "Please enter the new Proxy Access Token...\n(Leave this blank to disable access checks.)", "")
			if success {
				libproxy.SetAccessToken(newAccessToken)

				if len(newAccessToken) == 0 {
					_ = notifier.Notify("Proxyscotch", "Proxy Access check disabled.", "**Anyone can access your proxy server!** The Proxy Access Token check has been disabled.", notifier.GetIcon())
				} else {
					_ = notifier.Notify("Proxyscotch", "Proxy Access Token updated...", "The Proxy Access Token has been updated.", notifier.GetIcon())
				}
			}

		case <-mUpdateCheck.ClickedCh:
			// TODO: Add update check.
			_ = browser.OpenURL("https://github.com/hoppscotch/proxyscotch")

		case <-mQuit.ClickedCh:
			systray.Quit()
			return
		}
	}
}

func onExit() {
}

func runHoppscotchProxy() {
	libproxy.Initialize("hoppscotch", "127.0.0.1:9159", "https://hoppscotch.io", "", "", onProxyStateChange, true, nil)
}

func onProxyStateChange(status string, isListening bool) {
	mStatus.SetTitle(status)

	if isListening {
		mCopyAccessToken.Enable()
	}
}