package main

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

    "postwoman.io/proxy/icons"
    "postwoman.io/proxy/inputbox"
    "postwoman.io/proxy/libproxy"
    "postwoman.io/proxy/notifier"
);

var mStatus *systray.MenuItem;
var mCopyAccessToken *systray.MenuItem;

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

func onReady() {
    systray.SetIcon(icon.Data);
    systray.SetTooltip("Proxyman v1.0.0 - 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 Postwoman Interface
    mOpenPostwoman := systray.AddMenuItem("Open Postwoman", "");

    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 Proxywoman", "");

    /** Start proxy server. **/
    go runPostwomanProxy();

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

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

            case <- mViewHelp.ClickedCh:
                _ = browser.OpenURL("https://github.com/postwoman-io/proxywoman/wiki");

            case <- mSetAccessToken.ClickedCh:
                newAccessToken, success := inputbox.InputBox("Proxywoman", "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("Proxywoman", "Proxy Access check disabled.", "**Anyone can access your proxy server!** The Proxy Access Token check has been disabled.", notifier.GetIcon());
                    }else{
                        _ = notifier.Notify("Proxywoman", "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/NBTX/postwoman-proxy");

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

func onExit() {

}

func runPostwomanProxy() {
    libproxy.Initialize("postwoman", "127.0.0.1:9159", onProxyStateChange, true, nil);
}

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

    if isListening {
        mCopyAccessToken.Enable();
    }
}