main.go

package main

import (
	"xoon/ui"
	"xoon/utils"

	"github.com/rivo/tview"
)

var rootFlex *tview.Flex
var loginForm *tview.Form
var mainFlex *tview.Flex
var app *tview.Application

func main() {
	utils.PasswordProtectionInit()
	utils.InitJupiter()
	utils.XoosInit()

	app = tview.NewApplication()
	rootFlex = tview.NewFlex().SetDirection(tview.FlexRow)

	// Check for existing wallet
	partialPublicKey := utils.CheckExistingWallet()

	if partialPublicKey != "" {
		// Wallet exists, show login screen
		showLoginForm(app)
	} else {
		// No wallet exists, show main interface directly
		showMainInterface(app)
	}

	ui.SetupInputCapture(app, func() {
		// Clean up all UI elements
		rootFlex.Clear()
		loginForm = nil
		mainFlex = nil
	})

	app.SetRoot(rootFlex, true).EnableMouse(true)
	if err := app.Run(); err != nil {
		utils.ClearGlobalKeys()
		panic(err)
	}
}

func showErrorModal(message string) {
	modal := tview.NewModal().
		SetText(message).
		AddButtons([]string{"OK"}).
		SetDoneFunc(func(buttonIndex int, buttonLabel string) {
			if buttonLabel == "OK" {
				app.SetRoot(rootFlex, true)
			}
		})

	app.SetRoot(modal, false)
}

func showLoginForm(app *tview.Application) {
	var passwordFieldIndex int

	loginForm = tview.NewForm().
		AddTextView("Instructions", "Please input password for existing encrypted wallet to unlock unmineable solXEN Miner", 0, 2, false, false)

	passwordFieldIndex = loginForm.GetFormItemCount()
	loginForm.AddPasswordField("Password:", "", 32, '*', nil)

	loginForm.AddButton("Unlock", func() {
		password := loginForm.GetFormItem(passwordFieldIndex).(*tview.InputField).GetText()
		if utils.VerifyPassword(password) {
			showMainInterface(app)
		} else {
			showErrorModal("Invalid password")
		}
	}).
		AddButton("Quit", func() {
			app.Stop()
		})

	loginForm.SetBorder(true).SetTitle("Unlock umineable solXEN Miner")
	rootFlex.Clear()
	rootFlex.AddItem(loginForm, 0, 1, true)
}

func showMainInterface(app *tview.Application) {
	mainMenu := ui.CreateMainMenu()
	rightFlex := tview.NewFlex().SetDirection(tview.FlexRow)

	walletUI := ui.CreateWalletUI(app)
	solXENCPUUI := ui.CreateSolXENCPUUI(app)
	solXENNvidiaGPUUI := ui.CreateSolXENNvidiaGPUUI(app)
	solXENAMDGPUUI := ui.CreateSolXENAMDGPUUI(app)
	tokenharvestUI := ui.CreateTokenHarvestUI(app)
	switchView := ui.CreateSwitchViewFunc(rightFlex, mainMenu)

	modules := []ui.ModuleUI{
		{
			DashboardFlex: walletUI.DashboardFlex,
			ConfigFlex:    walletUI.ConfigFlex,
			LogView:       walletUI.LogView,
		},
		{
			DashboardFlex: solXENCPUUI.DashboardFlex,
			ConfigFlex:    solXENCPUUI.ConfigFlex,
			LogView:       solXENCPUUI.LogView,
		},
		{
			DashboardFlex: solXENNvidiaGPUUI.DashboardFlex,
			ConfigFlex:    solXENNvidiaGPUUI.ConfigFlex,
			LogView:       solXENNvidiaGPUUI.LogView,
		},
		{
			DashboardFlex: solXENAMDGPUUI.DashboardFlex,
			ConfigFlex:    solXENAMDGPUUI.ConfigFlex,
			LogView:       solXENAMDGPUUI.LogView,
		},
		{
			DashboardFlex: tokenharvestUI.DashboardFlex,
			ConfigFlex:    tokenharvestUI.ConfigFlex,
			LogView:       tokenharvestUI.LogView,
		},
	}

	ui.SetupMenuItemSelection(mainMenu, switchView, modules)

	mainFlex = tview.NewFlex().
		AddItem(mainMenu, 0, 1, true).
		AddItem(rightFlex, 0, 3, false)

	rootFlex.Clear()
	rootFlex.AddItem(mainFlex, 0, 1, true)
}