Add JSON formatter

This commit is contained in:
Dan Jones 2024-03-10 11:33:51 -05:00
commit 4c0edcd1a5
4 changed files with 132 additions and 1 deletions

63
formatters/json.go Normal file
View file

@ -0,0 +1,63 @@
package formatters
import (
"encoding/json"
"time"
"codeberg.org/danjones000/my-log/config"
"codeberg.org/danjones000/my-log/models"
//"codeberg.org/danjones000/my-log/tools"
)
func newJson(ff config.Formatters) (Formatter, error) {
// @todo pretty print
return &Json{ff.Json()}, nil
}
type Json struct {
jf config.JsonFormat
}
func (js *Json) Name() string {
return "json"
}
func (js *Json) Meta(m models.Meta) ([]byte, error) {
o := map[string]any{m.Key: m.Value}
return json.Marshal(o)
}
func (js *Json) entryMap(e models.Entry) map[string]any {
o := map[string]any{
"title": e.Title,
"date": e.Date.Format(time.RFC3339),
}
for _, m := range e.Fields {
o[m.Key] = m.Value
}
return o
}
func (js *Json) Entry(e models.Entry) ([]byte, error) {
return json.Marshal(js.entryMap(e))
}
func (js *Json) Log(l models.Log) ([]byte, error) {
return js.Logs([]models.Log{l})
}
func (js *Json) Logs(logs []models.Log) (out []byte, err error) {
if len(logs) == 0 {
return
}
o := map[string][]map[string]any{}
for _, l := range logs {
es := []map[string]any{}
for _, e := range l.Entries {
es = append(es, js.entryMap(e))
}
o[l.Name] = es
}
return json.Marshal(o)
}

67
formatters/json_test.go Normal file
View file

@ -0,0 +1,67 @@
package formatters
import (
//"bufio"
//"bytes"
"fmt"
"testing"
"time"
"codeberg.org/danjones000/my-log/models"
//"codeberg.org/danjones000/my-log/tools"
"github.com/stretchr/testify/assert"
//"github.com/stretchr/testify/require"
)
func TestJsonName(t *testing.T) {
f, _ := New("json")
assert.Equal(t, "json", f.Name())
}
func TestJsonMeta(t *testing.T) {
f, _ := New("json")
m := models.Meta{"foo", 42}
exp := `{"foo":42}`
o, err := f.Meta(m)
assert.NoError(t, err)
assert.JSONEq(t, exp, string(o))
}
func TestJsonEntry(t *testing.T) {
when := time.Now()
f, _ := New("json")
m := models.Meta{"foo", 42}
e := models.Entry{
Title: "Homer",
Date: when,
Fields: []models.Meta{m},
}
exp := fmt.Sprintf(`{"title":"%s","date":"%s","foo":42}`, e.Title, when.Format(time.RFC3339))
o, err := f.Entry(e)
assert.NoError(t, err)
assert.JSONEq(t, exp, string(o))
}
func TestJsonLog(t *testing.T) {
when := time.Now()
f, _ := New("json")
m := models.Meta{"foo", 42}
e := models.Entry{
Title: "Homer",
Date: when,
Fields: []models.Meta{m},
}
l := models.Log{"stuff", []models.Entry{e}}
exp := fmt.Sprintf(`{"%s":[{"title":"%s","date":"%s","foo":42}]}`, l.Name, e.Title, when.Format(time.RFC3339))
o, err := f.Log(l)
assert.NoError(t, err)
assert.JSONEq(t, exp, string(o))
}
func TestJsonNoLogs(t *testing.T) {
f, _ := New("json")
o, err := f.Logs([]models.Log{})
var exp []byte
assert.NoError(t, err)
assert.Equal(t, exp, o)
}

View file

@ -10,6 +10,7 @@ type formatMaker func(config.Formatters) (Formatter, error)
var formatterMap = map[string]formatMaker{
"plain": newPlain,
"json": newJson,
}
func Preferred() (f Formatter, err error) {

View file

@ -10,7 +10,7 @@ import (
)
func TestKinds(t *testing.T) {
assert.Equal(t, []string{"plain"}, Kinds())
assert.ElementsMatch(t, []string{"plain", "json"}, Kinds())
}
func TestNewUnsupported(t *testing.T) {