github.com/mohanarpit/terraform@v0.6.16-0.20160909104007-291f29853544/builtin/providers/cobbler/provider.go

package cobbler

import (
	"os"

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

func Provider() terraform.ResourceProvider {
	return &schema.Provider{
		Schema: map[string]*schema.Schema{
			"url": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				Description: "Cobbler URL",
				DefaultFunc: envDefaultFunc("COBBLER_URL"),
			},

			"username": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				Description: "The username for accessing Cobbler.",
				DefaultFunc: envDefaultFunc("COBBLER_USERNAME"),
			},

			"password": &schema.Schema{
				Type:        schema.TypeString,
				Required:    true,
				Description: "The password for accessing Cobbler.",
				DefaultFunc: envDefaultFunc("COBBLER_PASSWORD"),
			},
		},

		ResourcesMap: map[string]*schema.Resource{
			"cobbler_distro":         resourceDistro(),
			"cobbler_kickstart_file": resourceKickstartFile(),
			"cobbler_profile":        resourceProfile(),
			"cobbler_snippet":        resourceSnippet(),
			"cobbler_system":         resourceSystem(),
		},

		ConfigureFunc: configureProvider,
	}
}

func configureProvider(d *schema.ResourceData) (interface{}, error) {
	config := Config{
		Url:      d.Get("url").(string),
		Username: d.Get("username").(string),
		Password: d.Get("password").(string),
	}

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

	return &config, nil
}

func envDefaultFunc(k string) schema.SchemaDefaultFunc {
	return func() (interface{}, error) {
		if v := os.Getenv(k); v != "" {
			return v, nil
		}

		return nil, nil
	}
}

func envDefaultFuncAllowMissing(k string) schema.SchemaDefaultFunc {
	return func() (interface{}, error) {
		v := os.Getenv(k)
		return v, nil
	}
}