-
Notifications
You must be signed in to change notification settings - Fork 3
/
starlark_attr.go
89 lines (79 loc) · 1.94 KB
/
starlark_attr.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
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
// Copyright (c) ClaceIO, LLC
// SPDX-License-Identifier: Apache-2.0
package apptype
import (
"fmt"
"go.starlark.net/starlark"
)
func GetStringAttr(s starlark.HasAttrs, key string) (string, error) {
v, err := s.Attr(key)
if err != nil {
return "", fmt.Errorf("error getting %s: %s", key, err)
}
var vs starlark.String
var ok bool
if vs, ok = v.(starlark.String); !ok {
return "", fmt.Errorf("%s is not a string", key)
}
return vs.GoString(), nil
}
func GetIntAttr(s starlark.HasAttrs, key string) (int64, error) {
v, err := s.Attr(key)
if err != nil {
return 0, fmt.Errorf("error getting %s: %s", key, err)
}
var vi starlark.Int
var ok bool
if vi, ok = v.(starlark.Int); !ok {
return 0, fmt.Errorf("%s is not a integer", key)
}
intVal, ok := vi.Int64()
if !ok {
return 0, fmt.Errorf("%s is not a integer", key)
}
return intVal, nil
}
func GetBoolAttr(s starlark.HasAttrs, key string) (bool, error) {
v, err := s.Attr(key)
if err != nil {
return false, fmt.Errorf("error getting %s: %s", key, err)
}
var vb starlark.Bool
var ok bool
if vb, ok = v.(starlark.Bool); !ok {
return false, fmt.Errorf("%s is not a bool", key)
}
return bool(vb), nil
}
func GetListStringAttr(s starlark.HasAttrs, key string, optional bool) ([]string, error) {
v, err := s.Attr(key)
if err != nil {
if optional {
return []string{}, nil
} else {
return nil, fmt.Errorf("error getting %s: %s", key, err)
}
}
var list *starlark.List
var ok bool
if list, ok = v.(*starlark.List); !ok {
return nil, fmt.Errorf("%s is not a list", key)
}
return GetStringList(list)
}
func GetStringList(list *starlark.List) ([]string, error) {
ret := []string{}
iter := list.Iterate()
var ok bool
var val starlark.Value
var vs starlark.String
count := -1
for iter.Next(&val) {
count++
if vs, ok = val.(starlark.String); !ok {
return nil, fmt.Errorf("entry %d in list is not a string", count)
}
ret = append(ret, string(vs))
}
return ret, nil
}