refactor(routes): migrate lifecycle mutations to service

py/routes/base_model_routes.py

@@ -13,6 +13,7 @@ from ..services.downloader import get_downloader
 from ..services.metadata_service import get_default_metadata_provider, get_metadata_provider
 from ..services.metadata_sync_service import MetadataSyncService
 from ..services.model_file_service import ModelFileService, ModelMoveService
+from ..services.model_lifecycle_service import ModelLifecycleService
 from ..services.preview_asset_service import PreviewAssetService
 from ..services.server_i18n import server_i18n as default_server_i18n
 from ..services.service_registry import ServiceRegistry
@@ -75,6 +76,7 @@ class BaseModelRoutes(ABC):
 
         self.model_file_service: ModelFileService | None = None
         self.model_move_service: ModelMoveService | None = None
+        self.model_lifecycle_service: ModelLifecycleService | None = None
         self.websocket_progress_callback = WebSocketProgressCallback()
         self.metadata_progress_callback = WebSocketBroadcastCallback()
 
@@ -108,6 +110,12 @@ class BaseModelRoutes(ABC):
         self.model_type = service.model_type
         self.model_file_service = ModelFileService(service.scanner, service.model_type)
         self.model_move_service = ModelMoveService(service.scanner)
+        self.model_lifecycle_service = ModelLifecycleService(
+            scanner=service.scanner,
+            metadata_manager=MetadataManager,
+            metadata_loader=self._metadata_sync_service.load_local_metadata,
+            recipe_scanner_factory=ServiceRegistry.get_recipe_scanner,
+        )
         self._handler_set = None
         self._handler_mapping = None
 
@@ -139,6 +147,7 @@ class BaseModelRoutes(ABC):
             metadata_sync=self._metadata_sync_service,
             preview_service=self._preview_service,
             tag_update_service=self._tag_update_service,
+            lifecycle_service=self._ensure_lifecycle_service(),
         )
         query = ModelQueryHandler(service=service, logger=logger)
         download_use_case = DownloadModelUseCase(download_coordinator=self._download_coordinator)
@@ -248,6 +257,17 @@ class BaseModelRoutes(ABC):
             self.model_move_service = ModelMoveService(service.scanner)
         return self.model_move_service
 
+    def _ensure_lifecycle_service(self) -> ModelLifecycleService:
+        if self.model_lifecycle_service is None:
+            service = self._ensure_service()
+            self.model_lifecycle_service = ModelLifecycleService(
+                scanner=service.scanner,
+                metadata_manager=MetadataManager,
+                metadata_loader=self._metadata_sync_service.load_local_metadata,
+                recipe_scanner_factory=ServiceRegistry.get_recipe_scanner,
+            )
+        return self.model_lifecycle_service
+
     def _make_handler_proxy(self, name: str) -> Callable[[web.Request], web.StreamResponse]:
         async def proxy(request: web.Request) -> web.StreamResponse:
             try:

py/routes/handlers/model_handlers.py

@@ -30,7 +30,6 @@ from ...services.use_cases import (
 from ...services.websocket_manager import WebSocketManager
 from ...services.websocket_progress_callback import WebSocketProgressCallback
 from ...utils.file_utils import calculate_sha256
-from ...utils.routes_common import ModelRouteUtils
 
 
 class ModelPageView:
@@ -192,18 +191,44 @@ class ModelManagementHandler:
         metadata_sync: MetadataSyncService,
         preview_service: PreviewAssetService,
         tag_update_service: TagUpdateService,
+        lifecycle_service,
     ) -> None:
         self._service = service
         self._logger = logger
         self._metadata_sync = metadata_sync
         self._preview_service = preview_service
         self._tag_update_service = tag_update_service
+        self._lifecycle_service = lifecycle_service
 
     async def delete_model(self, request: web.Request) -> web.Response:
-        return await ModelRouteUtils.handle_delete_model(request, self._service.scanner)
+        try:
+            data = await request.json()
+            file_path = data.get("file_path")
+            if not file_path:
+                return web.Response(text="Model path is required", status=400)
+
+            result = await self._lifecycle_service.delete_model(file_path)
+            return web.json_response(result)
+        except ValueError as exc:
+            return web.json_response({"success": False, "error": str(exc)}, status=400)
+        except Exception as exc:
+            self._logger.error("Error deleting model: %s", exc, exc_info=True)
+            return web.Response(text=str(exc), status=500)
 
     async def exclude_model(self, request: web.Request) -> web.Response:
-        return await ModelRouteUtils.handle_exclude_model(request, self._service.scanner)
+        try:
+            data = await request.json()
+            file_path = data.get("file_path")
+            if not file_path:
+                return web.Response(text="Model path is required", status=400)
+
+            result = await self._lifecycle_service.exclude_model(file_path)
+            return web.json_response(result)
+        except ValueError as exc:
+            return web.json_response({"success": False, "error": str(exc)}, status=400)
+        except Exception as exc:
+            self._logger.error("Error excluding model: %s", exc, exc_info=True)
+            return web.Response(text=str(exc), status=500)
 
     async def fetch_civitai(self, request: web.Request) -> web.Response:
         try:
@@ -375,10 +400,58 @@ class ModelManagementHandler:
             return web.Response(text=str(exc), status=500)
 
     async def rename_model(self, request: web.Request) -> web.Response:
-        return await ModelRouteUtils.handle_rename_model(request, self._service.scanner)
+        try:
+            data = await request.json()
+            file_path = data.get("file_path")
+            new_file_name = data.get("new_file_name")
+
+            if not file_path or not new_file_name:
+                return web.json_response(
+                    {
+                        "success": False,
+                        "error": "File path and new file name are required",
+                    },
+                    status=400,
+                )
+
+            result = await self._lifecycle_service.rename_model(
+                file_path=file_path, new_file_name=new_file_name
+            )
+
+            return web.json_response(
+                {
+                    **result,
+                    "new_preview_path": config.get_preview_static_url(
+                        result.get("new_preview_path")
+                    ),
+                }
+            )
+        except ValueError as exc:
+            return web.json_response({"success": False, "error": str(exc)}, status=400)
+        except Exception as exc:
+            self._logger.error("Error renaming model: %s", exc, exc_info=True)
+            return web.json_response({"success": False, "error": str(exc)}, status=500)
 
     async def bulk_delete_models(self, request: web.Request) -> web.Response:
-        return await ModelRouteUtils.handle_bulk_delete_models(request, self._service.scanner)
+        try:
+            data = await request.json()
+            file_paths = data.get("file_paths", [])
+            if not file_paths:
+                return web.json_response(
+                    {
+                        "success": False,
+                        "error": "No file paths provided for deletion",
+                    },
+                    status=400,
+                )
+
+            result = await self._lifecycle_service.bulk_delete_models(file_paths)
+            return web.json_response(result)
+        except ValueError as exc:
+            return web.json_response({"success": False, "error": str(exc)}, status=400)
+        except Exception as exc:
+            self._logger.error("Error in bulk delete: %s", exc, exc_info=True)
+            return web.json_response({"success": False, "error": str(exc)}, status=500)
 
     async def verify_duplicates(self, request: web.Request) -> web.Response:
         try: