Shuffle things around out of new modules

2020-10-07 14:03:42 -04:00
1 parent 70a8051974
commit 5cbfcf8eb6
@@ -0,0 +1,408 @@
+package main
+
+import (
+	"text/template"
+
+	"github.com/hashicorp/boundary/internal/gen/controller/api"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/accounts"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/authmethods"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/authtokens"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/groups"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/hostcatalogs"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/hosts"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/hostsets"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/roles"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/scopes"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/sessions"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/targets"
+	"github.com/hashicorp/boundary/internal/gen/controller/api/resources/users"
+	"google.golang.org/protobuf/proto"
+)
+
+type structureInfo struct {
+	pkg    string
+	name   string
+	fields []fieldInfo
+}
+
+type fieldInfo struct {
+	Name              string
+	ProtoName         string
+	FieldType         string
+	GenerateSdkOption bool
+	SubtypeName       string
+	Query             bool
+	SkipDefault       bool
+}
+
+type structInfo struct {
+	inProto            proto.Message
+	outFile            string
+	generatedStructure structureInfo
+	templates          []*template.Template
+
+	// Subtype name for types implementing an abstract resource type. This is
+	// used as text to insert into With/Default function calls to separate out
+	// implementations of the same abstract type. This way e.g. a WithLoginName
+	// option turns into WithPasswordAccountLoginName which is wordy but
+	// unambiguous. It also switches the behavior of the functions to work on
+	// the attributes map.
+	subtypeName string
+
+	// For non-top-level collections, this can be used to indicate the name of
+	// the argument that should be used
+	parentTypeName string
+
+	// mappings of names of resources and param names for sub slice types, e.g.
+	// role principals and group members
+	sliceSubTypes map[string]string
+
+	// outputOnly indicates that we shouldn't create options for setting members
+	// for this struct
+	outputOnly bool
+
+	// versionEnabled indicates that we should build a Version handler in
+	// update. Some structs are embedded in others and shouldn't have version
+	// fields.
+	versionEnabled bool
+
+	// The parameters passed into the path.  These should be non-pluralized resource names.
+	// The templates will convert '-' to '_' and append an _id to them in the SDK param
+	// and append an 's' to it when building the path.
+	// The final value should be the path name of the resource since for single resource
+	// operations all values are used for the function argument.
+	// For collection based operations the last value is ignored for generating function argument.
+	pathArgs []string
+
+	// typeOnCreate indicates that create will be creating a concrete
+	// implementation of an abstract type and thus a type field is necessary
+	typeOnCreate bool
+
+	// extraOptions allows specifying extra options that will be created for a
+	// given type, e.g. arguments only valid for one call or purpose and not
+	// conveyed within the item itself
+	extraOptions []fieldInfo
+
+	// createResponseTypes controls for which structs response types are created
+	createResponseTypes bool
+
+	// fieldFilter is a set of field names that will not result in generated API
+	// fields
+	fieldFilter []string
+}
+
+var inputStructs = []*structInfo{
+	{
+		inProto:    &api.Error{},
+		outFile:    "error.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto:    &api.ErrorDetails{},
+		outFile:    "error_details.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto:    &api.FieldError{},
+		outFile:    "field_error.gen.go",
+		outputOnly: true,
+	},
+	// Scope related resources
+	{
+		inProto:    &scopes.ScopeInfo{},
+		outFile:    "scopes/scope_info.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto: &scopes.Scope{},
+		outFile: "scopes/scope.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs: []string{"scope"},
+		extraOptions: []fieldInfo{
+			{
+				Name:      "SkipAdminRoleCreation",
+				ProtoName: "skip_admin_role_creation",
+				FieldType: "bool",
+				Query:     true,
+			},
+			{
+				Name:      "SkipDefaultRoleCreation",
+				ProtoName: "skip_default_role_creation",
+				FieldType: "bool",
+				Query:     true,
+			},
+		},
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	// User related resources
+	{
+		inProto:    &users.Account{},
+		outFile:    "users/account.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto: &users.User{},
+		outFile: "users/user.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		sliceSubTypes: map[string]string{
+			"Accounts": "accountIds",
+		},
+		pathArgs:            []string{"user"},
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	// Group related resources
+	{
+		inProto:    &groups.Member{},
+		outFile:    "groups/member.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto: &groups.Group{},
+		outFile: "groups/group.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		sliceSubTypes: map[string]string{
+			"Members": "memberIds",
+		},
+		pathArgs:            []string{"group"},
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	// Role related resources
+	{
+		inProto:    &roles.Grant{},
+		outFile:    "roles/grant.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto:    &roles.Principal{},
+		outFile:    "roles/principal.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto:    &roles.GrantJson{},
+		outFile:    "roles/grant_json.gen.go",
+		outputOnly: true,
+	},
+	{
+		inProto: &roles.Role{},
+		outFile: "roles/role.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		sliceSubTypes: map[string]string{
+			"Principals": "principalIds",
+			"Grants":     "grantStrings",
+		},
+		pathArgs:            []string{"role"},
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	// Auth Methods related resources
+	{
+		inProto: &authmethods.AuthMethod{},
+		outFile: "authmethods/authmethods.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs:            []string{"auth-method"},
+		typeOnCreate:        true,
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	{
+		inProto:     &authmethods.PasswordAuthMethodAttributes{},
+		outFile:     "authmethods/password_auth_method_attributes.gen.go",
+		subtypeName: "PasswordAuthMethod",
+	},
+	// Accounts
+	{
+		inProto: &accounts.Account{},
+		outFile: "accounts/account.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs:            []string{"account"},
+		parentTypeName:      "auth-method",
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	{
+		inProto:     &accounts.PasswordAccountAttributes{},
+		outFile:     "accounts/password_account_attributes.gen.go",
+		subtypeName: "PasswordAccount",
+	},
+	// Auth Tokens
+	{
+		inProto: &authtokens.AuthToken{},
+		outFile: "authtokens/authtokens.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			readTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs:            []string{"auth-token"},
+		createResponseTypes: true,
+	},
+	// Host related resources
+	{
+		inProto: &hostcatalogs.HostCatalog{},
+		outFile: "hostcatalogs/host_catalog.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs:            []string{"host-catalog"},
+		typeOnCreate:        true,
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	{
+		inProto: &hosts.Host{},
+		outFile: "hosts/host.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs:            []string{"host"},
+		parentTypeName:      "host-catalog",
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	{
+		inProto:     &hosts.StaticHostAttributes{},
+		outFile:     "hosts/static_host_attributes.gen.go",
+		subtypeName: "StaticHost",
+	},
+	{
+		inProto: &hostsets.HostSet{},
+		outFile: "hostsets/host_set.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs:       []string{"host-set"},
+		parentTypeName: "host-catalog",
+		sliceSubTypes: map[string]string{
+			"Hosts": "hostIds",
+		},
+		versionEnabled:      true,
+		createResponseTypes: true,
+	},
+	{
+		inProto: &targets.HostSet{},
+		outFile: "targets/host_set.gen.go",
+	},
+	{
+		inProto: &targets.Target{},
+		outFile: "targets/target.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			createTemplate,
+			readTemplate,
+			updateTemplate,
+			deleteTemplate,
+			listTemplate,
+		},
+		pathArgs: []string{"target"},
+		sliceSubTypes: map[string]string{
+			"HostSets": "hostSetIds",
+		},
+		extraOptions: []fieldInfo{
+			{
+				Name:        "HostId",
+				ProtoName:   "host_id",
+				FieldType:   "string",
+				SkipDefault: true,
+			},
+		},
+		versionEnabled:      true,
+		typeOnCreate:        true,
+		createResponseTypes: true,
+	},
+	{
+		inProto:     &targets.TcpTargetAttributes{},
+		outFile:     "targets/tcp_target_attributes.gen.go",
+		subtypeName: "TcpTarget",
+	},
+	{
+		inProto: &sessions.Session{},
+		outFile: "sessions/session.gen.go",
+		templates: []*template.Template{
+			clientTemplate,
+			readTemplate,
+			listTemplate,
+		},
+		pathArgs:            []string{"session"},
+		createResponseTypes: true,
+		fieldFilter:         []string{"private_key"},
+	},
+	{
+		inProto: &sessions.SessionState{},
+		outFile: "sessions/state.gen.go",
+	},
+	{
+		inProto: &sessions.WorkerInfo{},
+		outFile: "sessions/workers.gen.go",
+	},
+	{
+		inProto:     &targets.SessionAuthorization{},
+		outFile:     "targets/session_authorization.gen.go",
+		subtypeName: "SessionAuthorization",
+	},
+	{
+		inProto:     &targets.WorkerInfo{},
+		outFile:     "targets/worker_info.gen.go",
+		subtypeName: "WorkerInfo",
+	},
+}