Files
terraform-provider-unifi/internal/provider/base/base.go
2025-03-21 11:52:55 +01:00

110 lines
2.7 KiB
Go

package base
import (
"context"
"fmt"
"github.com/hashicorp/terraform-plugin-framework/datasource"
"github.com/hashicorp/terraform-plugin-framework/diag"
"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 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))
}