(feat) add data source for all namespaces.

Author: wrmedford

internal/provider/namespaces_data_source.go

@@ -0,0 +1,108 @@
+package provider
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-framework/datasource"
+	"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+
+	client "github.com/pomerium/enterprise-client-go"
+	"github.com/pomerium/enterprise-client-go/pb"
+)
+
+var _ datasource.DataSource = &NamespacesDataSource{}
+
+func NewNamespacesDataSource() datasource.DataSource {
+	return &NamespacesDataSource{}
+}
+
+type NamespacesDataSource struct {
+	client *client.Client
+}
+
+type NamespaceModel struct {
+	ID       types.String `tfsdk:"id"`
+	Name     types.String `tfsdk:"name"`
+	ParentID types.String `tfsdk:"parent_id"`
+}
+
+type NamespacesDataSourceModel struct {
+	Namespaces []NamespaceModel `tfsdk:"namespaces"`
+}
+
+func (d *NamespacesDataSource) Metadata(_ context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
+	resp.TypeName = req.ProviderTypeName + "_namespaces"
+}
+
+func (d *NamespacesDataSource) Schema(_ context.Context, _ datasource.SchemaRequest, resp *datasource.SchemaResponse) {
+	resp.Schema = schema.Schema{
+		MarkdownDescription: "List all namespaces",
+
+		Attributes: map[string]schema.Attribute{
+			"namespaces": schema.ListNestedAttribute{
+				Computed: true,
+				NestedObject: schema.NestedAttributeObject{
+					Attributes: map[string]schema.Attribute{
+						"id": schema.StringAttribute{
+							Computed:    true,
+							Description: "Unique identifier for the namespace.",
+						},
+						"name": schema.StringAttribute{
+							Computed:    true,
+							Description: "Name of the namespace.",
+						},
+						"parent_id": schema.StringAttribute{
+							Computed:    true,
+							Description: "ID of the parent namespace.",
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func (d *NamespacesDataSource) Configure(_ context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
+	if req.ProviderData == nil {
+		return
+	}
+
+	client, ok := req.ProviderData.(*client.Client)
+	if !ok {
+		resp.Diagnostics.AddError(
+			"Unexpected Data Source Configure Type",
+			fmt.Sprintf("Expected *client.Client, got: %T.", req.ProviderData),
+		)
+		return
+	}
+
+	d.client = client
+}
+
+func (d *NamespacesDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
+	var data NamespacesDataSourceModel
+
+	namespacesResp, err := d.client.NamespaceService.ListNamespaces(ctx, &pb.ListNamespacesRequest{})
+	if err != nil {
+		resp.Diagnostics.AddError("Error reading namespaces", err.Error())
+		return
+	}
+
+	namespaces := make([]NamespaceModel, 0, len(namespacesResp.Namespaces))
+	for _, ns := range namespacesResp.Namespaces {
+		var namespace NamespaceModel
+		namespace.ID = types.StringValue(ns.Id)
+		namespace.Name = types.StringValue(ns.Name)
+		if ns.ParentId != "" {
+			namespace.ParentID = types.StringValue(ns.ParentId)
+		} else {
+			namespace.ParentID = types.StringNull()
+		}
+		namespaces = append(namespaces, namespace)
+	}
+
+	data.Namespaces = namespaces
+	resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
+}

internal/provider/provider.go

@@ -132,6 +132,7 @@ func (p *PomeriumProvider) DataSources(_ context.Context) []func() datasource.Da
 		NewRouteDataSource,
 		NewNamespaceDataSource,
 		NewPolicyDataSource,
+		NewNamespacesDataSource,
 	}
 }