connectivity_template_status.go

package blueprint

import (
	"context"
	"fmt"
	"strings"

	"github.com/Juniper/apstra-go-sdk/apstra"
	"github.com/Juniper/apstra-go-sdk/enum"
	"github.com/Juniper/terraform-provider-apstra/internal/rosetta"
	"github.com/Juniper/terraform-provider-apstra/internal/value"
	"github.com/hashicorp/terraform-plugin-framework/attr"
	"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
	"github.com/hashicorp/terraform-plugin-framework/diag"
	"github.com/hashicorp/terraform-plugin-framework/types"
)

type ConnectivityTemplateStatus struct {
	Id              types.String `tfsdk:"id"`
	Name            types.String `tfsdk:"name"`
	Description     types.String `tfsdk:"description"`
	Status          types.String `tfsdk:"status"`
	AssignmentCount types.Int64  `tfsdk:"assignment_count"`
	Tags            types.Set    `tfsdk:"tags"`
}

func (o ConnectivityTemplateStatus) AttrTypes() map[string]attr.Type {
	return map[string]attr.Type{
		"id":               types.StringType,
		"name":             types.StringType,
		"description":      types.StringType,
		"status":           types.StringType,
		"assignment_count": types.Int64Type,
		"tags":             types.SetType{ElemType: types.StringType},
	}
}

func (o ConnectivityTemplateStatus) DataSourceAttributes() map[string]schema.Attribute {
	return map[string]schema.Attribute{
		"id": schema.StringAttribute{
			MarkdownDescription: "Graph node ID of the Connectivity Template",
			Computed:            true,
		},
		"name": schema.StringAttribute{
			MarkdownDescription: "Name of the Connectivity Template, as displayed in the web UI",
			Computed:            true,
		},
		"description": schema.StringAttribute{
			MarkdownDescription: "Description of the Connectivity Template, as displayed in the web UI",
			Computed:            true,
		},
		"status": schema.StringAttribute{
			MarkdownDescription: fmt.Sprintf(
				"Status of the Connectivity Template - One of: [`%s`]",
				strings.Join(rosetta.StringersToFriendlyStrings(enum.EndpointPolicyStatuses.Members()), "`, `"),
			),
			Computed: true,
		},
		"assignment_count": schema.Int64Attribute{
			MarkdownDescription: "Count of Application Points to which the Connectivity Template has been assigned",
			Computed:            true,
		},
		"tags": schema.SetAttribute{
			MarkdownDescription: "Tags associated with the Connectivity Template",
			ElementType:         types.StringType,
			Computed:            true,
		},
	}
}

func (o *ConnectivityTemplateStatus) LoadApiData(ctx context.Context, in apstra.EndpointPolicyStatus, diags *diag.Diagnostics) {
	o.Id = types.StringValue(in.Id.String())
	o.Name = types.StringValue(in.Label)
	o.Description = types.StringValue(in.Description)
	o.Status = types.StringValue(in.Status.String())
	o.AssignmentCount = types.Int64Value(int64(in.AppPointsCount))
	o.Tags = value.SetOrNull(ctx, types.StringType, in.Tags, diags)
}