package display import ( "fmt" "strconv" "strings" "github.com/gdamore/tcell" "github.com/spf13/viper" "gitlab.com/tslocum/cview" ) // This file contains code for the popups / modals used in the display. // The bookmark modal is in bookmarks.go var infoModal = cview.NewModal(). SetTextColor(tcell.ColorWhite). AddButtons([]string{"Ok"}) var errorModal = cview.NewModal(). SetTextColor(tcell.ColorWhite). AddButtons([]string{"Ok"}) var inputModal = cview.NewModal(). SetTextColor(tcell.ColorWhite) //AddButtons([]string{"Send", "Cancel"}) - Added in func var inputCh = make(chan string) var inputModalText string // The current text of the input field in the modal var yesNoModal = cview.NewModal(). SetTextColor(tcell.ColorWhite). AddButtons([]string{"Yes", "No"}) // Channel to receive yesNo answer on var yesNoCh = make(chan bool) func modalInit() { tabPages.AddPage("info", infoModal, false, false). AddPage("error", errorModal, false, false). AddPage("input", inputModal, false, false). AddPage("yesno", yesNoModal, false, false). AddPage("bkmk", bkmkModal, false, false) // Color setup if viper.GetBool("a-general.color") { infoModal.SetBackgroundColor(tcell.ColorGray). SetButtonBackgroundColor(tcell.ColorNavy). SetButtonTextColor(tcell.ColorWhite) errorModal.SetBackgroundColor(tcell.ColorMaroon). SetButtonBackgroundColor(tcell.ColorNavy). SetButtonTextColor(tcell.ColorWhite) inputModal.SetBackgroundColor(tcell.ColorGreen). SetButtonBackgroundColor(tcell.ColorNavy). SetButtonTextColor(tcell.ColorWhite) yesNoModal.SetButtonBackgroundColor(tcell.ColorNavy). SetButtonTextColor(tcell.ColorWhite) } else { infoModal.SetBackgroundColor(tcell.ColorBlack). SetButtonBackgroundColor(tcell.ColorWhite). SetButtonTextColor(tcell.ColorBlack) errorModal.SetBackgroundColor(tcell.ColorBlack). SetButtonBackgroundColor(tcell.ColorWhite). SetButtonTextColor(tcell.ColorBlack) inputModal.SetBackgroundColor(tcell.ColorBlack). SetButtonBackgroundColor(tcell.ColorWhite). SetButtonTextColor(tcell.ColorBlack) inputModal.GetForm(). SetLabelColor(tcell.ColorWhite). SetFieldBackgroundColor(tcell.ColorWhite). SetFieldTextColor(tcell.ColorBlack) // YesNo background color is changed in funcs yesNoModal.SetButtonBackgroundColor(tcell.ColorWhite). SetButtonTextColor(tcell.ColorBlack) } // Modal functions that can't be added up above, because they return the wrong type infoModal.SetBorder(true) infoModal.SetBorderColor(tcell.ColorWhite) infoModal.SetDoneFunc(func(buttonIndex int, buttonLabel string) { tabPages.SwitchToPage(strconv.Itoa(curTab)) }) infoModal.GetFrame().SetTitleColor(tcell.ColorWhite) infoModal.GetFrame().SetTitleAlign(cview.AlignCenter) infoModal.GetFrame().SetTitle(" Info ") errorModal.SetBorder(true) errorModal.SetBorderColor(tcell.ColorWhite) errorModal.SetDoneFunc(func(buttonIndex int, buttonLabel string) { tabPages.SwitchToPage(strconv.Itoa(curTab)) }) errorModal.GetFrame().SetTitleColor(tcell.ColorWhite) errorModal.GetFrame().SetTitleAlign(cview.AlignCenter) inputModal.SetBorder(true) inputModal.SetBorderColor(tcell.ColorWhite) inputModal.SetDoneFunc(func(buttonIndex int, buttonLabel string) { if buttonLabel == "Send" { inputCh <- inputModalText return } // Empty string indicates no input inputCh <- "" //tabPages.SwitchToPage(strconv.Itoa(curTab)) - handled in Input() }) inputModal.GetFrame().SetTitleColor(tcell.ColorWhite) inputModal.GetFrame().SetTitleAlign(cview.AlignCenter) inputModal.GetFrame().SetTitle(" Input ") yesNoModal.SetBorder(true) yesNoModal.SetBorderColor(tcell.ColorWhite) yesNoModal.SetDoneFunc(func(buttonIndex int, buttonLabel string) { if buttonLabel == "Yes" { yesNoCh <- true return } yesNoCh <- false //tabPages.SwitchToPage(strconv.Itoa(curTab)) - Handled in YesNo() }) yesNoModal.GetFrame().SetTitleColor(tcell.ColorWhite) yesNoModal.GetFrame().SetTitleAlign(cview.AlignCenter) bkmkInit() } // Error displays an error on the screen in a modal. func Error(title, text string) { // Capitalize and add period if necessary - because most errors don't do that text = strings.ToUpper(string([]rune(text)[0])) + text[1:] if !strings.HasSuffix(text, ".") && !strings.HasSuffix(text, "!") && !strings.HasSuffix(text, "?") { text += "." } // Add spaces to title for aesthetic reasons title = " " + strings.TrimSpace(title) + " " errorModal.GetFrame().SetTitle(title) errorModal.SetText(text) tabPages.ShowPage("error") tabPages.SendToFront("error") App.SetFocus(errorModal) App.Draw() } // Info displays some info on the screen in a modal. func Info(s string) { infoModal.SetText(s) tabPages.ShowPage("info") tabPages.SendToFront("info") App.SetFocus(infoModal) App.Draw() } // Input pulls up a modal that asks for input, and returns the user's input. // It returns an bool indicating if the user chose to send input or not. func Input(prompt string) (string, bool) { // Remove elements and re-add them - to clear input text and keep input in focus inputModal.ClearButtons() inputModal.GetForm().Clear(false) inputModal.AddButtons([]string{"Send", "Cancel"}) inputModalText = "" inputModal.GetForm().AddInputField("", "", 0, nil, func(text string) { // Store for use later inputModalText = text }) inputModal.SetText(prompt) tabPages.ShowPage("input") tabPages.SendToFront("input") App.SetFocus(inputModal) App.Draw() resp := <-inputCh tabPages.SwitchToPage(strconv.Itoa(curTab)) if resp == "" { return "", false } return resp, true } // YesNo displays a modal asking a yes-or-no question. func YesNo(prompt string) bool { if viper.GetBool("a-general.color") { yesNoModal.SetBackgroundColor(tcell.ColorPurple) } else { yesNoModal.SetBackgroundColor(tcell.ColorBlack) } yesNoModal.SetText(prompt) tabPages.ShowPage("yesno") tabPages.SendToFront("yesno") App.SetFocus(yesNoModal) App.Draw() resp := <-yesNoCh tabPages.SwitchToPage(strconv.Itoa(curTab)) return resp } // Tofu displays the TOFU warning modal. // It returns a bool indicating whether the user wants to continue. func Tofu(host string) bool { // Reuses yesNoModal, with error colour if viper.GetBool("a-general.color") { yesNoModal.SetBackgroundColor(tcell.ColorMaroon) } else { yesNoModal.SetBackgroundColor(tcell.ColorBlack) } yesNoModal.SetText( fmt.Sprintf("%s's certificate has changed, possibly indicating an security issue. Are you sure you want to continue? ", host), ) tabPages.ShowPage("yesno") tabPages.SendToFront("yesno") App.SetFocus(yesNoModal) App.Draw() resp := <-yesNoCh tabPages.SwitchToPage(strconv.Itoa(curTab)) return resp }