provider.go

package digitalocean

import (
	"github.com/hashicorp/terraform/helper/schema"
	"github.com/hashicorp/terraform/terraform"
)

// Provider returns a schema.Provider for DigitalOcean.
func Provider() terraform.ResourceProvider {
	return &schema.Provider{
		Schema: map[string]*schema.Schema{
			"token": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				DefaultFunc: schema.EnvDefaultFunc("DIGITALOCEAN_TOKEN", nil),
				Description: "The token key for API operations.",
			},
		},

		ResourcesMap: map[string]*schema.Resource{
			"digitalocean_domain":      resourceDigitalOceanDomain(),
			"digitalocean_droplet":     resourceDigitalOceanDroplet(),
			"digitalocean_floating_ip": resourceDigitalOceanFloatingIp(),
			"digitalocean_record":      resourceDigitalOceanRecord(),
			"digitalocean_ssh_key":     resourceDigitalOceanSSHKey(),
			"digitalocean_tag":         resourceDigitalOceanTag(),
			"digitalocean_volume":      resourceDigitalOceanVolume(),
		},

		ConfigureFunc: providerConfigure,
	}
}

func providerConfigure(d *schema.ResourceData) (interface{}, error) {
	config := Config{
		Token: d.Get("token").(string),
	}

	return config.Client()
}