Skip to content

Commit

Permalink
add value.Tag and Tags method
Browse files Browse the repository at this point in the history
  • Loading branch information
rot1024 committed Dec 22, 2023
1 parent e882c68 commit 59a73f0
Show file tree
Hide file tree
Showing 3 changed files with 245 additions and 123 deletions.
60 changes: 56 additions & 4 deletions go/model_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -154,18 +154,70 @@ func TestField_ValueFloat(t *testing.T) {
}).GetValue().Float())
}

func TestField_ValueTag(t *testing.T) {
assert.Equal(t, &Tag{
ID: "xxx",
Name: "tag",
Color: "red",
}, (&Field{
Value: map[string]any{
"id": "xxx",
"name": "tag",
"color": "red",
},
}).GetValue().Tag())
assert.Nil(t, (&Field{
Value: 100,
}).GetValue().Tag())
}

func TestField_ValueTags(t *testing.T) {
assert.Equal(t, []Tag{
{ID: "xxx"},
{ID: "yyy"},
}, (&Field{
Value: []any{
map[string]any{"id": "xxx"}, map[string]any{"id": "yyy"},
},
}).GetValue().Tags())
assert.Equal(t, []Tag{
{ID: "xxx"},
{ID: "yyy"},
}, (&Field{
Value: []map[string]any{
{"id": "xxx"}, {"id": "yyy"},
},
}).GetValue().Tags())
assert.Nil(t, (&Field{
Value: map[string]any{
"id": "xxx",
"name": "tag",
"color": "red",
},
}).GetValue().Tags())
}

func TestField_ValueJSON(t *testing.T) {
r, err := (&Field{
var r any
err := (&Field{
Value: `{"foo":"bar"}`,
}).GetValue().JSON()
}).GetValue().JSON(&r)
assert.NoError(t, err)
assert.Equal(t, map[string]any{"foo": "bar"}, r)
}

func TestField_ValueJSONs(t *testing.T) {
r, err := (&Field{
r := []any{}
err := (&Field{
Value: []string{`{"foo":"bar"}`, `{"foo":"hoge"}`},
}).GetValue().JSONs(&r)
assert.NoError(t, err)
assert.Equal(t, []any{map[string]any{"foo": "bar"}, map[string]any{"foo": "hoge"}}, r)

r = []any{nil, nil}
err = (&Field{
Value: []string{`{"foo":"bar"}`, `{"foo":"hoge"}`},
}).GetValue().JSONs()
}).GetValue().JSONs(r)
assert.NoError(t, err)
assert.Equal(t, []any{map[string]any{"foo": "bar"}, map[string]any{"foo": "hoge"}}, r)
}
Expand Down
119 changes: 0 additions & 119 deletions go/model_value.go

This file was deleted.

189 changes: 189 additions & 0 deletions go/value.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,189 @@
package cms

import (
"encoding/json"
"fmt"

"github.com/samber/lo"
)

type Tag struct {
ID string `json:"id"`
Name string `json:"name"`
Color string `json:"color"`
}

func TagFrom(j any) *Tag {
if j == nil {
return nil
}

t := Tag{}
if id, ok := j.(map[string]any)["id"].(string); ok {
t.ID = id
}
if name, ok := j.(map[string]any)["name"].(string); ok {
t.Name = name
}
if color, ok := j.(map[string]any)["color"].(string); ok {
t.Color = color
}

return &t
}

type Value struct {
value any
}

func NewValue(value any) *Value {
return &Value{value: value}
}

func (v *Value) Interface() any {
if v == nil {
return nil
}
return v.value
}

func (v *Value) String() *string {
return getValue[string](v)
}

func (v *Value) Int() *int {
return getValue[int](v)
}

func (v *Value) Float() *float64 {
return getValue[float64](v)
}

func (v *Value) Bool() *bool {
return getValue[bool](v)
}

func (v *Value) Strings() []string {
return getValues[string](v)
}

func (v *Value) Ints() []int {
return getValues[int](v)
}

func (v *Value) Floats() []float64 {
return getValues[float64](v)
}

func (v *Value) Bools() []bool {
return getValues[bool](v)
}

func (v *Value) Tag() *Tag {
if v == nil {
return nil
}
return TagFrom(v.value)
}

func (v *Value) Tags() []Tag {
values, ok := v.value.([]any)
if !ok {
values2, ok := v.value.([]map[string]any)
if !ok {
return nil
}

values = make([]any, len(values2))
for i, value := range values2 {
values[i] = value
}
}

res := make([]Tag, len(values))
for i, value := range values {
if t := TagFrom(value); t != nil {
res[i] = *t
}
}

return res
}

func (f *Value) JSON(j any) error {
if f == nil {
return nil
}

s := f.String()
if s == nil {
return nil
}

err := json.Unmarshal([]byte(*s), &j)
return err
}

func (f *Value) JSONs(j any) error {
if f == nil {
return nil
}

values := f.Strings()
if values == nil {
return nil
}

if res, ok := j.(*[]any); ok {
*res = make([]any, len(values))
j = *res
}

res, ok := j.([]any)
if !ok {
return nil
}

if len(values) != len(res) {
return fmt.Errorf("length of values and j must be same")
}

for i, v := range values {
if err := json.Unmarshal([]byte(v), &res[i]); err != nil {
return fmt.Errorf("unmarshal json error at index %d: %w", i, err)
}
}

return nil
}

func getValue[T any](v *Value) *T {
if v == nil {
return nil
}

if v, ok := v.value.(T); ok {
return &v
}

return nil
}

func getValues[T any](v *Value) []T {
if v == nil {
return nil
}

if v, ok := v.value.([]T); ok {
return v
}

if v, ok := v.value.([]any); ok {
return lo.FilterMap(v, func(e any, _ int) (T, bool) {
s, ok := e.(T)
return s, ok
})
}

return nil
}

0 comments on commit 59a73f0

Please sign in to comment.