mirror of
https://github.com/oscarhermoso/terraform-provider-binarylane
synced 2026-10-18 21:39:17 +11:00
103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
package provider
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"terraform-provider-binarylane/internal/resources"
|
|
|
|
"github.com/hashicorp/terraform-plugin-framework/datasource"
|
|
d_schema "github.com/hashicorp/terraform-plugin-framework/datasource/schema"
|
|
"github.com/hashicorp/terraform-plugin-framework/types"
|
|
"github.com/hashicorp/terraform-plugin-log/tflog"
|
|
)
|
|
|
|
// Ensure the implementation satisfies the expected interfaces.
|
|
var (
|
|
_ datasource.DataSource = &serverDataSource{}
|
|
_ datasource.DataSourceWithConfigure = &serverDataSource{}
|
|
)
|
|
|
|
func NewServerDataSource() datasource.DataSource {
|
|
return &serverDataSource{}
|
|
}
|
|
|
|
type serverDataSource struct {
|
|
bc *BinarylaneClient
|
|
}
|
|
|
|
func (d *serverDataSource) Configure(_ context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
|
|
if req.ProviderData == nil {
|
|
return
|
|
}
|
|
|
|
bc, ok := req.ProviderData.(BinarylaneClient)
|
|
if !ok {
|
|
resp.Diagnostics.AddError(
|
|
"Unexpected Data Source Configure Type",
|
|
fmt.Sprintf("Expected *BinarylaneClient, got: %T.", req.ProviderData),
|
|
)
|
|
|
|
return
|
|
}
|
|
|
|
d.bc = &bc
|
|
}
|
|
|
|
func (d *serverDataSource) Metadata(ctx context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
|
|
resp.TypeName = req.ProviderTypeName + "_server"
|
|
}
|
|
|
|
func (d *serverDataSource) Schema(ctx context.Context, req datasource.SchemaRequest, resp *datasource.SchemaResponse) {
|
|
resp.Schema = *convertResourceSchemaToDataSourceSchema(ctx, resources.ServerResourceSchema(ctx))
|
|
resp.Schema.Description = "TODO"
|
|
|
|
// Overrides
|
|
id := resp.Schema.Attributes["id"]
|
|
resp.Schema.Attributes["id"] = d_schema.Int64Attribute{
|
|
Description: id.GetDescription(),
|
|
MarkdownDescription: id.GetMarkdownDescription(),
|
|
Required: true, // ID is required to find the server
|
|
}
|
|
}
|
|
|
|
func (d *serverDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
|
|
var data resources.ServerModel
|
|
|
|
// Read Terraform configuration data into the model
|
|
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
|
|
|
|
if resp.Diagnostics.HasError() {
|
|
return
|
|
}
|
|
|
|
// Read API call logic
|
|
serverResp, err := d.bc.client.GetServersServerIdWithResponse(ctx, data.Id.ValueInt64())
|
|
if err != nil {
|
|
resp.Diagnostics.AddError(
|
|
fmt.Sprintf("Error reading server: name=%s", data.Id.String()),
|
|
err.Error(),
|
|
)
|
|
return
|
|
}
|
|
|
|
// Set data values
|
|
tflog.Trace(ctx,
|
|
fmt.Sprintf(
|
|
"Reading server: id=%d, name=%s, image=%s, region=%s, size=%s",
|
|
*serverResp.JSON200.Server.Id,
|
|
*serverResp.JSON200.Server.Name,
|
|
*serverResp.JSON200.Server.Image.Slug,
|
|
*serverResp.JSON200.Server.Region.Slug,
|
|
*serverResp.JSON200.Server.Size.Slug,
|
|
),
|
|
)
|
|
data.Id = types.Int64Value(*serverResp.JSON200.Server.Id)
|
|
data.Name = types.StringValue(*serverResp.JSON200.Server.Name)
|
|
data.Image = types.StringValue(*serverResp.JSON200.Server.Image.Slug)
|
|
data.Region = types.StringValue(*serverResp.JSON200.Server.Region.Slug)
|
|
data.Size = types.StringValue(*serverResp.JSON200.Server.Size.Slug)
|
|
|
|
// Save data into Terraform state
|
|
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
|
|
}
|