mirror of https://github.com/hashicorp/boundary
Add initial API generation bits (#20)
* Add initial API generation bits * Revert accidental changes to oplogpull/22/head
Jeff Mitchell
6 years ago
committed by
GitHub
parent
7f03246c6f
commit
2926cc1000
@ -0,0 +1,44 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.4620286 -0400 EDT m=+0.042201401
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"github.com/fatih/structs"
|
||||
"github.com/hashicorp/watchtower/api/internal/strutil"
|
||||
)
|
||||
|
||||
type Error struct {
|
||||
defaultFields []string
|
||||
|
||||
// The HTTP Status code applicable to this error
|
||||
Status *int64 `json:"status,omitempty"`
|
||||
// An application-specific error string
|
||||
Code *string `json:"code,omitempty"`
|
||||
// A human readable explanation specific to this occurrence of the error
|
||||
Message *string `json:"message,omitempty"`
|
||||
// Additional metadata regarding the error. Depending on the error
|
||||
// different fields will be populated.
|
||||
Details *ErrorDetails `json:"details,omitempty"`
|
||||
}
|
||||
|
||||
func (s *Error) SetDefault(key string) {
|
||||
s.defaultFields = strutil.AppendIfMissing(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s *Error) UnsetDefault(key string) {
|
||||
s.defaultFields = strutil.StrListDelete(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s Error) 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)
|
||||
}
|
||||
@ -0,0 +1,38 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.462308 -0400 EDT m=+0.042480801
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"github.com/fatih/structs"
|
||||
"github.com/hashicorp/watchtower/api/internal/strutil"
|
||||
)
|
||||
|
||||
type ErrorDetails struct {
|
||||
defaultFields []string
|
||||
|
||||
TraceId *string `json:"trace_id,omitempty"`
|
||||
RequestId *string `json:"request_id,omitempty"`
|
||||
RequestFields []string `json:"request_fields,omitempty"`
|
||||
}
|
||||
|
||||
func (s *ErrorDetails) SetDefault(key string) {
|
||||
s.defaultFields = strutil.AppendIfMissing(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s *ErrorDetails) UnsetDefault(key string) {
|
||||
s.defaultFields = strutil.StrListDelete(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s ErrorDetails) 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)
|
||||
}
|
||||
@ -0,0 +1,42 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.4628286 -0400 EDT m=+0.043001401
|
||||
package hosts
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/mitchellh/mapstructure"
|
||||
)
|
||||
|
||||
type AwsEc2HostCatalog struct {
|
||||
*HostCatalog
|
||||
|
||||
// The AWS regions from which this catalog will retrieve the EC2 instances.
|
||||
Regions []string `json:"regions,omitempty"`
|
||||
// The access key used for authenticating with AWS when retrieving EC2 instance details.
|
||||
AccessKey *string `json:"access_key,omitempty"`
|
||||
// Input only.
|
||||
SecretKey *string `json:"secret_key,omitempty"`
|
||||
// This value will never be returned in a response.
|
||||
Rotate *bool `json:"rotate,omitempty"`
|
||||
}
|
||||
|
||||
func (s HostCatalog) AsAwsEc2HostCatalog() (*AwsEc2HostCatalog, error) {
|
||||
out := &AwsEc2HostCatalog{
|
||||
HostCatalog: &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
|
||||
}
|
||||
@ -0,0 +1,53 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.4624224 -0400 EDT m=+0.042595201
|
||||
package hosts
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"github.com/fatih/structs"
|
||||
"github.com/hashicorp/watchtower/api/internal/strutil"
|
||||
)
|
||||
|
||||
type Host struct {
|
||||
defaultFields []string
|
||||
|
||||
// Canonical path of the resource from the API's base URI
|
||||
// Output only.
|
||||
Path *string `json:"path,omitempty"`
|
||||
// The type of the resource, to help differentiate schemas
|
||||
Type *string `json:"type,omitempty"`
|
||||
// Friendly name, if set
|
||||
FriendlyName *string `json:"friendly_name,omitempty"`
|
||||
// The time this host was created
|
||||
// Ouput only.
|
||||
CreatedTime time.Time `json:"created_time,omitempty"`
|
||||
// The time this host was last updated
|
||||
// Output only.
|
||||
UpdatedTime time.Time `json:"updated_time,omitempty"`
|
||||
// Whether the host is disabled
|
||||
Disabled *bool `json:"disabled,omitempty"`
|
||||
// The address (DNS or IP name) used to reach the host
|
||||
Address *string `json:"address,omitempty"`
|
||||
}
|
||||
|
||||
func (s *Host) SetDefault(key string) {
|
||||
s.defaultFields = strutil.AppendIfMissing(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s *Host) UnsetDefault(key string) {
|
||||
s.defaultFields = strutil.StrListDelete(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s Host) 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)
|
||||
}
|
||||
@ -0,0 +1,54 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.4626029 -0400 EDT m=+0.042775701
|
||||
package hosts
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"github.com/fatih/structs"
|
||||
"github.com/hashicorp/watchtower/api/internal/strutil"
|
||||
)
|
||||
|
||||
type HostCatalog struct {
|
||||
defaultFields []string
|
||||
|
||||
// Canonical path of the resource from the API's base URI
|
||||
// Output only.
|
||||
Path *string `json:"path,omitempty"`
|
||||
// The type of the resource, to help differentiate schemas
|
||||
Type *string `json:"type,omitempty"`
|
||||
// Friendly name, if set
|
||||
FriendlyName *string `json:"friendly_name,omitempty"`
|
||||
// The time this host was created
|
||||
// Output only.
|
||||
CreatedTime time.Time `json:"created_time,omitempty"`
|
||||
// The time this host was last updated
|
||||
// Output only.
|
||||
UpdatedTime time.Time `json:"updated_time,omitempty"`
|
||||
// Whether the catalog is disabled
|
||||
Disabled *bool `json:"disabled,omitempty"`
|
||||
// Attributes specific to the catalog type
|
||||
Attributes map[string]interface {
|
||||
} `json:"attributes,omitempty"`
|
||||
}
|
||||
|
||||
func (s *HostCatalog) SetDefault(key string) {
|
||||
s.defaultFields = strutil.AppendIfMissing(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s *HostCatalog) UnsetDefault(key string) {
|
||||
s.defaultFields = strutil.StrListDelete(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s HostCatalog) 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)
|
||||
}
|
||||
@ -0,0 +1,59 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.4625345 -0400 EDT m=+0.042707301
|
||||
package hosts
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"github.com/fatih/structs"
|
||||
"github.com/hashicorp/watchtower/api/internal/strutil"
|
||||
)
|
||||
|
||||
type HostSet struct {
|
||||
defaultFields []string
|
||||
|
||||
// Canonical path of the resource from the API's base URI
|
||||
// Output only.
|
||||
Path *string `json:"path,omitempty"`
|
||||
// The type of the resource, to help differentiate schemas
|
||||
Type *string `json:"type,omitempty"`
|
||||
// Friendly name, if set
|
||||
FriendlyName *string `json:"friendly_name,omitempty"`
|
||||
// The time this host was created
|
||||
// Output only.
|
||||
CreatedTime time.Time `json:"created_time,omitempty"`
|
||||
// The time this host was last updated
|
||||
// Output only.
|
||||
UpdatedTime time.Time `json:"updated_time,omitempty"`
|
||||
// Whether the host set is disabled
|
||||
Disabled *bool `json:"disabled,omitempty"`
|
||||
// The total count of hosts in this host set
|
||||
// Output only.
|
||||
Size *int64 `json:"size,omitempty"`
|
||||
// A list of hosts in this host set
|
||||
// TODO: Figure out if this should be in the basic HostSet view and what
|
||||
// view to use on the Hosts.
|
||||
// Output only.
|
||||
Hosts []*Host `json:"hosts,omitempty"`
|
||||
}
|
||||
|
||||
func (s *HostSet) SetDefault(key string) {
|
||||
s.defaultFields = strutil.AppendIfMissing(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s *HostSet) UnsetDefault(key string) {
|
||||
s.defaultFields = strutil.StrListDelete(s.defaultFields, key)
|
||||
}
|
||||
|
||||
func (s HostSet) 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)
|
||||
}
|
||||
@ -0,0 +1,33 @@
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// 2020-05-03 14:33:51.4627143 -0400 EDT m=+0.042887101
|
||||
package hosts
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/mitchellh/mapstructure"
|
||||
)
|
||||
|
||||
type StaticHostCatalog struct {
|
||||
*HostCatalog
|
||||
}
|
||||
|
||||
func (s HostCatalog) AsStaticHostCatalog() (*StaticHostCatalog, error) {
|
||||
out := &StaticHostCatalog{
|
||||
HostCatalog: &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
|
||||
}
|
||||
@ -0,0 +1,488 @@
|
||||
package main
|
||||
|
||||
// +build genapi
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"go/ast"
|
||||
"go/format"
|
||||
"go/parser"
|
||||
"go/token"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"strings"
|
||||
"text/template"
|
||||
"time"
|
||||
)
|
||||
|
||||
type templateType int
|
||||
|
||||
const (
|
||||
templateTypeResource templateType = iota
|
||||
templateTypeDetail
|
||||
)
|
||||
|
||||
type generateInfo struct {
|
||||
inFile string
|
||||
inName string
|
||||
outFile string
|
||||
outName string
|
||||
outPkg string
|
||||
structFields string
|
||||
parentName string
|
||||
detailName string
|
||||
templateType templateType
|
||||
}
|
||||
|
||||
var (
|
||||
regex = regexp.MustCompile(`(json:".*")`)
|
||||
|
||||
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,
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
type visitFn func(node ast.Node)
|
||||
|
||||
func (fn visitFn) Visit(node ast.Node) ast.Visitor {
|
||||
fn(node)
|
||||
return fn
|
||||
}
|
||||
|
||||
func main() {
|
||||
createStructs()
|
||||
createUtilFuncs()
|
||||
}
|
||||
|
||||
func createStructs() {
|
||||
for _, inputStruct := range inputStructs {
|
||||
inFile, err := filepath.Abs(inputStruct.inFile)
|
||||
if err != nil {
|
||||
fmt.Printf("error opening file %q: %v\n", inputStruct.inFile, err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fset := token.NewFileSet()
|
||||
inAst, err := parser.ParseFile(fset, inFile, nil, parser.ParseComments)
|
||||
if err != nil {
|
||||
fmt.Printf("error parsing %s: %v\n", inFile, err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
ast.Walk(visitFn(func(n ast.Node) {
|
||||
spec, ok := n.(*ast.TypeSpec)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
if spec.Name == nil {
|
||||
return
|
||||
}
|
||||
if spec.Name.Name != inputStruct.inName {
|
||||
return
|
||||
}
|
||||
|
||||
spec.Name.Name = inputStruct.outName
|
||||
st, ok := spec.Type.(*ast.StructType)
|
||||
if !ok {
|
||||
fmt.Printf("expected struct type for identifier, got %t\n", spec.Type)
|
||||
os.Exit(1)
|
||||
return
|
||||
}
|
||||
|
||||
if st.Fields.List == nil {
|
||||
fmt.Printf("no fields found in %q\n", inputStruct.inName)
|
||||
os.Exit(1)
|
||||
return
|
||||
}
|
||||
|
||||
// Tee up unexported field deletion
|
||||
var elideList []int
|
||||
{
|
||||
defer func() {
|
||||
var cutCount int
|
||||
// Remove unexported proto stuff
|
||||
for _, val := range elideList {
|
||||
loc := val - cutCount
|
||||
st.Fields.List = append(st.Fields.List[:loc], st.Fields.List[loc+1:]...)
|
||||
cutCount++
|
||||
}
|
||||
|
||||
// Add default fields if a base resource
|
||||
if inputStruct.templateType == templateTypeResource {
|
||||
st.Fields.List = append([]*ast.Field{{
|
||||
Names: []*ast.Ident{
|
||||
{
|
||||
Name: "defaultFields",
|
||||
Obj: &ast.Object{
|
||||
Kind: ast.Var,
|
||||
Name: "defaultFields",
|
||||
},
|
||||
},
|
||||
},
|
||||
Type: &ast.ArrayType{
|
||||
Elt: &ast.Ident{
|
||||
Name: "string",
|
||||
},
|
||||
},
|
||||
}}, st.Fields.List...)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
for i, field := range st.Fields.List {
|
||||
if !field.Names[0].IsExported() {
|
||||
elideList = append(elideList, i)
|
||||
continue
|
||||
}
|
||||
|
||||
// Anything that isn't a basic type we expect to be a star
|
||||
// expression with a selector; that is, a wrapper value, timestamp
|
||||
// value, etc.
|
||||
//
|
||||
// TODO: this isn't necessarily a good assumption, which means we
|
||||
// might get failures with other types. This is an internal tools
|
||||
// only; we can revisit as needed!
|
||||
var selectorExpr *ast.SelectorExpr
|
||||
switch typ := field.Type.(type) {
|
||||
case *ast.Ident:
|
||||
typ.Name = "*" + typ.Name
|
||||
goto TAGMODIFY
|
||||
case *ast.ArrayType:
|
||||
goto TAGMODIFY
|
||||
case *ast.StarExpr:
|
||||
switch nextTyp := typ.X.(type) {
|
||||
case *ast.Ident:
|
||||
// Already a pointer, don't do anything
|
||||
goto TAGMODIFY
|
||||
case *ast.SelectorExpr:
|
||||
selectorExpr = nextTyp
|
||||
}
|
||||
case *ast.SelectorExpr:
|
||||
selectorExpr = typ
|
||||
}
|
||||
|
||||
switch {
|
||||
case selectorExpr != nil:
|
||||
xident, ok := selectorExpr.X.(*ast.Ident)
|
||||
if !ok {
|
||||
fmt.Printf("unexpected non-ident type in selector\n")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
switch xident.Name {
|
||||
case "wrappers":
|
||||
switch selectorExpr.Sel.Name {
|
||||
case "StringValue":
|
||||
st.Fields.List[i] = &ast.Field{
|
||||
Names: field.Names,
|
||||
Type: &ast.Ident{
|
||||
Name: "*string",
|
||||
},
|
||||
Tag: field.Tag,
|
||||
}
|
||||
case "BoolValue":
|
||||
st.Fields.List[i] = &ast.Field{
|
||||
Names: field.Names,
|
||||
Type: &ast.Ident{
|
||||
Name: "*bool",
|
||||
},
|
||||
Tag: field.Tag,
|
||||
}
|
||||
case "Int64Value":
|
||||
st.Fields.List[i] = &ast.Field{
|
||||
Names: field.Names,
|
||||
Type: &ast.Ident{
|
||||
Name: "*int64",
|
||||
},
|
||||
Tag: field.Tag,
|
||||
}
|
||||
default:
|
||||
fmt.Printf("unhandled wrappers selector sel name %q\n", selectorExpr.Sel.Name)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
case "timestamp":
|
||||
switch selectorExpr.Sel.Name {
|
||||
case "Timestamp":
|
||||
st.Fields.List[i] = &ast.Field{
|
||||
Names: field.Names,
|
||||
Type: &ast.Ident{
|
||||
Name: "time.Time",
|
||||
},
|
||||
Tag: field.Tag,
|
||||
}
|
||||
|
||||
default:
|
||||
fmt.Printf("unhandled timestamp selector sel name %q\n", selectorExpr.Sel.Name)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
case "_struct":
|
||||
switch selectorExpr.Sel.Name {
|
||||
case "Struct":
|
||||
st.Fields.List[i] = &ast.Field{
|
||||
Names: field.Names,
|
||||
Type: &ast.MapType{
|
||||
Key: &ast.Ident{
|
||||
Name: "string",
|
||||
},
|
||||
Value: &ast.InterfaceType{
|
||||
Methods: &ast.FieldList{},
|
||||
},
|
||||
},
|
||||
Tag: field.Tag,
|
||||
}
|
||||
|
||||
default:
|
||||
fmt.Printf("unhandled timestamp selector sel name %q\n", selectorExpr.Sel.Name)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
default:
|
||||
fmt.Printf("unhandled xident name %q\n", xident.Name)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
default:
|
||||
fmt.Println("unhandled non-ident, non-selector case")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
TAGMODIFY:
|
||||
st.Fields.List[i].Tag.Value = "`" + regex.FindString(st.Fields.List[i].Tag.Value) + "`"
|
||||
}
|
||||
}), inAst)
|
||||
|
||||
buf := new(bytes.Buffer)
|
||||
if err := format.Node(buf, fset, inAst); err != nil {
|
||||
fmt.Printf("error formatting new code: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// We have to manually cut out the lines we don't want because comments
|
||||
// won't be preserved otherwise. See the note about lossy comments in
|
||||
// https://arslan.io/2017/09/14/the-ultimate-guide-to-writing-a-go-tool/
|
||||
scanner := bufio.NewScanner(bytes.NewBufferString(buf.String()))
|
||||
var inType bool
|
||||
var outBuf []string
|
||||
for scanner.Scan() {
|
||||
if !inType {
|
||||
if strings.HasPrefix(scanner.Text(), "type "+inputStruct.outName+" struct") {
|
||||
inType = true
|
||||
// Don't add this line, we'll do it in the template
|
||||
continue
|
||||
}
|
||||
} else {
|
||||
if scanner.Text() == "}" {
|
||||
// We've reached the end of the type
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if inType {
|
||||
outBuf = append(outBuf, scanner.Text())
|
||||
}
|
||||
}
|
||||
|
||||
inputStruct.structFields = strings.Join(outBuf, "\n")
|
||||
}
|
||||
}
|
||||
|
||||
func createUtilFuncs() {
|
||||
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
|
||||
}{
|
||||
Timestamp: time.Now(),
|
||||
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
|
||||
}{
|
||||
Timestamp: time.Now(),
|
||||
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.
|
||||
// This file was generated by robots at
|
||||
// {{ .Timestamp }}
|
||||
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)
|
||||
}
|
||||
`))
|
||||
|
||||
// TODO: Figure out the right way to write out the specific fields
|
||||
var detailTemplate = template.Must(template.New("").Parse(`// Code generated by go generate; DO NOT EDIT.
|
||||
// This file was generated by robots at
|
||||
// {{ .Timestamp }}
|
||||
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
|
||||
}
|
||||
`))
|
||||
@ -0,0 +1,7 @@
|
||||
package main
|
||||
|
||||
// +build genapi
|
||||
|
||||
//go:generate go run genapi.go
|
||||
//go:generate goimports -w ../../hosts
|
||||
//go:generate goimports -w ../..
|
||||
@ -0,0 +1,34 @@
|
||||
package tests
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/hashicorp/watchtower/api"
|
||||
"github.com/hashicorp/watchtower/api/hosts"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestDetailTemplating(t *testing.T) {
|
||||
lt := time.Now()
|
||||
c := hosts.HostCatalog{
|
||||
Path: api.String("path"),
|
||||
CreatedTime: lt,
|
||||
Attributes: map[string]interface{}{
|
||||
"regions": []string{"a", "b"},
|
||||
"access_key": "access",
|
||||
"secret_key": "secret",
|
||||
"rotate": true,
|
||||
},
|
||||
}
|
||||
|
||||
ac, err := c.AsAwsEc2HostCatalog()
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, &hosts.AwsEc2HostCatalog{
|
||||
HostCatalog: &c,
|
||||
Regions: []string{"a", "b"},
|
||||
AccessKey: api.String("access"),
|
||||
SecretKey: api.String("secret"),
|
||||
Rotate: api.Bool(true),
|
||||
}, ac)
|
||||
}
|
||||
@ -0,0 +1,19 @@
|
||||
package api
|
||||
|
||||
func Bool(in bool) *bool {
|
||||
ret := new(bool)
|
||||
*ret = in
|
||||
return ret
|
||||
}
|
||||
|
||||
func String(in string) *string {
|
||||
ret := new(string)
|
||||
*ret = in
|
||||
return ret
|
||||
}
|
||||
|
||||
func Int(in int64) *int64 {
|
||||
ret := new(int64)
|
||||
*ret = in
|
||||
return ret
|
||||
}
|
||||
Loading…
Reference in new issue