Add Organisation and Project invitations

mongodbatlas/data_source_mongodbatlas_org_invitation.go

@@ -0,0 +1,97 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceMongoDBAtlasOrgInvitation() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceMongoDBAtlasOrgInvitationRead,
+		Schema: map[string]*schema.Schema{
+			"org_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"username": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"invitation_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"expires_at": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"created_at": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"inviter_username": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"roles": {
+				Type:     schema.TypeSet,
+				Optional: true,
+				Elem: &schema.Schema{
+					Type: schema.TypeString,
+				},
+			},
+		},
+	}
+}
+
+func dataSourceMongoDBAtlasOrgInvitationRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	// Get client connection.
+	conn := meta.(*MongoDBClient).Atlas
+	orgID := d.Get("org_id").(string)
+	username := d.Get("username").(string)
+	invitationID := d.Get("invitation_id").(string)
+
+	orgInvitation, _, err := conn.Organizations.Invitation(ctx, orgID, invitationID)
+	if err != nil {
+		return diag.FromErr(fmt.Errorf("error getting Organization Invitation information: %w", err))
+	}
+
+	if err := d.Set("username", orgInvitation.Username); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `username` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("org_id", orgInvitation.GroupID); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `username` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("invitation_id", orgInvitation.ID); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `invitation_id` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("expires_at", orgInvitation.ExpiresAt); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `expires_at` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("created_at", orgInvitation.CreatedAt); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `created_at` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("inviter_username", orgInvitation.InviterUsername); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `inviter_username` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("roles", orgInvitation.Roles); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `roles` for Organization Invitation (%s): %w", d.Id(), err))
+	}
+
+	d.SetId(encodeStateID(map[string]string{
+		"username":      username,
+		"org_id":        orgID,
+		"invitation_id": invitationID,
+	}))
+
+	return nil
+}

mongodbatlas/data_source_mongodbatlas_org_invitation_test.go

@@ -0,0 +1,54 @@
+package mongodbatlas
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/acctest"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccDataSourceMongoDBAtlasOrgInvitation_basic(t *testing.T) {
+	var (
+		dataSourceName = "mongodbatlas_org_invitations.test"
+		orgID          = os.Getenv("MONGODB_ATLAS_ORG_ID")
+		name           = fmt.Sprintf("test-acc-%[email protected]", acctest.RandString(10))
+		initialRole    = []string{"ORG_OWNER"}
+	)
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:          func() { testAccPreCheck(t) },
+		ProviderFactories: testAccProviderFactories,
+		CheckDestroy:      testAccCheckMongoDBAtlasOrgInvitationDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceMongoDBAtlasOrgInvitationConfig(orgID, name, initialRole),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrSet(dataSourceName, "org_id"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "username"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "invitation_id"),
+					resource.TestCheckResourceAttr(dataSourceName, "username", name),
+					resource.TestCheckResourceAttr(dataSourceName, "roles.#", "1"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceMongoDBAtlasOrgInvitationConfig(orgID, username string, roles []string) string {
+	return fmt.Sprintf(`
+		resource "mongodbatlas_org_invitation" "test" {
+			org_id 		= "%s"
+			username  = "%s"
+			roles  		= %s
+		}
+
+		data "mongodbatlas_org_invitation" "test" {
+			org_id   = mongodbatlas_org_invitation.test.org_id
+			username = mongodbatlas_org_invitation.test.username
+		}`, orgID, username,
+		strings.Join(roles, ","),
+	)
+}

mongodbatlas/data_source_mongodbatlas_project_invitation.go

@@ -0,0 +1,97 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceMongoDBAtlasProjectInvitation() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceMongoDBAtlasProjectInvitationRead,
+		Schema: map[string]*schema.Schema{
+			"project_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"username": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"invitation_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"expires_at": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"created_at": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"inviter_username": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"roles": {
+				Type:     schema.TypeSet,
+				Optional: true,
+				Elem: &schema.Schema{
+					Type: schema.TypeString,
+				},
+			},
+		},
+	}
+}
+
+func dataSourceMongoDBAtlasProjectInvitationRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	// Get client connection.
+	conn := meta.(*MongoDBClient).Atlas
+	projectID := d.Get("project_id").(string)
+	username := d.Get("username").(string)
+	invitationID := d.Get("invitation_id").(string)
+
+	projectInvitation, _, err := conn.Projects.Invitation(ctx, projectID, invitationID)
+	if err != nil {
+		return diag.FromErr(fmt.Errorf("error getting Project Invitation information: %w", err))
+	}
+
+	if err := d.Set("username", projectInvitation.Username); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `username` for Project Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("project_id", projectInvitation.GroupID); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `username` for Project Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("invitation_id", projectInvitation.ID); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `invitation_id` for Project Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("expires_at", projectInvitation.ExpiresAt); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `expires_at` for Project Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("created_at", projectInvitation.CreatedAt); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `created_at` for Project Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("inviter_username", projectInvitation.InviterUsername); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `inviter_username` for Project Invitation (%s): %w", d.Id(), err))
+	}
+
+	if err := d.Set("roles", projectInvitation.Roles); err != nil {
+		return diag.FromErr(fmt.Errorf("error getting `roles` for Project Invitation (%s): %s", d.Id(), err))
+	}
+
+	d.SetId(encodeStateID(map[string]string{
+		"username":      username,
+		"project_id":    projectID,
+		"invitation_id": invitationID,
+	}))
+
+	return nil
+}

mongodbatlas/data_source_mongodbatlas_project_invitation_test.go

@@ -0,0 +1,54 @@
+package mongodbatlas
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/acctest"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccDataSourceMongoDBAtlasProjectInvitation_basic(t *testing.T) {
+	var (
+		dataSourceName = "mongodbatlas_project_invitations.test"
+		projectID      = os.Getenv("MONGODB_ATLAS_PROJECT_ID")
+		name           = fmt.Sprintf("test-acc-%[email protected]", acctest.RandString(10))
+		initialRole    = []string{"GROUP_OWNER"}
+	)
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:          func() { testAccPreCheck(t) },
+		ProviderFactories: testAccProviderFactories,
+		CheckDestroy:      testAccCheckMongoDBAtlasProjectInvitationDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceMongoDBAtlasProjectInvitationConfig(projectID, name, initialRole),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrSet(dataSourceName, "project_id"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "username"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "invitation_id"),
+					resource.TestCheckResourceAttr(dataSourceName, "username", name),
+					resource.TestCheckResourceAttr(dataSourceName, "roles.#", "1"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceMongoDBAtlasProjectInvitationConfig(projectID, username string, roles []string) string {
+	return fmt.Sprintf(`
+		resource "mongodbatlas_project_invitation" "test" {
+			project_id	= "%s"
+			username  	= "%s"
+			roles  			= %s
+		}
+
+		data "mongodbatlas_project_invitation" "test" {
+			project_id = mongodbatlas_project_invitation.test.projectID
+			username 	 = mongodbatlas_project_invitation.test.username
+		}`, projectID, username,
+		strings.Join(roles, ","),
+	)
+}

mongodbatlas/provider.go

@@ -100,6 +100,8 @@ func Provider() *schema.Provider {
 			"mongodbatlas_data_lakes":                            dataSourceMongoDBAtlasDataLakes(),
 			"mongodbatlas_event_trigger":                         dataSourceMongoDBAtlasEventTrigger(),
 			"mongodbatlas_event_triggers":                        dataSourceMongoDBAtlasEventTriggers(),
+			"mongodbatlas_project_invitation":                    dataSourceMongoDBAtlasProjectInvitation(),
+			"mongodbatlas_org_invitation":                        dataSourceMongoDBAtlasOrgInvitation(),
 		},
 
 		ResourcesMap: map[string]*schema.Resource{
@@ -136,6 +138,8 @@ func Provider() *schema.Provider {
 			"mongodbatlas_data_lake":                             resourceMongoDBAtlasDataLake(),
 			"mongodbatlas_event_trigger":                         resourceMongoDBAtlasEventTriggers(),
 			"mongodbatlas_cloud_backup_schedule":                 resourceMongoDBAtlasCloudBackupSchedule(),
+			"mongodbatlas_project_invitation":                    resourceMongoDBAtlasProjectInvitation(),
+			"mongodbatlas_org_invitation":                        resourceMongoDBAtlasOrgInvitation(),
 		},
 
 		ConfigureContextFunc: providerConfigure,