refactor(api): decouple setup and spec endpoints (#40104)

api/extensions/ext_application_services.py

@@ -6,13 +6,21 @@ from typing import cast
 from flask import Flask, current_app
 from sqlalchemy.orm import Session, sessionmaker
 
+from configs import dify_config
 from constants.dsl_version import CURRENT_APP_DSL_VERSION
 from core.db.session_factory import get_session_maker
+from core.schemas.schema_manager import SchemaManager
+from enums.deployment_edition import DeploymentEdition
+from extensions.ext_redis import RedisClientWrapper, redis_client
+from repositories.installation_state_repository import InstallationStateRepository
 from repositories.workspace_member_query_repository import WorkspaceMemberQueryRepository
 from repositories.workspace_query_repository import WorkspaceQueryRepository
 from services.feature_query_service import FeatureQueryService
 from services.feature_service import FeatureService
 from services.feature_service_gateway import FeatureServiceGateway
+from services.schema_definition_service import SchemaDefinitionService
+from services.setup_adapters import RedisSetupLock, RegisterServiceAccountProvisioner
+from services.setup_service import SetupService
 from services.workspace_member_query_service import WorkspaceMemberQueryService
 from services.workspace_member_role_resolver import DeploymentWorkspaceMemberRoleResolver
 from services.workspace_plan_gateway import DeploymentWorkspacePlanGateway
@@ -23,6 +31,8 @@ _EXTENSION_KEY = "application_services"
 
 @dataclass(frozen=True, slots=True)
 class ApplicationServices:
+    schema_definitions: SchemaDefinitionService
+    setup: SetupService
     feature_queries: FeatureQueryService
     workspace_queries: WorkspaceQueryService
     workspace_member_queries: WorkspaceMemberQueryService
@@ -31,8 +41,18 @@ class ApplicationServices:
 def build_application_services(
     *,
     database_client: sessionmaker[Session],
+    deployment_edition: DeploymentEdition,
+    redis: RedisClientWrapper,
 ) -> ApplicationServices:
+    installation_state = InstallationStateRepository(client=database_client)
     return ApplicationServices(
+        schema_definitions=SchemaDefinitionService(source_factory=SchemaManager),
+        setup=SetupService(
+            state=installation_state,
+            accounts=RegisterServiceAccountProvisioner(client=database_client),
+            lock=RedisSetupLock(client=redis),
+            setup_required=deployment_edition != DeploymentEdition.CLOUD,
+        ),
         feature_queries=FeatureQueryService(
             features=FeatureServiceGateway(),
             trial_models=FeatureService.get_trial_models(),
@@ -56,6 +76,8 @@ def build_application_services(
 def init_app(app: Flask) -> None:
     app.extensions[_EXTENSION_KEY] = build_application_services(
         database_client=get_session_maker(),
+        deployment_edition=dify_config.DEPLOYMENT_EDITION,
+        redis=redis_client,
     )