134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
package scaleway
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"regexp"
|
|
|
|
"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 imageMatch) error {
|
|
d.Set("architecture", img.imageDefinition.Arch)
|
|
d.Set("organization", img.marketImage.Organization)
|
|
d.Set("public", img.marketImage.Public)
|
|
d.Set("creation_date", img.marketImage.CreationDate)
|
|
d.Set("name", img.marketImage.Name)
|
|
d.SetId(img.imageDefinition.ID)
|
|
|
|
return nil
|
|
}
|
|
|
|
type imageMatch struct {
|
|
marketImage api.MarketImage
|
|
imageDefinition api.MarketLocalImageDefinition
|
|
}
|
|
|
|
func dataSourceScalewayImageRead(d *schema.ResourceData, meta interface{}) error {
|
|
scaleway := meta.(*Client).scaleway
|
|
|
|
images, err := scaleway.GetImages()
|
|
log.Printf("[DEBUG] %#v", images)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var isNameMatch = func(api.MarketImage) bool { return true }
|
|
var isArchMatch = func(api.MarketLocalImageDefinition) bool { return true }
|
|
|
|
if name, ok := d.GetOk("name"); ok {
|
|
isNameMatch = func(img api.MarketImage) bool {
|
|
return img.Name == name.(string)
|
|
}
|
|
} else if nameFilter, ok := d.GetOk("name_filter"); ok {
|
|
exp, err := regexp.Compile(nameFilter.(string))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
isNameMatch = func(img api.MarketImage) bool {
|
|
return exp.MatchString(img.Name)
|
|
}
|
|
}
|
|
|
|
var architecture = d.Get("architecture").(string)
|
|
if architecture != "" {
|
|
isArchMatch = func(img api.MarketLocalImageDefinition) bool {
|
|
return img.Arch == architecture
|
|
}
|
|
}
|
|
|
|
var matches []imageMatch
|
|
for _, img := range *images {
|
|
if !isNameMatch(img) {
|
|
continue
|
|
}
|
|
|
|
var imageDefinition *api.MarketLocalImageDefinition
|
|
for _, version := range img.Versions {
|
|
for _, def := range version.LocalImages {
|
|
if isArchMatch(def) {
|
|
imageDefinition = &def
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if imageDefinition == nil {
|
|
continue
|
|
}
|
|
matches = append(matches, imageMatch{
|
|
marketImage: img,
|
|
imageDefinition: *imageDefinition,
|
|
})
|
|
}
|
|
|
|
if len(matches) > 1 {
|
|
return fmt.Errorf("The query returned more than one result. Please refine your query.")
|
|
}
|
|
if len(matches) == 0 {
|
|
return fmt.Errorf("The query returned no result. Please refine your query.")
|
|
}
|
|
|
|
return scalewayImageAttributes(d, matches[0])
|
|
}
|