Consolidate how we import resource pair IDs across resources

auth0 · May 8, 2023 · f198556 · f198556
1 parent d0573e4
commit f198556
Show file tree

Hide file tree

Showing 10 changed files with 151 additions and 329 deletions.
diff --git a/internal/auth0/connection/resource_client.go b/internal/auth0/connection/resource_client.go
@@ -11,6 +11,7 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 
 	"github.com/auth0/terraform-provider-auth0/internal/mutex"
+	internalSchema "github.com/auth0/terraform-provider-auth0/internal/schema"
 )
 
 var (
@@ -49,7 +50,7 @@ func NewClientResource() *schema.Resource {
 		ReadContext:   readConnectionClient,
 		DeleteContext: deleteConnectionClient,
 		Importer: &schema.ResourceImporter{
-			StateContext: importConnectionClient,
+			StateContext: internalSchema.ImportResourcePairID("connection_id", "client_id"),
 		},
 		Description: "With this resource, you can enable a single client on a connection.",
 	}

diff --git a/internal/auth0/connection/resource_client_import.go b/internal/auth0/connection/resource_client_import.go
diff --git a/internal/auth0/connection/resource_client_import_test.go b/internal/auth0/connection/resource_client_import_test.go
diff --git a/internal/auth0/organization/resource_connection.go b/internal/auth0/organization/resource_connection.go
@@ -11,6 +11,7 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 
 	"github.com/auth0/terraform-provider-auth0/internal/mutex"
+	internalSchema "github.com/auth0/terraform-provider-auth0/internal/schema"
 )
 
 var (
@@ -27,7 +28,7 @@ func NewConnectionResource() *schema.Resource {
 		UpdateContext: updateOrganizationConnection,
 		DeleteContext: deleteOrganizationConnection,
 		Importer: &schema.ResourceImporter{
-			StateContext: importOrganizationConnection,
+			StateContext: internalSchema.ImportResourcePairID("organization_id", "connection_id"),
 		},
 		Schema: map[string]*schema.Schema{
 			"organization_id": {

diff --git a/internal/auth0/organization/resource_import.go b/internal/auth0/organization/resource_import.go
diff --git a/internal/auth0/organization/resource_import_test.go b/internal/auth0/organization/resource_import_test.go
diff --git a/internal/auth0/organization/resource_member.go b/internal/auth0/organization/resource_member.go
@@ -10,14 +10,10 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 
 	"github.com/auth0/terraform-provider-auth0/internal/mutex"
+	internalSchema "github.com/auth0/terraform-provider-auth0/internal/schema"
 	"github.com/auth0/terraform-provider-auth0/internal/value"
 )
 
-var (
-	errEmptyOrganizationMemberID         = fmt.Errorf("ID cannot be empty")
-	errInvalidOrganizationMemberIDFormat = fmt.Errorf("ID must be formated as <organizationID>:<userID>")
-)
-
 // NewMemberResource will return a new auth0_organization_member resource.
 func NewMemberResource() *schema.Resource {
 	return &schema.Resource{
@@ -27,7 +23,7 @@ func NewMemberResource() *schema.Resource {
 		UpdateContext: updateOrganizationMember,
 		DeleteContext: deleteOrganizationMember,
 		Importer: &schema.ResourceImporter{
-			StateContext: importOrganizationMember,
+			StateContext: internalSchema.ImportResourcePairID("organization_id", "user_id"),
 		},
 		Schema: map[string]*schema.Schema{
 			"organization_id": {