resource_pagerduty_addon.go

package pagerduty

import (
	"log"
	"time"

	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
	"github.com/heimweh/go-pagerduty/pagerduty"
)

func resourcePagerDutyAddon() *schema.Resource {
	return &schema.Resource{
		Create: resourcePagerDutyAddonCreate,
		Read:   resourcePagerDutyAddonRead,
		Update: resourcePagerDutyAddonUpdate,
		Delete: resourcePagerDutyAddonDelete,
		Importer: &schema.ResourceImporter{
			State: schema.ImportStatePassthrough,
		},
		Schema: map[string]*schema.Schema{
			"name": {
				Type:     schema.TypeString,
				Required: true,
			},
			"src": {
				Type:     schema.TypeString,
				Required: true,
			},
		},
	}
}

func buildAddonStruct(d *schema.ResourceData) *pagerduty.Addon {
	addon := &pagerduty.Addon{
		Name: d.Get("name").(string),
		Src:  d.Get("src").(string),
		Type: "full_page_addon",
	}

	return addon
}

func resourcePagerDutyAddonCreate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*pagerduty.Client)

	addon := buildAddonStruct(d)

	log.Printf("[INFO] Creating PagerDuty add-on %s", addon.Name)

	addon, _, err := client.Addons.Install(addon)
	if err != nil {
		return err
	}

	d.SetId(addon.ID)

	return resourcePagerDutyAddonRead(d, meta)
}

func resourcePagerDutyAddonRead(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*pagerduty.Client)

	log.Printf("[INFO] Reading PagerDuty add-on %s", d.Id())

	return resource.Retry(2*time.Minute, func() *resource.RetryError {
		addon, _, err := client.Addons.Get(d.Id())
		if err != nil {
			errResp := handleNotFoundError(err, d)
			if errResp != nil {
				time.Sleep(2 * time.Second)
				return resource.RetryableError(errResp)
			}

			return nil
		}

		d.Set("name", addon.Name)
		d.Set("src", addon.Src)

		return nil
	})
}

func resourcePagerDutyAddonUpdate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*pagerduty.Client)

	addon := buildAddonStruct(d)

	log.Printf("[INFO] Updating PagerDuty add-on %s", d.Id())

	if _, _, err := client.Addons.Update(d.Id(), addon); err != nil {
		return err
	}

	return nil
}

func resourcePagerDutyAddonDelete(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*pagerduty.Client)

	log.Printf("[INFO] Deleting PagerDuty add-on %s", d.Id())

	if _, err := client.Addons.Delete(d.Id()); err != nil {
		return err
	}

	d.SetId("")

	return nil
}