[rbdr/mobius] / hotline / ui.go

package hotline

import (
	"errors"
	"fmt"
	"github.com/davecgh/go-spew/spew"
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
	"gopkg.in/yaml.v2"
	"io"
	"io/ioutil"
	"os"
	"strconv"
	"strings"
)

type UI struct {
	chatBox      *tview.TextView
	chatInput    *tview.InputField
	App          *tview.Application
	Pages        *tview.Pages
	userList     *tview.TextView
	agreeModal   *tview.Modal
	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(),
		agreeModal:  tview.NewModal(),
		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 {
		spew.Dump(err)
	}

	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()
		list.AddItem(string(srv.Name), string(srv.Description), rune(shortcut+i), func() {
			ui.Pages.RemovePage("joinServer")

			newJS := ui.renderJoinServerForm("", 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.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()

		out, err := yaml.Marshal(&ui.HLClient.pref)
		if err != nil {
			// TODO: handle err
		}
		// TODO: handle err
		err = ioutil.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.JoinServer(addr, login, password); err != nil {
		return errors.New(fmt.Sprintf("Error joining server: %v\n", err))
	}

	go func() {
		for {
			err := ui.HLClient.ReadLoop()
			if err != nil {
				ui.HLClient.Logger.Errorw("read error", "err", err)

				if err == io.EOF {
					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
				}
				ui.Pages.SwitchToPage("home")

				return
			}
		}
	}()

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