Add device group parent specification and custom codegen

Author: kklimonda-cl

pkg/commands/codegen/codegen.go

@@ -119,6 +119,8 @@ func (c *Command) Execute() error {
 					resourceTyp = properties.ResourceEntry
 				case properties.TerraformResourceUuid:
 					resourceTyp = properties.ResourceUuid
+				case properties.TerraformResourceCustom:
+					resourceTyp = properties.ResourceCustom
 				case properties.TerraformResourceConfig:
 					panic("missing implementation for config type resources")
 				}
@@ -140,6 +142,8 @@ func (c *Command) Execute() error {
 					resourceTyp = properties.ResourceEntryPlural
 				case properties.TerraformResourceUuid:
 					resourceTyp = properties.ResourceUuidPlural
+				case properties.TerraformResourceCustom:
+					resourceTyp = properties.ResourceCustom
 				case properties.TerraformResourceConfig:
 					panic("missing implementation for config type resources")
 				}
@@ -153,7 +157,7 @@ func (c *Command) Execute() error {
 				resourceList = append(resourceList, resources...)
 				dataSourceList = append(dataSourceList, dataSources...)
 			}
-		} else if c.commandType == properties.CommandTypeSDK {
+		} else if c.commandType == properties.CommandTypeSDK && !spec.GoSdkSkip {
 			generator := generate.NewCreator(config.Output.GoSdk, c.templatePath, spec)
 			if err = generator.RenderTemplate(); err != nil {
 				return fmt.Errorf("error rendering %s - %s", specPath, err)

pkg/generate/generator.go

@@ -64,7 +64,7 @@ func (c *Creator) RenderTerraformProviderFile(spec *properties.Normalization, ty
 		name = fmt.Sprintf("%s_%s", spec.TerraformProviderConfig.Suffix, spec.TerraformProviderConfig.PluralName)
 	case properties.ResourceEntryPlural:
 		name = spec.TerraformProviderConfig.PluralSuffix
-	case properties.ResourceEntry, properties.ResourceUuid:
+	case properties.ResourceEntry, properties.ResourceUuid, properties.ResourceCustom:
 		name = spec.Name
 	}
 
@@ -95,7 +95,7 @@ func (c *Creator) RenderTerraformProviderFile(spec *properties.Normalization, ty
 	case properties.ResourceEntryPlural:
 		name = spec.TerraformProviderConfig.PluralSuffix
 		filePath = c.createTerraformProviderFilePath(name)
-	case properties.ResourceEntry, properties.ResourceUuid:
+	case properties.ResourceEntry, properties.ResourceUuid, properties.ResourceCustom:
 		filePath = c.createTerraformProviderFilePath(spec.TerraformProviderConfig.Suffix)
 	}
 

pkg/properties/normalized.go

@@ -18,6 +18,7 @@ import (
 type Normalization struct {
 	Name                    string                  `json:"name" yaml:"name"`
 	TerraformProviderConfig TerraformProviderConfig `json:"terraform_provider_config" yaml:"terraform_provider_config"`
+	GoSdkSkip               bool                    `json:"go_sdk_skip" yaml:"go_sdk_skip"`
 	GoSdkPath               []string                `json:"go_sdk_path" yaml:"go_sdk_path"`
 	XpathSuffix             []string                `json:"xpath_suffix" yaml:"xpath_suffix"`
 	Locations               map[string]*Location    `json:"locations" yaml:"locations"`
@@ -53,6 +54,7 @@ const (
 	TerraformResourceEntry  TerraformResourceType = "entry"
 	TerraformResourceUuid   TerraformResourceType = "uuid"
 	TerraformResourceConfig TerraformResourceType = "config"
+	TerraformResourceCustom TerraformResourceType = "custom"
 )
 
 type TerraformResourceVariant string
@@ -67,6 +69,7 @@ type TerraformProviderConfig struct {
 	SkipDatasource        bool                       `json:"skip_datasource" yaml:"skip_datasource"`
 	SkipDatasourceListing bool                       `json:"skip_datasource_listing" yaml:"skip_datasource_listing"`
 	ResourceType          TerraformResourceType      `json:"resource_type" yaml:"resource_type"`
+	CustomFuncs           map[string]string          `json:"custom_functions" yaml:"custom_functions"`
 	ResourceVariants      []TerraformResourceVariant `json:"resource_variants" yaml:"resource_variants"`
 	Suffix                string                     `json:"suffix" yaml:"suffix"`
 	PluralSuffix          string                     `json:"plural_suffix" yaml:"plural_suffix"`
@@ -473,12 +476,14 @@ func schemaToSpec(object object.Object) (*Normalization, error) {
 			SkipDatasource:        object.TerraformConfig.SkipDatasource,
 			SkipDatasourceListing: object.TerraformConfig.SkipdatasourceListing,
 			ResourceType:          TerraformResourceType(object.TerraformConfig.ResourceType),
+			CustomFuncs:           object.TerraformConfig.CustomFunctions,
 			ResourceVariants:      resourceVariants,
 			Suffix:                object.TerraformConfig.Suffix,
 			PluralSuffix:          object.TerraformConfig.PluralSuffix,
 			PluralName:            object.TerraformConfig.PluralName,
 		},
 		Locations:   make(map[string]*Location),
+		GoSdkSkip:   object.GoSdkConfig.Skip,
 		GoSdkPath:   object.GoSdkConfig.Package,
 		XpathSuffix: object.XpathSuffix,
 		Version:     object.Version,

pkg/properties/resourcetype.go

@@ -4,6 +4,7 @@ type ResourceType int
 
 const (
 	ResourceEntry       ResourceType = iota
+	ResourceCustom      ResourceType = iota
 	ResourceEntryPlural ResourceType = iota
 	ResourceUuid        ResourceType = iota
 	ResourceUuidPlural  ResourceType = iota

pkg/schema/object/object.go

@@ -15,6 +15,7 @@ const (
 	TerraformResourceEntry  TerraformResourceType = "entry"
 	TerraformResourceUuid   TerraformResourceType = "uuid"
 	TerraformResourceConfig TerraformResourceType = "config"
+	TerraformResourceCustom TerraformResourceType = "custom"
 )
 
 type TerraformResourceVariant string
@@ -29,14 +30,16 @@ type TerraformConfig struct {
 	SkipDatasource        bool                       `yaml:"skip_datasource"`
 	SkipdatasourceListing bool                       `yaml:"skip_datasource_listing"`
 	ResourceType          TerraformResourceType      `yaml:"resource_type"`
+	CustomFunctions       map[string]string          `yaml:"custom_functions"`
 	ResourceVariants      []TerraformResourceVariant `yaml:"resource_variants"`
 	Suffix                string                     `yaml:"suffix"`
 	PluralSuffix          string                     `yaml:"plural_suffix"`
 	PluralName            string                     `yaml:"plural_name"`
 }
 
 type GoSdkConfig struct {
-	Package []string
+	Skip    bool     `yaml:"skip"`
+	Package []string `yaml:"package"`
 }
 
 type Entry struct {

pkg/translate/terraform_provider/device_group_parent_crud.go

@@ -0,0 +1,115 @@
+package terraform_provider
+
+const deviceGroupParentImports = `
+import (
+  "encoding/xml"
+  "github.com/PaloAltoNetworks/pango/xmlapi"
+)
+`
+
+const deviceGroupParentCommon = `
+var _ = tflog.Warn
+type _ = diag.Diagnostics
+type dgpReq struct {
+	XMLName xml.Name ` + "`" + `xml:"show"` + "`" + `
+	Cmd     string   ` + "`" + `xml:"dg-hierarchy"` + "`" + `
+}
+
+type dgpResp struct {
+	Result *dgHierarchy ` + "`" + `xml:"result>dg-hierarchy"` + "`" + `
+}
+
+func (o *dgpResp) results() map[string]string {
+	ans := make(map[string]string)
+
+	if o.Result != nil {
+		for _, v := range o.Result.Info {
+			ans[v.Name] = ""
+			v.results(ans)
+		}
+	}
+
+	return ans
+}
+
+type dgHierarchy struct {
+	Info []dghInfo ` + "`" + `xml:"dg"` + "`" + `
+}
+
+type dghInfo struct {
+	Name     string    ` + "`" + `xml:"name,attr"` + "`" + `
+	Children []dghInfo ` + "`" + `xml:"dg"` + "`" + `
+}
+
+func (o *dghInfo) results(ans map[string]string) {
+	for _, v := range o.Children {
+		ans[v.Name] = o.Name
+		v.results(ans)
+	}
+}
+`
+
+const deviceGroupParentDataSourceRead = `
+var state DeviceGroupParentResourceModel
+resp.Diagnostics.Append(req.Config.Get(ctx, &state)...)
+if resp.Diagnostics.HasError() {
+	return
+}
+
+cmd := &xmlapi.Op{
+	Command: dgpReq{},
+}
+
+var ans dgpResp
+if _, _, err := o.client.Communicate(ctx, cmd, false, &ans); err != nil {
+	resp.Diagnostics.AddError("Failed to query for device group parents", err.Error())
+	return
+}
+
+
+hierarchy := ans.results()
+
+name := state.DeviceGroup.ValueString()
+parent, ok := hierarchy[name]
+if !ok {
+	state.Parent = types.StringValue("")
+} else {
+	state.Parent = types.StringValue(parent)
+}
+
+resp.Diagnostics.Append(resp.State.Set(ctx, &state)...)
+`
+
+const deviceGroupParentResourceRead = `
+var state DeviceGroupParentResourceModel
+resp.Diagnostics.Append(req.State.Get(ctx, &state)...)
+if resp.Diagnostics.HasError() {
+	return
+}
+
+cmd := &xmlapi.Op{
+	Command: dgpReq{},
+}
+
+var ans dgpResp
+if _, _, err := o.client.Communicate(ctx, cmd, false, &ans); err != nil {
+	resp.Diagnostics.AddError("Failed to query for device group parents", err.Error())
+	return
+}
+
+hierarchy := ans.results()
+
+name := state.DeviceGroup.ValueString()
+parent, ok := hierarchy[name]
+if !ok {
+	state.Parent = types.StringValue("")
+} else {
+	state.Parent = types.StringValue(parent)
+}
+
+resp.Diagnostics.Append(resp.State.Set(ctx, &state)...)
+`
+
+const deviceGroupParentResourceCreate = ``
+const deviceGroupParentResourceUpdate = ``
+const deviceGroupParentResourceDelete = ``