-
Notifications
You must be signed in to change notification settings - Fork 87
/
embed.go
107 lines (91 loc) · 2.01 KB
/
embed.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
package mappers
import (
"fmt"
"github.com/rancher/wrangler/pkg/data"
types "github.com/rancher/wrangler/pkg/schemas"
)
type Embed struct {
Field string
Optional bool
ReadOnly bool
Ignore []string
ignoreOverride bool
embeddedFields []string
EmptyValueOk bool
}
func (e *Embed) FromInternal(data data.Object) {
sub := data.Map(e.Field)
for _, fieldName := range e.embeddedFields {
if v, ok := sub[fieldName]; ok {
data[fieldName] = v
}
}
delete(data, e.Field)
}
func (e *Embed) ToInternal(data data.Object) error {
if data == nil {
return nil
}
sub := map[string]interface{}{}
for _, fieldName := range e.embeddedFields {
if v, ok := data[fieldName]; ok {
sub[fieldName] = v
}
delete(data, fieldName)
}
if len(sub) == 0 {
if e.EmptyValueOk {
data[e.Field] = nil
}
return nil
}
data[e.Field] = sub
return nil
}
func (e *Embed) ModifySchema(schema *types.Schema, schemas *types.Schemas) error {
err := ValidateField(e.Field, schema)
if err != nil {
if e.Optional {
return nil
}
return err
}
e.embeddedFields = []string{}
embeddedSchemaID := schema.ResourceFields[e.Field].Type
embeddedSchema := schemas.Schema(embeddedSchemaID)
if embeddedSchema == nil {
if e.Optional {
return nil
}
return fmt.Errorf("failed to find schema %s for embedding", embeddedSchemaID)
}
deleteField := true
outer:
for name, field := range embeddedSchema.ResourceFields {
for _, ignore := range e.Ignore {
if ignore == name {
continue outer
}
}
if name == e.Field {
deleteField = false
} else {
if !e.ignoreOverride {
if _, ok := schema.ResourceFields[name]; ok {
return fmt.Errorf("embedding field %s on %s will overwrite the field %s",
e.Field, schema.ID, name)
}
}
}
if e.ReadOnly {
field.Create = false
field.Update = false
}
schema.ResourceFields[name] = field
e.embeddedFields = append(e.embeddedFields, name)
}
if deleteField {
delete(schema.ResourceFields, e.Field)
}
return nil
}