chore(integrations): register integration domains in initializer

src/sentry/integrations/api/endpoints/organization_integrations_index.py

@@ -21,7 +21,8 @@
     OrganizationIntegrationBaseEndpoint,
 )
 from sentry.integrations.api.serializers.models.integration import OrganizationIntegrationResponse
-from sentry.integrations.base import INTEGRATION_TYPE_TO_PROVIDER, IntegrationDomain
+from sentry.integrations.base import IntegrationDomain
+from sentry.integrations.manager import default_manager as integrations
 from sentry.integrations.models.integration import Integration
 from sentry.integrations.models.organization_integration import OrganizationIntegration
 from sentry.organizations.services.organization.model import (
@@ -117,7 +118,8 @@ def get(
             except ValueError:
                 return Response({"detail": "Invalid integration type"}, status=400)
             provider_slugs = [
-                provider for provider in INTEGRATION_TYPE_TO_PROVIDER.get(integration_domain, [])
+                provider
+                for provider in integrations.get_integrations_for_domain(integration_domain)
             ]
             queryset = queryset.filter(integration__provider__in=provider_slugs)
 

src/sentry/integrations/base.py

@@ -135,49 +135,6 @@ class IntegrationDomain(StrEnum):
     IDENTITY = "identity"  # for identity pipelines
 
 
-class IntegrationProviderSlug(StrEnum):
-    SLACK = "slack"
-    DISCORD = "discord"
-    MSTeams = "msteams"
-    JIRA = "jira"
-    JIRA_SERVER = "jira_server"
-    AZURE_DEVOPS = "vsts"
-    GITHUB = "github"
-    GITHUB_ENTERPRISE = "github_enterprise"
-    GITLAB = "gitlab"
-    BITBUCKET = "bitbucket"
-    PAGERDUTY = "pagerduty"
-    OPSGENIE = "opsgenie"
-
-
-INTEGRATION_TYPE_TO_PROVIDER = {
-    IntegrationDomain.MESSAGING: [
-        IntegrationProviderSlug.SLACK,
-        IntegrationProviderSlug.DISCORD,
-        IntegrationProviderSlug.MSTeams,
-    ],
-    IntegrationDomain.PROJECT_MANAGEMENT: [
-        IntegrationProviderSlug.JIRA,
-        IntegrationProviderSlug.JIRA_SERVER,
-        IntegrationProviderSlug.GITHUB,
-        IntegrationProviderSlug.GITHUB_ENTERPRISE,
-        IntegrationProviderSlug.GITLAB,
-        IntegrationProviderSlug.AZURE_DEVOPS,
-    ],
-    IntegrationDomain.SOURCE_CODE_MANAGEMENT: [
-        IntegrationProviderSlug.GITHUB,
-        IntegrationProviderSlug.GITHUB_ENTERPRISE,
-        IntegrationProviderSlug.GITLAB,
-        IntegrationProviderSlug.BITBUCKET,
-        IntegrationProviderSlug.AZURE_DEVOPS,
-    ],
-    IntegrationDomain.ON_CALL_SCHEDULING: [
-        IntegrationProviderSlug.PAGERDUTY,
-        IntegrationProviderSlug.OPSGENIE,
-    ],
-}
-
-
 class IntegrationProvider(PipelineProvider, abc.ABC):
     """
     An integration provider describes a third party that can be registered within Sentry.
@@ -245,6 +202,8 @@ class is just a descriptor for how that object functions, and what behavior
     requires_feature_flag = False
     """if this is hidden without the feature flag"""
 
+    domain: IntegrationDomain | None = None
+
     @classmethod
     def get_installation(
         cls, model: RpcIntegration | Integration, organization_id: int, **kwargs: Any
@@ -593,8 +552,10 @@ def disable_integration(
 
 
 def get_integration_types(provider: str):
+    from sentry.integrations.manager import default_manager as integrations
+
     types = []
-    for integration_type, providers in INTEGRATION_TYPE_TO_PROVIDER.items():
+    for integration_type, providers in integrations.get_integrations_by_domain().items():
         if provider in providers:
             types.append(integration_type)
     return types

src/sentry/integrations/bitbucket/integration.py

@@ -174,6 +174,7 @@ def username(self):
 class BitbucketIntegrationProvider(IntegrationProvider):
     key = "bitbucket"
     name = "Bitbucket"
+    domain = IntegrationDomain.SOURCE_CODE_MANAGEMENT
     metadata = metadata
     scopes = scopes
     integration_cls = BitbucketIntegration

src/sentry/integrations/bitbucket_server/integration.py

@@ -330,6 +330,7 @@ def username(self):
 class BitbucketServerIntegrationProvider(IntegrationProvider):
     key = "bitbucket_server"
     name = "Bitbucket Server"
+    domain = IntegrationDomain.SOURCE_CODE_MANAGEMENT
     metadata = metadata
     integration_cls = BitbucketServerIntegration
     needs_default_identity = True

src/sentry/integrations/discord/integration.py

@@ -10,6 +10,7 @@
 from sentry.constants import ObjectStatus
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationInstallation,
     IntegrationMetadata,
@@ -114,6 +115,7 @@ def uninstall(self) -> None:
 class DiscordIntegrationProvider(IntegrationProvider):
     key = "discord"
     name = "Discord"
+    domain = IntegrationDomain.MESSAGING
     metadata = metadata
     integration_cls = DiscordIntegration
     features = frozenset([IntegrationFeatures.CHAT_UNFURL, IntegrationFeatures.ALERT_RULE])

src/sentry/integrations/github/integration.py

@@ -311,6 +311,7 @@ def search_issues(self, query: str | None, **kwargs) -> dict[str, Any]:
 class GitHubIntegrationProvider(IntegrationProvider):
     key = "github"
     name = "GitHub"
+    domain = IntegrationDomain.SOURCE_CODE_MANAGEMENT
     metadata = metadata
     integration_cls = GitHubIntegration
     features = frozenset(

src/sentry/integrations/github_enterprise/integration.py

@@ -13,6 +13,7 @@
 from sentry.identity.pipeline import IdentityProviderPipeline
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatureNotImplementedError,
     IntegrationFeatures,
     IntegrationMetadata,
@@ -318,6 +319,7 @@ def dispatch(self, request: Request, pipeline) -> HttpResponse:
 class GitHubEnterpriseIntegrationProvider(GitHubIntegrationProvider):
     key = "github_enterprise"
     name = "GitHub Enterprise"
+    domain = IntegrationDomain.SOURCE_CODE_MANAGEMENT
     metadata = metadata
     integration_cls = GitHubEnterpriseIntegration
     features = frozenset(

src/sentry/integrations/gitlab/integration.py

@@ -13,6 +13,7 @@
 from sentry.identity.pipeline import IdentityProviderPipeline
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationMetadata,
     IntegrationProvider,
@@ -313,6 +314,7 @@ def dispatch(self, request: Request, pipeline) -> HttpResponse:
 class GitlabIntegrationProvider(IntegrationProvider):
     key = "gitlab"
     name = "GitLab"
+    domain = IntegrationDomain.SOURCE_CODE_MANAGEMENT
     metadata = metadata
     integration_cls = GitlabIntegration
 

src/sentry/integrations/jira/integration.py

@@ -16,6 +16,7 @@
 from sentry.eventstore.models import GroupEvent
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationMetadata,
     IntegrationProvider,
@@ -995,6 +996,7 @@ def parse_jira_issue_metadata(self, meta: dict[str, Any]) -> list[JiraIssueTypeM
 class JiraIntegrationProvider(IntegrationProvider):
     key = "jira"
     name = "Jira"
+    domain = IntegrationDomain.PROJECT_MANAGEMENT
     metadata = metadata
     integration_cls = JiraIntegration
 

src/sentry/integrations/jira_server/integration.py

@@ -20,6 +20,7 @@
 from sentry import features
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationMetadata,
     IntegrationProvider,
@@ -1167,6 +1168,7 @@ def migrate_issues(self):
 class JiraServerIntegrationProvider(IntegrationProvider):
     key = "jira_server"
     name = "Jira Server"
+    domain = IntegrationDomain.PROJECT_MANAGEMENT
     metadata = metadata
     integration_cls = JiraServerIntegration
 

src/sentry/integrations/manager.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
+from collections import defaultdict
 from collections.abc import Iterable, Iterator
 from typing import Any
 
 from sentry.exceptions import NotRegistered
-from sentry.integrations.base import IntegrationProvider
+from sentry.integrations.base import IntegrationDomain, IntegrationProvider
 
 __all__ = ["IntegrationManager"]
 
@@ -14,6 +15,12 @@
 class IntegrationManager:
     def __init__(self) -> None:
         self.__values: dict[str, type[IntegrationProvider]] = {}
+        self.__domain_integrations: dict[IntegrationDomain, list[str]] = defaultdict(
+            list
+        )  # integrations by domain
+        self.__integration_domains: dict[str, IntegrationDomain] = (
+            {}
+        )  # map of integration to domain
 
     def __iter__(self) -> Iterator[IntegrationProvider]:
         return iter(self.all())
@@ -36,6 +43,9 @@ def exists(self, key: str) -> bool:
 
     def register(self, cls: type[IntegrationProvider]) -> None:
         self.__values[cls.key] = cls
+        if cls.domain:
+            self.__domain_integrations[cls.domain].append(cls.key)
+            self.__integration_domains[cls.key] = cls.domain
 
     def unregister(self, cls: type[IntegrationProvider]) -> None:
         try:
@@ -47,6 +57,19 @@ def unregister(self, cls: type[IntegrationProvider]) -> None:
             return
         del self.__values[cls.key]
 
+    def get_integrations_for_domain(self, domain: IntegrationDomain) -> list[str]:
+        if domain not in self.__domain_integrations:
+            return []
+        return self.__domain_integrations[domain]
+
+    def get_integrations_by_domain(self) -> dict[IntegrationDomain, list[str]]:
+        return self.__domain_integrations
+
+    def get_integration_domain(self, key: str) -> IntegrationDomain:
+        if key not in self.__integration_domains:
+            raise NotRegistered(key)
+        return self.__integration_domains[key]
+
 
 default_manager = IntegrationManager()
 all = default_manager.all

src/sentry/integrations/msteams/integration.py

@@ -10,6 +10,7 @@
 from sentry import options
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationInstallation,
     IntegrationMetadata,
@@ -79,6 +80,7 @@ def get_client(self) -> MsTeamsClient:
 class MsTeamsIntegrationProvider(IntegrationProvider):
     key = "msteams"
     name = "Microsoft Teams"
+    domain = IntegrationDomain.MESSAGING
     can_add = False
     metadata = metadata
     integration_cls = MsTeamsIntegration

src/sentry/integrations/opsgenie/integration.py

@@ -13,6 +13,7 @@
 from sentry.constants import ObjectStatus
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationInstallation,
     IntegrationMetadata,
@@ -229,6 +230,7 @@ def schedule_migrate_opsgenie_plugin(self):
 class OpsgenieIntegrationProvider(IntegrationProvider):
     key = "opsgenie"
     name = "Opsgenie"
+    domain = IntegrationDomain.ON_CALL_SCHEDULING
     metadata = metadata
     integration_cls = OpsgenieIntegration
     features = frozenset([IntegrationFeatures.INCIDENT_MANAGEMENT, IntegrationFeatures.ALERT_RULE])

src/sentry/integrations/pagerduty/integration.py

@@ -12,6 +12,7 @@
 from sentry import options
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationInstallation,
     IntegrationMetadata,
@@ -166,6 +167,7 @@ def services(self) -> list[PagerDutyServiceDict]:
 class PagerDutyIntegrationProvider(IntegrationProvider):
     key = "pagerduty"
     name = "PagerDuty"
+    domain = IntegrationDomain.ON_CALL_SCHEDULING
     metadata = metadata
     features = frozenset([IntegrationFeatures.ALERT_RULE, IntegrationFeatures.INCIDENT_MANAGEMENT])
     integration_cls = PagerDutyIntegration

src/sentry/integrations/slack/integration.py

@@ -13,6 +13,7 @@
 from sentry.identity.pipeline import IdentityProviderPipeline
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationInstallation,
     IntegrationMetadata,
@@ -88,6 +89,7 @@ def get_config_data(self) -> Mapping[str, str]:
 class SlackIntegrationProvider(IntegrationProvider):
     key = "slack"
     name = "Slack"
+    domain = IntegrationDomain.MESSAGING
     metadata = metadata
     features = frozenset([IntegrationFeatures.CHAT_UNFURL, IntegrationFeatures.ALERT_RULE])
     integration_cls = SlackIntegration

src/sentry/integrations/vsts/integration.py

@@ -20,6 +20,7 @@
 from sentry.identity.vsts.provider import get_user_info
 from sentry.integrations.base import (
     FeatureDescription,
+    IntegrationDomain,
     IntegrationFeatures,
     IntegrationMetadata,
     IntegrationProvider,
@@ -381,6 +382,7 @@ def default_project(self) -> str | None:
 class VstsIntegrationProvider(IntegrationProvider):
     key = "vsts"
     name = "Azure DevOps"
+    domain = IntegrationDomain.SOURCE_CODE_MANAGEMENT
     metadata = metadata
     api_version = "4.1"
     oauth_redirect_url = "/extensions/vsts/setup/"