feat(codegen): add Custom structure for custom resources that can be used to store extra state

Author: kklimonda-cl

assets/terraform/internal/provider/api_key_crud.go

@@ -7,6 +7,12 @@ import (
 	"github.com/hashicorp/terraform-plugin-framework/types"
 )
 
+type ApiKeyCustom struct{}
+
+func NewApiKeyCustom(provider *ProviderData) (*ApiKeyCustom, error) {
+	return &ApiKeyCustom{}, nil
+}
+
 func (o *ApiKeyResource) OpenCustom(ctx context.Context, req ephemeral.OpenRequest, resp *ephemeral.OpenResponse) {
 	var data ApiKeyResourceModel
 	resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)

assets/terraform/internal/provider/device_group_parent_crud.go

@@ -13,6 +13,12 @@ import (
 	"github.com/hashicorp/terraform-plugin-framework/types"
 )
 
+type DeviceGroupParentCustom struct{}
+
+func NewDeviceGroupParentCustom(provider *ProviderData) (*DeviceGroupParentCustom, error) {
+	return &DeviceGroupParentCustom{}, nil
+}
+
 type dgpReq struct {
 	XMLName xml.Name `xml:"show"`
 	Cmd     string   `xml:"dg-hierarchy"`

assets/terraform/internal/provider/vm_auth_key_crud.go

@@ -10,6 +10,12 @@ import (
 	"github.com/hashicorp/terraform-plugin-framework/types"
 )
 
+type VmAuthKeyCustom struct{}
+
+func NewVmAuthKeyCustom(data *ProviderData) (*VmAuthKeyCustom, error) {
+	return &VmAuthKeyCustom{}, nil
+}
+
 type vmAuthKeyRequest struct {
 	XMLName  xml.Name `xml:"request"`
 	Lifetime int64    `xml:"bootstrap>vm-auth-key>generate>lifetime"`

pkg/translate/terraform_provider/template.go

@@ -252,7 +252,7 @@ func New{{ resourceStructName }}() resource.Resource {
 type {{ resourceStructName }} struct {
 	client *pango.Client
 {{- if IsCustom }}
-
+	custom *{{ structName }}Custom
 {{- else if and IsEntry HasImports }}
 	manager *sdkmanager.ImportableEntryObjectManager[*{{ resourceSDKName }}.Entry, {{ resourceSDKName }}.Location, {{ resourceSDKName }}.ImportLocation, *{{ resourceSDKName }}.Service]
 {{- else if IsEntry }}
@@ -367,7 +367,12 @@ func (o *{{ resourceStructName }}) Configure(ctx context.Context, req {{ tfresou
 	o.client = providerData.Client
 
 {{- if IsCustom }}
-
+	custom, err := New{{ structName }}Custom(providerData)
+	if err != nil {
+		resp.Diagnostics.AddError("Failed to configure SDK client", err.Error())
+		return
+	}
+	o.custom = custom
 {{- else if and IsEntry HasImports }}
 	specifier, _, err := {{ resourceSDKName }}.Versioning(o.client.Versioning())
 	if err != nil {
@@ -2016,7 +2021,7 @@ type {{ dataSourceStructName }} struct {
     client *pango.Client
 
 {{- if IsCustom }}
-
+	custom *{{ structName }}Custom
 {{- else if and IsEntry HasImports }}
 	manager *sdkmanager.ImportableEntryObjectManager[*{{ resourceSDKName }}.Entry, {{ resourceSDKName }}.Location, {{ resourceSDKName }}.ImportLocation, *{{ resourceSDKName }}.Service]
 {{- else if IsEntry }}
@@ -2070,7 +2075,12 @@ func (d *{{ dataSourceStructName }}) Configure(_ context.Context, req datasource
 	d.client = providerData.Client
 
 {{- if IsCustom }}
-
+	custom, err := New{{ structName }}Custom(providerData)
+	if err != nil {
+		resp.Diagnostics.AddError("Failed to configure SDK client", err.Error())
+		return
+	}
+	d.custom = custom
 {{- else if and IsEntry HasImports }}
 	specifier, _, err := {{ resourceSDKName }}.Versioning(d.client.Versioning())
 	if err != nil {