package scaleway

import (
	"fmt"

	"github.com/hashicorp/terraform/helper/schema"
	"github.com/scaleway/scaleway-cli/pkg/api"
)

func dataSourceScalewayImage() *schema.Resource {
	return &schema.Resource{
		Read: dataSourceScalewayImageRead,

		Schema: map[string]*schema.Schema{
			"name": &schema.Schema{
				Type:     schema.TypeString,
				Optional: true,
				ForceNew: true,
				Computed: true,
			},
			"name_filter": &schema.Schema{
				Type:     schema.TypeString,
				Optional: true,
				ForceNew: true,
			},
			"architecture": &schema.Schema{
				Type:     schema.TypeString,
				Required: true,
				ForceNew: true,
			},
			// Computed values.
			"organization": &schema.Schema{
				Type:     schema.TypeString,
				Computed: true,
			},
			"public": &schema.Schema{
				Type:     schema.TypeBool,
				Computed: true,
			},
			"creation_date": &schema.Schema{
				Type:     schema.TypeString,
				Computed: true,
			},
		},
	}
}

func scalewayImageAttributes(d *schema.ResourceData, img *api.ScalewayImage) error {
	d.Set("architecture", img.Arch)
	d.Set("organization", img.Organization)
	d.Set("public", img.Public)
	d.Set("creation_date", img.CreationDate)
	d.Set("name", img.Name)
	d.SetId(img.Identifier)

	return nil
}

func dataSourceScalewayImageRead(d *schema.ResourceData, meta interface{}) error {
	scaleway := meta.(*Client).scaleway

	var needle string
	if name, ok := d.GetOk("name"); ok {
		needle = name.(string)
	} else if nameFilter, ok := d.GetOk("name_filter"); ok {
		needle = nameFilter.(string)
	}

	images, err := scaleway.ResolveImage(needle)
	if err != nil {
		return err
	}
	images = api.FilterImagesByArch(images, d.Get("architecture").(string))
	images = api.FilterImagesByRegion(images, scaleway.Region)

	if len(images) > 1 {
		return fmt.Errorf("The query returned more than one result. Please refine your query.")
	}
	if len(images) == 0 {
		return fmt.Errorf("The query returned no result. Please refine your query.")
	}

	img, err := scaleway.GetImage(images[0].Identifier)
	if err != nil {
		return err
	}

	return scalewayImageAttributes(d, img)
}