Dan Anglin
ccdd8b6530
Add a new internal printer package for printing resources to the screen or pager. With the new printer in place, most of the settings such as the pager command, colour theme, whether or not colour output is disabled, etc are defined in one place which saves us the trouble of passing an increasing number of parameters to an increasing number of Display methods throughout the code base. The old Displayer interface and associated Display methods in the model package are removed as this is now handled by the printer. The format functions in the utilities package has essentially been rewritten as methods to the Printer type. Additional changes: - All indentation when displaying information about resources (e.g. statuses, instance, accounts) are removed. - The application's build information now has colour output.
185 lines
3.6 KiB
Go
185 lines
3.6 KiB
Go
// SPDX-FileCopyrightText: 2024 Dan Anglin <d.n.i.anglin@gmail.com>
|
|
//
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package printer
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
minTerminalWidth = 40
|
|
)
|
|
|
|
type theme struct {
|
|
reset string
|
|
boldblue string
|
|
boldmagenta string
|
|
green string
|
|
boldgreen string
|
|
grey string
|
|
red string
|
|
boldred string
|
|
}
|
|
|
|
type Printer struct {
|
|
theme theme
|
|
noColor bool
|
|
maxTerminalWidth int
|
|
pager string
|
|
statusSeparator string
|
|
bullet string
|
|
pollMeterSymbol string
|
|
successSymbol string
|
|
failureSymbol string
|
|
dateFormat string
|
|
dateTimeFormat string
|
|
}
|
|
|
|
func NewPrinter(
|
|
noColor bool,
|
|
pager string,
|
|
maxTerminalWidth int,
|
|
) *Printer {
|
|
theme := theme{
|
|
reset: "\033[0m",
|
|
boldblue: "\033[34;1m",
|
|
boldmagenta: "\033[35;1m",
|
|
green: "\033[32m",
|
|
boldgreen: "\033[32;1m",
|
|
grey: "\033[90m",
|
|
red: "\033[31m",
|
|
boldred: "\033[31;1m",
|
|
}
|
|
|
|
if maxTerminalWidth < minTerminalWidth {
|
|
maxTerminalWidth = minTerminalWidth
|
|
}
|
|
|
|
return &Printer{
|
|
noColor: noColor,
|
|
maxTerminalWidth: maxTerminalWidth,
|
|
pager: pager,
|
|
statusSeparator: strings.Repeat("\u2501", maxTerminalWidth),
|
|
bullet: "\u2022",
|
|
pollMeterSymbol: "\u2501",
|
|
successSymbol: "\u2714",
|
|
failureSymbol: "\u2717",
|
|
dateFormat: "02 Jan 2006",
|
|
dateTimeFormat: "02 Jan 2006, 15:04 (MST)",
|
|
theme: theme,
|
|
}
|
|
}
|
|
|
|
func (p Printer) PrintSuccess(text string) {
|
|
success := p.theme.boldgreen + p.successSymbol + p.theme.reset
|
|
if p.noColor {
|
|
success = p.successSymbol
|
|
}
|
|
|
|
printToStdout(success + " " + text + "\n")
|
|
}
|
|
|
|
func (p Printer) PrintFailure(text string) {
|
|
failure := p.theme.boldred + p.failureSymbol + p.theme.reset
|
|
if p.noColor {
|
|
failure = p.failureSymbol
|
|
}
|
|
|
|
printToStderr(failure + " " + text + "\n")
|
|
}
|
|
|
|
func (p Printer) PrintInfo(text string) {
|
|
printToStdout(text)
|
|
}
|
|
|
|
func (p Printer) headerFormat(text string) string {
|
|
if p.noColor {
|
|
return text
|
|
}
|
|
|
|
return p.theme.boldblue + text + p.theme.reset
|
|
}
|
|
|
|
func (p Printer) fieldFormat(text string) string {
|
|
if p.noColor {
|
|
return text
|
|
}
|
|
|
|
return p.theme.green + text + p.theme.reset
|
|
}
|
|
|
|
func (p Printer) fullDisplayNameFormat(displayName, acct string) string {
|
|
// use this pattern to remove all emoji strings
|
|
pattern := regexp.MustCompile(`\s:[A-Za-z0-9_]*:`)
|
|
|
|
var builder strings.Builder
|
|
|
|
if p.noColor {
|
|
builder.WriteString(pattern.ReplaceAllString(displayName, ""))
|
|
} else {
|
|
builder.WriteString(p.theme.boldmagenta + pattern.ReplaceAllString(displayName, "") + p.theme.reset)
|
|
}
|
|
|
|
builder.WriteString(" (@" + acct + ")")
|
|
|
|
return builder.String()
|
|
}
|
|
|
|
func (p Printer) formatDate(date time.Time) string {
|
|
return date.Local().Format(p.dateFormat)
|
|
}
|
|
|
|
func (p Printer) formatDateTime(date time.Time) string {
|
|
return date.Local().Format(p.dateTimeFormat)
|
|
}
|
|
|
|
func (p Printer) print(text string) {
|
|
if p.pager == "" {
|
|
printToStdout(text)
|
|
|
|
return
|
|
}
|
|
|
|
cmdSplit := strings.Split(p.pager, " ")
|
|
|
|
pager := new(exec.Cmd)
|
|
|
|
if len(cmdSplit) == 1 {
|
|
pager = exec.Command(cmdSplit[0]) //nolint:gosec
|
|
} else {
|
|
pager = exec.Command(cmdSplit[0], cmdSplit[1:]...) //nolint:gosec
|
|
}
|
|
|
|
pipe, err := pager.StdinPipe()
|
|
if err != nil {
|
|
printToStdout(text)
|
|
|
|
return
|
|
}
|
|
|
|
pager.Stdout = os.Stdout
|
|
pager.Stderr = os.Stderr
|
|
|
|
_ = pager.Start()
|
|
|
|
defer func() {
|
|
_ = pipe.Close()
|
|
_ = pager.Wait()
|
|
}()
|
|
|
|
_, _ = pipe.Write([]byte(text))
|
|
}
|
|
|
|
func printToStdout(text string) {
|
|
os.Stdout.WriteString(text)
|
|
}
|
|
|
|
func printToStderr(text string) {
|
|
os.Stderr.WriteString(text)
|
|
}
|