mirror of https://github.com/hashicorp/boundary
Jeff Mitchell
6 years ago
parent
9325a6422d
commit
be1063d168
@ -0,0 +1,9 @@
|
||||
# Determine this makefile's path.
|
||||
# Be sure to place this BEFORE `include` directives, if any.
|
||||
THIS_FILE := $(lastword $(MAKEFILE_LIST))
|
||||
|
||||
api:
|
||||
go run -tags genapi .
|
||||
goimports -w ${APIGEN_BASEPATH}/api
|
||||
|
||||
.PHONY: api
|
||||
@ -0,0 +1,97 @@
|
||||
// +build genapi
|
||||
|
||||
package main
|
||||
|
||||
import "os"
|
||||
|
||||
type generateInfo struct {
|
||||
inFile string
|
||||
inName string
|
||||
outFile string
|
||||
outName string
|
||||
outPkg string
|
||||
structFields string
|
||||
parentName string
|
||||
detailName string
|
||||
templateType templateType
|
||||
}
|
||||
|
||||
var inputStructs = []*generateInfo{
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/error.pb.go",
|
||||
"Error",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/error.go",
|
||||
"Error",
|
||||
"api",
|
||||
"",
|
||||
"",
|
||||
"",
|
||||
templateTypeResource,
|
||||
},
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/error.pb.go",
|
||||
"ErrorDetails",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/error_details.go",
|
||||
"ErrorDetails",
|
||||
"api",
|
||||
"",
|
||||
"",
|
||||
"",
|
||||
templateTypeResource,
|
||||
},
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/resources/hosts/host.pb.go",
|
||||
"Host",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/hosts/host.go",
|
||||
"Host",
|
||||
"hosts",
|
||||
"",
|
||||
"",
|
||||
"",
|
||||
templateTypeResource,
|
||||
},
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/resources/hosts/host_set.pb.go",
|
||||
"HostSet",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/hosts/host_set.go",
|
||||
"HostSet",
|
||||
"hosts",
|
||||
"",
|
||||
"",
|
||||
"",
|
||||
templateTypeResource,
|
||||
},
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/resources/hosts/host_catalog.pb.go",
|
||||
"HostCatalog",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/hosts/host_catalog.go",
|
||||
"HostCatalog",
|
||||
"hosts",
|
||||
"",
|
||||
"",
|
||||
"",
|
||||
templateTypeResource,
|
||||
},
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/resources/hosts/host_catalog.pb.go",
|
||||
"StaticHostCatalogDetails",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/hosts/static_host_catalog.go",
|
||||
"StaticHostCatalogDetails",
|
||||
"hosts",
|
||||
"",
|
||||
"HostCatalog",
|
||||
"StaticHostCatalog",
|
||||
templateTypeDetail,
|
||||
},
|
||||
{
|
||||
os.Getenv("APIGEN_BASEPATH") + "/internal/gen/controller/api/resources/hosts/host_catalog.pb.go",
|
||||
"AwsEc2HostCatalogDetails",
|
||||
os.Getenv("APIGEN_BASEPATH") + "/api/hosts/awsec2_host_catalog.go",
|
||||
"AwsEc2HostCatalogDetails",
|
||||
"hosts",
|
||||
"",
|
||||
"HostCatalog",
|
||||
"AwsEc2HostCatalog",
|
||||
templateTypeDetail,
|
||||
},
|
||||
}
|
||||
@ -0,0 +1,8 @@
|
||||
// +build genapi
|
||||
|
||||
package main
|
||||
|
||||
func main() {
|
||||
parsePBs()
|
||||
writeTemplates()
|
||||
}
|
||||
@ -1,7 +0,0 @@
|
||||
package main
|
||||
|
||||
// +build genapi
|
||||
|
||||
//go:generate go run genapi.go
|
||||
//go:generate goimports -w ../../hosts
|
||||
//go:generate goimports -w ../..
|
||||
@ -0,0 +1,132 @@
|
||||
// +build genapi
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"text/template"
|
||||
"time"
|
||||
)
|
||||
|
||||
type templateType int
|
||||
|
||||
const (
|
||||
templateTypeResource templateType = iota
|
||||
templateTypeDetail
|
||||
)
|
||||
|
||||
func writeTemplates() {
|
||||
for _, inputStruct := range inputStructs {
|
||||
outBuf := new(bytes.Buffer)
|
||||
switch inputStruct.templateType {
|
||||
case templateTypeResource:
|
||||
utilFuncsTemplate.Execute(outBuf, struct {
|
||||
Timestamp time.Time
|
||||
Name string
|
||||
Package string
|
||||
StructFields string
|
||||
}{
|
||||
Name: inputStruct.outName,
|
||||
Package: inputStruct.outPkg,
|
||||
StructFields: inputStruct.structFields,
|
||||
})
|
||||
|
||||
case templateTypeDetail:
|
||||
detailTemplate.Execute(outBuf, struct {
|
||||
Timestamp time.Time
|
||||
Package string
|
||||
StructFields string
|
||||
ParentName string
|
||||
DetailName string
|
||||
}{
|
||||
Package: inputStruct.outPkg,
|
||||
StructFields: inputStruct.structFields,
|
||||
ParentName: inputStruct.parentName,
|
||||
DetailName: inputStruct.detailName,
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
outFile, err := filepath.Abs(inputStruct.outFile)
|
||||
if err != nil {
|
||||
fmt.Printf("error opening file %q: %v\n", inputStruct.outFile, err)
|
||||
os.Exit(1)
|
||||
}
|
||||
if err := ioutil.WriteFile(outFile, outBuf.Bytes(), 0644); err != nil {
|
||||
fmt.Printf("error writing file %q: %v\n", outFile, err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var utilFuncsTemplate = template.Must(template.New("").Parse(`// Code generated by go generate; DO NOT EDIT.
|
||||
package {{ .Package }}
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"github.com/fatih/structs"
|
||||
"github.com/hashicorp/watchtower/api/internal/strutil"
|
||||
)
|
||||
|
||||
type {{ .Name }} struct {
|
||||
{{ .StructFields }}
|
||||
}
|
||||
|
||||
func (s *{{ .Name }}) SetDefault(key string) {
|
||||
s.defaultFields = strutil.AppendIfMissing(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s *{{ .Name }}) UnsetDefault(key string) {
|
||||
s.defaultFields = strutil.StrListDelete(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s {{ .Name }}) MarshalJSON() ([]byte, error) {
|
||||
m := structs.Map(s)
|
||||
if m == nil {
|
||||
m = make(map[string]interface{})
|
||||
}
|
||||
for _, k := range s.defaultFields {
|
||||
m[k] = nil
|
||||
}
|
||||
return json.Marshal(m)
|
||||
}
|
||||
`))
|
||||
|
||||
var detailTemplate = template.Must(template.New("").Parse(`// Code generated by go generate; DO NOT EDIT.
|
||||
package {{ .Package }}
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/mitchellh/mapstructure"
|
||||
)
|
||||
|
||||
type {{ .DetailName }} struct {
|
||||
*{{ .ParentName }}
|
||||
{{ .StructFields }}
|
||||
}
|
||||
|
||||
func (s {{ .ParentName }}) As{{ .DetailName }}() (*{{ .DetailName }}, error) {
|
||||
out := &{{ .DetailName }}{
|
||||
{{ .ParentName }}: &s,
|
||||
}
|
||||
decoder, err := mapstructure.NewDecoder(&mapstructure.DecoderConfig{
|
||||
Result: out,
|
||||
TagName: "json",
|
||||
})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("error creating map decoder: %w", err)
|
||||
}
|
||||
|
||||
if err := decoder.Decode(s.Attributes); err != nil {
|
||||
return nil, fmt.Errorf("error decoding attributes map: %w", err)
|
||||
}
|
||||
|
||||
return out, nil
|
||||
}
|
||||
`))
|
||||
Loading…
Reference in new issue