Some convenience methods for Metas

This commit is contained in:
Dan Jones 2024-03-11 16:04:29 -05:00
commit 79fa957d02
4 changed files with 39 additions and 16 deletions

View file

@ -42,17 +42,17 @@ var dropCmd = &cobra.Command{
RunE: func(cmd *cobra.Command, args []string) error { RunE: func(cmd *cobra.Command, args []string) error {
log := args[0] log := args[0]
title := args[1] title := args[1]
ms := models.Metas{} ms := &models.Metas{}
if len(j.RawMessage) > 8 { if len(j.RawMessage) > 8 {
err := json.Unmarshal([]byte(j.RawMessage), &ms) err := json.Unmarshal([]byte(j.RawMessage), ms)
if err != nil { if err != nil {
return err return err
} }
} }
for k, v := range fields { for k, v := range fields {
ms = append(ms, models.Meta{k, tools.ParseString(v)}) ms.AppendTo(k, tools.ParseString(v))
} }
e := models.Entry{title, d.t, ms} e := models.Entry{title, d.t, *ms}
l := models.Log{log, []models.Entry{e}} l := models.Log{log, []models.Entry{e}}
err := files.Append(l) err := files.Append(l)
if err != nil { if err != nil {

View file

@ -186,7 +186,7 @@ func (e Entry) MarshalJSON() ([]byte, error) {
out := map[string]any{} out := map[string]any{}
out["title"] = e.Title out["title"] = e.Title
out["date"] = e.Date.Format(time.RFC3339) out["date"] = e.Date.Format(time.RFC3339)
for k, v := range e.Fields.toMap() { for k, v := range e.Fields.Map() {
out[k] = v out[k] = v
} }
return json.Marshal(out) return json.Marshal(out)

View file

@ -13,6 +13,8 @@ import (
// Type assertions // Type assertions
var _ encoding.TextMarshaler = Meta{} var _ encoding.TextMarshaler = Meta{}
var _ encoding.TextUnmarshaler = new(Meta) var _ encoding.TextUnmarshaler = new(Meta)
var _ json.Marshaler = Metas{}
var _ json.Unmarshaler = new(Metas)
var skipMarshalTest = errors.New("skip marshal") var skipMarshalTest = errors.New("skip marshal")
@ -125,3 +127,17 @@ func TestMetasJsonError(t *testing.T) {
assert.Error(t, err) assert.Error(t, err)
assert.Len(t, ms, 0) assert.Len(t, ms, 0)
} }
func TestMetasAppend(t *testing.T) {
ms := Metas{}
ms = ms.Append("foo", 42)
assert.Len(t, ms, 1)
assert.Equal(t, Meta{"foo", 42}, ms[0])
}
func TestMetasAppendTo(t *testing.T) {
ms := &Metas{}
ms.AppendTo("foo", 42)
assert.Len(t, *ms, 1)
assert.Equal(t, Meta{"foo", 42}, (*ms)[0])
}

View file

@ -1,21 +1,15 @@
package models package models
import ( import (
//"bufio"
//"bytes"
"encoding/json" "encoding/json"
//"fmt"
//"errors"
//"regexp"
//"strings"
//"sync"
"time" "time"
//"codeberg.org/danjones000/my-log/tools"
) )
// A slice of Meta
type Metas []Meta type Metas []Meta
func (ms Metas) toMap() map[string]any { // Returns a single map containing all the Meta. Is useful when encoding to JSON
func (ms Metas) Map() map[string]any {
out := map[string]any{} out := map[string]any{}
for _, f := range ms { for _, f := range ms {
if _, found := out[f.Key]; found || f.Key == "title" || f.Key == "date" { if _, found := out[f.Key]; found || f.Key == "title" || f.Key == "date" {
@ -42,12 +36,14 @@ func (ms Metas) toMap() map[string]any {
return out return out
} }
// Implements json.Marshaler
func (ms Metas) MarshalJSON() ([]byte, error) { func (ms Metas) MarshalJSON() ([]byte, error) {
return json.Marshal(ms.toMap()) return json.Marshal(ms.Map())
} }
// Implements json.Unmarshaler
func (ms *Metas) UnmarshalJSON(in []byte) error { func (ms *Metas) UnmarshalJSON(in []byte) error {
old := (*ms).toMap() old := (*ms).Map()
out := map[string]any{} out := map[string]any{}
err := json.Unmarshal(in, &out) err := json.Unmarshal(in, &out)
if err != nil { if err != nil {
@ -62,3 +58,14 @@ func (ms *Metas) UnmarshalJSON(in []byte) error {
*ms = ret *ms = ret
return nil return nil
} }
// Returns a new Metas with a new Meta appended
func (ms Metas) Append(k string, v any) Metas {
return append(ms, Meta{k, v})
}
// Appends a new Meta to this Metas
func (ms *Metas) AppendTo(k string, v any) {
n := (*ms).Append(k, v)
*ms = n
}