|  | 
|  | 1 | +package datawarehouse | 
|  | 2 | + | 
|  | 3 | +import ( | 
|  | 4 | +	"context" | 
|  | 5 | +	"fmt" | 
|  | 6 | +	"strings" | 
|  | 7 | + | 
|  | 8 | +	"github.com/hashicorp/terraform-plugin-sdk/v2/diag" | 
|  | 9 | +	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema" | 
|  | 10 | +	datawarehouseapi "github.com/scaleway/scaleway-sdk-go/api/datawarehouse/v1beta1" | 
|  | 11 | +	"github.com/scaleway/scaleway-sdk-go/scw" | 
|  | 12 | +	"github.com/scaleway/terraform-provider-scaleway/v2/internal/httperrors" | 
|  | 13 | +	"github.com/scaleway/terraform-provider-scaleway/v2/internal/locality/regional" | 
|  | 14 | +) | 
|  | 15 | + | 
|  | 16 | +func ResourceDatabase() *schema.Resource { | 
|  | 17 | +	return &schema.Resource{ | 
|  | 18 | +		CreateContext: resourceDatabaseCreate, | 
|  | 19 | +		ReadContext:   resourceDatabaseRead, | 
|  | 20 | +		DeleteContext: resourceDatabaseDelete, | 
|  | 21 | +		Importer: &schema.ResourceImporter{ | 
|  | 22 | +			StateContext: schema.ImportStatePassthroughContext, | 
|  | 23 | +		}, | 
|  | 24 | +		Schema: map[string]*schema.Schema{ | 
|  | 25 | +			"region": regional.Schema(), | 
|  | 26 | +			"deployment_id": { | 
|  | 27 | +				Type:        schema.TypeString, | 
|  | 28 | +				Required:    true, | 
|  | 29 | +				ForceNew:    true, | 
|  | 30 | +				Description: "ID of the Datawarehouse deployment to which this database belongs.", | 
|  | 31 | +			}, | 
|  | 32 | +			"name": { | 
|  | 33 | +				Type:        schema.TypeString, | 
|  | 34 | +				Required:    true, | 
|  | 35 | +				ForceNew:    true, | 
|  | 36 | +				Description: "Name of the database.", | 
|  | 37 | +			}, | 
|  | 38 | +			"size": { | 
|  | 39 | +				Type:        schema.TypeInt, | 
|  | 40 | +				Computed:    true, | 
|  | 41 | +				Description: "Size of the database (in GB).", | 
|  | 42 | +			}, | 
|  | 43 | +		}, | 
|  | 44 | +	} | 
|  | 45 | +} | 
|  | 46 | + | 
|  | 47 | +func resourceDatabaseCreate(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics { | 
|  | 48 | +	api := NewAPI(meta) | 
|  | 49 | + | 
|  | 50 | +	region := scw.Region(d.Get("region").(string)) | 
|  | 51 | +	deploymentID := d.Get("deployment_id").(string) | 
|  | 52 | +	name := d.Get("name").(string) | 
|  | 53 | + | 
|  | 54 | +	req := &datawarehouseapi.CreateDatabaseRequest{ | 
|  | 55 | +		Region:       region, | 
|  | 56 | +		DeploymentID: deploymentID, | 
|  | 57 | +		Name:         name, | 
|  | 58 | +	} | 
|  | 59 | + | 
|  | 60 | +	_, err := api.CreateDatabase(req, scw.WithContext(ctx)) | 
|  | 61 | +	if err != nil { | 
|  | 62 | +		return diag.FromErr(err) | 
|  | 63 | +	} | 
|  | 64 | + | 
|  | 65 | +	d.SetId(ResourceDatabaseID(region, deploymentID, name)) | 
|  | 66 | + | 
|  | 67 | +	return resourceDatabaseRead(ctx, d, meta) | 
|  | 68 | +} | 
|  | 69 | + | 
|  | 70 | +func resourceDatabaseRead(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics { | 
|  | 71 | +	api := NewAPI(meta) | 
|  | 72 | + | 
|  | 73 | +	region, deploymentID, name, err := ResourceDatabaseParseID(d.Id()) | 
|  | 74 | +	if err != nil { | 
|  | 75 | +		return diag.FromErr(err) | 
|  | 76 | +	} | 
|  | 77 | + | 
|  | 78 | +	resp, err := api.ListDatabases(&datawarehouseapi.ListDatabasesRequest{ | 
|  | 79 | +		Region:       region, | 
|  | 80 | +		DeploymentID: deploymentID, | 
|  | 81 | +		Name:         scw.StringPtr(name), | 
|  | 82 | +	}, scw.WithContext(ctx)) | 
|  | 83 | +	if err != nil { | 
|  | 84 | +		if httperrors.Is404(err) { | 
|  | 85 | +			d.SetId("") | 
|  | 86 | + | 
|  | 87 | +			return nil | 
|  | 88 | +		} | 
|  | 89 | + | 
|  | 90 | +		return diag.FromErr(err) | 
|  | 91 | +	} | 
|  | 92 | + | 
|  | 93 | +	var found *datawarehouseapi.Database | 
|  | 94 | + | 
|  | 95 | +	for _, db := range resp.Databases { | 
|  | 96 | +		if db.Name == name { | 
|  | 97 | +			found = db | 
|  | 98 | + | 
|  | 99 | +			break | 
|  | 100 | +		} | 
|  | 101 | +	} | 
|  | 102 | + | 
|  | 103 | +	if found == nil { | 
|  | 104 | +		d.SetId("") | 
|  | 105 | + | 
|  | 106 | +		return nil | 
|  | 107 | +	} | 
|  | 108 | + | 
|  | 109 | +	_ = d.Set("deployment_id", deploymentID) | 
|  | 110 | +	_ = d.Set("name", found.Name) | 
|  | 111 | +	_ = d.Set("size", int(found.Size)) | 
|  | 112 | + | 
|  | 113 | +	return nil | 
|  | 114 | +} | 
|  | 115 | + | 
|  | 116 | +func resourceDatabaseDelete(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics { | 
|  | 117 | +	api := NewAPI(meta) | 
|  | 118 | + | 
|  | 119 | +	region, deploymentID, name, err := ResourceDatabaseParseID(d.Id()) | 
|  | 120 | +	if err != nil { | 
|  | 121 | +		return diag.FromErr(err) | 
|  | 122 | +	} | 
|  | 123 | + | 
|  | 124 | +	err = api.DeleteDatabase(&datawarehouseapi.DeleteDatabaseRequest{ | 
|  | 125 | +		Region:       region, | 
|  | 126 | +		DeploymentID: deploymentID, | 
|  | 127 | +		Name:         name, | 
|  | 128 | +	}, scw.WithContext(ctx)) | 
|  | 129 | +	if err != nil && !httperrors.Is404(err) { | 
|  | 130 | +		return diag.FromErr(err) | 
|  | 131 | +	} | 
|  | 132 | + | 
|  | 133 | +	d.SetId("") | 
|  | 134 | + | 
|  | 135 | +	return nil | 
|  | 136 | +} | 
|  | 137 | + | 
|  | 138 | +func ResourceDatabaseID(region scw.Region, deploymentID, name string) string { | 
|  | 139 | +	return fmt.Sprintf("%s/%s/%s", region, deploymentID, name) | 
|  | 140 | +} | 
|  | 141 | + | 
|  | 142 | +func ResourceDatabaseParseID(id string) (scw.Region, string, string, error) { | 
|  | 143 | +	parts := strings.Split(id, "/") | 
|  | 144 | +	if len(parts) != 3 { | 
|  | 145 | +		return "", "", "", fmt.Errorf("unexpected format of ID (%s), expected region/deployment_id/name", id) | 
|  | 146 | +	} | 
|  | 147 | + | 
|  | 148 | +	return scw.Region(parts[0]), parts[1], parts[2], nil | 
|  | 149 | +} | 
0 commit comments