package cmd import ( "fmt" "log/slog" "github.com/ayn2op/discordo/internal/config" "github.com/ayn2op/discordo/internal/consts" "github.com/ayn2op/discordo/internal/login" "github.com/ayn2op/tview" "github.com/gdamore/tcell/v2" "github.com/zalando/go-keyring" ) type application struct { cfg *config.Config *tview.Application pages *tview.Pages flex *tview.Flex guildsTree *guildsTree messagesText *messagesText messageInput *messageInput flexPage int autocompletePage int } func newApplication(cfg *config.Config) *application { app := &application{ cfg: cfg, Application: tview.NewApplication(), pages: tview.NewPages(), flex: tview.NewFlex(), guildsTree: newGuildsTree(cfg), messagesText: newMessagesText(cfg), messageInput: newMessageInput(cfg), } app.EnableMouse(cfg.Mouse) app.SetInputCapture(app.onInputCapture) app.flex.SetInputCapture(app.onFlexInputCapture) return app } func (a *application) show(token string) error { if token == "" { loginForm := login.NewForm(a.cfg, func(token string) { if err := a.show(token); err != nil { slog.Error("failed to show app", "err", err) return } }) a.SetRoot(loginForm, true) } else { if err := openState(token); err != nil { return err } a.init() a.SetRoot(a.pages, true) } return nil } func (a *application) run(token string) error { if err := a.show(token); err != nil { return err } if err := a.Run(); err != nil { return fmt.Errorf("failed to run application: %w", err) } return nil } func (a *application) quit() { if discordState != nil { if err := discordState.Close(); err != nil { slog.Error("failed to close the session", "err", err) } } a.Stop() } func (a *application) init() { a.pages.Clear() a.flex.Clear() right := tview.NewFlex() right.SetDirection(tview.FlexRow) right.AddItem(a.messagesText, 0, 1, false) right.AddItem(a.messageInput, 3, 1, false) // The guilds tree is always focused first at start-up. a.flex.AddItem(a.guildsTree, 0, 1, true) a.flex.AddItem(right, 0, 4, false) a.flexPage = a.pages.AddAndSwitchToPage(a.flex, true) a.autocompletePage = a.pages.AddPage(a.messageInput.autocomplete, false, false) } func (a *application) onInputCapture(event *tcell.EventKey) *tcell.EventKey { switch event.Name() { case a.cfg.Keys.Quit: a.quit() return nil case "Ctrl+C": // https://github.com/ayn2op/tview/blob/a64fc48d7654432f71922c8b908280cdb525805c/application.go#L153 return tcell.NewEventKey(tcell.KeyCtrlC, 0, tcell.ModNone) } return event } func (a *application) onFlexInputCapture(event *tcell.EventKey) *tcell.EventKey { switch event.Name() { case a.cfg.Keys.FocusGuildsTree: a.pages.HidePage(a.autocompletePage) a.SetFocus(app.guildsTree) return nil case a.cfg.Keys.FocusMessagesText: a.pages.HidePage(a.autocompletePage) a.SetFocus(app.messagesText) return nil case a.cfg.Keys.FocusMessageInput: a.SetFocus(app.messageInput) return nil case a.cfg.Keys.Logout: a.quit() if err := keyring.Delete(consts.Name, "token"); err != nil { slog.Error("failed to delete token from keyring", "err", err) return nil } return nil case a.cfg.Keys.ToggleGuildsTree: // The guilds tree is visible if the numbers of items is two. if a.flex.GetItemCount() == 2 { a.flex.RemoveItem(a.guildsTree) if a.guildsTree.HasFocus() { a.SetFocus(a.flex) } } else { a.init() a.SetFocus(a.guildsTree) } return nil } return event }