1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
package mt
import (
"encoding/json"
"strings"
)
type ItemMeta string
var sanitizer = strings.NewReplacer(
string(1), "",
string(2), "",
string(3), "",
)
func NewItemMeta(fields []Field) ItemMeta {
if len(fields) == 0 {
return ""
}
b := new(strings.Builder)
b.WriteByte(1)
for _, f := range fields {
sanitizer.WriteString(b, f.Name)
b.WriteByte(2)
sanitizer.WriteString(b, f.Value)
b.WriteByte(3)
}
return ItemMeta(b.String())
}
func (m ItemMeta) Fields() []Field {
var f []Field
if len(m) > 0 && m[0] == 1 {
m = m[1:]
eat := func(stop byte) string {
if i := strings.IndexByte(string(m), stop); i != -1 {
defer func() {
m = m[i+1:]
}()
return string(m[:i])
}
defer func() {
m = ""
}()
return string(m)
}
for len(m) > 0 {
f = append(f, Field{eat(2), eat(3)})
}
return f
}
return []Field{{"", string(m)}}
}
func (m ItemMeta) Field(name string) (s string, ok bool) {
for _, f := range m.Fields() {
if f.Name == name {
s, ok = f.Value, true
}
}
return
}
func (m *ItemMeta) SetField(name, value string) {
var fields []Field
for _, f := range m.Fields() {
if f.Name != name {
fields = append(fields, f)
}
}
fields = append(fields, Field{name, value})
*m = NewItemMeta(fields)
}
func (m ItemMeta) ToolCaps() (ToolCaps, bool) {
f, ok := m.Field("tool_capabilities")
if !ok {
return ToolCaps{}, false
}
var tc ToolCaps
if err := json.Unmarshal([]byte(f), &tc); err != nil {
return tc, false
}
return tc, true
}
func (m *ItemMeta) SetToolCaps(tc ToolCaps) {
b, err := tc.MarshalJSON()
if err != nil {
panic(err)
}
m.SetField("tool_capabilities", string(b))
}
|
