feat(example-images): add use case orchestration

py/routes/handlers/example_images_handlers.py

@@ -6,37 +6,101 @@ from typing import Callable, Mapping
 
 from aiohttp import web
 
+from ...services.use_cases.example_images import (
+    DownloadExampleImagesConfigurationError,
+    DownloadExampleImagesInProgressError,
+    DownloadExampleImagesUseCase,
+    ImportExampleImagesUseCase,
+    ImportExampleImagesValidationError,
+)
+from ...utils.example_images_download_manager import (
+    DownloadConfigurationError,
+    DownloadInProgressError,
+    DownloadNotRunningError,
+    ExampleImagesDownloadError,
+)
+from ...utils.example_images_processor import ExampleImagesImportError
+
 
 class ExampleImagesDownloadHandler:
     """HTTP adapters for download-related example image endpoints."""
 
-    def __init__(self, download_manager) -> None:
+    def __init__(
+        self,
+        download_use_case: DownloadExampleImagesUseCase,
+        download_manager,
+    ) -> None:
+        self._download_use_case = download_use_case
         self._download_manager = download_manager
 
     async def download_example_images(self, request: web.Request) -> web.StreamResponse:
-        return await self._download_manager.start_download(request)
+        try:
+            payload = await request.json()
+            result = await self._download_use_case.execute(payload)
+            return web.json_response(result)
+        except DownloadExampleImagesInProgressError as exc:
+            response = {
+                'success': False,
+                'error': str(exc),
+                'status': exc.progress,
+            }
+            return web.json_response(response, status=400)
+        except DownloadExampleImagesConfigurationError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=400)
+        except ExampleImagesDownloadError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=500)
 
     async def get_example_images_status(self, request: web.Request) -> web.StreamResponse:
-        return await self._download_manager.get_status(request)
+        result = await self._download_manager.get_status(request)
+        return web.json_response(result)
 
     async def pause_example_images(self, request: web.Request) -> web.StreamResponse:
-        return await self._download_manager.pause_download(request)
+        try:
+            result = await self._download_manager.pause_download(request)
+            return web.json_response(result)
+        except DownloadNotRunningError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=400)
 
     async def resume_example_images(self, request: web.Request) -> web.StreamResponse:
-        return await self._download_manager.resume_download(request)
+        try:
+            result = await self._download_manager.resume_download(request)
+            return web.json_response(result)
+        except DownloadNotRunningError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=400)
 
     async def force_download_example_images(self, request: web.Request) -> web.StreamResponse:
-        return await self._download_manager.start_force_download(request)
+        try:
+            payload = await request.json()
+            result = await self._download_manager.start_force_download(payload)
+            return web.json_response(result)
+        except DownloadInProgressError as exc:
+            response = {
+                'success': False,
+                'error': str(exc),
+                'status': exc.progress_snapshot,
+            }
+            return web.json_response(response, status=400)
+        except DownloadConfigurationError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=400)
+        except ExampleImagesDownloadError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=500)
 
 
 class ExampleImagesManagementHandler:
     """HTTP adapters for import/delete endpoints."""
 
-    def __init__(self, processor) -> None:
+    def __init__(self, import_use_case: ImportExampleImagesUseCase, processor) -> None:
+        self._import_use_case = import_use_case
         self._processor = processor
 
     async def import_example_images(self, request: web.Request) -> web.StreamResponse:
-        return await self._processor.import_images(request)
+        try:
+            result = await self._import_use_case.execute(request)
+            return web.json_response(result)
+        except ImportExampleImagesValidationError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=400)
+        except ExampleImagesImportError as exc:
+            return web.json_response({'success': False, 'error': str(exc)}, status=500)
 
     async def delete_example_image(self, request: web.Request) -> web.StreamResponse:
         return await self._processor.delete_custom_image(request)