tui.go

package main

import (
	"fmt"

	"github.com/AtlasInsideCorp/UTMStackInstaller/utils"

	"github.com/rivo/tview"
)

const (
	actionPageIndex     = "0"
	masterPageIndex     = "1"
	probePageIndex      = "2"
	uninstallPageIndex  = "3"
	errTag              = "ERROR: "
	logFileAnnouncement = "Please see the logs in /var/log/utm-setup.log for more details."
)

func tui() {
	app := tview.NewApplication()
	pages := tview.NewPages()

	pages.AddPage(actionPageIndex, actionPage(pages, app), true, true)
	pages.AddPage(masterPageIndex, masterPage(pages, app), true, false)
	pages.AddPage(probePageIndex, probePage(pages, app), true, false)
	pages.AddPage(uninstallPageIndex, uninstallPage(pages, app), false, false)

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

func center(width, height int, p tview.Primitive) tview.Primitive {
	return tview.NewFlex().
		AddItem(nil, 0, 1, false).
		AddItem(tview.NewFlex().
			SetDirection(tview.FlexRow).
			AddItem(nil, 0, 1, false).
			AddItem(p, height, 1, true).
			AddItem(nil, 0, 1, false), width, 1, true).
		AddItem(nil, 0, 1, false)
}

func alert(pages *tview.Pages, message string) *tview.Pages {
	id := "alert-dialog"
	return pages.AddPage(
		id,
		tview.NewModal().
			SetText(message).
			AddButtons([]string{"Accept"}).
			SetDoneFunc(func(buttonIndex int, buttonLabel string) {
				pages.HidePage(id).RemovePage(id)
			}),
		false,
		true,
	)
}

func showResults(pages *tview.Pages, app *tview.Application, message string) *tview.Pages {
	id := "results-dialog"
	return pages.AddPage(
		id,
		tview.NewModal().
			SetText(message).
			AddButtons([]string{"Quit"}).
			SetDoneFunc(func(buttonIndex int, buttonLabel string) {
				app.Stop()
			}),
		false,
		true,
	)
}

func actionPage(pages *tview.Pages, app *tview.Application) tview.Primitive {
	list := tview.NewList().
		AddItem("Install Master", "", 'a', func() {
			pages.SwitchToPage(masterPageIndex)
		}).
		AddItem("Install Probe", "", 'b', func() {
			pages.SwitchToPage(probePageIndex)
		}).
		AddItem("Remove UTMStack", "", 'c', func() {
			pages.SwitchToPage(uninstallPageIndex)
		}).
		AddItem("Quit", "", 'q', func() {
			app.Stop()
		})
	list.SetBorder(true).SetTitle("UTMStack - Select Operation").SetTitleAlign(tview.AlignCenter).SetBorderPadding(1, 0, 1, 0)
	return center(33, 12, list)
}

func uninstallPage(pages *tview.Pages, app *tview.Application) tview.Primitive {
	return tview.NewModal().SetText("Are you sure you want to remove UTMStack?").
		AddButtons([]string{"Remove", "Cancel"}).
		SetDoneFunc(func(buttonIndex int, buttonLabel string) {
			if buttonIndex == 0 {
				pages.AddPage(
					"uninstalling-dialog",
					tview.NewModal().SetText("Removing... This may take several minutes. Please wait."),
					false,
					true,
				)
				pages.HidePage(uninstallPageIndex)
				go func() {
					err := utils.Uninstall("ui")
					var msg string
					if err != nil {
						msg = errTag + err.Error() + ". " + logFileAnnouncement
					} else {
						msg = "Successfully removed. " + logFileAnnouncement
					}
					showResults(pages, app, msg)
					app.Draw()
				}()
			} else {
				pages.SwitchToPage(actionPageIndex)
			}
		})
}

func masterPage(pages *tview.Pages, app *tview.Application) tview.Primitive {
	form := tview.NewForm()
	form.AddButton("Back", func() {
		pages.SwitchToPage(actionPageIndex)
	}).AddButton("Install", func() {
		datadir := "/utmstack"
		dbPass := utils.GenerateSecret()

		pages.AddPage(
			"installing-dialog",
			tview.NewModal().SetText("Installing... This may take several minutes, please wait."),
			false,
			true,
		)

		pages.HidePage(masterPageIndex)
		go func() {
			err := utils.InstallMaster("ui", datadir, dbPass, tag, lite)
			var msg string
			if err != nil {
				msg = errTag + err.Error() + ". " + logFileAnnouncement
			} else {
				msg = fmt.Sprintf("Successfully installed. Master password: %s", dbPass)
			}
			showResults(pages, app, msg)
			app.Draw()
		}()
	}).AddButton("Quit", func() {
		app.Stop()
	})
	form.SetBorder(true).SetTitle("Install Master").SetTitleAlign(tview.AlignCenter)
	return center(46, 15, form)
}

func probePage(pages *tview.Pages, app *tview.Application) tview.Primitive {
	form := tview.NewForm().
		AddPasswordField("Master password", "", 25, '*', nil).
		AddPasswordField("Confirm master password", "", 25, '*', nil).
		AddInputField("Master IP or FQDN", "", 25, nil, nil)
	form.AddButton("Back", func() {
		pages.SwitchToPage(actionPageIndex)
	}).AddButton("Install", func() {
		datadir := "/utmstack"
		dbPass := form.GetFormItem(0).(*tview.InputField).GetText()
		confirmDBPass := form.GetFormItem(1).(*tview.InputField).GetText()
		host := form.GetFormItem(2).(*tview.InputField).GetText()

		if dbPass == "" || confirmDBPass == "" || host == "" {
			alert(pages, "You must provide all requested data.")
		} else if dbPass != confirmDBPass {
			alert(pages, "Password confirmation does not match")
		} else {
			pages.AddPage(
				"installing-dialog",
				tview.NewModal().SetText("Installing... This may take several minutes. Please wait."),
				false,
				true,
			)
			pages.HidePage(probePageIndex)
			go func() {
				err := utils.InstallProbe("ui", datadir, dbPass, host, tag, lite)
				var msg string
				if err != nil {
					msg = errTag + err.Error() + ". " + logFileAnnouncement
				} else {
					msg = "Successfully installed. " + logFileAnnouncement
				}
				showResults(pages, app, msg)
				app.Draw()
			}()
		}
	}).AddButton("Quit", func() {
		app.Stop()
	})
	form.SetBorder(true).SetTitle("Install Probe").SetTitleAlign(tview.AlignCenter)
	return center(46, 11, form)
}