42 lines
742 B
Go
42 lines
742 B
Go
package output
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
type Printer struct {
|
|
format string
|
|
writer io.Writer
|
|
}
|
|
|
|
func New(format string, w io.Writer) *Printer {
|
|
return &Printer{format: format, writer: w}
|
|
}
|
|
|
|
func (p *Printer) Print(data any) error {
|
|
switch p.format {
|
|
case "table":
|
|
return printTable(p.writer, data)
|
|
case "json":
|
|
return printJSON(p.writer, data)
|
|
case "yaml":
|
|
return printYAML(p.writer, data)
|
|
case "raw":
|
|
b, err := json.Marshal(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = fmt.Fprintln(p.writer, string(b))
|
|
return err
|
|
default:
|
|
return fmt.Errorf("unknown output format: %s", p.format)
|
|
}
|
|
}
|
|
|
|
func (p *Printer) PrintRaw(raw []byte) error {
|
|
_, err := fmt.Fprintln(p.writer, string(raw))
|
|
return err
|
|
}
|