INTMDB-346: Feature add: add support for programmatic API keys

mongodbatlas/data_source_mongodbatlas_accesslist_api_key.go

@@ -0,0 +1,89 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
+)
+
+func dataSourceMongoDBAtlasAccessListAPIKey() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceMongoDBAtlasAccessListAPIKeyRead,
+		Schema: map[string]*schema.Schema{
+			"org_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"api_key_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"ip_address": {
+				Type:         schema.TypeString,
+				Required:     true,
+				ValidateFunc: validation.IsIPAddress,
+			},
+			"cidr_block": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"created": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"access_count": {
+				Type:     schema.TypeInt,
+				Computed: true,
+			},
+			"last_used": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"last_used_address": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+		},
+	}
+}
+
+func dataSourceMongoDBAtlasAccessListAPIKeyRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	// Get client connection.
+	conn := meta.(*MongoDBClient).Atlas
+
+	orgID := d.Get("org_id").(string)
+	apiKeyID := d.Get("api_key_id").(string)
+	ipAddress := d.Get("ip_address").(string)
+	accessListAPIKey, _, err := conn.AccessListAPIKeys.Get(ctx, orgID, apiKeyID, ipAddress)
+	if err != nil {
+		return diag.FromErr(fmt.Errorf("error getting access list api key information: %s", err))
+	}
+
+	if err := d.Set("cidr_block", accessListAPIKey.CidrBlock); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `cidr_block`: %s", err))
+	}
+
+	if err := d.Set("last_used_address", accessListAPIKey.LastUsedAddress); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `last_used_address`: %s", err))
+	}
+
+	if err := d.Set("last_used", accessListAPIKey.LastUsed); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `last_used`: %s", err))
+	}
+
+	if err := d.Set("created", accessListAPIKey.Created); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `created`: %s", err))
+	}
+
+	if err := d.Set("access_count", accessListAPIKey.Count); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `access_count`: %s", err))
+	}
+
+	d.SetId(resource.UniqueId())
+
+	return nil
+}

mongodbatlas/data_source_mongodbatlas_accesslist_api_key_test.go

@@ -0,0 +1,63 @@
+package mongodbatlas
+
+import (
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/acctest"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccConfigDSAccesslistAPIKey_basic(t *testing.T) {
+	resourceName := "mongodbatlas_access_list_api_key.test"
+	dataSourceName := "data.mongodbatlas_access_list_api_key.test"
+	orgID := os.Getenv("MONGODB_ATLAS_ORG_ID")
+	description := fmt.Sprintf("test-acc-accesslist-api_key-%s", acctest.RandString(5))
+	ipAddress := fmt.Sprintf("179.154.226.%d", acctest.RandIntRange(0, 255))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:          func() { testAccPreCheck(t) },
+		ProviderFactories: testAccProviderFactories,
+		//CheckDestroy:      testAccCheckMongoDBAtlasNetworkPeeringDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDSMongoDBAtlasAccesslistAPIKeyConfig(orgID, description, ipAddress),
+				Check: resource.ComposeTestCheckFunc(
+					// Test for Resource
+					testAccCheckMongoDBAtlasAccessListAPIKeyExists(resourceName),
+					resource.TestCheckResourceAttrSet(resourceName, "org_id"),
+					resource.TestCheckResourceAttrSet(resourceName, "ip_address"),
+					resource.TestCheckResourceAttr(resourceName, "org_id", orgID),
+					resource.TestCheckResourceAttr(resourceName, "ip_address", ipAddress),
+					// Test for Data source
+					resource.TestCheckResourceAttrSet(dataSourceName, "org_id"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "ip_address"),
+					resource.TestCheckResourceAttr(dataSourceName, "ip_address", ipAddress),
+				),
+			},
+		},
+	})
+}
+
+func testAccDSMongoDBAtlasAccesslistAPIKeyConfig(orgID, description, ipAddress string) string {
+	return fmt.Sprintf(`
+	data "mongodbatlas_access_list_api_key" "test" {
+		org_id     = %[1]q
+		api_key_id = mongodbatlas_access_list_api_key.test.api_key_id
+		ip_address = %[3]q
+	  }
+
+	  resource "mongodbatlas_api_key" "test" {
+		org_id = %[1]q
+		description = %[2]q
+		role_names  = ["ORG_MEMBER","ORG_BILLING_ADMIN"]
+	  }
+
+	  resource "mongodbatlas_access_list_api_key" "test" {
+		org_id     = %[1]q
+		ip_address = %[3]q
+	    api_key_id = mongodbatlas_api_key.test.api_key_id
+	  }
+	`, orgID, description, ipAddress)
+}

mongodbatlas/data_source_mongodbatlas_accesslist_api_keys.go

@@ -0,0 +1,93 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+
+	matlas "go.mongodb.org/atlas/mongodbatlas"
+)
+
+func dataSourceMongoDBAtlasAccessListAPIKeys() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceMongoDBAtlasAccessListAPIKeysRead,
+		Schema: map[string]*schema.Schema{
+			"org_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"api_key_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"page_num": {
+				Type:     schema.TypeInt,
+				Optional: true,
+			},
+			"items_per_page": {
+				Type:     schema.TypeInt,
+				Optional: true,
+			},
+			"results": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"ip_address": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"cidr_block": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"created": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"access_count": {
+							Type:     schema.TypeInt,
+							Computed: true,
+						},
+						"last_used": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"last_used_address": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func dataSourceMongoDBAtlasAccessListAPIKeysRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	// Get client connection.
+	conn := meta.(*MongoDBClient).Atlas
+	options := &matlas.ListOptions{
+		PageNum:      d.Get("page_num").(int),
+		ItemsPerPage: d.Get("items_per_page").(int),
+	}
+
+	orgID := d.Get("org_id").(string)
+	apiKeyID := d.Get("api_key_id").(string)
+
+	accessListAPIKeys, _, err := conn.AccessListAPIKeys.List(ctx, orgID, apiKeyID, options)
+	if err != nil {
+		return diag.FromErr(fmt.Errorf("error getting access list api keys information: %s", err))
+	}
+
+	if err := d.Set("results", flattenAccessListAPIKeys(ctx, conn, orgID, accessListAPIKeys.Results)); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `results`: %s", err))
+	}
+
+	d.SetId(resource.UniqueId())
+
+	return nil
+}

mongodbatlas/data_source_mongodbatlas_accesslist_api_keys_test.go

@@ -0,0 +1,61 @@
+package mongodbatlas
+
+import (
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/acctest"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccConfigDSAccesslistAPIKeys_basic(t *testing.T) {
+	resourceName := "mongodbatlas_access_list_api_key.test"
+	dataSourceName := "data.mongodbatlas_access_list_api_keys.test"
+	orgID := os.Getenv("MONGODB_ATLAS_ORG_ID")
+	description := fmt.Sprintf("test-acc-accesslist-api_keys-%s", acctest.RandString(5))
+	ipAddress := fmt.Sprintf("179.154.226.%d", acctest.RandIntRange(0, 255))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:          func() { testAccPreCheck(t) },
+		ProviderFactories: testAccProviderFactories,
+		//CheckDestroy:      testAccCheckMongoDBAtlasNetworkPeeringDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDSMongoDBAtlasAccesslistAPIKeysConfig(orgID, description, ipAddress),
+				Check: resource.ComposeTestCheckFunc(
+					// Test for Resource
+					testAccCheckMongoDBAtlasAccessListAPIKeyExists(resourceName),
+					resource.TestCheckResourceAttrSet(resourceName, "org_id"),
+					resource.TestCheckResourceAttrSet(resourceName, "ip_address"),
+					resource.TestCheckResourceAttr(resourceName, "org_id", orgID),
+					resource.TestCheckResourceAttr(resourceName, "ip_address", ipAddress),
+					// Test for Data source
+					resource.TestCheckResourceAttrSet(dataSourceName, "org_id"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "results.#"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDSMongoDBAtlasAccesslistAPIKeysConfig(orgID, description, ipAddress string) string {
+	return fmt.Sprintf(`
+	data "mongodbatlas_access_list_api_keys" "test" {
+		org_id     = %[1]q
+		api_key_id = mongodbatlas_access_list_api_key.test.api_key_id
+	  }
+
+	  resource "mongodbatlas_api_key" "test" {
+		org_id = %[1]q
+		description = %[2]q
+		role_names  = ["ORG_MEMBER","ORG_BILLING_ADMIN"]
+	  }
+
+	  resource "mongodbatlas_access_list_api_key" "test" {
+		org_id     = %[1]q
+		ip_address = %[3]q
+	    api_key_id = mongodbatlas_api_key.test.api_key_id
+	  }
+	`, orgID, description, ipAddress)
+}

mongodbatlas/data_source_mongodbatlas_api_key.go

@@ -0,0 +1,69 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceMongoDBAtlasAPIKey() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceMongoDBAtlasAPIKeyRead,
+		Schema: map[string]*schema.Schema{
+			"org_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"api_key_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"description": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"public_key": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"role_names": {
+				Type:     schema.TypeSet,
+				Computed: true,
+				Elem: &schema.Schema{
+					Type: schema.TypeString,
+				},
+			},
+		},
+	}
+}
+
+func dataSourceMongoDBAtlasAPIKeyRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	// Get client connection.
+	conn := meta.(*MongoDBClient).Atlas
+
+	orgID := d.Get("org_id").(string)
+	apiKeyID := d.Get("api_key_id").(string)
+	apiKey, _, err := conn.APIKeys.Get(ctx, orgID, apiKeyID)
+	if err != nil {
+		return diag.FromErr(fmt.Errorf("error getting api key information: %s", err))
+	}
+
+	if err := d.Set("description", apiKey.Desc); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `description`: %s", err))
+	}
+
+	if err := d.Set("public_key", apiKey.PublicKey); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `public_key`: %s", err))
+	}
+
+	if err := d.Set("role_names", flattenOrgAPIKeyRoles(orgID, apiKey.Roles)); err != nil {
+		return diag.FromErr(fmt.Errorf("error setting `roles`: %s", err))
+	}
+
+	d.SetId(resource.UniqueId())
+
+	return nil
+}