http: api: blueprints: implement validate

Author: rnovatorov

src/enapter/cli/http/api/blueprint_command.py

@@ -4,6 +4,7 @@
 
 from .blueprint_download_command import BlueprintDownloadCommand
 from .blueprint_upload_command import BlueprintUploadCommand
+from .blueprint_validate_command import BlueprintValidateCommand
 
 
 class BlueprintCommand(cli.Command):
@@ -17,6 +18,7 @@ def register(parent: cli.Subparsers) -> None:
         for command in [
             BlueprintDownloadCommand,
             BlueprintUploadCommand,
+            BlueprintValidateCommand,
         ]:
             command.register(subparsers)
 
@@ -27,5 +29,7 @@ async def run(args: argparse.Namespace) -> None:
                 await BlueprintDownloadCommand.run(args)
             case "upload":
                 await BlueprintUploadCommand.run(args)
+            case "validate":
+                await BlueprintValidateCommand.run(args)
             case _:
                 raise NotImplementedError(args.command_command)

src/enapter/cli/http/api/blueprint_validate_command.py

@@ -0,0 +1,27 @@
+import argparse
+import logging
+import pathlib
+
+from enapter import cli, http
+
+LOGGER = logging.getLogger(__name__)
+
+
+class BlueprintValidateCommand(cli.Command):
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        parser = parent.add_parser(
+            "validate", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument(
+            "path", type=pathlib.Path, help="Path to a directory or a zip file"
+        )
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            if args.path.is_dir():
+                await client.blueprints.validate_directory(args.path)
+            else:
+                await client.blueprints.validate_file(args.path)

src/enapter/http/api/blueprints/client.py

@@ -20,11 +20,8 @@ async def upload_file(self, path: pathlib.Path) -> Blueprint:
         return await self.upload(data)
 
     async def upload_directory(self, path: pathlib.Path) -> Blueprint:
-        buffer = io.BytesIO()
-        with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
-            for file_path in path.rglob("*"):
-                zip_file.write(file_path, arcname=file_path.relative_to(path))
-        return await self.upload(buffer.getvalue())
+        data = await self._zip_directory(path)
+        return await self.upload(data)
 
     async def upload(self, data: bytes) -> Blueprint:
         url = "v3/blueprints/upload"
@@ -39,3 +36,29 @@ async def download(
         response = await self._client.get(url, params={"view": view.value})
         api.check_error(response)
         return response.content
+
+    async def validate_file(self, path: pathlib.Path) -> None:
+        with path.open("rb") as file:
+            data = file.read()
+        await self.validate(data)
+
+    async def validate_directory(self, path: pathlib.Path) -> None:
+        data = await self._zip_directory(path)
+        await self.validate(data)
+
+    async def validate(self, data: bytes) -> None:
+        url = "v3/blueprints/validate"
+        response = await self._client.post(url, content=data)
+        api.check_error(response)
+        validation_errors = response.json().get("validation_errors", [])
+        if validation_errors:
+            raise api.MultiError(
+                [api.Error(msg, code=None, details=None) for msg in validation_errors]
+            )
+
+    async def _zip_directory(self, path: pathlib.Path) -> bytes:
+        buffer = io.BytesIO()
+        with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
+            for file_path in path.rglob("*"):
+                zip_file.write(file_path, arcname=file_path.relative_to(path))
+        return buffer.getvalue()