Add Account Provisioning related capabilities (#250)

ConductorOne · Nov 7, 2024 · 060a1a7 · 060a1a7
1 parent 9bcd6a7
commit 060a1a7
Show file tree

Hide file tree

Showing 3 changed files with 70 additions and 31 deletions.
diff --git a/pb/c1/connector/v2/connector.pb.go b/pb/c1/connector/v2/connector.pb.go
diff --git a/pkg/connectorbuilder/connectorbuilder.go b/pkg/connectorbuilder/connectorbuilder.go
@@ -560,6 +560,21 @@ func getCapabilities(ctx context.Context, b *builderImpl) *v2.ConnectorCapabilit
 			resourceTypeCapability.Capabilities = append(resourceTypeCapability.Capabilities, v2.Capability_CAPABILITY_PROVISION)
 			connectorCaps[v2.Capability_CAPABILITY_PROVISION] = struct{}{}
 		}
+		if _, ok := rb.(AccountManager); ok {
+			resourceTypeCapability.Capabilities = append(resourceTypeCapability.Capabilities, v2.Capability_CAPABILITY_ACCOUNT_PROVISIONING)
+			connectorCaps[v2.Capability_CAPABILITY_ACCOUNT_PROVISIONING] = struct{}{}
+		}
+
+		if _, ok := rb.(CredentialManager); ok {
+			resourceTypeCapability.Capabilities = append(resourceTypeCapability.Capabilities, v2.Capability_CAPABILITY_CREDENTIAL_ROTATION)
+			connectorCaps[v2.Capability_CAPABILITY_CREDENTIAL_ROTATION] = struct{}{}
+		}
+
+		if _, ok := rb.(ResourceManager); ok {
+			resourceTypeCapability.Capabilities = append(resourceTypeCapability.Capabilities, v2.Capability_CAPABILITY_RESOURCE_CREATE, v2.Capability_CAPABILITY_RESOURCE_DELETE)
+			connectorCaps[v2.Capability_CAPABILITY_RESOURCE_CREATE] = struct{}{}
+			connectorCaps[v2.Capability_CAPABILITY_RESOURCE_DELETE] = struct{}{}
+		}
 		resourceTypeCapabilities = append(resourceTypeCapabilities, resourceTypeCapability)
 	}
 	sort.Slice(resourceTypeCapabilities, func(i, j int) bool {

diff --git a/proto/c1/connector/v2/connector.proto b/proto/c1/connector/v2/connector.proto
@@ -45,6 +45,10 @@ enum Capability {
   CAPABILITY_SYNC = 2;
   CAPABILITY_EVENT_FEED = 3;
   CAPABILITY_TICKETING = 4;
+  CAPABILITY_ACCOUNT_PROVISIONING = 5;
+  CAPABILITY_CREDENTIAL_ROTATION = 6;
+  CAPABILITY_RESOURCE_CREATE = 7;
+  CAPABILITY_RESOURCE_DELETE = 8;
 }
 
 message ConnectorCapabilities {