From 982943510819ebea863fd800963961bd51946351 Mon Sep 17 00:00:00 2001 From: unkin-agent Date: Fri, 16 Oct 2026 23:37:38 +1100 Subject: [PATCH] Add artifactapi_local_puppet resource --- README.md | 1 + .../artifactapi_local_puppet/main.tf | 4 + internal/provider/provider.go | 1 + internal/provider/provider_test.go | 5 +- internal/provider/resource_local_puppet.go | 164 ++++++++++++++++++ .../provider/resource_local_puppet_test.go | 125 +++++++++++++ 6 files changed, 298 insertions(+), 2 deletions(-) create mode 100644 examples/resources/artifactapi_local_puppet/main.tf create mode 100644 internal/provider/resource_local_puppet.go create mode 100644 internal/provider/resource_local_puppet_test.go diff --git a/README.md b/README.md index 4d9266e..9d6c240 100644 --- a/README.md +++ b/README.md @@ -196,6 +196,7 @@ Available resource types: - `artifactapi_local_rpm` - `artifactapi_local_deb` - `artifactapi_local_alpine` +- `artifactapi_local_puppet` — a Puppet Forge (v3 API for `puppet module install`, r10k and g10k) - `artifactapi_local_terraform` Each takes just `name` (required, forces replacement) and an optional diff --git a/examples/resources/artifactapi_local_puppet/main.tf b/examples/resources/artifactapi_local_puppet/main.tf new file mode 100644 index 0000000..60f54a4 --- /dev/null +++ b/examples/resources/artifactapi_local_puppet/main.tf @@ -0,0 +1,4 @@ +resource "artifactapi_local_puppet" "internal" { + name = "puppet-internal" + description = "Internal Puppet Forge" +} diff --git a/internal/provider/provider.go b/internal/provider/provider.go index 16e4a24..8cee550 100644 --- a/internal/provider/provider.go +++ b/internal/provider/provider.go @@ -89,6 +89,7 @@ func (p *ArtifactAPIProvider) Resources(_ context.Context) []func() resource.Res NewLocalRPMResource, NewLocalDebResource, NewLocalAlpineResource, + NewLocalPuppetResource, NewLocalDockerResource, NewLocalGenericResource, } diff --git a/internal/provider/provider_test.go b/internal/provider/provider_test.go index 2a6f93d..04cca51 100644 --- a/internal/provider/provider_test.go +++ b/internal/provider/provider_test.go @@ -66,8 +66,8 @@ func TestProvider_Resources(t *testing.T) { p := &ArtifactAPIProvider{version: "1.0.0"} resources := p.Resources(context.Background()) - // 15 remote resource types + 1 virtual + 7 local = 23 - expectedCount := 23 + // 15 remote resource types + 1 virtual + 8 local = 24 + expectedCount := 24 if len(resources) != expectedCount { t.Fatalf("expected %d resources, got %d", expectedCount, len(resources)) } @@ -117,6 +117,7 @@ func TestProvider_Resources_ContainsExpectedTypes(t *testing.T) { "artifactapi_local_rpm", "artifactapi_local_deb", "artifactapi_local_alpine", + "artifactapi_local_puppet", "artifactapi_local_docker", "artifactapi_local_generic", } diff --git a/internal/provider/resource_local_puppet.go b/internal/provider/resource_local_puppet.go new file mode 100644 index 0000000..e6bd167 --- /dev/null +++ b/internal/provider/resource_local_puppet.go @@ -0,0 +1,164 @@ +package provider + +import ( + "context" + "fmt" + + "github.com/hashicorp/terraform-plugin-framework/path" + "github.com/hashicorp/terraform-plugin-framework/resource" + "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/stringdefault" + "github.com/hashicorp/terraform-plugin-framework/resource/schema/stringplanmodifier" + "github.com/hashicorp/terraform-plugin-framework/types" +) + +var ( + _ resource.Resource = &localPuppetResource{} + _ resource.ResourceWithImportState = &localPuppetResource{} +) + +type localPuppetResource struct { + client *apiClient +} + +type localPuppetResourceModel struct { + Name types.String `tfsdk:"name"` + Description types.String `tfsdk:"description"` +} + +func NewLocalPuppetResource() resource.Resource { + return &localPuppetResource{} +} + +func (r *localPuppetResource) Metadata(_ context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) { + resp.TypeName = req.ProviderTypeName + "_local_puppet" +} + +func (r *localPuppetResource) Schema(_ context.Context, _ resource.SchemaRequest, resp *resource.SchemaResponse) { + resp.Schema = schema.Schema{ + Description: "Manages a local ArtifactAPI Puppet Forge for hosting Puppet modules directly.", + Attributes: map[string]schema.Attribute{ + "name": schema.StringAttribute{ + Description: "Unique name of the local Puppet Forge.", + Required: true, + PlanModifiers: []planmodifier.String{ + stringplanmodifier.RequiresReplace(), + }, + }, + "description": schema.StringAttribute{ + Description: "Human-readable description.", + Optional: true, + Computed: true, + Default: stringdefault.StaticString(""), + }, + }, + } +} + +func (r *localPuppetResource) Configure(_ context.Context, req resource.ConfigureRequest, resp *resource.ConfigureResponse) { + if req.ProviderData == nil { + return + } + client, ok := req.ProviderData.(*apiClient) + if !ok { + resp.Diagnostics.AddError("unexpected provider data type", fmt.Sprintf("got %T", req.ProviderData)) + return + } + r.client = client +} + +func (r *localPuppetResource) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) { + var plan localPuppetResourceModel + resp.Diagnostics.Append(req.Plan.Get(ctx, &plan)...) + if resp.Diagnostics.HasError() { + return + } + + api := localPuppetModelToAPI(plan) + api.ManagedBy = "terraform" + + var created remoteAPI + if err := r.client.post(ctx, "/api/v2/remotes", api, &created); err != nil { + resp.Diagnostics.AddError("create local puppet failed", err.Error()) + return + } + + state := localPuppetAPIToModel(created) + resp.Diagnostics.Append(resp.State.Set(ctx, state)...) +} + +func (r *localPuppetResource) Read(ctx context.Context, req resource.ReadRequest, resp *resource.ReadResponse) { + var state localPuppetResourceModel + resp.Diagnostics.Append(req.State.Get(ctx, &state)...) + if resp.Diagnostics.HasError() { + return + } + + var remote remoteAPI + err := r.client.get(ctx, "/api/v2/remotes/"+state.Name.ValueString(), &remote) + if err != nil { + if isNotFound(err) { + resp.State.RemoveResource(ctx) + return + } + resp.Diagnostics.AddError("read local puppet failed", err.Error()) + return + } + + newState := localPuppetAPIToModel(remote) + resp.Diagnostics.Append(resp.State.Set(ctx, newState)...) +} + +func (r *localPuppetResource) Update(ctx context.Context, req resource.UpdateRequest, resp *resource.UpdateResponse) { + var plan localPuppetResourceModel + resp.Diagnostics.Append(req.Plan.Get(ctx, &plan)...) + if resp.Diagnostics.HasError() { + return + } + + api := localPuppetModelToAPI(plan) + api.ManagedBy = "terraform" + + var updated remoteAPI + if err := r.client.put(ctx, "/api/v2/remotes/"+plan.Name.ValueString(), api, &updated); err != nil { + resp.Diagnostics.AddError("update local puppet failed", err.Error()) + return + } + + state := localPuppetAPIToModel(updated) + resp.Diagnostics.Append(resp.State.Set(ctx, state)...) +} + +func (r *localPuppetResource) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) { + var state localPuppetResourceModel + resp.Diagnostics.Append(req.State.Get(ctx, &state)...) + if resp.Diagnostics.HasError() { + return + } + + if err := r.client.del(ctx, "/api/v2/remotes/"+state.Name.ValueString()); err != nil { + resp.Diagnostics.AddError("delete local puppet failed", err.Error()) + return + } +} + +func (r *localPuppetResource) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) { + resource.ImportStatePassthroughID(ctx, path.Root("name"), req, resp) +} + +func localPuppetModelToAPI(m localPuppetResourceModel) remoteAPI { + return remoteAPI{ + Name: m.Name.ValueString(), + PackageType: "puppet", + RepoType: "local", + Description: m.Description.ValueString(), + } +} + +func localPuppetAPIToModel(api remoteAPI) localPuppetResourceModel { + return localPuppetResourceModel{ + Name: types.StringValue(api.Name), + Description: types.StringValue(api.Description), + } +} diff --git a/internal/provider/resource_local_puppet_test.go b/internal/provider/resource_local_puppet_test.go new file mode 100644 index 0000000..e2f5a89 --- /dev/null +++ b/internal/provider/resource_local_puppet_test.go @@ -0,0 +1,125 @@ +package provider + +import ( + "context" + "testing" + + "github.com/hashicorp/terraform-plugin-framework/resource" + "github.com/hashicorp/terraform-plugin-framework/types" +) + +func TestLocalPuppetModelToAPI(t *testing.T) { + model := localPuppetResourceModel{ + Name: types.StringValue("puppet-internal"), + Description: types.StringValue("Internal Puppet forge"), + } + + api := localPuppetModelToAPI(model) + + if api.Name != "puppet-internal" { + t.Errorf("Name: expected puppet-internal, got %s", api.Name) + } + if api.PackageType != "puppet" { + t.Errorf("PackageType: expected puppet, got %s", api.PackageType) + } + if api.RepoType != "local" { + t.Errorf("RepoType: expected local, got %s", api.RepoType) + } + if api.Description != "Internal Puppet forge" { + t.Errorf("Description: expected 'Internal Puppet forge', got %s", api.Description) + } +} + +func TestLocalPuppetModelToAPI_EmptyDescription(t *testing.T) { + model := localPuppetResourceModel{ + Name: types.StringValue("puppet-empty"), + Description: types.StringValue(""), + } + + api := localPuppetModelToAPI(model) + + if api.Name != "puppet-empty" { + t.Errorf("Name: expected puppet-empty, got %s", api.Name) + } + if api.Description != "" { + t.Errorf("Description: expected empty string, got %s", api.Description) + } + if api.PackageType != "puppet" { + t.Errorf("PackageType: expected puppet, got %s", api.PackageType) + } + if api.RepoType != "local" { + t.Errorf("RepoType: expected local, got %s", api.RepoType) + } +} + +func TestLocalPuppetAPIToModel(t *testing.T) { + api := remoteAPI{ + Name: "puppet-internal", + PackageType: "puppet", + RepoType: "local", + Description: "Internal Puppet forge", + ManagedBy: "terraform", + } + + model := localPuppetAPIToModel(api) + + if model.Name.ValueString() != "puppet-internal" { + t.Errorf("Name: expected puppet-internal, got %s", model.Name.ValueString()) + } + if model.Description.ValueString() != "Internal Puppet forge" { + t.Errorf("Description: expected 'Internal Puppet forge', got %s", model.Description.ValueString()) + } +} + +func TestLocalPuppetRoundTrip(t *testing.T) { + original := localPuppetResourceModel{ + Name: types.StringValue("roundtrip-puppet"), + Description: types.StringValue("Round trip test"), + } + + api := localPuppetModelToAPI(original) + result := localPuppetAPIToModel(api) + + if result.Name.ValueString() != original.Name.ValueString() { + t.Errorf("Name: expected %s, got %s", original.Name.ValueString(), result.Name.ValueString()) + } + if result.Description.ValueString() != original.Description.ValueString() { + t.Errorf("Description: expected %s, got %s", original.Description.ValueString(), result.Description.ValueString()) + } +} + +func TestLocalPuppetResource_Metadata(t *testing.T) { + r := NewLocalPuppetResource() + req := resource.MetadataRequest{ProviderTypeName: "artifactapi"} + var resp resource.MetadataResponse + r.Metadata(context.Background(), req, &resp) + if resp.TypeName != "artifactapi_local_puppet" { + t.Errorf("expected artifactapi_local_puppet, got %s", resp.TypeName) + } +} + +func TestLocalPuppetResource_Schema(t *testing.T) { + r := NewLocalPuppetResource() + req := resource.SchemaRequest{} + var resp resource.SchemaResponse + r.Schema(context.Background(), req, &resp) + + expectedAttrs := []string{"name", "description"} + for _, attr := range expectedAttrs { + if _, ok := resp.Schema.Attributes[attr]; !ok { + t.Errorf("missing expected attribute: %s", attr) + } + } + + if len(resp.Schema.Attributes) != len(expectedAttrs) { + t.Errorf("expected %d attributes, got %d", len(expectedAttrs), len(resp.Schema.Attributes)) + } +} + +func TestNewLocalPuppetResource_Type(t *testing.T) { + r := NewLocalPuppetResource() + _, ok := r.(*localPuppetResource) + if !ok { + t.Error("expected *localPuppetResource") + } +} -- 2.47.3