[rbdr/mobius] / hotline / ui.go

package hotline

import (
	"fmt"
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
	"gopkg.in/yaml.v3"
	"os"
	"strconv"
	"strings"
)

type UI struct {
	chatBox     *tview.TextView
	chatInput   *tview.InputField
	App         *tview.Application
	Pages       *tview.Pages
	userList    *tview.TextView
	trackerList *tview.List
	HLClient    *Client
}

// pages
const (
	pageServerUI = "serverUI"
)

func NewUI(c *Client) *UI {
	app := tview.NewApplication()
	chatBox := tview.NewTextView().
		SetScrollable(true).
		SetDynamicColors(true).
		SetWordWrap(true).
		SetChangedFunc(func() {
			app.Draw() // TODO: docs say this is bad but it's the only way to show content during initial render??
		})
	chatBox.Box.SetBorder(true).SetTitle("| Chat |")

	chatInput := tview.NewInputField()
	chatInput.
		SetLabel("> ").
		SetFieldBackgroundColor(tcell.ColorDimGray).
		SetDoneFunc(func(key tcell.Key) {
			// skip send if user hit enter with no other text
			if len(chatInput.GetText()) == 0 {
				return
			}

			_ = c.Send(
				*NewTransaction(TranChatSend, nil,
					NewField(FieldData, []byte(chatInput.GetText())),
				),
			)
			chatInput.SetText("") // clear the input field after chat send
		})

	chatInput.Box.SetBorder(true).SetTitle("Send")

	userList := tview.
		NewTextView().
		SetDynamicColors(true).
		SetChangedFunc(func() {
			app.Draw() // TODO: docs say this is bad but it's the only way to show content during initial render??
		})
	userList.Box.SetBorder(true).SetTitle("Users")

	return &UI{
		App:         app,
		chatBox:     chatBox,
		Pages:       tview.NewPages(),
		chatInput:   chatInput,
		userList:    userList,
		trackerList: tview.NewList(),
		HLClient:    c,
	}
}

func (ui *UI) showBookmarks() *tview.List {
	list := tview.NewList()
	list.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if event.Key() == tcell.KeyEsc {
			ui.Pages.SwitchToPage("home")
		}
		return event
	})
	list.Box.SetBorder(true).SetTitle("| Bookmarks |")

	shortcut := 97 // rune for "a"
	for i, srv := range ui.HLClient.Pref.Bookmarks {
		addr := srv.Addr
		login := srv.Login
		pass := srv.Password
		list.AddItem(srv.Name, srv.Addr, rune(shortcut+i), func() {
			ui.Pages.RemovePage("joinServer")

			newJS := ui.renderJoinServerForm("", addr, login, pass, "bookmarks", true, true)

			ui.Pages.AddPage("joinServer", newJS, true, true)
		})
	}

	return list
}

func (ui *UI) getTrackerList() *tview.List {
	listing, err := GetListing(ui.HLClient.Pref.Tracker)
	if err != nil {
		// TODO
	}

	list := tview.NewList()
	list.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if event.Key() == tcell.KeyEsc {
			ui.Pages.SwitchToPage("home")
		}
		return event
	})
	list.Box.SetBorder(true).SetTitle("| Servers |")

	shortcut := 97 // rune for "a"
	for i, srv := range listing {
		addr := srv.Addr()
		srvName := srv.Name
		list.AddItem(string(srv.Name), string(srv.Description), rune(shortcut+i), func() {
			ui.Pages.RemovePage("joinServer")

			newJS := ui.renderJoinServerForm(string(srvName), addr, GuestAccount, "", trackerListPage, false, true)

			ui.Pages.AddPage("joinServer", newJS, true, true)
			ui.Pages.ShowPage("joinServer")
		})
	}

	return list
}

func (ui *UI) renderSettingsForm() *tview.Flex {
	iconStr := strconv.Itoa(ui.HLClient.Pref.IconID)
	settingsForm := tview.NewForm()
	settingsForm.AddInputField("Your Name", ui.HLClient.Pref.Username, 0, nil, nil)
	settingsForm.AddInputField("IconID", iconStr, 0, func(idStr string, _ rune) bool {
		_, err := strconv.Atoi(idStr)
		return err == nil
	}, nil)
	settingsForm.AddInputField("Tracker", ui.HLClient.Pref.Tracker, 0, nil, nil)
	settingsForm.AddCheckbox("Enable Terminal Bell", ui.HLClient.Pref.EnableBell, nil)
	settingsForm.AddButton("Save", func() {
		usernameInput := settingsForm.GetFormItem(0).(*tview.InputField).GetText()
		if len(usernameInput) == 0 {
			usernameInput = "unnamed"
		}
		ui.HLClient.Pref.Username = usernameInput
		iconStr = settingsForm.GetFormItem(1).(*tview.InputField).GetText()
		ui.HLClient.Pref.IconID, _ = strconv.Atoi(iconStr)
		ui.HLClient.Pref.Tracker = settingsForm.GetFormItem(2).(*tview.InputField).GetText()
		ui.HLClient.Pref.EnableBell = settingsForm.GetFormItem(3).(*tview.Checkbox).IsChecked()

		out, err := yaml.Marshal(&ui.HLClient.Pref)
		if err != nil {
			// TODO: handle err
		}
		// TODO: handle err
		err = os.WriteFile(ui.HLClient.cfgPath, out, 0666)
		if err != nil {
			println(ui.HLClient.cfgPath)
			panic(err)
		}
		ui.Pages.RemovePage("settings")
	})
	settingsForm.SetBorder(true)
	settingsForm.SetCancelFunc(func() {
		ui.Pages.RemovePage("settings")
	})
	settingsPage := tview.NewFlex().SetDirection(tview.FlexRow)
	settingsPage.Box.SetBorder(true).SetTitle("Settings")
	settingsPage.AddItem(settingsForm, 0, 1, true)

	centerFlex := tview.NewFlex().
		AddItem(nil, 0, 1, false).
		AddItem(tview.NewFlex().
			SetDirection(tview.FlexRow).
			AddItem(nil, 0, 1, false).
			AddItem(settingsForm, 15, 1, true).
			AddItem(nil, 0, 1, false), 40, 1, true).
		AddItem(nil, 0, 1, false)

	return centerFlex
}

func (ui *UI) joinServer(addr, login, password string) error {
	// append default port to address if no port supplied
	if len(strings.Split(addr, ":")) == 1 {
		addr += ":5500"
	}
	if err := ui.HLClient.Connect(addr, login, password); err != nil {
		return fmt.Errorf("Error joining server: %v\n", err)
	}

	go func() {
		if err := ui.HLClient.HandleTransactions(); err != nil {
			ui.Pages.SwitchToPage("home")
		}

		loginErrModal := tview.NewModal().
			AddButtons([]string{"Ok"}).
			SetText("The server connection has closed.").
			SetDoneFunc(func(buttonIndex int, buttonLabel string) {
				ui.Pages.SwitchToPage("home")
			})
		loginErrModal.Box.SetTitle("Server Connection Error")

		ui.Pages.AddPage("loginErr", loginErrModal, false, true)
		ui.App.Draw()
	}()

	return nil
}

func (ui *UI) renderJoinServerForm(name, server, login, password, backPage string, save, defaultConnect bool) *tview.Flex {
	joinServerForm := tview.NewForm()
	joinServerForm.
		//	AddInputField("Name", server, 0, func(textToCheck string, lastChar rune) bool {
		//	return false
		// }, nil).
		AddInputField("Server", server, 0, nil, nil).
		AddInputField("Login", login, 0, nil, nil).
		AddPasswordField("Password", password, 0, '*', nil).
		AddCheckbox("Save", save, func(checked bool) {
			ui.HLClient.Logger.Infow("saving bookmark")
			// TODO: Implement bookmark saving

			ui.HLClient.Pref.AddBookmark(joinServerForm.GetFormItem(0).(*tview.InputField).GetText(), joinServerForm.GetFormItem(0).(*tview.InputField).GetText(), joinServerForm.GetFormItem(1).(*tview.InputField).GetText(), joinServerForm.GetFormItem(2).(*tview.InputField).GetText())
			out, err := yaml.Marshal(ui.HLClient.Pref)
			if err != nil {
				panic(err)
			}

			err = os.WriteFile(ui.HLClient.cfgPath, out, 0666)
			if err != nil {
				panic(err)
			}
			// 		Pref := ui.HLClient.Pref
		}).
		AddButton("Cancel", func() {
			ui.Pages.SwitchToPage(backPage)
		}).
		AddButton("Connect", func() {
			srvAddr := joinServerForm.GetFormItem(0).(*tview.InputField).GetText()
			loginInput := joinServerForm.GetFormItem(1).(*tview.InputField).GetText()
			err := ui.joinServer(
				srvAddr,
				loginInput,
				joinServerForm.GetFormItem(2).(*tview.InputField).GetText(),
			)
			if name == "" {
				name = fmt.Sprintf("%s@%s", loginInput, srvAddr)
			}
			ui.HLClient.serverName = name

			if err != nil {
				ui.HLClient.Logger.Errorw("login error", "err", err)
				loginErrModal := tview.NewModal().
					AddButtons([]string{"Oh no"}).
					SetText(err.Error()).
					SetDoneFunc(func(buttonIndex int, buttonLabel string) {
						ui.Pages.SwitchToPage(backPage)
					})

				ui.Pages.AddPage("loginErr", loginErrModal, false, true)
			}

			// Save checkbox
			if joinServerForm.GetFormItem(3).(*tview.Checkbox).IsChecked() {
				// TODO: implement bookmark saving
			}
		})

	joinServerForm.Box.SetBorder(true).SetTitle("| Connect |")
	joinServerForm.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if event.Key() == tcell.KeyEscape {
			ui.Pages.SwitchToPage(backPage)
		}
		return event
	})

	if defaultConnect {
		joinServerForm.SetFocus(5)
	}

	joinServerPage := tview.NewFlex().
		AddItem(nil, 0, 1, false).
		AddItem(tview.NewFlex().
			SetDirection(tview.FlexRow).
			AddItem(nil, 0, 1, false).
			AddItem(joinServerForm, 14, 1, true).
			AddItem(nil, 0, 1, false), 40, 1, true).
		AddItem(nil, 0, 1, false)

	return joinServerPage
}

func (ui *UI) renderServerUI() *tview.Flex {
	ui.chatBox.SetText("") // clear any previously existing chatbox text
	commandList := tview.NewTextView().SetDynamicColors(true)
	commandList.
		SetText("[yellow]^n[-::]: Read News   [yellow]^p[-::]: Post News\n[yellow]^l[-::]: View Logs   [yellow]^f[-::]: View Files\n").
		SetBorder(true).
		SetTitle("| Keyboard Shortcuts| ")

	modal := tview.NewModal().
		SetText("Disconnect from the server?").
		AddButtons([]string{"Cancel", "Exit"}).
		SetFocus(1)
	modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
		if buttonIndex == 1 {
			_ = ui.HLClient.Disconnect()
			ui.Pages.RemovePage(pageServerUI)
			ui.Pages.SwitchToPage("home")
		} else {
			ui.Pages.HidePage("modal")
		}
	})

	serverUI := tview.NewFlex().
		AddItem(tview.NewFlex().
			SetDirection(tview.FlexRow).
			AddItem(commandList, 4, 0, false).
			AddItem(ui.chatBox, 0, 8, false).
			AddItem(ui.chatInput, 3, 0, true), 0, 1, true).
		AddItem(ui.userList, 25, 1, false)
	serverUI.SetBorder(true).SetTitle("| Mobius - Connected to " + ui.HLClient.serverName + " |").SetTitleAlign(tview.AlignLeft)
	serverUI.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if event.Key() == tcell.KeyEscape {
			ui.Pages.AddPage("modal", modal, false, true)
		}

		// List files
		if event.Key() == tcell.KeyCtrlF {
			if err := ui.HLClient.Send(*NewTransaction(TranGetFileNameList, nil)); err != nil {
				ui.HLClient.Logger.Errorw("err", "err", err)
			}
		}

		// Show News
		if event.Key() == tcell.KeyCtrlN {
			if err := ui.HLClient.Send(*NewTransaction(TranGetMsgs, nil)); err != nil {
				ui.HLClient.Logger.Errorw("err", "err", err)
			}
		}

		// Post news
		if event.Key() == tcell.KeyCtrlP {
			newsFlex := tview.NewFlex()
			newsFlex.SetBorderPadding(0, 0, 1, 1)
			newsPostTextArea := tview.NewTextView()
			newsPostTextArea.SetBackgroundColor(tcell.ColorDarkSlateGrey)
			newsPostTextArea.SetChangedFunc(func() {
				ui.App.Draw() // TODO: docs say this is bad but it's the only way to show content during initial render??
			})

			newsPostForm := tview.NewForm().
				SetButtonsAlign(tview.AlignRight).
				// AddButton("Cancel", nil). // TODO: implement cancel button behavior
				AddButton("Send", nil)
			newsPostForm.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
				switch event.Key() {
				case tcell.KeyEscape:
					ui.Pages.RemovePage("newsInput")
				case tcell.KeyTab:
					ui.App.SetFocus(newsPostTextArea)
				case tcell.KeyEnter:
					newsText := strings.ReplaceAll(newsPostTextArea.GetText(true), "\n", "\r")
					if len(newsText) == 0 {
						return event
					}
					err := ui.HLClient.Send(
						*NewTransaction(TranOldPostNews, nil,
							NewField(FieldData, []byte(newsText)),
						),
					)
					if err != nil {
						ui.HLClient.Logger.Errorw("Error posting news", "err", err)
						// TODO: display errModal to user
					}
					ui.Pages.RemovePage("newsInput")
				}

				return event
			})

			newsFlex.
				SetDirection(tview.FlexRow).
				SetBorder(true).
				SetTitle("| Post Message |")

			newsPostTextArea.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
				switch event.Key() {
				case tcell.KeyEscape:
					ui.Pages.RemovePage("newsInput")
				case tcell.KeyTab:
					ui.App.SetFocus(newsPostForm)
				case tcell.KeyEnter:
					_, _ = fmt.Fprintf(newsPostTextArea, "\n")
				default:
					const windowsBackspaceRune = 8
					const macBackspaceRune = 127
					switch event.Rune() {
					case macBackspaceRune, windowsBackspaceRune:
						curTxt := newsPostTextArea.GetText(true)
						if len(curTxt) > 0 {
							curTxt = curTxt[:len(curTxt)-1]
							newsPostTextArea.SetText(curTxt)
						}
					default:
						_, _ = fmt.Fprintf(newsPostTextArea, string(event.Rune()))
					}
				}

				return event
			})

			newsFlex.AddItem(newsPostTextArea, 10, 0, true)
			newsFlex.AddItem(newsPostForm, 3, 0, false)

			newsPostPage := tview.NewFlex().
				AddItem(nil, 0, 1, false).
				AddItem(tview.NewFlex().
					SetDirection(tview.FlexRow).
					AddItem(nil, 0, 1, false).
					AddItem(newsFlex, 15, 1, true).
					// AddItem(newsPostForm, 3, 0, false).
					AddItem(nil, 0, 1, false), 40, 1, false).
				AddItem(nil, 0, 1, false)

			ui.Pages.AddPage("newsInput", newsPostPage, true, true)
			ui.App.SetFocus(newsPostTextArea)
		}

		return event
	})
	return serverUI
}

func (ui *UI) Start() {
	home := tview.NewFlex().SetDirection(tview.FlexRow)
	home.Box.SetBorder(true).SetTitle("| Mobius v" + VERSION + " |").SetTitleAlign(tview.AlignLeft)
	mainMenu := tview.NewList()

	bannerItem := tview.NewTextView().
		SetText(randomBanner()).
		SetDynamicColors(true).
		SetTextAlign(tview.AlignCenter)

	home.AddItem(
		tview.NewFlex().AddItem(bannerItem, 0, 1, false),
		14, 1, false)
	home.AddItem(tview.NewFlex().
		AddItem(nil, 0, 1, false).
		AddItem(mainMenu, 0, 1, true).
		AddItem(nil, 0, 1, false),
		0, 1, true,
	)

	mainMenu.AddItem("Join Server", "", 'j', func() {
		joinServerPage := ui.renderJoinServerForm("", "", GuestAccount, "", "home", false, false)
		ui.Pages.AddPage("joinServer", joinServerPage, true, true)
	}).
		AddItem("Bookmarks", "", 'b', func() {
			ui.Pages.AddAndSwitchToPage("bookmarks", ui.showBookmarks(), true)
		}).
		AddItem("Browse Tracker", "", 't', func() {
			ui.trackerList = ui.getTrackerList()
			ui.Pages.AddAndSwitchToPage("trackerList", ui.trackerList, true)
		}).
		AddItem("Settings", "", 's', func() {
			ui.Pages.AddPage("settings", ui.renderSettingsForm(), true, true)
		}).
		AddItem("Quit", "", 'q', func() {
			ui.App.Stop()
		})

	ui.Pages.AddPage("home", home, true, true)

	// App level input capture
	ui.App.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if event.Key() == tcell.KeyCtrlC {
			ui.HLClient.Logger.Infow("Exiting")
			ui.App.Stop()
			os.Exit(0)
		}
		// Show Logs
		if event.Key() == tcell.KeyCtrlL {
			ui.HLClient.DebugBuf.TextView.ScrollToEnd()
			ui.HLClient.DebugBuf.TextView.SetBorder(true).SetTitle("Logs")
			ui.HLClient.DebugBuf.TextView.SetDoneFunc(func(key tcell.Key) {
				if key == tcell.KeyEscape {
					ui.Pages.RemovePage("logs")
				}
			})

			ui.Pages.AddPage("logs", ui.HLClient.DebugBuf.TextView, true, true)
		}
		return event
	})

	if err := ui.App.SetRoot(ui.Pages, true).SetFocus(ui.Pages).Run(); err != nil {
		ui.App.Stop()
		os.Exit(1)
	}
}
Commit	Line	Data
d80b37ca	1	package hotline
b198b22b JH	2
b198b22b JH	3	import (
b198b22b	4	"fmt"
b198b22b JH	5	"github.com/gdamore/tcell/v2"
b198b22b JH	6	"github.com/rivo/tview"
0197c3f5	7	"gopkg.in/yaml.v3"
b198b22b JH	8	"os"
	9	"strconv"
	10	"strings"
	11	)
	12
	13	type UI struct {
e005c191 JH	14	chatBox *tview.TextView
	15	chatInput *tview.InputField
	16	App *tview.Application
	17	Pages *tview.Pages
	18	userList *tview.TextView
e005c191 JH	19	trackerList *tview.List
e005c191 JH	20	HLClient *Client
b198b22b JH	21	}
b198b22b JH	22
e75ba43a JH	23	// pages
	24	const (
	25	pageServerUI = "serverUI"
	26	)
	27
b198b22b JH	28	func NewUI(c Client) UI {
	29	app := tview.NewApplication()
	30	chatBox := tview.NewTextView().
	31	SetScrollable(true).
	32	SetDynamicColors(true).
	33	SetWordWrap(true).
	34	SetChangedFunc(func() {
	35	app.Draw() // TODO: docs say this is bad but it's the only way to show content during initial render??
	36	})
40afb444	37	chatBox.Box.SetBorder(true).SetTitle("\| Chat \|")
b198b22b JH	38
	39	chatInput := tview.NewInputField()
	40	chatInput.
	41	SetLabel("> ").
	42	SetFieldBackgroundColor(tcell.ColorDimGray).
	43	SetDoneFunc(func(key tcell.Key) {
	44	// skip send if user hit enter with no other text
	45	if len(chatInput.GetText()) == 0 {
	46	return
	47	}
	48
e75ba43a	49	_ = c.Send(
d005ef04 JH	50	*NewTransaction(TranChatSend, nil,
d005ef04 JH	51	NewField(FieldData, []byte(chatInput.GetText())),
b198b22b JH	52	),
	53	)
	54	chatInput.SetText("") // clear the input field after chat send
	55	})
	56
	57	chatInput.Box.SetBorder(true).SetTitle("Send")
	58
	59	userList := tview.
	60	NewTextView().
	61	SetDynamicColors(true).
	62	SetChangedFunc(func() {
	63	app.Draw() // TODO: docs say this is bad but it's the only way to show content during initial render??
	64	})
	65	userList.Box.SetBorder(true).SetTitle("Users")
	66
	67	return &UI{
	68	App: app,
	69	chatBox: chatBox,
	70	Pages: tview.NewPages(),
	71	chatInput: chatInput,
	72	userList: userList,
	73	trackerList: tview.NewList(),
b198b22b JH	74	HLClient: c,
	75	}
	76	}
	77
	78	func (ui UI) showBookmarks() tview.List {
	79	list := tview.NewList()
	80	list.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
	81	if event.Key() == tcell.KeyEsc {
	82	ui.Pages.SwitchToPage("home")
	83	}
	84	return event
	85	})
	86	list.Box.SetBorder(true).SetTitle("\| Bookmarks \|")
	87
	88	shortcut := 97 // rune for "a"
d005ef04	89	for i, srv := range ui.HLClient.Pref.Bookmarks {
b198b22b JH	90	addr := srv.Addr
	91	login := srv.Login
	92	pass := srv.Password
	93	list.AddItem(srv.Name, srv.Addr, rune(shortcut+i), func() {
	94	ui.Pages.RemovePage("joinServer")
	95
da1e0d79	96	newJS := ui.renderJoinServerForm("", addr, login, pass, "bookmarks", true, true)
b198b22b JH	97
	98	ui.Pages.AddPage("joinServer", newJS, true, true)
	99	})
	100	}
	101
	102	return list
	103	}
	104
	105	func (ui UI) getTrackerList() tview.List {
d005ef04	106	listing, err := GetListing(ui.HLClient.Pref.Tracker)
b198b22b	107	if err != nil {
d4c152a4	108	// TODO
b198b22b JH	109	}
	110
	111	list := tview.NewList()
	112	list.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
	113	if event.Key() == tcell.KeyEsc {
	114	ui.Pages.SwitchToPage("home")
	115	}
	116	return event
	117	})
	118	list.Box.SetBorder(true).SetTitle("\| Servers \|")
	119
	120	shortcut := 97 // rune for "a"
	121	for i, srv := range listing {
	122	addr := srv.Addr()
e005c191	123	srvName := srv.Name
b198b22b JH	124	list.AddItem(string(srv.Name), string(srv.Description), rune(shortcut+i), func() {
	125	ui.Pages.RemovePage("joinServer")
	126
e005c191	127	newJS := ui.renderJoinServerForm(string(srvName), addr, GuestAccount, "", trackerListPage, false, true)
b198b22b JH	128
	129	ui.Pages.AddPage("joinServer", newJS, true, true)
	130	ui.Pages.ShowPage("joinServer")
	131	})
	132	}
	133
	134	return list
	135	}
	136
	137	func (ui UI) renderSettingsForm() tview.Flex {
d005ef04	138	iconStr := strconv.Itoa(ui.HLClient.Pref.IconID)
b198b22b	139	settingsForm := tview.NewForm()
d005ef04	140	settingsForm.AddInputField("Your Name", ui.HLClient.Pref.Username, 0, nil, nil)
b198b22b JH	141	settingsForm.AddInputField("IconID", iconStr, 0, func(idStr string, _ rune) bool {
	142	_, err := strconv.Atoi(idStr)
	143	return err == nil
	144	}, nil)
d005ef04 JH	145	settingsForm.AddInputField("Tracker", ui.HLClient.Pref.Tracker, 0, nil, nil)
d005ef04 JH	146	settingsForm.AddCheckbox("Enable Terminal Bell", ui.HLClient.Pref.EnableBell, nil)
b198b22b JH	147	settingsForm.AddButton("Save", func() {
	148	usernameInput := settingsForm.GetFormItem(0).(*tview.InputField).GetText()
	149	if len(usernameInput) == 0 {
	150	usernameInput = "unnamed"
	151	}
d005ef04	152	ui.HLClient.Pref.Username = usernameInput
b198b22b	153	iconStr = settingsForm.GetFormItem(1).(*tview.InputField).GetText()
d005ef04 JH	154	ui.HLClient.Pref.IconID, _ = strconv.Atoi(iconStr)
	155	ui.HLClient.Pref.Tracker = settingsForm.GetFormItem(2).(*tview.InputField).GetText()
	156	ui.HLClient.Pref.EnableBell = settingsForm.GetFormItem(3).(*tview.Checkbox).IsChecked()
b198b22b	157
d005ef04	158	out, err := yaml.Marshal(&ui.HLClient.Pref)
b198b22b JH	159	if err != nil {
	160	// TODO: handle err
	161	}
	162	// TODO: handle err
7152b7e5	163	err = os.WriteFile(ui.HLClient.cfgPath, out, 0666)
b198b22b JH	164	if err != nil {
	165	println(ui.HLClient.cfgPath)
	166	panic(err)
	167	}
	168	ui.Pages.RemovePage("settings")
	169	})
	170	settingsForm.SetBorder(true)
	171	settingsForm.SetCancelFunc(func() {
	172	ui.Pages.RemovePage("settings")
	173	})
	174	settingsPage := tview.NewFlex().SetDirection(tview.FlexRow)
	175	settingsPage.Box.SetBorder(true).SetTitle("Settings")
	176	settingsPage.AddItem(settingsForm, 0, 1, true)
	177
	178	centerFlex := tview.NewFlex().
	179	AddItem(nil, 0, 1, false).
	180	AddItem(tview.NewFlex().
	181	SetDirection(tview.FlexRow).
	182	AddItem(nil, 0, 1, false).
	183	AddItem(settingsForm, 15, 1, true).
	184	AddItem(nil, 0, 1, false), 40, 1, true).
	185	AddItem(nil, 0, 1, false)
	186
	187	return centerFlex
	188	}
	189
	190	func (ui *UI) joinServer(addr, login, password string) error {
58efb95f JH	191	// append default port to address if no port supplied
	192	if len(strings.Split(addr, ":")) == 1 {
	193	addr += ":5500"
	194	}
d005ef04	195	if err := ui.HLClient.Connect(addr, login, password); err != nil {
8b7908e6	196	return fmt.Errorf("Error joining server: %v\n", err)
b198b22b JH	197	}
	198
	199	go func() {
d005ef04 JH	200	if err := ui.HLClient.HandleTransactions(); err != nil {
d005ef04 JH	201	ui.Pages.SwitchToPage("home")
b198b22b	202	}
3178ae58	203
d005ef04 JH	204	loginErrModal := tview.NewModal().
	205	AddButtons([]string{"Ok"}).
	206	SetText("The server connection has closed.").
	207	SetDoneFunc(func(buttonIndex int, buttonLabel string) {
	208	ui.Pages.SwitchToPage("home")
	209	})
	210	loginErrModal.Box.SetTitle("Server Connection Error")
3178ae58	211
d005ef04 JH	212	ui.Pages.AddPage("loginErr", loginErrModal, false, true)
d005ef04 JH	213	ui.App.Draw()
b198b22b JH	214	}()
	215
	216	return nil
	217	}
	218
da1e0d79	219	func (ui UI) renderJoinServerForm(name, server, login, password, backPage string, save, defaultConnect bool) tview.Flex {
b198b22b JH	220	joinServerForm := tview.NewForm()
b198b22b JH	221	joinServerForm.
da1e0d79 JH	222	// AddInputField("Name", server, 0, func(textToCheck string, lastChar rune) bool {
da1e0d79 JH	223	// return false
aebc4d36	224	// }, nil).
b198b22b JH	225	AddInputField("Server", server, 0, nil, nil).
	226	AddInputField("Login", login, 0, nil, nil).
	227	AddPasswordField("Password", password, 0, '*', nil).
	228	AddCheckbox("Save", save, func(checked bool) {
da1e0d79	229	ui.HLClient.Logger.Infow("saving bookmark")
b198b22b	230	// TODO: Implement bookmark saving
da1e0d79	231
d005ef04 JH	232	ui.HLClient.Pref.AddBookmark(joinServerForm.GetFormItem(0).(tview.InputField).GetText(), joinServerForm.GetFormItem(0).(tview.InputField).GetText(), joinServerForm.GetFormItem(1).(tview.InputField).GetText(), joinServerForm.GetFormItem(2).(tview.InputField).GetText())
d005ef04 JH	233	out, err := yaml.Marshal(ui.HLClient.Pref)
da1e0d79 JH	234	if err != nil {
	235	panic(err)
	236	}
	237
7152b7e5	238	err = os.WriteFile(ui.HLClient.cfgPath, out, 0666)
da1e0d79 JH	239	if err != nil {
	240	panic(err)
	241	}
d005ef04	242	// Pref := ui.HLClient.Pref
b198b22b JH	243	}).
	244	AddButton("Cancel", func() {
	245	ui.Pages.SwitchToPage(backPage)
	246	}).
	247	AddButton("Connect", func() {
e005c191 JH	248	srvAddr := joinServerForm.GetFormItem(0).(*tview.InputField).GetText()
e005c191 JH	249	loginInput := joinServerForm.GetFormItem(1).(*tview.InputField).GetText()
b198b22b	250	err := ui.joinServer(
e005c191 JH	251	srvAddr,
e005c191 JH	252	loginInput,
b198b22b JH	253	joinServerForm.GetFormItem(2).(*tview.InputField).GetText(),
b198b22b JH	254	)
e005c191 JH	255	if name == "" {
	256	name = fmt.Sprintf("%s@%s", loginInput, srvAddr)
	257	}
	258	ui.HLClient.serverName = name
	259
b198b22b JH	260	if err != nil {
	261	ui.HLClient.Logger.Errorw("login error", "err", err)
	262	loginErrModal := tview.NewModal().
	263	AddButtons([]string{"Oh no"}).
	264	SetText(err.Error()).
	265	SetDoneFunc(func(buttonIndex int, buttonLabel string) {
	266	ui.Pages.SwitchToPage(backPage)
	267	})
	268
	269	ui.Pages.AddPage("loginErr", loginErrModal, false, true)
	270	}
	271
	272	// Save checkbox
	273	if joinServerForm.GetFormItem(3).(*tview.Checkbox).IsChecked() {
	274	// TODO: implement bookmark saving
	275	}
	276	})
	277
	278	joinServerForm.Box.SetBorder(true).SetTitle("\| Connect \|")
	279	joinServerForm.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
	280	if event.Key() == tcell.KeyEscape {
	281	ui.Pages.SwitchToPage(backPage)
	282	}
	283	return event
	284	})
	285
	286	if defaultConnect {
	287	joinServerForm.SetFocus(5)
	288	}
	289
	290	joinServerPage := tview.NewFlex().
	291	AddItem(nil, 0, 1, false).
	292	AddItem(tview.NewFlex().
	293	SetDirection(tview.FlexRow).
	294	AddItem(nil, 0, 1, false).
	295	AddItem(joinServerForm, 14, 1, true).
	296	AddItem(nil, 0, 1, false), 40, 1, true).
	297	AddItem(nil, 0, 1, false)
	298
	299	return joinServerPage
	300	}
	301
	302	func (ui UI) renderServerUI() tview.Flex {
e75ba43a	303	ui.chatBox.SetText("") // clear any previously existing chatbox text
b198b22b JH	304	commandList := tview.NewTextView().SetDynamicColors(true)
b198b22b JH	305	commandList.
e75ba43a	306	SetText("[yellow]^n[-::]: Read News [yellow]^p[-::]: Post News\n[yellow]^l[-::]: View Logs [yellow]^f[-::]: View Files\n").
b198b22b	307	SetBorder(true).
43ecc0f4	308	SetTitle("\| Keyboard Shortcuts\| ")
b198b22b JH	309
	310	modal := tview.NewModal().
	311	SetText("Disconnect from the server?").
	312	AddButtons([]string{"Cancel", "Exit"}).
	313	SetFocus(1)
	314	modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
	315	if buttonIndex == 1 {
	316	_ = ui.HLClient.Disconnect()
e75ba43a	317	ui.Pages.RemovePage(pageServerUI)
3178ae58	318	ui.Pages.SwitchToPage("home")
b198b22b JH	319	} else {
	320	ui.Pages.HidePage("modal")
	321	}
	322	})
	323
	324	serverUI := tview.NewFlex().
	325	AddItem(tview.NewFlex().
	326	SetDirection(tview.FlexRow).
	327	AddItem(commandList, 4, 0, false).
	328	AddItem(ui.chatBox, 0, 8, false).
	329	AddItem(ui.chatInput, 3, 0, true), 0, 1, true).
	330	AddItem(ui.userList, 25, 1, false)
e005c191	331	serverUI.SetBorder(true).SetTitle("\| Mobius - Connected to " + ui.HLClient.serverName + " \|").SetTitleAlign(tview.AlignLeft)
b198b22b JH	332	serverUI.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
	333	if event.Key() == tcell.KeyEscape {
	334	ui.Pages.AddPage("modal", modal, false, true)
	335	}
	336
43ecc0f4 JH	337	// List files
43ecc0f4 JH	338	if event.Key() == tcell.KeyCtrlF {
d005ef04	339	if err := ui.HLClient.Send(*NewTransaction(TranGetFileNameList, nil)); err != nil {
43ecc0f4 JH	340	ui.HLClient.Logger.Errorw("err", "err", err)
	341	}
	342	}
	343
b198b22b JH	344	// Show News
b198b22b JH	345	if event.Key() == tcell.KeyCtrlN {
d005ef04	346	if err := ui.HLClient.Send(*NewTransaction(TranGetMsgs, nil)); err != nil {
b198b22b JH	347	ui.HLClient.Logger.Errorw("err", "err", err)
	348	}
	349	}
	350
	351	// Post news
	352	if event.Key() == tcell.KeyCtrlP {
b198b22b	353	newsFlex := tview.NewFlex()
e75ba43a	354	newsFlex.SetBorderPadding(0, 0, 1, 1)
b198b22b	355	newsPostTextArea := tview.NewTextView()
e75ba43a	356	newsPostTextArea.SetBackgroundColor(tcell.ColorDarkSlateGrey)
b198b22b JH	357	newsPostTextArea.SetChangedFunc(func() {
	358	ui.App.Draw() // TODO: docs say this is bad but it's the only way to show content during initial render??
	359	})
b198b22b JH	360
	361	newsPostForm := tview.NewForm().
	362	SetButtonsAlign(tview.AlignRight).
aebc4d36	363	// AddButton("Cancel", nil). // TODO: implement cancel button behavior
c1ab228e	364	AddButton("Send", nil)
b198b22b JH	365	newsPostForm.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
b198b22b JH	366	switch event.Key() {
e75ba43a JH	367	case tcell.KeyEscape:
e75ba43a JH	368	ui.Pages.RemovePage("newsInput")
b198b22b JH	369	case tcell.KeyTab:
	370	ui.App.SetFocus(newsPostTextArea)
	371	case tcell.KeyEnter:
	372	newsText := strings.ReplaceAll(newsPostTextArea.GetText(true), "\n", "\r")
e75ba43a JH	373	if len(newsText) == 0 {
	374	return event
	375	}
b198b22b	376	err := ui.HLClient.Send(
d005ef04 JH	377	*NewTransaction(TranOldPostNews, nil,
d005ef04 JH	378	NewField(FieldData, []byte(newsText)),
b198b22b JH	379	),
	380	)
	381	if err != nil {
	382	ui.HLClient.Logger.Errorw("Error posting news", "err", err)
	383	// TODO: display errModal to user
	384	}
b198b22b JH	385	ui.Pages.RemovePage("newsInput")
	386	}
	387
	388	return event
	389	})
	390
	391	newsFlex.
	392	SetDirection(tview.FlexRow).
	393	SetBorder(true).
c1ab228e	394	SetTitle("\| Post Message \|")
b198b22b JH	395
b198b22b JH	396	newsPostTextArea.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
b198b22b JH	397	switch event.Key() {
	398	case tcell.KeyEscape:
	399	ui.Pages.RemovePage("newsInput")
	400	case tcell.KeyTab:
	401	ui.App.SetFocus(newsPostForm)
	402	case tcell.KeyEnter:
e75ba43a	403	_, _ = fmt.Fprintf(newsPostTextArea, "\n")
b198b22b JH	404	default:
	405	const windowsBackspaceRune = 8
	406	const macBackspaceRune = 127
	407	switch event.Rune() {
	408	case macBackspaceRune, windowsBackspaceRune:
	409	curTxt := newsPostTextArea.GetText(true)
	410	if len(curTxt) > 0 {
	411	curTxt = curTxt[:len(curTxt)-1]
	412	newsPostTextArea.SetText(curTxt)
	413	}
	414	default:
e75ba43a	415	_, _ = fmt.Fprintf(newsPostTextArea, string(event.Rune()))
b198b22b JH	416	}
	417	}
	418
	419	return event
	420	})
	421
	422	newsFlex.AddItem(newsPostTextArea, 10, 0, true)
	423	newsFlex.AddItem(newsPostForm, 3, 0, false)
	424
	425	newsPostPage := tview.NewFlex().
	426	AddItem(nil, 0, 1, false).
	427	AddItem(tview.NewFlex().
	428	SetDirection(tview.FlexRow).
	429	AddItem(nil, 0, 1, false).
	430	AddItem(newsFlex, 15, 1, true).
aebc4d36	431	// AddItem(newsPostForm, 3, 0, false).
b198b22b JH	432	AddItem(nil, 0, 1, false), 40, 1, false).
	433	AddItem(nil, 0, 1, false)
	434
	435	ui.Pages.AddPage("newsInput", newsPostPage, true, true)
	436	ui.App.SetFocus(newsPostTextArea)
	437	}
	438
	439	return event
	440	})
	441	return serverUI
	442	}
	443
	444	func (ui *UI) Start() {
	445	home := tview.NewFlex().SetDirection(tview.FlexRow)
	446	home.Box.SetBorder(true).SetTitle("\| Mobius v" + VERSION + " \|").SetTitleAlign(tview.AlignLeft)
	447	mainMenu := tview.NewList()
	448
	449	bannerItem := tview.NewTextView().
	450	SetText(randomBanner()).
	451	SetDynamicColors(true).
	452	SetTextAlign(tview.AlignCenter)
	453
	454	home.AddItem(
	455	tview.NewFlex().AddItem(bannerItem, 0, 1, false),
	456	14, 1, false)
	457	home.AddItem(tview.NewFlex().
	458	AddItem(nil, 0, 1, false).
	459	AddItem(mainMenu, 0, 1, true).
	460	AddItem(nil, 0, 1, false),
	461	0, 1, true,
	462	)
	463
	464	mainMenu.AddItem("Join Server", "", 'j', func() {
da1e0d79	465	joinServerPage := ui.renderJoinServerForm("", "", GuestAccount, "", "home", false, false)
b198b22b JH	466	ui.Pages.AddPage("joinServer", joinServerPage, true, true)
	467	}).
	468	AddItem("Bookmarks", "", 'b', func() {
	469	ui.Pages.AddAndSwitchToPage("bookmarks", ui.showBookmarks(), true)
	470	}).
	471	AddItem("Browse Tracker", "", 't', func() {
	472	ui.trackerList = ui.getTrackerList()
	473	ui.Pages.AddAndSwitchToPage("trackerList", ui.trackerList, true)
	474	}).
	475	AddItem("Settings", "", 's', func() {
	476	ui.Pages.AddPage("settings", ui.renderSettingsForm(), true, true)
	477	}).
	478	AddItem("Quit", "", 'q', func() {
	479	ui.App.Stop()
	480	})
	481
	482	ui.Pages.AddPage("home", home, true, true)
	483
	484	// App level input capture
	485	ui.App.SetInputCapture(func(event tcell.EventKey) tcell.EventKey {
	486	if event.Key() == tcell.KeyCtrlC {
	487	ui.HLClient.Logger.Infow("Exiting")
	488	ui.App.Stop()
	489	os.Exit(0)
	490	}
	491	// Show Logs
	492	if event.Key() == tcell.KeyCtrlL {
	493	ui.HLClient.DebugBuf.TextView.ScrollToEnd()
	494	ui.HLClient.DebugBuf.TextView.SetBorder(true).SetTitle("Logs")
	495	ui.HLClient.DebugBuf.TextView.SetDoneFunc(func(key tcell.Key) {
	496	if key == tcell.KeyEscape {
	497	ui.Pages.RemovePage("logs")
	498	}
	499	})
	500
	501	ui.Pages.AddPage("logs", ui.HLClient.DebugBuf.TextView, true, true)
	502	}
	503	return event
	504	})
	505
	506	if err := ui.App.SetRoot(ui.Pages, true).SetFocus(ui.Pages).Run(); err != nil {
	507	ui.App.Stop()
	508	os.Exit(1)
	509	}
	510	}