Skip to content

Commit

Permalink
Merge pull request #14 from replicatedhq/preflight-ui
Browse files Browse the repository at this point in the history 
preflight ui
  • Loading branch information
@marccampbell
marccampbell committed Jul 19, 2019
2 parents fe8f2be + 1150e10 commit 997a220
Show file tree
Hide file tree
Showing 3 changed files with 237 additions and 85 deletions.
264 changes: 222 additions & 42 deletions cmd/preflight/cli/interactive_results.go
View file
Expand Up @@ -2,103 +2,283 @@ package cli

import (
"fmt"
"io/ioutil"
"os"
"path"
"strings"
"time"

ui "github.com/gizak/termui/v3"
"github.com/gizak/termui/v3/widgets"
analyzerunner "github.com/replicatedhq/troubleshoot/pkg/analyze"
)

type nodeValue string

func (nv nodeValue) String() string {
return string(nv)
}
var (
selectedResult = 0
isShowingSaved = false
)

func showInteractiveResults(analyzeResults []*analyzerunner.AnalyzeResult) error {
func showInteractiveResults(preflightName string, analyzeResults []*analyzerunner.AnalyzeResult) error {
if err := ui.Init(); err != nil {
return err
}
defer ui.Close()

selectedResult := 0

preflightTable := getPreflightTable(analyzeResults)
details := getDetails(analyzeResults[selectedResult])

grid := ui.NewGrid()
termWidth, termHeight := ui.TerminalDimensions()
grid.SetRect(0, 0, termWidth, termHeight)

grid.Set(
ui.NewRow(1.0,
ui.NewCol(1.0/2, preflightTable),
ui.NewCol(1.0/2, details),
),
)

ui.Render(grid)
drawUI(preflightName, analyzeResults)

uiEvents := ui.PollEvents()
for {
select {
case e := <-uiEvents:
switch e.ID {
case "q", "<C-c>":
case "<C-c>":
return nil
case "q":
if isShowingSaved == true {
isShowingSaved = false
ui.Clear()
drawUI(preflightName, analyzeResults)
} else {
return nil
}
case "s":
filename, err := save(preflightName, analyzeResults)
if err != nil {
// show
} else {
showSaved(filename)
go func() {
time.Sleep(time.Second * 5)
isShowingSaved = false
ui.Clear()
drawUI(preflightName, analyzeResults)
}()
}
case "<Resize>":
payload := e.Payload.(ui.Resize)
grid.SetRect(0, 0, payload.Width, payload.Height)
ui.Clear()
ui.Render(grid)
drawUI(preflightName, analyzeResults)
case "<Down>":
if selectedResult < len(analyzeResults)-1 {
selectedResult++
} else {
selectedResult = 0
}
ui.Clear()
drawUI(preflightName, analyzeResults)
case "<Up>":
if selectedResult > 0 {
selectedResult--
} else {
selectedResult = len(analyzeResults) - 1
}
ui.Clear()
drawUI(preflightName, analyzeResults)
}
}
}
}

func getPreflightTable(analyzeResults []*analyzerunner.AnalyzeResult) *widgets.Table {
func drawUI(preflightName string, analyzeResults []*analyzerunner.AnalyzeResult) {
drawGrid(analyzeResults)
drawHeader(preflightName)
drawFooter()
}

func drawGrid(analyzeResults []*analyzerunner.AnalyzeResult) {
drawPreflightTable(analyzeResults)
drawDetails(analyzeResults[selectedResult])
}

func drawHeader(preflightName string) {
termWidth, _ := ui.TerminalDimensions()

title := widgets.NewParagraph()
title.Text = fmt.Sprintf("%s Preflight Checks", appName(preflightName))
title.TextStyle.Fg = ui.ColorWhite
title.TextStyle.Bg = ui.ColorClear
title.TextStyle.Modifier = ui.ModifierBold
title.Border = false

left := termWidth/2 - 2*len(title.Text)/3
right := termWidth/2 + (termWidth/2 - left)

title.SetRect(left, 0, right, 1)
ui.Render(title)
}

func drawFooter() {
termWidth, termHeight := ui.TerminalDimensions()

instructions := widgets.NewParagraph()
instructions.Text = "[q] quit [s] save [↑][↓] scroll"
instructions.Border = false

left := 0
right := termWidth
top := termHeight - 1
bottom := termHeight

instructions.SetRect(left, top, right, bottom)
ui.Render(instructions)
}

func drawPreflightTable(analyzeResults []*analyzerunner.AnalyzeResult) {
termWidth, termHeight := ui.TerminalDimensions()

table := widgets.NewTable()
table.SetRect(0, 3, termWidth/2, termHeight-6)
table.FillRow = true
table.Border = true
table.Rows = [][]string{}
table.ColumnWidths = []int{termWidth}

for i, analyzeResult := range analyzeResults {
title := analyzeResult.Title
if analyzeResult.IsPass {
title = fmt.Sprintf("✔ %s", title)
} else if analyzeResult.IsWarn {
title = fmt.Sprintf("⚠️ %s", title)
} else if analyzeResult.IsFail {
title = fmt.Sprintf("✘ %s", title)
}
table.Rows = append(table.Rows, []string{
analyzeResult.Title,
title,
})

if analyzeResult.IsPass {
table.RowStyles[i] = ui.NewStyle(ui.ColorGreen, ui.ColorClear, ui.ModifierBold)
if i == selectedResult {
table.RowStyles[i] = ui.NewStyle(ui.ColorGreen, ui.ColorClear, ui.ModifierReverse)
} else {
table.RowStyles[i] = ui.NewStyle(ui.ColorGreen, ui.ColorClear)
}
} else if analyzeResult.IsWarn {
table.RowStyles[i] = ui.NewStyle(ui.ColorYellow, ui.ColorClear, ui.ModifierBold)
if i == selectedResult {
table.RowStyles[i] = ui.NewStyle(ui.ColorYellow, ui.ColorClear, ui.ModifierReverse)
} else {
table.RowStyles[i] = ui.NewStyle(ui.ColorYellow, ui.ColorClear)
}
} else if analyzeResult.IsFail {
table.RowStyles[i] = ui.NewStyle(ui.ColorRed, ui.ColorClear)
if i == selectedResult {
table.RowStyles[i] = ui.NewStyle(ui.ColorRed, ui.ColorClear, ui.ModifierReverse)
} else {
table.RowStyles[i] = ui.NewStyle(ui.ColorRed, ui.ColorClear)
}
}
}

return table
ui.Render(table)
}

func getDetails(analysisResult *analyzerunner.AnalyzeResult) *ui.Grid {
grid := ui.NewGrid()

entries := []interface{}{}
func drawDetails(analysisResult *analyzerunner.AnalyzeResult) {
termWidth, _ := ui.TerminalDimensions()

currentTop := 4
title := widgets.NewParagraph()
title.Text = analysisResult.Title
title.Border = false
entries = append(entries, ui.NewRow(0.2, ui.NewCol(1.0, title)))
if analysisResult.IsPass {
title.TextStyle = ui.NewStyle(ui.ColorGreen, ui.ColorClear, ui.ModifierBold)
} else if analysisResult.IsWarn {
title.TextStyle = ui.NewStyle(ui.ColorYellow, ui.ColorClear, ui.ModifierBold)
} else if analysisResult.IsFail {
title.TextStyle = ui.NewStyle(ui.ColorRed, ui.ColorClear, ui.ModifierBold)
}
height := estimateNumberOfLines(title.Text, termWidth/2)
title.SetRect(termWidth/2, currentTop, termWidth, currentTop+height)
ui.Render(title)
currentTop = currentTop + height + 1

message := widgets.NewParagraph()
message.Text = analysisResult.Message
message.Border = false
entries = append(entries, ui.NewRow(0.2, ui.NewCol(1.0, message)))
height = estimateNumberOfLines(message.Text, termWidth/2) + 2
message.SetRect(termWidth/2, currentTop, termWidth, currentTop+height)
ui.Render(message)
currentTop = currentTop + height + 1

if analysisResult.URI != "" {
uri := widgets.NewParagraph()
uri.Text = fmt.Sprintf("For more information: %s", analysisResult.URI)
uri.Border = false
entries = append(entries, ui.NewRow(0.2, ui.NewCol(1.0, uri)))
height = estimateNumberOfLines(uri.Text, termWidth/2)
uri.SetRect(termWidth/2, currentTop, termWidth, currentTop+height)
ui.Render(uri)
currentTop = currentTop + height + 1
}
}

func estimateNumberOfLines(text string, width int) int {
lines := len(text)/width + 1
return lines
}

func save(preflightName string, analyzeResults []*analyzerunner.AnalyzeResult) (string, error) {
filename := path.Join(homeDir(), fmt.Sprintf("%s-results.txt", preflightName))
_, err := os.Stat(filename)
if err == nil {
os.Remove(filename)
}

results := fmt.Sprintf("%s Preflight Checks\n\n", appName(preflightName))
for _, analyzeResult := range analyzeResults {
result := ""

if analyzeResult.IsPass {
result = "Check PASS\n"
} else if analyzeResult.IsWarn {
result = "Check WARN\n"
} else if analyzeResult.IsFail {
result = "Check FAIL\n"
}

result = result + fmt.Sprintf("Title: %s\n", analyzeResult.Title)
result = result + fmt.Sprintf("Message: %s\n", analyzeResult.Message)

if analyzeResult.URI != "" {
result = result + fmt.Sprintf("URI: %s\n", analyzeResult.URI)
}

result = result + "\n------------\n"

results = results + result
}

if err := ioutil.WriteFile(filename, []byte(results), 0644); err != nil {
return "", err
}

return filename, nil
}

func showSaved(filename string) {
termWidth, termHeight := ui.TerminalDimensions()

savedMessage := widgets.NewParagraph()
savedMessage.Text = fmt.Sprintf("Preflight results saved to\n\n%s", filename)
savedMessage.WrapText = true
savedMessage.Border = true

left := termWidth/2 - 20
right := termWidth/2 + 20
top := termHeight/2 - 4
bottom := termHeight/2 + 4

savedMessage.SetRect(left, top, right, bottom)
ui.Render(savedMessage)

isShowingSaved = true
}

func appName(preflightName string) string {
words := strings.Split(strings.Title(strings.Replace(preflightName, "-", " ", -1)), " ")
casedWords := []string{}
for _, word := range words {
if strings.ToLower(word) == "ai" {
casedWords = append(casedWords, "AI")
} else {
casedWords = append(casedWords, word)
}
}

grid.Set(entries...)
return grid
return strings.Join(casedWords, " ")
}
2 changes: 1 addition & 1 deletion cmd/preflight/cli/run_nocrd.go
View file
Expand Up @@ -88,7 +88,7 @@ func runPreflightsNoCRD(v *viper.Viper, arg string) error {
}

if v.GetBool("interactive") {
return showInteractiveResults(analyzeResults)
return showInteractiveResults(preflight.Name, analyzeResults)
}

fmt.Printf("only interactive results are supported\n")
Expand Down

0 comments on commit 997a220

Please sign in to comment.