json_ui.go

package ui

import (
	"encoding/json"
	"fmt"
	"reflect"

	boshlog "github.com/cloudfoundry/bosh-utils/logger"

	. "github.com/cloudfoundry/bosh-cli/ui/table"
	"strconv"
)

type jsonUI struct {
	parent UI
	uiResp uiResp

	logTag string
	logger boshlog.Logger
}

type uiResp struct {
	Tables []tableResp
	Blocks []string
	Lines  []string
}

type tableResp struct {
	Content string
	Header  map[string]string
	Rows    []map[string]string
	Notes   []string
}

func NewJSONUI(parent UI, logger boshlog.Logger) UI {
	return &jsonUI{parent: parent, logTag: "JSONUI", logger: logger}
}

func (ui *jsonUI) ErrorLinef(pattern string, args ...interface{}) {
	ui.addLine(pattern, args)
}

func (ui *jsonUI) PrintLinef(pattern string, args ...interface{}) {
	ui.addLine(pattern, args)
}

func (ui *jsonUI) BeginLinef(pattern string, args ...interface{}) {
	ui.addLine(pattern, args)
}

func (ui *jsonUI) EndLinef(pattern string, args ...interface{}) {
	ui.addLine(pattern, args)
}

func (ui *jsonUI) PrintBlock(block []byte) {
	ui.uiResp.Blocks = append(ui.uiResp.Blocks, string(block))
}

func (ui *jsonUI) PrintErrorBlock(block string) {
	ui.uiResp.Blocks = append(ui.uiResp.Blocks, block)
}

func (ui *jsonUI) PrintTable(table Table) {
	resp := ui.printTableHeader(&table)
	ui.uiResp.Tables = append(ui.uiResp.Tables, resp)
}

func (ui *jsonUI) PrintTableFiltered(table Table, filterHeader []Header) {
	resp := ui.printTableHeader(&table)
	ui.uiResp.Tables = append(ui.uiResp.Tables, resp)
}

func (ui *jsonUI) AskForText(_ string) (string, error) {
	panic("Cannot ask for input in JSON UI")
}

func (ui *jsonUI) AskForChoice(_ string, _ []string) (int, error) {
	panic("Cannot ask for a choice in JSON UI")
}

func (ui *jsonUI) AskForPassword(_ string) (string, error) {
	panic("Cannot ask for password in JSON UI")
}

func (ui *jsonUI) AskForConfirmation() error {
	panic("Cannot ask for confirmation in JSON UI")
}

func (ui *jsonUI) IsInteractive() bool {
	return ui.parent.IsInteractive()
}

func (ui *jsonUI) Flush() {
	defer ui.parent.Flush()

	if !reflect.DeepEqual(ui.uiResp, uiResp{}) {
		bytes, err := json.MarshalIndent(ui.uiResp, "", "    ")
		if err != nil {
			ui.logger.Error(ui.logTag, "Failed to marshal UI response")
			return
		}

		ui.parent.PrintBlock(bytes)
	}
}

func (ui *jsonUI) printTableHeader(table *Table) tableResp {
	table.FillFirstColumn = true

	header := map[string]string{}

	if len(table.Header) > 0 {
		for i, val := range table.Header {
			if val.Hidden {
				continue
			}

			if val.Key == string(UNKNOWN_HEADER_MAPPING) {
				table.Header[i].Key = strconv.Itoa(i)
			}

			header[table.Header[i].Key] = val.Title
		}
	} else if len(table.AsRows()) > 0 {
		var rawHeaders []Header
		for i, _ := range table.AsRows()[0] {
			val := Header{
				Key:    fmt.Sprintf("col_%d", i),
				Hidden: false,
			}
			header[val.Key] = val.Title
			rawHeaders = append(rawHeaders, val)
		}
		table.Header = rawHeaders
	}

	resp := tableResp{
		Content: table.Content,
		Header:  header,
		Rows:    ui.stringRows(table.Header, table.AsRows()),
		Notes:   table.Notes,
	}

	return resp
}

func (ui *jsonUI) stringRows(header []Header, rows [][]Value) []map[string]string {
	result := []map[string]string{}

	for _, row := range rows {
		data := map[string]string{}

		for i, col := range row {
			if header[i].Hidden {
				continue
			}

			data[header[i].Key] = col.String()
		}

		result = append(result, data)
	}

	return result
}

func (ui *jsonUI) addLine(pattern string, args []interface{}) {
	msg := fmt.Sprintf(pattern, args...)
	ui.uiResp.Lines = append(ui.uiResp.Lines, msg)
	ui.logger.Debug(ui.logTag, msg)
}