mirror of
https://codeberg.org/scip/anydb.git
synced 2025-12-17 04:20:59 +01:00
added -N flag, added -m template support to get and list commands
This commit is contained in:
@@ -1,11 +1,14 @@
|
||||
package output
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
tpl "text/template"
|
||||
|
||||
"github.com/dustin/go-humanize"
|
||||
"github.com/olekukonko/tablewriter"
|
||||
@@ -16,14 +19,12 @@ import (
|
||||
func List(writer io.Writer, conf *cfg.Config, entries app.DbEntries) error {
|
||||
// FIXME: call sort here
|
||||
switch conf.Mode {
|
||||
case "wide":
|
||||
fallthrough
|
||||
case "":
|
||||
fallthrough
|
||||
case "table":
|
||||
case "wide", "", "table":
|
||||
return ListTable(writer, conf, entries)
|
||||
case "json":
|
||||
return ListJson(writer, conf, entries)
|
||||
case "template":
|
||||
return ListTemplate(writer, conf, entries)
|
||||
default:
|
||||
return errors.New("unsupported mode")
|
||||
}
|
||||
@@ -39,43 +40,67 @@ func ListJson(writer io.Writer, conf *cfg.Config, entries app.DbEntries) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func ListTemplate(writer io.Writer, conf *cfg.Config, entries app.DbEntries) error {
|
||||
tmpl, err := tpl.New("list").Parse(conf.Template)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to parse output template: %w", err)
|
||||
}
|
||||
|
||||
buf := bytes.Buffer{}
|
||||
|
||||
for _, row := range entries {
|
||||
row.Normalize()
|
||||
|
||||
buf.Reset()
|
||||
err = tmpl.Execute(&buf, row)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to execute output template: %w", err)
|
||||
}
|
||||
|
||||
if buf.Len() > 0 {
|
||||
fmt.Fprintln(writer, buf.String())
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func ListTable(writer io.Writer, conf *cfg.Config, entries app.DbEntries) error {
|
||||
tableString := &strings.Builder{}
|
||||
table := tablewriter.NewWriter(tableString)
|
||||
|
||||
if !conf.NoHeaders {
|
||||
if conf.Mode == "wide" {
|
||||
table.SetHeader([]string{"KEY", "TAGS", "SIZE", "AGE", "VALUE"})
|
||||
table.SetHeader([]string{"KEY", "TAGS", "SIZE", "UPDATED", "VALUE"})
|
||||
} else {
|
||||
table.SetHeader([]string{"KEY", "VALUE"})
|
||||
}
|
||||
}
|
||||
|
||||
for _, row := range entries {
|
||||
size := len(row.Value)
|
||||
|
||||
if row.Encrypted {
|
||||
row.Value = "<encrypted-content>"
|
||||
}
|
||||
|
||||
if len(row.Bin) > 0 {
|
||||
row.Value = "<binary-content>"
|
||||
size = len(row.Bin)
|
||||
}
|
||||
|
||||
if len(row.Value) > 60 {
|
||||
row.Value = row.Value[0:60] + "..."
|
||||
}
|
||||
row.Normalize()
|
||||
|
||||
if conf.Mode == "wide" {
|
||||
table.Append([]string{
|
||||
row.Key,
|
||||
strings.Join(row.Tags, ","),
|
||||
humanize.Bytes(uint64(size)),
|
||||
//row.Created.Format("02.01.2006T03:04.05"),
|
||||
humanize.Time(row.Created),
|
||||
row.Value,
|
||||
})
|
||||
switch conf.NoHumanize {
|
||||
case true:
|
||||
table.Append([]string{
|
||||
row.Key,
|
||||
strings.Join(row.Tags, ","),
|
||||
strconv.Itoa(row.Size),
|
||||
row.Created.Format("02.01.2006T03:04.05"),
|
||||
row.Value,
|
||||
})
|
||||
default:
|
||||
table.Append([]string{
|
||||
row.Key,
|
||||
strings.Join(row.Tags, ","),
|
||||
humanize.Bytes(uint64(row.Size)),
|
||||
//row.Created.Format("02.01.2006T03:04.05"),
|
||||
humanize.Time(row.Created),
|
||||
row.Value,
|
||||
})
|
||||
}
|
||||
|
||||
} else {
|
||||
table.Append([]string{row.Key, row.Value})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user