/
set.go
50 lines (42 loc) · 1.11 KB
/
set.go
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
package constructors_common
import (
"strings"
"github.com/Laughs-In-Flowers/countfloyd/lib/feature"
"github.com/Laughs-In-Flowers/data"
)
// A set constructor that takes provided keys and links them to multiple features in a return map.
func Set() feature.Constructor {
return feature.NewConstructor("SET", 50, set)
}
func set(tag string, r *feature.RawFeature, e feature.CEnv) (feature.Informer, feature.Emitter, feature.Mapper) {
raw := r.MustGetValues()
values := argsMustBeLength(e, raw, 1)
ef := func() data.Item {
d := data.New("")
kfs := extractKFS(e, strings.ToLower(tag), values...)
for _, v := range kfs {
i := v.f.Emit()
i.NewKey(v.k)
switch i.(type) {
case data.VectorItem:
if dv, err := v.f.EmitVector(); err == nil {
vi := dv.ToVector()
ldv := vi.List()
for _, ldvi := range ldv {
o := ldvi.Key()
nk := smoothKey(v.k, o)
ldvi.NewKey(nk)
}
d.Set(ldv...)
}
default:
d.Set(i)
}
}
return data.NewVectorItem(tag, d)
}
mf := func(d *data.Vector) {
d.Set(ef())
}
return construct("SET", r.Group, tag, raw, values, ef, mf)
}