112 lines
2.7 KiB
Go
112 lines
2.7 KiB
Go
// Copyright 2018 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/olekukonko/tablewriter"
|
|
)
|
|
|
|
var (
|
|
showLog bool
|
|
)
|
|
|
|
// Println println content according the flag
|
|
func Println(a ...interface{}) {
|
|
if showLog {
|
|
fmt.Println(a...)
|
|
}
|
|
}
|
|
|
|
// Printf printf content according the flag
|
|
func Printf(format string, a ...interface{}) {
|
|
if showLog {
|
|
fmt.Printf(format, a...)
|
|
}
|
|
}
|
|
|
|
// Error println content as an error information
|
|
func Error(a ...interface{}) {
|
|
fmt.Println(a...)
|
|
}
|
|
|
|
// Errorf printf content as an error information
|
|
func Errorf(format string, a ...interface{}) {
|
|
fmt.Printf(format, a...)
|
|
}
|
|
|
|
// OutputTable prints structured data as table
|
|
func OutputTable(headers []string, values [][]string) {
|
|
table := tablewriter.NewWriter(os.Stdout)
|
|
if len(headers) > 0 {
|
|
table.SetHeader(headers)
|
|
}
|
|
for _, value := range values {
|
|
table.Append(value)
|
|
}
|
|
table.Render()
|
|
}
|
|
|
|
// OutputSimple prints structured data as space delimited value
|
|
func OutputSimple(headers []string, values [][]string) {
|
|
for _, value := range values {
|
|
fmt.Printf(strings.Join(value, " "))
|
|
fmt.Printf("\n")
|
|
}
|
|
}
|
|
|
|
// OutputDsv prints structured data as delimiter separated value format
|
|
func OutputDsv(headers []string, values [][]string, delimiterOpt ...string) {
|
|
delimiter := ","
|
|
if len(delimiterOpt) > 0 {
|
|
delimiter = delimiterOpt[0]
|
|
}
|
|
fmt.Println("\"" + strings.Join(headers, "\""+delimiter+"\"") + "\"")
|
|
for _, value := range values {
|
|
fmt.Printf("\"")
|
|
fmt.Printf(strings.Join(value, "\""+delimiter+"\""))
|
|
fmt.Printf("\"")
|
|
fmt.Printf("\n")
|
|
}
|
|
}
|
|
|
|
// OutputYaml prints structured data as yaml
|
|
func OutputYaml(headers []string, values [][]string) {
|
|
for _, value := range values {
|
|
fmt.Println("-")
|
|
for j, val := range value {
|
|
intVal, _ := strconv.Atoi(val)
|
|
if strconv.Itoa(intVal) == val {
|
|
fmt.Printf(" %s: %s\n", headers[j], val)
|
|
} else {
|
|
fmt.Printf(" %s: '%s'\n", headers[j], val)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Output provides general function to convert given information
|
|
// into several outputs
|
|
func Output(output string, headers []string, values [][]string) {
|
|
switch {
|
|
case output == "" || output == "table":
|
|
OutputTable(headers, values)
|
|
case output == "csv":
|
|
OutputDsv(headers, values, ",")
|
|
case output == "simple":
|
|
OutputSimple(headers, values)
|
|
case output == "tsv":
|
|
OutputDsv(headers, values, "\t")
|
|
case output == "yaml":
|
|
OutputYaml(headers, values)
|
|
default:
|
|
Errorf("unknown output type '" + output + "', available types are:\n- csv: comma-separated values\n- simple: space-separated values\n- table: auto-aligned table format (default)\n- tsv: tab-separated values\n- yaml: YAML format\n")
|
|
}
|
|
}
|