terraform-provider-threshold/internal/provider/resource_managed_group.go

package provider

import (
	"context"
	"net/http"

	"github.com/hashicorp/boundary/api"
	"github.com/hashicorp/boundary/api/managedgroups"
	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
)

const (
	managedGroupFilterKey = "filter"
)

func resourceManagedGroup() *schema.Resource {
	return &schema.Resource{
		Description: "The managed group resource allows you to configure a Boundary group.",

		CreateContext: resourceManagedGroupCreate,
		ReadContext:   resourceManagedGroupRead,
		UpdateContext: resourceManagedGroupUpdate,
		DeleteContext: resourceManagedGroupDelete,
		Importer: &schema.ResourceImporter{
			StateContext: schema.ImportStatePassthroughContext,
		},

		Schema: map[string]*schema.Schema{
			IDKey: {
				Description: "The ID of the group.",
				Type:        schema.TypeString,
				Computed:    true,
			},
			NameKey: {
				Description: "The managed group name. Defaults to the resource name.",
				Type:        schema.TypeString,
				Optional:    true,
			},
			DescriptionKey: {
				Description: "The managed group description.",
				Type:        schema.TypeString,
				Optional:    true,
			},
			AuthMethodIdKey: {
				Description: "The resource ID for the auth method.",
				Type:        schema.TypeString,
				Required:    true,
				ForceNew:    true,
			},
			managedGroupFilterKey: {
				Description: "Boolean expression to filter the workers for this managed group.",
				Type:        schema.TypeString,
				Required:    true,
			},
		},
	}
}

func setFromManagedGroupResponseMap(d *schema.ResourceData, raw map[string]interface{}) error {
	if err := d.Set(NameKey, raw[NameKey]); err != nil {
		return err
	}
	if err := d.Set(DescriptionKey, raw[DescriptionKey]); err != nil {
		return err
	}
	if err := d.Set(AuthMethodIdKey, raw[AuthMethodIdKey]); err != nil {
		return err
	}

	if attrsVal, ok := raw["attributes"]; ok {
		attrs := attrsVal.(map[string]interface{})
		if v, ok := attrs[managedGroupFilterKey]; ok {
			if err := d.Set(managedGroupFilterKey, v); err != nil {
				return err
			}
		}
	}

	d.SetId(raw[IDKey].(string))

	return nil
}

func resourceManagedGroupCreate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	md := meta.(*metaData)
	grpClient := managedgroups.NewClient(md.client)

	var authMethodId string
	authMethodVal, ok := d.GetOk(AuthMethodIdKey)
	if !ok {
		return diag.Errorf("no auth method ID provided")
	}
	authMethodId = authMethodVal.(string)

	var opts []managedgroups.Option
	nameVal, ok := d.GetOk(NameKey)
	if ok {
		nameStr := nameVal.(string)
		opts = append(opts, managedgroups.WithName(nameStr))
	}

	descVal, ok := d.GetOk(DescriptionKey)
	if ok {
		descStr := descVal.(string)
		opts = append(opts, managedgroups.WithDescription(descStr))
	}

	v, ok := d.GetOk(managedGroupFilterKey)
	if ok {
		str := v.(string)
		opts = append(opts, managedgroups.WithOidcManagedGroupFilter(str))
	}

	grp, err := grpClient.Create(ctx, authMethodId, opts...)
	if err != nil {
		return diag.Errorf("error creating managed group: %v", err)
	}
	if grp == nil {
		return diag.Errorf("managed group nil after create")
	}
	raw := grp.GetResponse().Map

	if err := setFromManagedGroupResponseMap(d, raw); err != nil {
		return diag.FromErr(err)
	}

	return nil
}

func resourceManagedGroupRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	md := meta.(*metaData)
	grpClient := managedgroups.NewClient(md.client)

	grp, err := grpClient.Read(ctx, d.Id())
	if err != nil {
		if apiErr := api.AsServerError(err); apiErr.Response().StatusCode() == http.StatusNotFound {
			d.SetId("")
			return nil
		}
		return diag.Errorf("error reading managed group: %v", err)
	}
	if grp == nil {
		return diag.Errorf("managed group nil after read")
	}

	if err := setFromManagedGroupResponseMap(d, grp.GetResponse().Map); err != nil {
		return diag.FromErr(err)
	}

	return nil
}

func resourceManagedGroupUpdate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	md := meta.(*metaData)
	grpClient := managedgroups.NewClient(md.client)

	var opts []managedgroups.Option
	var name *string
	if d.HasChange(NameKey) {
		opts = append(opts, managedgroups.DefaultName())
		nameVal, ok := d.GetOk(NameKey)
		if ok {
			nameStr := nameVal.(string)
			name = &nameStr
			opts = append(opts, managedgroups.WithName(nameStr))
		}
	}

	var desc *string
	if d.HasChange(DescriptionKey) {
		opts = append(opts, managedgroups.DefaultDescription())
		descVal, ok := d.GetOk(DescriptionKey)
		if ok {
			descStr := descVal.(string)
			desc = &descStr
			opts = append(opts, managedgroups.WithDescription(descStr))
		}
	}

	var filter *string
	if d.HasChange(managedGroupFilterKey) {
		if f, ok := d.GetOk(managedGroupFilterKey); ok {
			filterStr := f.(string)
			filter = &filterStr
			opts = append(opts, managedgroups.WithOidcManagedGroupFilter(filterStr))
		}
	}

	if len(opts) > 0 {
		opts = append(opts, managedgroups.WithAutomaticVersioning(true))
		_, err := grpClient.Update(ctx, d.Id(), 0, opts...)
		if err != nil {
			return diag.Errorf("error updating managed group: %v", err)
		}
	}

	if d.HasChange(NameKey) {
		if err := d.Set(NameKey, name); err != nil {
			return diag.FromErr(err)
		}
	}
	if d.HasChange(DescriptionKey) {
		if err := d.Set(DescriptionKey, desc); err != nil {
			return diag.FromErr(err)
		}
	}
	if d.HasChange(managedGroupFilterKey) {
		if err := d.Set(managedGroupFilterKey, filter); err != nil {
			return diag.FromErr(err)
		}
	}

	return nil
}

func resourceManagedGroupDelete(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	md := meta.(*metaData)
	grpClient := managedgroups.NewClient(md.client)

	_, err := grpClient.Delete(ctx, d.Id())
	if err != nil {
		return diag.Errorf("error calling delete managed group: %s", err.Error())
	}

	return nil
}