resource_cloud_project_database_ip_restriction.go

package ovh

import (
	"context"
	"fmt"
	"log"
	"net/url"
	"strings"
	"time"

	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	"github.com/ovh/go-ovh/ovh"
	"github.com/ovh/terraform-provider-ovh/ovh/helpers"
	"github.com/ovh/terraform-provider-ovh/ovh/helpers/hashcode"
)

func resourceCloudProjectDatabaseIpRestriction() *schema.Resource {
	return &schema.Resource{
		CreateContext: resourceCloudProjectDatabaseIpRestrictionCreate,
		ReadContext:   resourceCloudProjectDatabaseIpRestrictionRead,
		DeleteContext: resourceCloudProjectDatabaseIpRestrictionDelete,
		UpdateContext: resourceCloudProjectDatabaseIpRestrictionUpdate,

		Importer: &schema.ResourceImporter{
			State: resourceCloudProjectDatabaseIpRestrictionImportState,
		},

		Timeouts: &schema.ResourceTimeout{
			Create: schema.DefaultTimeout(20 * time.Minute),
			Update: schema.DefaultTimeout(20 * time.Minute),
			Delete: schema.DefaultTimeout(20 * time.Minute),
		},

		Schema: map[string]*schema.Schema{
			"service_name": {
				Type:        schema.TypeString,
				ForceNew:    true,
				Required:    true,
				DefaultFunc: schema.EnvDefaultFunc("OVH_CLOUD_PROJECT_SERVICE", nil),
			},
			"engine": {
				Type:        schema.TypeString,
				Description: "Name of the engine of the service",
				ForceNew:    true,
				Required:    true,
			},
			"cluster_id": {
				Type:        schema.TypeString,
				Description: "Id of the database cluster",
				ForceNew:    true,
				Required:    true,
			},
			"ip": {
				Type:        schema.TypeString,
				Description: "Authorized IP",
				ForceNew:    true,
				Required:    true,
			},
			"description": {
				Type:        schema.TypeString,
				Description: "Description of the IP restriction",
				Optional:    true,
			},

			//Computed
			"status": {
				Type:        schema.TypeString,
				Description: "Current status of the IP restriction",
				Computed:    true,
			},
		},
	}
}

func resourceCloudProjectDatabaseIpRestrictionImportState(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
	givenId := d.Id()
	n := 4
	splitId := strings.SplitN(givenId, "/", n)
	if len(splitId) != n {
		return nil, fmt.Errorf("Import Id is not service_name/engine/cluster_id/ip formatted")
	}
	serviceName := splitId[0]
	engine := splitId[1]
	clusterId := splitId[2]
	ip := splitId[3]
	d.SetId(hashcode.Strings([]string{ip}))
	d.Set("engine", engine)
	d.Set("service_name", serviceName)
	d.Set("cluster_id", clusterId)
	d.Set("ip", ip)

	results := make([]*schema.ResourceData, 1)
	results[0] = d
	return results, nil
}

func resourceCloudProjectDatabaseIpRestrictionCreate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	config := meta.(*Config)
	serviceName := d.Get("service_name").(string)
	engine := d.Get("engine").(string)
	clusterId := d.Get("cluster_id").(string)

	endpoint := fmt.Sprintf("/cloud/project/%s/database/%s/%s/ipRestriction",
		url.PathEscape(serviceName),
		url.PathEscape(engine),
		url.PathEscape(clusterId),
	)
	params := (&CloudProjectDatabaseIpRestrictionCreateOpts{}).FromResource(d)
	res := &CloudProjectDatabaseIpRestrictionResponse{}

	return diag.FromErr(
		resource.RetryContext(ctx, d.Timeout(schema.TimeoutCreate),
			func() *resource.RetryError {
				log.Printf("[DEBUG] Will create IP restriction: %+v for cluster %s from project %s", params, clusterId, serviceName)
				rErr := config.OVHClient.Post(endpoint, params, res)
				if rErr != nil {
					if errOvh, ok := rErr.(*ovh.APIError); ok && (errOvh.Code == 409) {
						return resource.RetryableError(rErr)
					}
					return resource.NonRetryableError(fmt.Errorf("calling Post %s with params %+v:\n\t %q", endpoint, params, rErr))
				}

				log.Printf("[DEBUG] Waiting for IP restriction %s to be READY", res.Ip)
				rErr = waitForCloudProjectDatabaseIpRestrictionReady(ctx, config.OVHClient, serviceName, engine, clusterId, res.Ip, d.Timeout(schema.TimeoutCreate))
				if rErr != nil {
					return resource.NonRetryableError(fmt.Errorf("timeout while waiting IP restriction %s to be READY: %w", res.Ip, rErr))
				}
				log.Printf("[DEBUG] IP restriction %s is READY", res.Ip)

				d.SetId(hashcode.Strings([]string{res.Ip}))

				readDiags := resourceCloudProjectDatabaseIpRestrictionRead(ctx, d, meta)
				rErr = diagnosticsToError(readDiags)
				if rErr != nil {
					return resource.NonRetryableError(rErr)
				}
				return nil
			},
		),
	)
}

func resourceCloudProjectDatabaseIpRestrictionRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	config := meta.(*Config)
	serviceName := d.Get("service_name").(string)
	engine := d.Get("engine").(string)
	clusterId := d.Get("cluster_id").(string)
	ip := d.Get("ip").(string)

	endpoint := fmt.Sprintf("/cloud/project/%s/database/%s/%s/ipRestriction/%s",
		url.PathEscape(serviceName),
		url.PathEscape(engine),
		url.PathEscape(clusterId),
		url.PathEscape(ip),
	)
	res := &CloudProjectDatabaseIpRestrictionResponse{}

	log.Printf("[DEBUG] Will read IP restriction %s from cluster %s from project %s", ip, clusterId, serviceName)
	if err := config.OVHClient.Get(endpoint, res); err != nil {
		return diag.FromErr(helpers.CheckDeleted(d, err, endpoint))
	}

	d.SetId(hashcode.Strings([]string{res.Ip}))
	for k, v := range res.ToMap() {
		d.Set(k, v)
	}

	log.Printf("[DEBUG] Read IP restriction %+v", res)
	return nil
}

func resourceCloudProjectDatabaseIpRestrictionUpdate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	config := meta.(*Config)
	serviceName := d.Get("service_name").(string)
	engine := d.Get("engine").(string)
	clusterId := d.Get("cluster_id").(string)
	ip := d.Get("ip").(string)

	endpoint := fmt.Sprintf("/cloud/project/%s/database/%s/%s/ipRestriction/%s",
		url.PathEscape(serviceName),
		url.PathEscape(engine),
		url.PathEscape(clusterId),
		url.PathEscape(ip),
	)
	params := (&CloudProjectDatabaseIpRestrictionUpdateOpts{}).FromResource(d)

	err := resource.Retry(d.Timeout(schema.TimeoutUpdate), func() *resource.RetryError {
		log.Printf("[DEBUG] Will update IP restriction: %+v from cluster %s from project %s", params, clusterId, serviceName)
		rErr := config.OVHClient.Put(endpoint, params, nil)
		if rErr != nil {
			if errOvh, ok := rErr.(*ovh.APIError); ok && (errOvh.Code == 409) {
				return resource.RetryableError(rErr)
			}
			return resource.NonRetryableError(fmt.Errorf("calling Put %s with params %+v:\n\t %q", endpoint, params, rErr))
		}

		log.Printf("[DEBUG] Waiting for IP restriction %s to be READY", ip)
		rErr = waitForCloudProjectDatabaseIpRestrictionReady(ctx, config.OVHClient, serviceName, engine, clusterId, ip, d.Timeout(schema.TimeoutUpdate))
		if rErr != nil {
			return resource.NonRetryableError(fmt.Errorf("timeout while waiting IP restriction %s to be READY: %w", ip, rErr))
		}
		log.Printf("[DEBUG] IP restriction %s is READY", ip)

		readDiags := resourceCloudProjectDatabaseIpRestrictionRead(ctx, d, meta)
		rErr = diagnosticsToError(readDiags)
		if rErr != nil {
			return resource.NonRetryableError(rErr)
		}
		return nil
	})
	return diag.FromErr(err)
}

func resourceCloudProjectDatabaseIpRestrictionDelete(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
	config := meta.(*Config)
	serviceName := d.Get("service_name").(string)
	engine := d.Get("engine").(string)
	clusterId := d.Get("cluster_id").(string)
	ip := d.Get("ip").(string)

	endpoint := fmt.Sprintf("/cloud/project/%s/database/%s/%s/ipRestriction/%s",
		url.PathEscape(serviceName),
		url.PathEscape(engine),
		url.PathEscape(clusterId),
		url.PathEscape(ip),
	)

	err := resource.Retry(d.Timeout(schema.TimeoutDelete), func() *resource.RetryError {
		log.Printf("[DEBUG] Will delete IP restriction %s from cluster %s from project %s", ip, clusterId, serviceName)
		rErr := config.OVHClient.Delete(endpoint, nil)
		if rErr != nil {
			if errOvh, ok := rErr.(*ovh.APIError); ok && (errOvh.Code == 409) {
				return resource.RetryableError(rErr)
			}
			rErr = helpers.CheckDeleted(d, rErr, endpoint)
			if rErr != nil {
				return resource.NonRetryableError(rErr)
			}
			return nil
		}

		log.Printf("[DEBUG] Waiting for IP restriction %s to be DELETED", clusterId)
		rErr = waitForCloudProjectDatabaseIpRestrictionDeleted(ctx, config.OVHClient, serviceName, engine, clusterId, ip, d.Timeout(schema.TimeoutDelete))
		if rErr != nil {
			return resource.NonRetryableError(fmt.Errorf("timeout while waiting IP restriction %s to be DELETED: %w", clusterId, rErr))
		}
		log.Printf("[DEBUG] IP restriction %s is DELETED", clusterId)

		d.SetId("")

		return nil
	})
	return diag.FromErr(err)
}