tag_resource.go

package provider

import (
	"context"
	"fmt"
	"regexp"
	"strconv"

	"github.com/devopsarr/radarr-go/radarr"
	"github.com/devopsarr/terraform-provider-radarr/internal/helpers"
	"github.com/hashicorp/terraform-plugin-framework-validators/stringvalidator"
	"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/int64planmodifier"
	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
	"github.com/hashicorp/terraform-plugin-framework/schema/validator"
	"github.com/hashicorp/terraform-plugin-framework/types"
	"github.com/hashicorp/terraform-plugin-log/tflog"
)

const tagResourceName = "tag"

// Ensure provider defined types fully satisfy framework interfaces.
var (
	_ resource.Resource                = &TagResource{}
	_ resource.ResourceWithImportState = &TagResource{}
)

func NewTagResource() resource.Resource {
	return &TagResource{}
}

// TagResource defines the tag implementation.
type TagResource struct {
	client *radarr.APIClient
}

// Tag describes the tag data model.
type Tag struct {
	Label types.String `tfsdk:"label"`
	ID    types.Int64  `tfsdk:"id"`
}

func (r *TagResource) Metadata(ctx context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) {
	resp.TypeName = req.ProviderTypeName + "_" + tagResourceName
}

func (r *TagResource) Schema(ctx context.Context, req resource.SchemaRequest, resp *resource.SchemaResponse) {
	resp.Schema = schema.Schema{
		MarkdownDescription: "<!-- subcategory:Tags -->Tag resource.\nFor more information refer to [Tags](https://wiki.servarr.com/radarr/settings#tags) documentation.",
		Attributes: map[string]schema.Attribute{
			"label": schema.StringAttribute{
				MarkdownDescription: "Tag label. It must be lowercase.",
				Required:            true,
				Validators: []validator.String{
					stringvalidator.RegexMatches(
						regexp.MustCompile(`^.*[^A-Z]+.*$`),
						"String cannot contains uppercase values",
					),
				},
			},
			"id": schema.Int64Attribute{
				MarkdownDescription: "Tag ID.",
				Computed:            true,
				PlanModifiers: []planmodifier.Int64{
					int64planmodifier.UseStateForUnknown(),
				},
			},
		},
	}
}

func (r *TagResource) Configure(ctx context.Context, req resource.ConfigureRequest, resp *resource.ConfigureResponse) {
	if client := helpers.ResourceConfigure(ctx, req, resp); client != nil {
		r.client = client
	}
}

func (r *TagResource) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) {
	// Retrieve values from plan
	var tag *Tag

	resp.Diagnostics.Append(req.Plan.Get(ctx, &tag)...)

	if resp.Diagnostics.HasError() {
		return
	}

	// Create new Tag
	request := *radarr.NewTagResource()
	request.SetLabel(tag.Label.ValueString())

	response, _, err := r.client.TagApi.CreateTag(ctx).TagResource(request).Execute()
	if err != nil {
		resp.Diagnostics.AddError(helpers.ClientError, helpers.ParseClientError(helpers.Create, tagResourceName, err))

		return
	}

	tflog.Trace(ctx, "created tag: "+strconv.Itoa(int(response.GetId())))
	// Generate resource state struct
	tag.write(response)
	resp.Diagnostics.Append(resp.State.Set(ctx, &tag)...)
}

func (r *TagResource) Read(ctx context.Context, req resource.ReadRequest, resp *resource.ReadResponse) {
	// Get current state
	var tag *Tag

	resp.Diagnostics.Append(req.State.Get(ctx, &tag)...)

	if resp.Diagnostics.HasError() {
		return
	}

	// Get tag current value
	response, _, err := r.client.TagApi.GetTagById(ctx, int32(tag.ID.ValueInt64())).Execute()
	if err != nil {
		resp.Diagnostics.AddError(helpers.ClientError, helpers.ParseClientError(helpers.Read, tagResourceName, err))

		return
	}

	tflog.Trace(ctx, "read "+tagResourceName+": "+strconv.Itoa(int(response.GetId())))
	// Map response body to resource schema attribute
	tag.write(response)
	resp.Diagnostics.Append(resp.State.Set(ctx, &tag)...)
}

func (r *TagResource) Update(ctx context.Context, req resource.UpdateRequest, resp *resource.UpdateResponse) {
	// Get plan values
	var tag *Tag

	resp.Diagnostics.Append(req.Plan.Get(ctx, &tag)...)

	if resp.Diagnostics.HasError() {
		return
	}

	// Update Tag
	tagResource := *radarr.NewTagResource()
	tagResource.SetLabel(tag.Label.ValueString())
	tagResource.SetId(int32(tag.ID.ValueInt64()))

	response, _, err := r.client.TagApi.UpdateTag(ctx, fmt.Sprint(tagResource.GetId())).TagResource(tagResource).Execute()
	if err != nil {
		resp.Diagnostics.AddError(helpers.ClientError, helpers.ParseClientError(helpers.Update, tagResourceName, err))

		return
	}

	tflog.Trace(ctx, "updated "+tagResourceName+": "+strconv.Itoa(int(response.GetId())))
	// Generate resource state struct
	tag.write(response)
	resp.Diagnostics.Append(resp.State.Set(ctx, &tag)...)
}

func (r *TagResource) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) {
	var tag *Tag

	resp.Diagnostics.Append(req.State.Get(ctx, &tag)...)

	if resp.Diagnostics.HasError() {
		return
	}

	// Delete tag current value
	_, err := r.client.TagApi.DeleteTag(ctx, int32(tag.ID.ValueInt64())).Execute()
	if err != nil {
		resp.Diagnostics.AddError(helpers.ClientError, helpers.ParseClientError(helpers.Read, tagResourceName, err))

		return
	}

	tflog.Trace(ctx, "deleted "+tagResourceName+": "+strconv.Itoa(int(tag.ID.ValueInt64())))
	resp.State.RemoveResource(ctx)
}

func (r *TagResource) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) {
	helpers.ImportStatePassthroughIntID(ctx, path.Root("id"), req, resp)
	tflog.Trace(ctx, "imported "+tagResourceName+": "+req.ID)
}

func (t *Tag) write(tag *radarr.TagResource) {
	t.ID = types.Int64Value(int64(tag.GetId()))
	t.Label = types.StringValue(tag.GetLabel())
}