-
Notifications
You must be signed in to change notification settings - Fork 13
/
main.go
99 lines (86 loc) · 2.75 KB
/
main.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
// Copyright 2016-2021 The Libsacloud Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package main
import (
"log"
"path/filepath"
"github.com/sacloud/libsacloud/v2/internal/define"
"github.com/sacloud/libsacloud/v2/internal/tools"
)
const destination = "sacloud/zz_api_transformers.go"
func init() {
log.SetFlags(0)
log.SetPrefix("gen-api-transformer: ")
}
func main() {
outputPath := destination
tools.WriteFileWithTemplate(&tools.TemplateConfig{
OutputPath: filepath.Join(tools.ProjectRootPath(), outputPath),
Template: tmpl,
Parameter: define.APIs,
})
log.Printf("generated: %s\n", outputPath)
}
const tmpl = `// generated by 'github.com/sacloud/libsacloud/internal/tools/gen-api-transformer'; DO NOT EDIT
package sacloud
import (
"encoding/json"
"github.com/sacloud/libsacloud/v2/pkg/mapconv"
"github.com/sacloud/libsacloud/v2/sacloud/types"
)
{{ range . }}{{ $typeName := .TypeName }}{{$resource := .}}
{{ range .Operations }}{{$returnErrStatement := .ReturnErrorStatement}}{{ $operationName := .MethodName }}
{{ if .HasRequestEnvelope }}
func (o *{{ $typeName }}Op) transform{{.MethodName}}Args({{ range .Arguments }}{{ .ArgName }} {{ .TypeName }},{{ end }}) (*{{.RequestEnvelopeStructName}}, error) {
{{- range $i, $v := .Arguments }}
if {{.ArgName}} == {{.ZeroValueOnSource}} {
{{.ArgName}} = {{.ZeroInitializer}}
}
var arg{{$i}} interface{} = {{.ArgName}}
if v , ok := arg{{$i}}.(argumentDefaulter); ok {
arg{{$i}} = v.setDefaults()
}
{{- end }}
args := &struct {
{{- range $i, $v := .Arguments }}
Arg{{ $i }} interface{} {{.MapConvTagSrc}}
{{- end }}
}{
{{- range $i, $v := .Arguments }}
Arg{{ $i }}: arg{{ $i }},
{{- end }}
}
v := &{{.RequestEnvelopeStructName}}{}
if err := mapconv.ConvertTo(args, v); err != nil {
return nil, err
}
return v, nil
}
{{ end -}}
{{ if .HasResponseEnvelope }}
func (o *{{ $typeName }}Op) transform{{.MethodName}}Results(data []byte) (*{{.ResultTypeName}}, error) {
nakedResponse := &{{.ResponseEnvelopeStructName}}{}
if err := json.Unmarshal(data, nakedResponse); err != nil {
return nil, err
}
results := &{{.ResultTypeName}}{}
if err := mapconv.ConvertFrom(nakedResponse, results); err != nil {
return nil, err
}
return results, nil
}
{{ end -}}
{{ end -}}
{{ end -}}
`