terraform/builtin/providers/triton/provider.go

package triton

import (
	"fmt"
	"log"
	"os"

	"github.com/hashicorp/go-multierror"
	"github.com/hashicorp/terraform/helper/schema"
	"github.com/hashicorp/terraform/terraform"
	"github.com/joyent/gocommon/client"
	"github.com/joyent/gosdc/cloudapi"
	"github.com/joyent/gosign/auth"
)

// Provider returns a terraform.ResourceProvider.
func Provider() terraform.ResourceProvider {
	return &schema.Provider{
		Schema: map[string]*schema.Schema{
			"account": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				DefaultFunc: schema.EnvDefaultFunc("SDC_ACCOUNT", ""),
			},

			"url": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				DefaultFunc: schema.EnvDefaultFunc("SDC_URL", "https://us-west-1.api.joyentcloud.com"),
			},

			"key_material": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				DefaultFunc: schema.EnvDefaultFunc("SDC_KEY_MATERIAL", ""),
			},

			"key_id": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				DefaultFunc: schema.EnvDefaultFunc("SDC_KEY_ID", ""),
			},
		},

		ResourcesMap: map[string]*schema.Resource{
			"triton_firewall_rule": resourceFirewallRule(),
			"triton_machine":       resourceMachine(),
			"triton_key":           resourceKey(),
			"triton_vlan":          resourceVLAN(),
		},
		ConfigureFunc: providerConfigure,
	}
}

type SDCConfig struct {
	Account     string
	KeyMaterial string
	KeyID       string
	URL         string
}

func (c SDCConfig) validate() error {
	var err *multierror.Error

	if c.URL == "" {
		err = multierror.Append(err, fmt.Errorf("URL must be configured for the Triton provider"))
	}
	if c.KeyMaterial == "" {
		err = multierror.Append(err, fmt.Errorf("Key Material must be configured for the Triton provider"))
	}
	if c.KeyID == "" {
		err = multierror.Append(err, fmt.Errorf("Key ID must be configured for the Triton provider"))
	}
	if c.Account == "" {
		err = multierror.Append(err, fmt.Errorf("Account must be configured for the Triton provider"))
	}

	return err.ErrorOrNil()
}

func (c SDCConfig) getSDCClient() (*cloudapi.Client, error) {
	userauth, err := auth.NewAuth(c.Account, c.KeyMaterial, "rsa-sha256")
	if err != nil {
		return nil, err
	}

	creds := &auth.Credentials{
		UserAuthentication: userauth,
		SdcKeyId:           c.KeyID,
		SdcEndpoint:        auth.Endpoint{URL: c.URL},
	}

	client := cloudapi.New(client.NewClient(
		c.URL,
		cloudapi.DefaultAPIVersion,
		creds,
		log.New(os.Stderr, "", log.LstdFlags),
	))

	return client, nil
}

func providerConfigure(d *schema.ResourceData) (interface{}, error) {
	config := SDCConfig{
		Account:     d.Get("account").(string),
		URL:         d.Get("url").(string),
		KeyMaterial: d.Get("key_material").(string),
		KeyID:       d.Get("key_id").(string),
	}

	if err := config.validate(); err != nil {
		return nil, err
	}

	client, err := config.getSDCClient()
	if err != nil {
		return nil, err
	}

	return client, nil
}