package google

import (
	"fmt"
	"log"
	"time"

	"github.com/hashicorp/terraform/helper/schema"
	"google.golang.org/api/compute/v1"
)

const computeImageCreateTimeoutDefault = 4

func resourceComputeImage() *schema.Resource {
	return &schema.Resource{
		Create: resourceComputeImageCreate,
		Read:   resourceComputeImageRead,
		Update: resourceComputeImageUpdate,
		Delete: resourceComputeImageDelete,
		Importer: &schema.ResourceImporter{
			State: schema.ImportStatePassthrough,
		},

		Timeouts: &schema.ResourceTimeout{
			Create: schema.DefaultTimeout(computeImageCreateTimeoutDefault * time.Minute),
			Update: schema.DefaultTimeout(computeImageCreateTimeoutDefault * time.Minute),
			Delete: schema.DefaultTimeout(computeImageCreateTimeoutDefault * time.Minute),
		},

		Schema: map[string]*schema.Schema{
			// TODO(cblecker): one of source_disk or raw_disk is required

			"name": {
				Type:     schema.TypeString,
				Required: true,
				ForceNew: true,
			},

			"description": {
				Type:     schema.TypeString,
				Optional: true,
				ForceNew: true,
			},

			"family": {
				Type:     schema.TypeString,
				Optional: true,
				ForceNew: true,
			},

			"project": {
				Type:     schema.TypeString,
				Optional: true,
				Computed: true,
				ForceNew: true,
			},

			"source_disk": {
				Type:     schema.TypeString,
				Optional: true,
				ForceNew: true,
			},

			"raw_disk": {
				Type:     schema.TypeList,
				Optional: true,
				ForceNew: true,
				MaxItems: 1,
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"source": {
							Type:     schema.TypeString,
							Required: true,
							ForceNew: true,
						},
						"sha1": {
							Type:     schema.TypeString,
							Optional: true,
							ForceNew: true,
						},
						"container_type": {
							Type:     schema.TypeString,
							Optional: true,
							Default:  "TAR",
							ForceNew: true,
						},
					},
				},
			},

			"self_link": {
				Type:     schema.TypeString,
				Computed: true,
			},

			"create_timeout": {
				Type:     schema.TypeInt,
				Optional: true,
				Removed:  "Use timeouts block instead. See https://www.terraform.io/docs/configuration/resources.html#timeouts.",
			},

			"labels": {
				Type:     schema.TypeMap,
				Optional: true,
				Elem:     &schema.Schema{Type: schema.TypeString},
				Set:      schema.HashString,
			},

			"licenses": {
				Type:     schema.TypeList,
				Optional: true,
				ForceNew: true,
				Elem:     &schema.Schema{Type: schema.TypeString},
				Computed: true,
			},

			"label_fingerprint": {
				Type:     schema.TypeString,
				Computed: true,
			},
		},
	}
}

func resourceComputeImageCreate(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	// Build the image
	image := &compute.Image{
		Name: d.Get("name").(string),
	}

	if v, ok := d.GetOk("description"); ok {
		image.Description = v.(string)
	}

	if v, ok := d.GetOk("family"); ok {
		image.Family = v.(string)
	}

	// Load up the source_disk for this image if specified
	if v, ok := d.GetOk("source_disk"); ok {
		image.SourceDisk = v.(string)
	}

	// Load up the raw_disk for this image if specified
	if v, ok := d.GetOk("raw_disk"); ok {
		rawDiskEle := v.([]interface{})[0].(map[string]interface{})
		imageRawDisk := &compute.ImageRawDisk{
			Source:        rawDiskEle["source"].(string),
			ContainerType: rawDiskEle["container_type"].(string),
		}
		if val, ok := rawDiskEle["sha1"]; ok {
			imageRawDisk.Sha1Checksum = val.(string)
		}

		image.RawDisk = imageRawDisk
	}

	if _, ok := d.GetOk("labels"); ok {
		image.Labels = expandLabels(d)
	}

	// Load up the licenses for this image if specified
	if _, ok := d.GetOk("licenses"); ok {
		image.Licenses = licenses(d)
	}

	// Read create timeout
	createTimeout := int(d.Timeout(schema.TimeoutCreate).Minutes())

	// Insert the image
	op, err := config.clientCompute.Images.Insert(
		project, image).Do()
	if err != nil {
		return fmt.Errorf("Error creating image: %s", err)
	}

	// Store the ID
	d.SetId(image.Name)

	err = computeOperationWaitTime(config.clientCompute, op, project, "Creating Image", createTimeout)
	if err != nil {
		return err
	}

	return resourceComputeImageRead(d, meta)
}

func resourceComputeImageRead(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	image, err := config.clientCompute.Images.Get(
		project, d.Id()).Do()
	if err != nil {
		return handleNotFoundError(err, d, fmt.Sprintf("Image %q", d.Get("name").(string)))
	}

	if image.SourceDisk != "" {
		d.Set("source_disk", image.SourceDisk)
	} else if image.RawDisk != nil {
		// `raw_disk.*.source` is only used at image creation but is not returned when calling Get.
		// `raw_disk.*.sha1` is not supported, the value is simply discarded by the server.
		// Leaving `raw_disk` to current state value.
	} else {
		return fmt.Errorf("Either raw_disk or source_disk configuration is required.")
	}

	d.Set("name", image.Name)
	d.Set("description", image.Description)
	d.Set("family", image.Family)
	d.Set("self_link", image.SelfLink)
	d.Set("labels", image.Labels)
	d.Set("licenses", image.Licenses)
	d.Set("label_fingerprint", image.LabelFingerprint)
	d.Set("project", project)

	return nil
}

func resourceComputeImageUpdate(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	// Technically we are only updating one attribute, but setting d.Partial here makes it easier to add updates later
	d.Partial(true)

	if d.HasChange("labels") {
		labels := expandLabels(d)
		labelFingerprint := d.Get("label_fingerprint").(string)
		setLabelsRequest := compute.GlobalSetLabelsRequest{
			LabelFingerprint: labelFingerprint,
			Labels:           labels,
			ForceSendFields:  []string{"Labels"},
		}

		op, err := config.clientCompute.Images.SetLabels(project, d.Id(), &setLabelsRequest).Do()
		if err != nil {
			return err
		}

		d.SetPartial("labels")

		err = computeOperationWaitTime(config.clientCompute, op, project, "Setting labels", int(d.Timeout(schema.TimeoutUpdate).Minutes()))
		if err != nil {
			return err
		}
		// Perform a read to see the new label_fingerprint value
		image, err := config.clientCompute.Images.Get(project, d.Id()).Do()
		if err != nil {
			return err
		}
		d.Set("label_fingerprint", image.LabelFingerprint)
		d.SetPartial("label_fingerprint")
	}

	d.Partial(false)
	return nil
}

func resourceComputeImageDelete(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	// Delete the image
	log.Printf("[DEBUG] image delete request")
	op, err := config.clientCompute.Images.Delete(
		project, d.Id()).Do()
	if err != nil {
		return fmt.Errorf("Error deleting image: %s", err)
	}

	err = computeOperationWaitTime(config.clientCompute, op, project, "Deleting image", int(d.Timeout(schema.TimeoutDelete).Minutes()))
	if err != nil {
		return err
	}

	d.SetId("")
	return nil
}

func licenses(d *schema.ResourceData) []string {
	licensesCount := d.Get("licenses.#").(int)
	data := make([]string, licensesCount)
	for i := 0; i < licensesCount; i++ {
		data[i] = d.Get(fmt.Sprintf("licenses.%d", i)).(string)
	}
	return data
}