-
Notifications
You must be signed in to change notification settings - Fork 4
/
object.go
94 lines (77 loc) · 2.22 KB
/
object.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
package types
import (
"github.com/dave/jennifer/jen"
"github.com/dipdup-net/go-lib/tzkt/data"
)
// Object -
type Object struct{}
// AsField -
func (Object) AsField(name, path string, schema data.JSONSchema, isRequired bool, result *ContractTypeResult) (jen.Code, error) {
tags := map[string]string{
"json": name,
}
if isRequired {
tags["validate"] = TagRequired
}
return jen.Id(fieldName("Object", name)).Add(jen.Id("Unit")).Tag(tags), nil
}
// AsCode -
func (obj Object) AsCode(name, path string, schema data.JSONSchema, result *ContractTypeResult) (Code, error) {
code, err := obj.create(name, path, schema, result)
if err != nil {
return Code{}, err
}
statement := jen.Comment(code.Name).Line().Type().Id(code.Name)
return Code{
Statement: statement.Add(code.Statement),
Name: code.Name,
}, nil
}
// AsType -
func (obj Object) AsType(name, path string, schema data.JSONSchema, result *ContractTypeResult) (Code, error) {
return obj.create(name, path, schema, result)
}
func (Object) create(name, path string, schema data.JSONSchema, result *ContractTypeResult) (Code, error) {
fields := make([]jen.Code, 0)
statements := make([]jen.Code, 0)
for propName, props := range schema.Properties {
var required bool
for j := range schema.Required {
if schema.Required[j] == propName {
required = true
break
}
}
typ, err := selectType(props)
if err != nil {
return Code{}, err
}
newPath := getPath(path, name)
if isSimpleType(props.Comment) || props.Comment == "big_map" {
field, err := typ.AsField(propName, newPath, props, required, result)
if err != nil {
return Code{}, err
}
fields = append(fields, field)
} else {
code, err := typ.AsCode(propName, newPath, props, result)
if err != nil {
return Code{}, err
}
if code.Statement != nil {
statements = append(statements, code.Statement)
tags := map[string]string{
"json": propName,
}
if required {
tags["validate"] = TagRequired
}
fields = append(fields, jen.Id(fieldName(props.Comment, propName)).Op(code.Name).Tag(tags))
}
}
}
return Code{
Statement: jen.Struct(fields...).Line().Add(statements...).Line(),
Name: result.GetName("Object", name),
}, nil
}