http: api: devices: add create vucm and create lua

Author: rnovatorov

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

@@ -3,7 +3,9 @@
 from enapter import cli
 
 from .device_assign_blueprint_command import DeviceAssignBlueprintCommand
+from .device_create_lua_command import DeviceCreateLuaCommand
 from .device_create_standalone_command import DeviceCreateStandaloneCommand
+from .device_create_vucm_command import DeviceCreateVUCMCommand
 from .device_delete_command import DeviceDeleteCommand
 from .device_generate_communication_config_command import (
     DeviceGenerateCommunicationConfigCommand,
@@ -24,7 +26,9 @@ def register(parent: cli.Subparsers) -> None:
         for command in [
             DeviceAssignBlueprintCommand,
             DeviceCreateStandaloneCommand,
+            DeviceCreateVUCMCommand,
             DeviceDeleteCommand,
+            DeviceCreateLuaCommand,
             DeviceGenerateCommunicationConfigCommand,
             DeviceGetCommand,
             DeviceListCommand,
@@ -35,6 +39,8 @@ def register(parent: cli.Subparsers) -> None:
     @staticmethod
     async def run(args: argparse.Namespace) -> None:
         match args.device_command:
+            case "create-lua":
+                await DeviceCreateLuaCommand.run(args)
             case "assign-blueprint":
                 await DeviceAssignBlueprintCommand.run(args)
             case "create-standalone":
@@ -49,5 +55,7 @@ async def run(args: argparse.Namespace) -> None:
                 await DeviceListCommand.run(args)
             case "update":
                 await DeviceUpdateCommand.run(args)
+            case "create-vucm":
+                await DeviceCreateVUCMCommand.run(args)
             case _:
                 raise NotImplementedError(args.device_command)

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

@@ -0,0 +1,30 @@
+import argparse
+import json
+
+from enapter import cli, http
+
+
+class DeviceCreateLuaCommand(cli.Command):
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        parser = parent.add_parser(
+            "create-lua", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument("-r", "--runtime-id", help="Runtime ID of the Lua device")
+        parser.add_argument(
+            "-b", "--blueprint-id", help="Blueprint ID of the Lua device"
+        )
+        parser.add_argument("-s", "--slug", help="Slug of the Lua device")
+        parser.add_argument("name", help="Name of the Lua device to create")
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            device = await client.devices.create_lua(
+                name=args.name,
+                runtime_id=args.runtime_id,
+                blueprint_id=args.blueprint_id,
+                slug=args.slug,
+            )
+            print(json.dumps(device.to_dto()))

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

@@ -0,0 +1,24 @@
+import argparse
+import json
+
+from enapter import cli, http
+
+
+class DeviceCreateVUCMCommand(cli.Command):
+
+    @staticmethod
+    def register(parent: cli.Subparsers) -> None:
+        parser = parent.add_parser(
+            "create-vucm", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        parser.add_argument("-s", "--site-id", help="Site ID to create device in")
+        parser.add_argument("--hardware-id", help="Hardware ID of the VUCM device")
+        parser.add_argument("name", help="Name of the VUCM device to create")
+
+    @staticmethod
+    async def run(args: argparse.Namespace) -> None:
+        async with http.api.Client(http.api.Config.from_env()) as client:
+            device = await client.devices.create_vucm(
+                name=args.name, site_id=args.site_id, hardware_id=args.hardware_id
+            )
+            print(json.dumps(device.to_dto()))

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

@@ -1,3 +1,4 @@
+import random
 from typing import AsyncGenerator
 
 import httpx
@@ -17,7 +18,37 @@ def __init__(self, client: httpx.AsyncClient) -> None:
 
     async def create_standalone(self, name: str, site_id: str | None = None) -> Device:
         url = "v3/provisioning/standalone"
-        response = await self._client.post(url, json={"name": name, "site_id": site_id})
+        response = await self._client.post(url, json={"slug": name, "site_id": site_id})
+        api.check_error(response)
+        return await self.get(device_id=response.json()["device_id"])
+
+    async def create_vucm(
+        self, name: str, hardware_id: str | None = None, site_id: str | None = None
+    ) -> Device:
+        if hardware_id is None:
+            hardware_id = random_hardware_id()
+        url = "v3/provisioning/vucm"
+        response = await self._client.post(
+            url, json={"name": name, "hardware_id": hardware_id, "site_id": site_id}
+        )
+        api.check_error(response)
+        return await self.get(
+            device_id=response.json()["device_id"], expand_communication=True
+        )
+
+    async def create_lua(
+        self, name: str, runtime_id: str, blueprint_id: str, slug: str | None = None
+    ) -> Device:
+        url = "v3/provisioning/lua_device"
+        response = await self._client.post(
+            url,
+            json={
+                "name": name,
+                "runtime_id": runtime_id,
+                "blueprint_id": blueprint_id,
+                "slug": slug,
+            },
+        )
         api.check_error(response)
         return await self.get(device_id=response.json()["device_id"])
 
@@ -99,3 +130,7 @@ async def generate_communication_config(
         response = await self._client.post(url, json={"protocol": mqtt_protocol.value})
         api.check_error(response)
         return CommunicationConfig.from_dto(response.json()["config"])
+
+
+def random_hardware_id() -> str:
+    return "V" + "".join(f"{b:02X}" for b in random.randbytes(16))