terraform/builtin/providers/spotinst/resource_spotinst_subscription.go

package spotinst

import (
	"fmt"
	"log"
	"strings"

	"github.com/hashicorp/terraform/helper/schema"
	"github.com/spotinst/spotinst-sdk-go/spotinst"
	"github.com/spotinst/spotinst-sdk-go/spotinst/util/stringutil"
)

func resourceSpotinstSubscription() *schema.Resource {
	return &schema.Resource{
		Create: resourceSpotinstSubscriptionCreate,
		Update: resourceSpotinstSubscriptionUpdate,
		Read:   resourceSpotinstSubscriptionRead,
		Delete: resourceSpotinstSubscriptionDelete,

		Schema: map[string]*schema.Schema{
			"resource_id": &schema.Schema{
				Type:     schema.TypeString,
				Required: true,
			},

			"event_type": &schema.Schema{
				Type:     schema.TypeString,
				Required: true,
				StateFunc: func(v interface{}) string {
					value := v.(string)
					return strings.ToUpper(value)
				},
			},

			"protocol": &schema.Schema{
				Type:     schema.TypeString,
				Required: true,
			},

			"endpoint": &schema.Schema{
				Type:     schema.TypeString,
				Required: true,
			},

			"format": &schema.Schema{
				Type:     schema.TypeMap,
				Optional: true,
			},
		},
	}
}

func resourceSpotinstSubscriptionCreate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*spotinst.Client)
	newSubscription, err := buildSubscriptionOpts(d, meta)
	if err != nil {
		return err
	}
	log.Printf("[DEBUG] Subscription create configuration: %s\n", stringutil.Stringify(newSubscription))
	input := &spotinst.CreateSubscriptionInput{Subscription: newSubscription}
	resp, err := client.SubscriptionService.Create(input)
	if err != nil {
		return fmt.Errorf("Error creating subscription: %s", err)
	}
	d.SetId(spotinst.StringValue(resp.Subscription.ID))
	log.Printf("[INFO] Subscription created successfully: %s\n", d.Id())
	return resourceSpotinstSubscriptionRead(d, meta)
}

func resourceSpotinstSubscriptionRead(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*spotinst.Client)
	input := &spotinst.ReadSubscriptionInput{ID: spotinst.String(d.Id())}
	resp, err := client.SubscriptionService.Read(input)
	if err != nil {
		return fmt.Errorf("Error retrieving subscription: %s", err)
	}
	if s := resp.Subscription; s != nil {
		d.Set("resource_id", s.ResourceID)
		d.Set("event_type", s.EventType)
		d.Set("protocol", s.Protocol)
		d.Set("endpoint", s.Endpoint)
		d.Set("format", s.Format)
	} else {
		d.SetId("")
	}
	return nil
}

func resourceSpotinstSubscriptionUpdate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*spotinst.Client)
	subscription := &spotinst.Subscription{ID: spotinst.String(d.Id())}
	update := false

	if d.HasChange("resource_id") {
		subscription.ResourceID = spotinst.String(d.Get("resource_id").(string))
		update = true
	}

	if d.HasChange("event_type") {
		subscription.EventType = spotinst.String(d.Get("event_type").(string))
		update = true
	}

	if d.HasChange("protocol") {
		subscription.Protocol = spotinst.String(d.Get("protocol").(string))
		update = true
	}

	if d.HasChange("endpoint") {
		subscription.Endpoint = spotinst.String(d.Get("endpoint").(string))
		update = true
	}

	if d.HasChange("format") {
		subscription.Format = d.Get("format").(map[string]interface{})
		update = true
	}

	if update {
		log.Printf("[DEBUG] Subscription update configuration: %s\n", stringutil.Stringify(subscription))
		input := &spotinst.UpdateSubscriptionInput{Subscription: subscription}
		if _, err := client.SubscriptionService.Update(input); err != nil {
			return fmt.Errorf("Error updating subscription %s: %s", d.Id(), err)
		}
	}

	return resourceSpotinstSubscriptionRead(d, meta)
}

func resourceSpotinstSubscriptionDelete(d *schema.ResourceData, meta interface{}) error {
	d.SetId("")
	return nil
}

// buildSubscriptionOpts builds the Spotinst Subscription options.
func buildSubscriptionOpts(d *schema.ResourceData, meta interface{}) (*spotinst.Subscription, error) {
	subscription := &spotinst.Subscription{
		ResourceID: spotinst.String(d.Get("resource_id").(string)),
		EventType:  spotinst.String(strings.ToUpper(d.Get("event_type").(string))),
		Protocol:   spotinst.String(d.Get("protocol").(string)),
		Endpoint:   spotinst.String(d.Get("endpoint").(string)),
		Format:     d.Get("format").(map[string]interface{}),
	}
	return subscription, nil
}