terraform-provider-unifi/internal/provider/base/base.go

package base

import (
	"context"
	"fmt"
	"github.com/hashicorp/terraform-plugin-framework/attr"
	"github.com/hashicorp/terraform-plugin-framework/diag"
	"github.com/hashicorp/terraform-plugin-framework/types/basetypes"

	"github.com/hashicorp/terraform-plugin-framework/datasource"
	"github.com/hashicorp/terraform-plugin-framework/resource"
	"github.com/hashicorp/terraform-plugin-framework/types"
)

type SiteAware interface {
	GetSite() string
	SetSite(string)
	GetRawSite() types.String
}

type Identifiable interface {
	GetID() string
	SetID(string)
	GetRawID() types.String
}

type Resource interface {
	SetClient(client *Client)
	SetVersionValidator(validator ControllerVersionValidator)
	SetFeatureValidator(validator FeatureValidator)
}

// ResourceModel defines the interface that all setting models must implement
type ResourceModel interface {
	Identifiable
	SiteAware
	Merge(context.Context, interface{}) diag.Diagnostics
	AsUnifiModel(context.Context) (interface{}, diag.Diagnostics)
}

// ResourceModel defines the interface that all setting models must implement
type DatasourceModel interface {
	SiteAware
	Merge(context.Context, interface{}) diag.Diagnostics
}

type NestedObject interface {
	AttributeTypes() map[string]attr.Type
}

func ObjectNull(obj interface{}) (basetypes.ObjectValue, diag.Diagnostics) {
	diags := diag.Diagnostics{}
	if nested, ok := obj.(NestedObject); ok {
		obj := types.ObjectNull(nested.AttributeTypes())
		return obj, diags
	}
	diags.AddError("Invalid object type", fmt.Sprintf("Expected NestedObject, got: %T", obj))
	return types.ObjectNull(map[string]attr.Type{}), diags
}

type Model struct {
	ID   types.String `tfsdk:"id"`
	Site types.String `tfsdk:"site"`
}

func (b *Model) GetID() string {
	return b.ID.ValueString()
}

func (b *Model) GetRawID() types.String {
	return b.ID
}

func (b *Model) SetID(id string) {
	b.ID = types.StringValue(id)
}

func (b *Model) GetSite() string {
	return b.Site.ValueString()
}

func (b *Model) GetRawSite() types.String {
	return b.Site
}

func (b *Model) SetSite(site string) {
	b.Site = types.StringValue(site)
}

func ConfigureDatasource(base Resource, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
	if req.ProviderData == nil {
		return
	}

	cfg, ok := req.ProviderData.(*Client)
	if !ok {
		resp.Diagnostics.AddError("Unexpected Datasource Configure Type", fmt.Sprintf("Expected provider.Client, got: %T", req.ProviderData))
		return
	}
	if cfg == nil {
		resp.Diagnostics.AddError("Empty configuration", "provider.Client is nil")
		return
	}
	base.SetClient(cfg)
	base.SetVersionValidator(NewControllerVersionValidator(cfg))
	base.SetFeatureValidator(NewFeatureValidator(cfg))
}

func ConfigureResource(base Resource, req resource.ConfigureRequest, resp *resource.ConfigureResponse) {
	if req.ProviderData == nil {
		return
	}

	cfg, ok := req.ProviderData.(*Client)
	if !ok {
		resp.Diagnostics.AddError("Unexpected Resource Configure Type", fmt.Sprintf("Expected provider.Client, got: %T", req.ProviderData))
		return
	}
	if cfg == nil {
		resp.Diagnostics.AddError("Empty configuration", "provider.Client is nil")
		return
	}
	base.SetClient(cfg)
	base.SetVersionValidator(NewControllerVersionValidator(cfg))
	base.SetFeatureValidator(NewFeatureValidator(cfg))
}