[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
	settingsPage *tview.Box
	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)

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