feat: add ssh key data source (scaleway#262)

* feat: add ssh key data source

Author: kindermoumoute

scaleway/data_source_account_ssh.go

@@ -0,0 +1,72 @@
+package scaleway
+
+import (
+	"fmt"
+
+	"github.com/hashicorp/terraform/helper/schema"
+	account "github.com/scaleway/scaleway-sdk-go/api/account/v2alpha1"
+)
+
+func dataSourceScalewayAccountSSH() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceScalewayAccountSSHRead,
+
+		Schema: map[string]*schema.Schema{
+			"name": {
+				Type:        schema.TypeString,
+				Optional:    true,
+				Computed:    true,
+				Description: "The name of the SSH key",
+			},
+			"ssh_key_id": {
+				Type:         schema.TypeString,
+				Optional:     true,
+				Computed:     true,
+				Description:  "The name of the SSH key",
+				ValidateFunc: validationUUIDorUUIDWithLocality(),
+			},
+			"public_key": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "The public SSH key",
+			},
+			"organization_id": organizationIDSchema(),
+		},
+	}
+}
+
+func dataSourceScalewayAccountSSHRead(d *schema.ResourceData, m interface{}) error {
+	accountAPI := getAccountAPI(m)
+
+	var sshKey *account.SSHKey
+	sshKeyID, ok := d.GetOk("ssh_key_id")
+	if ok {
+		res, err := accountAPI.GetSSHKey(&account.GetSSHKeyRequest{SSHKeyID: expandID(sshKeyID)})
+		if err != nil {
+			return err
+		}
+		sshKey = res
+	} else {
+		res, err := accountAPI.ListSSHKeys(&account.ListSSHKeysRequest{
+			Name: String(d.Get("name").(string)),
+		})
+		if err != nil {
+			return err
+		}
+		if len(res.SSHKeys) == 0 {
+			return fmt.Errorf("no SSH Key found with the name %s", d.Get("name"))
+		}
+		if len(res.SSHKeys) > 1 {
+			return fmt.Errorf("%d SSH Keys found with the same name %s", len(res.SSHKeys), d.Get("name"))
+		}
+		sshKey = res.SSHKeys[0]
+	}
+
+	d.SetId(sshKey.ID)
+	d.Set("name", sshKey.Name)
+	d.Set("ssh_key_id", sshKey.ID)
+	d.Set("public_key", sshKey.PublicKey)
+	d.Set("organization_id", sshKey.OrganizationID)
+
+	return nil
+}

scaleway/data_source_account_ssh_test.go

@@ -0,0 +1,48 @@
+package scaleway
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccScalewayDataSourceAccountSSH_Basic(t *testing.T) {
+	sshKeyName := acctest.RandString(10)
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckScalewayAccountSSHKeyDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceScalewayAccountSSHKeyConfig(sshKeyName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckScalewayAccountSSHKeyExists("data.scaleway_account_ssh_key.prod"),
+					resource.TestCheckResourceAttr("data.scaleway_account_ssh_key.prod", "name", sshKeyName),
+					resource.TestCheckResourceAttr("data.scaleway_account_ssh_key.prod", "public_key", dataSourceAccountSSHKey),
+					testAccCheckScalewayAccountSSHKeyExists("data.scaleway_account_ssh_key.stg"),
+					resource.TestCheckResourceAttr("data.scaleway_account_ssh_key.stg", "name", sshKeyName),
+					resource.TestCheckResourceAttr("data.scaleway_account_ssh_key.stg", "public_key", dataSourceAccountSSHKey),
+				),
+			},
+		},
+	})
+}
+
+const dataSourceAccountSSHKey = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDCtMlb2pIlcCC3Exui1z77NfFWtlf59P/L38OwfHCKnxgHZfcjWZZEOdJPxeA6/iRCUrJo+mUKOedNJTCR9Wgg5zhUM1dd/fiyCS+STLf9fzoA1evPEEIHp0iobIpVQ0pGIdNqipL2n8BcV7f1oC2AkELCUp4gogkeUkKtK71DtsjGJQJBRlg01U2gTAnU0Q3kf5wxhCIELke9J3eblTpvdnNonqpXsQuy+InpT51NDtMbBdgkwNsgw6wDxI64NY92mEXO7PK/uhSAmxIM9a4evLhgFxSr8vFNTwqq5fbSyDeeQwHG23U0CmhM0tOwLuJAbQotWrfYsLCpinnBb+sp [email protected]"
+
+func testAccDataSourceScalewayAccountSSHKeyConfig(name string) string {
+	return `
+resource "scaleway_account_ssh_key" "main" {
+	name 	   = "` + name + `"
+	public_key = "` + dataSourceAccountSSHKey + `"
+}
+
+data "scaleway_account_ssh_key" "prod" {
+	name = "${scaleway_account_ssh_key.main.name}"
+}
+
+data "scaleway_account_ssh_key" "stg" {
+	ssh_key_id = "${scaleway_account_ssh_key.main.id}"
+}`
+}

scaleway/provider.go

@@ -209,10 +209,11 @@ func Provider() terraform.ResourceProvider {
 		},
 
 		DataSourcesMap: map[string]*schema.Resource{
-			"scaleway_bootscript":     dataSourceScalewayBootscript(),
-			"scaleway_image":          dataSourceScalewayImage(),
-			"scaleway_security_group": dataSourceScalewaySecurityGroup(),
-			"scaleway_volume":         dataSourceScalewayVolume(),
+			"scaleway_bootscript":      dataSourceScalewayBootscript(),
+			"scaleway_image":           dataSourceScalewayImage(),
+			"scaleway_security_group":  dataSourceScalewaySecurityGroup(),
+			"scaleway_volume":          dataSourceScalewayVolume(),
+			"scaleway_account_ssh_key": dataSourceScalewayAccountSSH(),
 		},
 
 		ConfigureFunc: providerConfigure,