feat(cockpit): add plan datasource (#1939)

Author: quantumsheep

scaleway/data_source_cockpit_plan.go

@@ -0,0 +1,54 @@
+package scaleway
+
+import (
+	"context"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	cockpit "github.com/scaleway/scaleway-sdk-go/api/cockpit/v1beta1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+func dataSourceScalewayCockpitPlan() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceScalewayCockpitPlanRead,
+		Schema: map[string]*schema.Schema{
+			"name": {
+				Type:        schema.TypeString,
+				Description: "The name of the plan",
+				Required:    true,
+			},
+		},
+	}
+}
+
+func dataSourceScalewayCockpitPlanRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	api, err := cockpitAPI(meta)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	name := d.Get("name").(string)
+
+	res, err := api.ListPlans(&cockpit.ListPlansRequest{}, scw.WithContext(ctx), scw.WithAllPages())
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	var plan *cockpit.Plan
+	for _, p := range res.Plans {
+		if p.Name.String() == name {
+			plan = p
+			break
+		}
+	}
+
+	if plan == nil {
+		return diag.Errorf("could not find plan with name %s", name)
+	}
+
+	d.SetId(plan.ID)
+	_ = d.Set("name", plan.Name.String())
+
+	return nil
+}

scaleway/data_source_cockpit_plan_test.go

@@ -0,0 +1,48 @@
+package scaleway
+
+import (
+	"regexp"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccScalewayDataSourceCockpitPlan_Basic(t *testing.T) {
+	tt := NewTestTools(t)
+	defer tt.Cleanup()
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:          func() { testAccPreCheck(t) },
+		ProviderFactories: tt.ProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: `
+					data "scaleway_cockpit_plan" "free" {
+						name = "free"
+					}
+
+					data "scaleway_cockpit_plan" "premium" {
+						name = "premium"
+					}
+
+					data "scaleway_cockpit_plan" "custom" {
+						name = "custom"
+					}
+				`,
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrSet("data.scaleway_cockpit_plan.free", "id"),
+					resource.TestCheckResourceAttrSet("data.scaleway_cockpit_plan.premium", "id"),
+					resource.TestCheckResourceAttrSet("data.scaleway_cockpit_plan.custom", "id"),
+				),
+			},
+			{
+				Config: `
+					data "scaleway_cockpit_plan" "random" {
+						name = "plan? there ain't no plan"
+					}
+				`,
+				ExpectError: regexp.MustCompile("could not find plan"),
+			},
+		},
+	})
+}

scaleway/provider.go

@@ -172,6 +172,7 @@ func Provider(config *ProviderConfig) plugin.ProviderFunc {
 				"scaleway_baremetal_os":                        dataSourceScalewayBaremetalOs(),
 				"scaleway_baremetal_server":                    dataSourceScalewayBaremetalServer(),
 				"scaleway_cockpit":                             dataSourceScalewayCockpit(),
+				"scaleway_cockpit_plan":                        dataSourceScalewayCockpitPlan(),
 				"scaleway_domain_record":                       dataSourceScalewayDomainRecord(),
 				"scaleway_domain_zone":                         dataSourceScalewayDomainZone(),
 				"scaleway_container_namespace":                 dataSourceScalewayContainerNamespace(),