-
Notifications
You must be signed in to change notification settings - Fork 3
/
store_type.go
113 lines (94 loc) · 2.3 KB
/
store_type.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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright (c) ClaceIO, LLC
// SPDX-License-Identifier: Apache-2.0
package utils
import (
"fmt"
"go.starlark.net/starlark"
)
type TypeName string
const (
INT TypeName = "INT"
FLOAT TypeName = "FLOAT"
DATETIME TypeName = "DATETIME"
STRING TypeName = "STRING"
BOOLEAN TypeName = "BOOLEAN"
LIST TypeName = "LIST"
DICT TypeName = "DICT"
)
type StoreInfo struct {
Bytes []byte // The raw bytes for the schema.star file
Types []StoreType
}
type StoreType struct {
Name string
Fields []StoreField
Indexes []Index
}
type StoreField struct {
Name string
Type TypeName
Default any
}
type Index struct {
Fields []string
Unique bool
}
type TypeBuilder struct {
Name string
Fields []StoreField
}
func (s *TypeBuilder) CreateType(thread *starlark.Thread, _ *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
unpackArgs := make([]any, 0, 2*len(s.Fields))
for _, f := range s.Fields {
// Unpack takes field name followed by a pointer to the value
unpackArgs = append(unpackArgs, f.Name)
var value starlark.Value
switch f.Type {
case INT:
var v starlark.Int
value = v
case FLOAT:
var v starlark.Float
value = v
case DATETIME:
var v starlark.Int
value = v
case STRING:
var v starlark.String
value = v
case BOOLEAN:
var v starlark.Bool
value = v
case LIST:
var v *starlark.List
value = v
case DICT:
var v *starlark.Dict
value = v
default:
return nil, fmt.Errorf("unknown type %s for %s", f.Type, f.Name)
}
// Add value pointer
unpackArgs = append(unpackArgs, &value)
}
if err := starlark.UnpackArgs(s.Name, args, kwargs, unpackArgs...); err != nil {
return nil, err
}
valueMap := make(map[string]starlark.Value)
for i := 0; i < len(unpackArgs); i += 2 {
argName := unpackArgs[i].(string)
var ok bool
val, ok := (unpackArgs[i+1]).(*starlark.Value)
if !ok {
return nil, fmt.Errorf("invalid type for %s", argName)
}
valueMap[argName] = *val
}
valueMap["_id"] = starlark.MakeInt(0)
valueMap["_version"] = starlark.MakeInt(0)
valueMap["_created_by"] = starlark.String("")
valueMap["_updated_by"] = starlark.String("")
valueMap["_created_at"] = starlark.MakeInt(0)
valueMap["_updated_at"] = starlark.MakeInt(0)
return NewStarlarkType(s.Name, valueMap), nil
}