feat: build taint datasource and taint resource with minor patch to taint Response DTO

terraform-provider-autoglue/internal/provider/client.go

@@ -4,7 +4,7 @@ import (
 	"context"
 	"net/http"
 
-	"github.com/glueops/autoglue-sdk"
+	"github.com/glueops/autoglue-sdk-go"
 	"github.com/hashicorp/terraform-plugin-framework/diag"
 )
 

terraform-provider-autoglue/internal/provider/datasource_taints.go

@@ -0,0 +1,107 @@
+package provider
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-framework/datasource"
+	dschema "github.com/hashicorp/terraform-plugin-framework/datasource/schema"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+)
+
+var _ datasource.DataSource = &TaintsDataSource{}
+var _ datasource.DataSourceWithConfigure = &TaintsDataSource{}
+
+type TaintsDataSource struct{ client *Client }
+
+func NewTaintsDataSource() datasource.DataSource { return &TaintsDataSource{} }
+
+type taintsDSModel struct {
+	Items []taintItem `tfsdk:"items"`
+}
+
+type taintItem struct {
+	ID             types.String `tfsdk:"id"`
+	OrganizationID types.String `tfsdk:"organization_id"`
+	CreatedAt      types.String `tfsdk:"created_at"`
+	UpdatedAt      types.String `tfsdk:"updated_at"`
+	Key            types.String `tfsdk:"key"`
+	Value          types.String `tfsdk:"value"`
+	Effect         types.String `tfsdk:"effect"`
+	Raw            types.String `tfsdk:"raw"`
+}
+
+func (d *TaintsDataSource) Metadata(_ context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
+	resp.TypeName = req.ProviderTypeName + "_taints"
+}
+
+func (d *TaintsDataSource) Schema(_ context.Context, _ datasource.SchemaRequest, resp *datasource.SchemaResponse) {
+	resp.Schema = dschema.Schema{
+		Description: "List taints for the organization (org-scoped).",
+		Attributes: map[string]dschema.Attribute{
+			"items": dschema.ListNestedAttribute{
+				Computed:    true,
+				Description: "Taints returned by the API.",
+				NestedObject: dschema.NestedAttributeObject{
+					Attributes: map[string]dschema.Attribute{
+						"id":              dschema.StringAttribute{Computed: true, Description: "Taint ID (UUID)."},
+						"organization_id": dschema.StringAttribute{Computed: true},
+						"key":             dschema.StringAttribute{Computed: true},
+						"value":           dschema.StringAttribute{Computed: true},
+						"effect":          dschema.StringAttribute{Computed: true},
+						"created_at":      dschema.StringAttribute{Computed: true, Description: "RFC3339, UTC."},
+						"updated_at":      dschema.StringAttribute{Computed: true, Description: "RFC3339, UTC."},
+						"raw":             dschema.StringAttribute{Computed: true, Description: "Full JSON for the item."},
+					},
+				},
+			},
+		},
+	}
+}
+
+func (d *TaintsDataSource) Configure(_ context.Context, req datasource.ConfigureRequest, _ *datasource.ConfigureResponse) {
+	if req.ProviderData == nil {
+		return
+	}
+	d.client = req.ProviderData.(*Client)
+}
+
+func (d *TaintsDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
+	if d.client == nil || d.client.SDK == nil {
+		resp.Diagnostics.AddError("Client not configured", "Provider configuration missing")
+		return
+	}
+
+	var conf taintsDSModel
+	resp.Diagnostics.Append(req.Config.Get(ctx, &conf)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	call := d.client.SDK.TaintsAPI.ListTaints(ctx)
+	items, httpResp, err := call.Execute()
+	if err != nil {
+		resp.Diagnostics.AddError("List taints failed", fmt.Sprintf("%v", httpErr(err, httpResp)))
+		return
+	}
+
+	out := taintsDSModel{
+		Items: make([]taintItem, 0, len(items)),
+	}
+
+	for _, item := range items {
+		raw, _ := json.Marshal(item)
+		out.Items = append(out.Items, taintItem{
+			ID:             types.StringPointerValue(item.Id),
+			OrganizationID: types.StringPointerValue(item.OrganizationId),
+			Key:            types.StringPointerValue(item.Key),
+			Value:          types.StringPointerValue(item.Value),
+			Effect:         types.StringPointerValue(item.Effect),
+			CreatedAt:      types.StringPointerValue(item.CreatedAt),
+			UpdatedAt:      types.StringPointerValue(item.UpdatedAt),
+			Raw:            types.StringValue(string(raw)),
+		})
+	}
+	resp.Diagnostics.Append(resp.State.Set(ctx, &out)...)
+}

terraform-provider-autoglue/internal/provider/provider.go

@@ -49,6 +49,7 @@ func (p *AutoglueProvider) DataSources(_ context.Context) []func() datasource.Da
 	return []func() datasource.DataSource{
 		NewSshDataSource,
 		NewServersDataSource,
+		NewTaintsDataSource,
 	}
 }
 
@@ -56,5 +57,6 @@ func (p *AutoglueProvider) Resources(_ context.Context) []func() resource.Resour
 	return []func() resource.Resource{
 		NewSshResource,
 		NewServerResource,
+		NewTaintResource,
 	}
 }

terraform-provider-autoglue/internal/provider/resource_server.go

@@ -7,7 +7,7 @@ import (
 	"regexp"
 	"strings"
 
-	"github.com/glueops/autoglue-sdk"
+	"github.com/glueops/autoglue-sdk-go"
 	"github.com/hashicorp/terraform-plugin-framework-validators/stringvalidator"
 	"github.com/hashicorp/terraform-plugin-framework/path"
 	"github.com/hashicorp/terraform-plugin-framework/resource"

terraform-provider-autoglue/internal/provider/resource_ssh_key.go

@@ -4,7 +4,7 @@ import (
 	"context"
 	"fmt"
 
-	"github.com/glueops/autoglue-sdk"
+	"github.com/glueops/autoglue-sdk-go"
 	"github.com/hashicorp/terraform-plugin-framework-validators/int64validator"
 	"github.com/hashicorp/terraform-plugin-framework-validators/stringvalidator"
 	"github.com/hashicorp/terraform-plugin-framework/path"

terraform-provider-autoglue/internal/provider/resource_taint.go

@@ -0,0 +1,256 @@
+package provider
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+
+	"github.com/glueops/autoglue-sdk-go"
+	"github.com/hashicorp/terraform-plugin-framework-validators/stringvalidator"
+	"github.com/hashicorp/terraform-plugin-framework/path"
+	"github.com/hashicorp/terraform-plugin-framework/resource"
+	rschema "github.com/hashicorp/terraform-plugin-framework/resource/schema"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/stringplanmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/schema/validator"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+)
+
+var _ resource.Resource = &TaintResource{}
+var _ resource.ResourceWithConfigure = &TaintResource{}
+var _ resource.ResourceWithImportState = &TaintResource{}
+
+type TaintResource struct{ client *Client }
+
+func NewTaintResource() resource.Resource { return &TaintResource{} }
+
+type taintResModel struct {
+	ID             types.String `tfsdk:"id"`
+	OrganizationID types.String `tfsdk:"organization_id"`
+	CreatedAt      types.String `tfsdk:"created_at"`
+	UpdatedAt      types.String `tfsdk:"updated_at"`
+	Key            types.String `tfsdk:"key"`
+	Value          types.String `tfsdk:"value"`
+	Effect         types.String `tfsdk:"effect"`
+	Raw            types.String `tfsdk:"raw"`
+}
+
+func (r *TaintResource) Metadata(_ context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) {
+	resp.TypeName = req.ProviderTypeName + "_taint"
+}
+
+func (r *TaintResource) Schema(_ context.Context, _ resource.SchemaRequest, resp *resource.SchemaResponse) {
+	resp.Schema = rschema.Schema{
+		Description: "Create and manage a taint (org-scoped).",
+		Attributes: map[string]rschema.Attribute{
+			"id": rschema.StringAttribute{
+				Computed:    true,
+				Description: "Server ID (UUID).",
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.UseStateForUnknown(),
+				},
+			},
+			"organization_id": rschema.StringAttribute{Computed: true},
+			"created_at":      rschema.StringAttribute{Computed: true},
+			"updated_at":      rschema.StringAttribute{Computed: true},
+			"key": rschema.StringAttribute{
+				Required:    true,
+				Description: "Key.",
+			},
+			"value": rschema.StringAttribute{
+				Required:    true,
+				Description: "Value.",
+			},
+			"effect": rschema.StringAttribute{
+				Required:    true,
+				Description: "Effect.",
+				Validators: []validator.String{
+					stringvalidator.OneOf("NoSchedule", "NoExecute", "PreferNoSchedule"),
+				},
+			},
+			"raw": rschema.StringAttribute{
+				Computed:    true,
+				Description: "Full server JSON from API.",
+			},
+		},
+	}
+}
+
+func (r *TaintResource) Configure(_ context.Context, req resource.ConfigureRequest, _ *resource.ConfigureResponse) {
+	if req.ProviderData == nil {
+		return
+	}
+	r.client = req.ProviderData.(*Client)
+}
+
+func (r *TaintResource) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) {
+	if r.client == nil || r.client.SDK == nil {
+		resp.Diagnostics.AddError("Client not configured", "Provider configuration missing")
+		return
+	}
+
+	var plan taintResModel
+	resp.Diagnostics.Append(req.Plan.Get(ctx, &plan)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	payload := autoglue.NewDtoCreateTaintRequest()
+	if !plan.Key.IsNull() {
+		k := plan.Key.ValueString()
+		payload.SetKey(k)
+	}
+	if !plan.Value.IsNull() {
+		v := plan.Value.ValueString()
+		payload.SetValue(v)
+	}
+	if !plan.Effect.IsNull() {
+		e := plan.Effect.ValueString()
+		payload.SetEffect(e)
+	}
+
+	call := r.client.SDK.TaintsAPI.CreateTaint(ctx).Body(*payload)
+
+	out, httpResp, err := call.Execute()
+	if err != nil {
+		resp.Diagnostics.AddError("Create taint failed", fmt.Sprintf("%v", httpErr(err, httpResp)))
+		return
+	}
+
+	raw, _ := json.Marshal(out)
+
+	var state taintResModel
+	r.mapRespToState(out, &state)
+	state.Raw = types.StringValue(string(raw))
+
+	resp.Diagnostics.Append(resp.State.Set(ctx, &state)...)
+}
+
+func (r *TaintResource) Read(ctx context.Context, req resource.ReadRequest, resp *resource.ReadResponse) {
+	if r.client == nil || r.client.SDK == nil {
+		resp.Diagnostics.AddError("Client not configured", "Provider configuration missing")
+		return
+	}
+
+	var state taintResModel
+	resp.Diagnostics.Append(req.State.Get(ctx, &state)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	if state.ID.IsNull() || state.ID.ValueString() == "" {
+		// Nothing to read; treat as gone
+		resp.State.RemoveResource(ctx)
+		return
+	}
+
+	call := r.client.SDK.TaintsAPI.GetTaint(ctx, state.ID.ValueString())
+
+	out, httpResp, err := call.Execute()
+	if err != nil {
+		if httpResp != nil && httpResp.StatusCode == http.StatusNotFound {
+			resp.State.RemoveResource(ctx)
+			return
+		}
+		resp.Diagnostics.AddError("Read taint failed", fmt.Sprintf("%v", httpErr(err, httpResp)))
+		return
+	}
+
+	raw, _ := json.Marshal(out)
+	r.mapRespToState(out, &state)
+	state.Raw = types.StringValue(string(raw))
+
+	resp.Diagnostics.Append(resp.State.Set(ctx, &state)...)
+}
+
+func (r *TaintResource) Update(ctx context.Context, req resource.UpdateRequest, resp *resource.UpdateResponse) {
+	if r.client == nil || r.client.SDK == nil {
+		resp.Diagnostics.AddError("Client not configured", "Provider configuration missing")
+		return
+	}
+
+	var plan taintResModel
+	var prior taintResModel
+	resp.Diagnostics.Append(req.Plan.Get(ctx, &plan)...)
+	resp.Diagnostics.Append(req.State.Get(ctx, &prior)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	body := autoglue.NewDtoUpdateTaintRequest()
+	if !plan.Key.IsNull() {
+		k := plan.Key.ValueString()
+		body.SetKey(k)
+	}
+	if !plan.Value.IsNull() {
+		v := plan.Value.ValueString()
+		body.SetValue(v)
+	}
+	if !plan.Effect.IsNull() {
+		e := plan.Effect.ValueString()
+		body.SetEffect(e)
+	}
+
+	call := r.client.SDK.TaintsAPI.UpdateTaint(ctx, prior.ID.ValueString()).Body(*body)
+
+	out, httpResp, err := call.Execute()
+	if err != nil {
+		// If 404 on update, treat as gone
+		if httpResp != nil && httpResp.StatusCode == http.StatusNotFound {
+			resp.State.RemoveResource(ctx)
+			return
+		}
+		resp.Diagnostics.AddError("Update taint failed", fmt.Sprintf("%v", httpErr(err, httpResp)))
+		return
+	}
+
+	raw, _ := json.Marshal(out)
+
+	var newState taintResModel
+	r.mapRespToState(out, &newState)
+	newState.Raw = types.StringValue(string(raw))
+
+	resp.Diagnostics.Append(resp.State.Set(ctx, &newState)...)
+}
+
+func (r *TaintResource) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) {
+	if r.client == nil || r.client.SDK == nil {
+		resp.Diagnostics.AddError("Client not configured", "Provider configuration missing")
+		return
+	}
+
+	var state taintResModel
+	resp.Diagnostics.Append(req.State.Get(ctx, &state)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	call := r.client.SDK.TaintsAPI.DeleteTaint(ctx, state.ID.ValueString())
+
+	_, httpResp, err := call.Execute()
+	if err != nil {
+		// If already gone, that's fine
+		if httpResp != nil && httpResp.StatusCode == http.StatusNotFound {
+			return
+		}
+		resp.Diagnostics.AddError("Delete taint failed", fmt.Sprintf("%v", httpErr(err, httpResp)))
+		return
+	}
+}
+
+func (r *TaintResource) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) {
+	resp.Diagnostics.Append(resp.State.SetAttribute(ctx, path.Root("id"), req.ID)...)
+}
+
+// --- helpers ---
+
+func (r *TaintResource) mapRespToState(s *autoglue.DtoTaintResponse, out *taintResModel) {
+	out.ID = types.StringPointerValue(s.Id)
+	out.OrganizationID = types.StringPointerValue(s.OrganizationId)
+	out.Key = types.StringPointerValue(s.Key)
+	out.Value = types.StringPointerValue(s.Value)
+	out.Effect = types.StringPointerValue(s.Effect)
+	out.CreatedAt = types.StringPointerValue(s.CreatedAt)
+	out.UpdatedAt = types.StringPointerValue(s.UpdatedAt)
+}