resource_vmc.go

package main

import (
	"context"
	"time"

	"github.com/hashicorp/go-hclog"
	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	hcx "github.com/adeleporte/terraform-provider-hcx/hcx"

	"log"
)

func resourceVmc() *schema.Resource {
	return &schema.Resource{
		CreateContext: resourceVmcCreate,
		ReadContext:   resourceVmcRead,
		UpdateContext: resourceVmcUpdate,
		DeleteContext: resourceVmcDelete,

		Schema: map[string]*schema.Schema{
			"sddc_id": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"sddc_name": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"cloud_url": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"cloud_name": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"cloud_type": {
				Type:     schema.TypeString,
				Computed: true,
			},
		},
	}
}

func resourceVmcCreate(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {

	client := m.(*hcx.Client)

	token := client.Token
	sddc_name := d.Get("sddc_name").(string)
	sddcID := d.Get("sddc_id").(string)

	if sddc_name == "" && sddcID == "" {
		return diag.Errorf("SDDC name or Id must be specified")
	}

	// Authenticate with VMware Cloud Services
	access_token, err := hcx.VmcAuthenticate(token)
	if err != nil {
		return diag.FromErr(err)
	}

	err = hcx.HcxCloudAuthenticate(client, access_token)
	if err != nil {
		return diag.FromErr(err)
	}

	var sddc hcx.SDDC
	if sddcID != "" {
		sddc, err = hcx.GetSddcByID(client, sddcID)
	} else {
		sddc, err = hcx.GetSddcByName(client, sddc_name)
	}

	if err != nil {
		return diag.FromErr(err)
	}

	// Check if already activated
	if sddc.DeploymentStatus == "ACTIVE" {
		return diag.Errorf("Already activated")
	}

	// Activate HCX
	_, err = hcx.ActivateHcxOnSDDC(client, sddc.ID)
	if err != nil {
		return diag.FromErr(err)
	}

	// Wait for task to be completed
	errcount := 0
	for {
		if sddcID != "" {
			sddc, err = hcx.GetSddcByID(client, sddcID)
		} else {
			sddc, err = hcx.GetSddcByName(client, sddc_name)
		}
		if err != nil {
			// Attempt to bypass recurring situation where the HCX API
			// returns status 502 with a proxy server error, and an HTML response
			// instead of JSON.
			errcount += 1
			hclog.Default().Info("[INFO] - resourceVmcCreate() - Error retrieving SDDC status: ", "error", err.Error(), "Errcount:", errcount)
			if errcount > 12 {
				return diag.FromErr(err)
			}
		}

		if sddc.DeploymentStatus == "ACTIVE" {
			break
		}

		if sddc.DeploymentStatus == "ACTIVATION_FAILED" {
			return diag.Errorf("Activation failed")
		}

		time.Sleep(10 * time.Second)
	}

	return resourceVmcRead(ctx, d, m)
}

func resourceVmcRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
	var diags diag.Diagnostics

	client := m.(*hcx.Client)

	token := client.Token
	sddc_name := d.Get("sddc_name").(string)
	sddcID := d.Get("sddc_id").(string)

	log.Printf("******************************************************************\n")
	log.Printf("token: %s, sddc_name: %s,   sddc: %s   \n", token, sddc_name, sddcID)
	log.Printf("******************************************************************\n")

	if sddc_name == "" && sddcID == "" {
		return diag.Errorf("SDDC name or Id must be specified")
	}

	// Authenticate with VMware Cloud Services
	access_token, err := hcx.VmcAuthenticate(token)
	if err != nil {
		return diag.FromErr(err)
	}

	err = hcx.HcxCloudAuthenticate(client, access_token)
	if err != nil {
		return diag.FromErr(err)
	}

	log.Printf("****************")
	log.Printf("[Client inside]: %+v ", client)
	log.Printf("****************")

	var sddc hcx.SDDC
	if sddcID != "" {
		sddc, err = hcx.GetSddcByID(client, sddcID)
	} else {
		sddc, err = hcx.GetSddcByName(client, sddc_name)
	}
	if err != nil {
		return diag.FromErr(err)
	}

	d.SetId(sddc.ID)
	d.Set("cloud_url", sddc.CloudURL)
	d.Set("cloud_name", sddc.CloudName)
	d.Set("cloud_type", sddc.CloudType)

	return diags
}

func resourceVmcUpdate(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {

	return resourceVmcRead(ctx, d, m)
}

func resourceVmcDelete(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
	var diags diag.Diagnostics

	client := m.(*hcx.Client)

	token := client.Token
	sddc_name := d.Get("sddc_name").(string)
	sddcID := d.Get("sddc_id").(string)

	// Authenticate with VMware Cloud Services
	access_token, err := hcx.VmcAuthenticate(token)
	if err != nil {
		return diag.FromErr(err)
	}

	err = hcx.HcxCloudAuthenticate(client, access_token)
	if err != nil {
		return diag.FromErr(err)
	}

	var sddc hcx.SDDC
	if sddcID != "" {
		sddc, err = hcx.GetSddcByID(client, sddcID)
	} else {
		sddc, err = hcx.GetSddcByName(client, sddc_name)
	}
	if err != nil {
		return diag.FromErr(err)
	}

	// Deactivate HCX
	_, err = hcx.DeactivateHcxOnSDDC(client, sddc.ID)
	if err != nil {
		return diag.FromErr(err)
	}

	// Wait for task to be completed
	errcount := 0
	for {
		var sddc hcx.SDDC
		if sddcID != "" {
			sddc, err = hcx.GetSddcByID(client, sddcID)
		} else {
			sddc, err = hcx.GetSddcByName(client, sddc_name)
		}
		if err != nil {
			// Attempt to bypass recurring situation where the HCX API
			// returns status 502 with a proxy server error, and an HTML response
			// instead of JSON.
			errcount += 1
			hclog.Default().Info("[INFO] - resourceVmcDelete() - Error retrieving SDDC status: ", "error", err.Error(), "Errcount:", errcount)
			if errcount > 12 {
				return diag.FromErr(err)
			}
		}

		if sddc.DeploymentStatus == "DE-ACTIVATED" {
			break
		}

		if sddc.DeploymentStatus == "DEACTIVATION_FAILED" {
			return diag.Errorf("Deactivation failed")
		}

		time.Sleep(10 * time.Second)
	}

	return diags
}