package triton

import (
	"errors"
	"strings"

	"github.com/hashicorp/terraform/helper/schema"
	"github.com/joyent/gosdc/cloudapi"
)

var (
	// ErrNoKeyComment will be returned when the key name cannot be generated from
	// the key comment and is not otherwise specified.
	ErrNoKeyComment = errors.New("no key comment found to use as a name (and none specified)")
)

func resourceKey() *schema.Resource {
	return &schema.Resource{
		Create: resourceKeyCreate,
		Exists: resourceKeyExists,
		Read:   resourceKeyRead,
		Delete: resourceKeyDelete,
		Importer: &schema.ResourceImporter{
			State: resourceKeyImporter,
		},

		Schema: map[string]*schema.Schema{
			"name": &schema.Schema{
				Description: "name of this key (will be generated from the key comment, if not set and comment present)",
				Type:        schema.TypeString,
				Optional:    true,
				Computed:    true,
				ForceNew:    true,
			},
			"key": &schema.Schema{
				Description: "content of public key from disk",
				Type:        schema.TypeString,
				Required:    true,
				ForceNew:    true,
			},
		},
	}
}

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

	if d.Get("name").(string) == "" {
		parts := strings.SplitN(d.Get("key").(string), " ", 3)
		if len(parts) == 3 {
			d.Set("name", parts[2])
		} else {
			return ErrNoKeyComment
		}
	}

	_, err := client.CreateKey(cloudapi.CreateKeyOpts{
		Name: d.Get("name").(string),
		Key:  d.Get("key").(string),
	})
	if err != nil {
		return err
	}

	err = resourceKeyRead(d, meta)
	if err != nil {
		return err
	}

	return nil
}

func resourceKeyExists(d *schema.ResourceData, meta interface{}) (bool, error) {
	client := meta.(*cloudapi.Client)

	keys, err := client.ListKeys()
	if err != nil {
		return false, err
	}

	for _, key := range keys {
		if key.Name == d.Id() {
			return true, nil
		}
	}

	return false, nil
}

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

	key, err := client.GetKey(d.Id())
	if err != nil {
		return err
	}

	d.SetId(key.Name)
	d.Set("name", key.Name)
	d.Set("key", key.Key)

	return nil
}

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

	if err := client.DeleteKey(d.Get("name").(string)); err != nil {
		return err
	}

	return nil
}

func resourceKeyImporter(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
	return []*schema.ResourceData{d}, nil
}