[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(servers []ServerRecord) *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("| Servers |")

	const shortcut = 97 // rune for "a"
	for i, _ := range servers {
		srv := servers[i]
		list.AddItem(string(srv.Name), string(srv.Description), rune(shortcut+i), func() {
			ui.Pages.RemovePage("joinServer")

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