-
Notifications
You must be signed in to change notification settings - Fork 6
/
requests.go
179 lines (152 loc) · 4.1 KB
/
requests.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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
package hasura
import (
"os"
"github.com/pkg/errors"
)
type Request struct {
Type string `json:"type"`
Args interface{} `json:"args"`
}
type versionedRequest struct {
Type string `json:"type"`
Version int `json:"int"`
Args interface{} `json:"args"`
}
// Permission -
type Permission struct {
Columns Columns `json:"columns"`
Limit uint64 `json:"limit"`
AllowAggs bool `json:"allow_aggregations"`
Filter interface{} `json:"filter,omitempty"`
}
// Metadata -
type Metadata struct {
Version int `json:"version"`
Sources []Source `json:"sources"`
QueryCollections []QueryCollection `json:"query_collections,omitempty"`
RestEndpoints []interface{} `json:"rest_endpoints,omitempty"`
}
func newMetadata(version int, sources []Source) *Metadata {
return &Metadata{
Version: version,
Sources: sources,
}
}
// Configuration -
type Configuration struct {
ConnectionInfo ConnectionInfo `json:"connection_info"`
}
// ConnectionInfo -
type ConnectionInfo struct {
UsePreparedStatements bool `json:"use_prepared_statements"`
IsolationLevel string `json:"isolation_level"`
DatabaseUrl DatabaseUrl `json:"database_url"`
}
// DatabaseUrl -
type DatabaseUrl string
// DatabaseUrlFromEnv -
type DatabaseUrlFromEnv struct {
FromEnv string `json:"from_env"`
}
// UnmarshalJSON -
func (d *DatabaseUrl) UnmarshalJSON(data []byte) error {
var s string
if err := json.Unmarshal(data, &s); err == nil {
*d = DatabaseUrl(s)
return nil
}
var fromEnv DatabaseUrlFromEnv
if err := json.Unmarshal(data, &fromEnv); err != nil {
return err
}
*d = DatabaseUrl(os.Getenv(fromEnv.FromEnv))
return nil
}
// Source -
type Source struct {
Name string `json:"name"`
Kind string `json:"kind"`
Tables []Table `json:"tables"`
Configuration Configuration `json:"configuration"`
}
// Table -
type Table struct {
ObjectRelationships []interface{} `json:"object_relationships"`
ArrayRelationships []interface{} `json:"array_relationships"`
SelectPermissions []SelectPermission `json:"select_permissions"`
Configuration TableConfiguration `json:"configuration"`
Schema TableSchema `json:"table"`
}
func newMetadataTable(name, schema string) Table {
return Table{
ObjectRelationships: make([]interface{}, 0),
ArrayRelationships: make([]interface{}, 0),
SelectPermissions: make([]SelectPermission, 0),
Schema: TableSchema{
Name: name,
Schema: schema,
},
}
}
// TableConfiguration -
type TableConfiguration struct {
Comment *string `json:"comment"`
CustomRootFields map[string]string `json:"custom_root_fields"`
CustomColumnNames map[string]string `json:"custom_column_names"`
}
// TableSchema -
type TableSchema struct {
Schema string `json:"schema"`
Name string `json:"name"`
}
// SelectPermission -
type SelectPermission struct {
Role string `json:"role"`
Permission Permission `json:"permission"`
}
// Columns -
type Columns []string
// UnmarshalJSON -
func (columns *Columns) UnmarshalJSON(data []byte) error {
var val interface{}
if err := json.Unmarshal(data, &val); err != nil {
return err
}
*columns = make(Columns, 0)
switch typ := val.(type) {
case string:
*columns = append(*columns, typ)
case []interface{}:
for i := range typ {
if s, ok := typ[i].(string); ok {
*columns = append(*columns, s)
}
}
default:
return errors.Errorf("Invalid columns type: %T", typ)
}
return nil
}
// MarshalJSON -
func (columns Columns) MarshalJSON() ([]byte, error) {
if len(columns) == 1 && columns[0] == "*" {
return []byte(`"*"`), nil
}
s := []string(columns)
return json.Marshal(s)
}
// QueryCollection -
type QueryCollection struct {
Definition Definition `json:"definition"`
Name string `json:"name"`
}
// Definition -
type Definition struct {
Queries []Query `json:"queries"`
}
// Query -
type Query struct {
Name string `json:"name"`
Query string `json:"query,omitempty"`
CollectionName string `json:"collection_name,omitempty"`
}