import_lists_data_source.go

package provider

import (
	"context"
	"strconv"

	"github.com/devopsarr/lidarr-go/lidarr"
	"github.com/devopsarr/terraform-provider-lidarr/internal/helpers"
	"github.com/hashicorp/terraform-plugin-framework/datasource"
	"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
	"github.com/hashicorp/terraform-plugin-framework/types"
	"github.com/hashicorp/terraform-plugin-log/tflog"
)

const importListsDataSourceName = "import_lists"

// Ensure provider defined types fully satisfy framework interfaces.
var _ datasource.DataSource = &ImportListsDataSource{}

func NewImportListsDataSource() datasource.DataSource {
	return &ImportListsDataSource{}
}

// ImportListsDataSource defines the import lists implementation.
type ImportListsDataSource struct {
	client *lidarr.APIClient
}

// ImportLists describes the import lists data model.
type ImportLists struct {
	ImportLists types.Set    `tfsdk:"import_lists"`
	ID          types.String `tfsdk:"id"`
}

func (d *ImportListsDataSource) Metadata(_ context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
	resp.TypeName = req.ProviderTypeName + "_" + importListsDataSourceName
}

func (d *ImportListsDataSource) Schema(_ context.Context, _ datasource.SchemaRequest, resp *datasource.SchemaResponse) {
	resp.Schema = schema.Schema{
		// This description is used by the documentation generator and the delay server.
		MarkdownDescription: "<!-- subcategory:Import Lists -->List all available [Import Lists](../resources/import_list).",
		Attributes: map[string]schema.Attribute{
			// TODO: remove ID once framework support tests without ID https://www.terraform.io/plugin/framework/acctests#implement-id-attribute
			"id": schema.StringAttribute{
				Computed: true,
			},
			"import_lists": schema.SetNestedAttribute{
				MarkdownDescription: "Import List list.",
				Computed:            true,
				NestedObject: schema.NestedAttributeObject{
					Attributes: map[string]schema.Attribute{
						"enable_automatic_add": schema.BoolAttribute{
							MarkdownDescription: "Enable automatic add flag.",
							Computed:            true,
						},
						"should_monitor_existing": schema.BoolAttribute{
							MarkdownDescription: "Should monitor existing flag.",
							Computed:            true,
						},
						"should_search": schema.BoolAttribute{
							MarkdownDescription: "Should search flag.",
							Computed:            true,
						},
						"quality_profile_id": schema.Int64Attribute{
							MarkdownDescription: "Quality profile ID.",
							Computed:            true,
						},
						"metadata_profile_id": schema.Int64Attribute{
							MarkdownDescription: "Metadata profile ID.",
							Computed:            true,
						},
						"list_order": schema.Int64Attribute{
							MarkdownDescription: "List order.",
							Computed:            true,
						},
						"root_folder_path": schema.StringAttribute{
							MarkdownDescription: "Root folder path.",
							Computed:            true,
						},
						"should_monitor": schema.StringAttribute{
							MarkdownDescription: "Should monitor.",
							Computed:            true,
						},
						"monitor_new_items": schema.StringAttribute{
							MarkdownDescription: "Monitor new items.",
							Computed:            true,
						},
						"implementation": schema.StringAttribute{
							MarkdownDescription: "ImportList implementation name.",
							Computed:            true,
						},
						"config_contract": schema.StringAttribute{
							MarkdownDescription: "ImportList configuration template.",
							Computed:            true,
						},
						"list_type": schema.StringAttribute{
							MarkdownDescription: "List type.",
							Computed:            true,
						},
						"name": schema.StringAttribute{
							MarkdownDescription: "Import List name.",
							Computed:            true,
						},
						"tags": schema.SetAttribute{
							MarkdownDescription: "List of associated tags.",
							Computed:            true,
							ElementType:         types.Int64Type,
						},
						"id": schema.Int64Attribute{
							MarkdownDescription: "Import List ID.",
							Computed:            true,
						},
						// Field values
						"count_list": schema.Int64Attribute{
							MarkdownDescription: "Elements to pull from list.",
							Computed:            true,
						},
						"access_token": schema.StringAttribute{
							MarkdownDescription: "Access token.",
							Computed:            true,
							Sensitive:           true,
						},
						"refresh_token": schema.StringAttribute{
							MarkdownDescription: "Refresh token.",
							Computed:            true,
							Sensitive:           true,
						},
						"api_key": schema.StringAttribute{
							MarkdownDescription: "API key.",
							Computed:            true,
							Sensitive:           true,
						},
						"user_id": schema.StringAttribute{
							MarkdownDescription: "User ID.",
							Computed:            true,
						},
						"tag_id": schema.StringAttribute{
							MarkdownDescription: "Tag ID.",
							Computed:            true,
						},
						"list_id": schema.StringAttribute{
							MarkdownDescription: "List ID.",
							Computed:            true,
						},
						"series_id": schema.StringAttribute{
							MarkdownDescription: "Series ID.",
							Computed:            true,
						},
						"base_url": schema.StringAttribute{
							MarkdownDescription: "Base URL.",
							Computed:            true,
						},
						"expires": schema.StringAttribute{
							MarkdownDescription: "Expires.",
							Computed:            true,
						},
						"profile_ids": schema.SetAttribute{
							MarkdownDescription: "Profile IDs.",
							Computed:            true,
							ElementType:         types.Int64Type,
						},
						"tag_ids": schema.SetAttribute{
							MarkdownDescription: "Tag IDs.",
							Computed:            true,
							ElementType:         types.Int64Type,
						},
						"playlist_ids": schema.SetAttribute{
							MarkdownDescription: "Playlist IDs.",
							Computed:            true,
							ElementType:         types.StringType,
						},
					},
				},
			},
		},
	}
}

func (d *ImportListsDataSource) Configure(ctx context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
	if client := helpers.DataSourceConfigure(ctx, req, resp); client != nil {
		d.client = client
	}
}

func (d *ImportListsDataSource) Read(ctx context.Context, _ datasource.ReadRequest, resp *datasource.ReadResponse) {
	// Get import lists current value
	response, _, err := d.client.ImportListAPI.ListImportList(ctx).Execute()
	if err != nil {
		resp.Diagnostics.AddError(helpers.ClientError, helpers.ParseClientError(helpers.Read, importListsDataSourceName, err))

		return
	}

	tflog.Trace(ctx, "read "+importListsDataSourceName)
	// Map response body to resource schema attribute
	importLists := make([]ImportList, len(response))
	for i, d := range response {
		importLists[i].write(ctx, d, &resp.Diagnostics)
	}

	listList, diags := types.SetValueFrom(ctx, ImportList{}.getType(), importLists)
	resp.Diagnostics.Append(diags...)
	resp.Diagnostics.Append(resp.State.Set(ctx, ImportLists{ImportLists: listList, ID: types.StringValue(strconv.Itoa(len(response)))})...)
}