mainframe.go

package main

import (
	"encoding/hex"
	"fmt"

	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
)

var verString = "VER 1.0"
var mainFrameTitle = fmt.Sprintf("- SAFEBOX KEY MANGEMENT SYSTEM %v -", verString)

func showExporterWindow(row int, col int) {
	windowName := "showExporterWindow"
	exporterLabel := "Select an exporter (hit Enter):"
	var exportorNames []string
	for k := range exports {
		exportorNames = append(exportorNames, exports[k].Name())
	}

	idx := uint16(row - 1)
	selected := 0
	form := tview.NewForm()
	form.SetTitle("EXPORT KEY")
	form.AddDropDown(exporterLabel, exportorNames, 0, func(option string, optionIndex int) {
		selected = optionIndex
	})
	form.AddButton("Export", func() {
		key, _ := masterKey.deriveKey(idx, exports[selected].KeySize())
		bts, _ := exports[selected].Export(key)

		// output page
		outputBox := tview.NewTextView()
		outputBox.SetScrollable(true)
		outputBox.SetWrap(true)
		outputBox.Write(bts)
		root.AddPage("output", outputBox, true, true)
	})
	form.SetFocus(0)

	view := tview.NewFlex()
	view.SetBorder(true)
	view.SetDirection(tview.FlexRow).
		AddItem(form, 0, 1, true)

	root.AddPage(windowName, popup(80, 10, view), true, true)
}

func showSetLabelWindow(row int, col int) {
	windowName := "showSetLabelWindow"
	idx := uint16(row - 1)

	form := tview.NewForm()
	form.SetBorder(true)
	form.SetTitle("SETTING KEY LABEL")
	form.AddInputField("Label", masterKey.lables[idx], 16, nil, nil)
	form.AddButton("Update", func() {
		//update key
		masterKey.lables[idx] = form.GetFormItemByLabel("Label").(*tview.InputField).GetText()
		masterKey.store(masterKey.password, masterKey.path)
		table.SetCell(int(idx)+1, 1,
			tview.NewTableCell(masterKey.lables[idx]).
				SetTextColor(tcell.ColorRed).
				SetAlign(tview.AlignLeft).
				SetSelectable(true))

		root.RemovePage(windowName)
		refreshInfo()
	})
	form.SetFocus(0)

	root.AddPage(windowName, popup(40, 10, form), true, true)
}

// main operation frame
func mainFrameWindow() (content *tview.Flex) {
	if masterKey == nil {
		// if master key has not loaded
		text := tview.NewTextView()
		text.SetDynamicColors(true).
			SetTextAlign(tview.AlignCenter)
		fmt.Fprintf(text, `[red]KEY NOT LOADED
PLEASE LOAD A MASTER KEY[yellow][F2][red] OR GENERATE ONE[yellow][F1][red] FIRST`)

		flex := tview.NewFlex()
		flex.SetDirection(tview.FlexRow).
			SetBorder(true).
			SetTitle(mainFrameTitle)

		flex.AddItem(tview.NewBox(), 0, 8, false)
		flex.AddItem(text, 0, 1, true)
		flex.AddItem(tview.NewBox(), 0, 8, false)
		return flex
	}

	// key table
	table = tview.NewTable().SetBorders(true)
	table.SetTitle(mainFrameTitle)

	// table header
	table.SetCell(0, 0,
		tview.NewTableCell("ID").
			SetTextColor(tcell.ColorDarkOrange).
			SetSelectable(false).
			SetAlign(tview.AlignLeft))

	table.SetCell(0, 1,
		tview.NewTableCell("NAME").
			SetTextColor(tcell.ColorDarkOrange).
			SetSelectable(false).
			SetAlign(tview.AlignLeft))

	table.SetCell(0, 2,
		tview.NewTableCell("DERIVED KEY").
			SetTextColor(tcell.ColorDarkOrange).
			SetSelectable(false).
			SetAlign(tview.AlignLeft))

	// fix table header & first column
	table.SetFixed(1, 1)
	table.SetSelectable(true, true)

	addDerivedKeys := func(start uint16) {
		for i := uint16(0); i < 64; i++ {
			idx := start + i
			if idx >= MaxKeys {
				return
			}
			// we derive and show the part of the key
			key, err := masterKey.deriveKey(idx, 32)
			if err != nil {
				panic(err)
			}

			table.SetCell(int(idx)+1, 0,
				tview.NewTableCell(fmt.Sprint(idx)).
					SetAlign(tview.AlignLeft).
					SetSelectable(false))

			table.SetCell(int(idx)+1, 1,
				tview.NewTableCell(masterKey.lables[idx]).
					SetAlign(tview.AlignLeft).
					SetSelectable(true))

			table.SetCell(int(idx)+1, 2,
				tview.NewTableCell(mask(hex.EncodeToString(key), 4, '*')).
					SetTextColor(tcell.ColorDarkCyan).
					SetAlign(tview.AlignLeft))
		}
	}

	// add initial derived keys
	addDerivedKeys(0)

	var lastRow int
	var lastCol int
	table.SetSelectionChangedFunc(func(row, column int) {
		// moved to last
		idx := uint16(row) - 1
		if row == table.GetRowCount()-1 {
			addDerivedKeys(idx)
		}

		// mask previous key again
		// derive key again
		if lastRow > 0 {
			key, err := masterKey.deriveKey(uint16(lastRow)-1, 32)
			if err != nil {
				panic(err)
			}

			table.SetCell(lastRow, lastCol,
				tview.NewTableCell(mask(hex.EncodeToString(key), 4, '*')).
					SetTextColor(tcell.ColorDarkCyan).
					SetAlign(tview.AlignLeft))
		}

		// mask previous selection
		if column == 2 && row > 0 {
			// derive key again
			key, err := masterKey.deriveKey(idx, 32)
			if err != nil {
				panic(err)
			}

			// uncover mask
			table.SetCell(row, column,
				tview.NewTableCell(hex.EncodeToString(key)).
					SetTextColor(tcell.ColorDarkCyan).
					SetAlign(tview.AlignLeft))
			// remember last selection
			lastRow = row
			lastCol = column
		}
	})

	// key selection
	table.SetSelectedFunc(func(row, column int) {
		if column == 1 {
			showSetLabelWindow(row, column)
		} else if column == 2 {
			showExporterWindow(row, column)
		}
	})

	flex := tview.NewFlex()
	flex.SetDirection(tview.FlexRow).
		SetBorder(true).
		SetTitle(mainFrameTitle)

	flex.AddItem(table, 0, 1, true)
	return flex
}