e7164c0460
* add DNS record * revamp tests * lint * cleanup * feat dns test * chore: add DNS Record tests * linting * f
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package base
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/hashicorp/terraform-plugin-framework/datasource"
|
|
"github.com/hashicorp/terraform-plugin-framework/resource"
|
|
)
|
|
|
|
type BaseData interface {
|
|
SetClient(client *Client)
|
|
}
|
|
|
|
func ConfigureDatasource(base BaseData, 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)
|
|
}
|
|
|
|
func ConfigureResource(base BaseData, 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)
|
|
}
|