Use FastAPI to provide rest endpoints

.devcontainer/Dockerfile

@@ -34,4 +34,4 @@ ENV PATH=/venv/bin:$PATH
 
 # change this entrypoint if it is not the same as the repo
 ENTRYPOINT ["blueapi"]
-CMD ["worker"]
+CMD ["run"]

.vscode/launch.json

@@ -4,6 +4,18 @@
     // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
     "version": "0.2.0",
     "configurations": [
+        {
+            "name": "Python: FastAPI",
+            "type": "python",
+            "request": "launch",
+            "module": "uvicorn",
+            "args": [
+                "src.blueapi.rest.main:app",
+                "--reload"
+            ],
+            "jinja": true,
+            "justMyCode": true
+        },
         {
             "name": "Debug Unit Test",
             "type": "python",
@@ -68,4 +80,4 @@
             ]
         }
     ]
-}
+}

.vscode/settings.json

@@ -13,5 +13,5 @@
         "source.organizeImports": true
     },
     "esbonio.server.enabled": true,
-    "esbonio.sphinx.confDir": ""
-}
+    "esbonio.sphinx.confDir": "",
+}

pyproject.toml

@@ -22,6 +22,9 @@ dependencies = [
     "scanspec",
     "PyYAML",
     "click",
+    "fastapi",
+    "uvicorn",
+    "httpx",
 ]
 dynamic = ["version"]
 license.file = "LICENSE"
@@ -44,6 +47,8 @@ dev = [
     "tox-direct",
     "types-mock",
     "types-PyYAML",
+    "types-requests",
+    "types-urllib3",
 ]
 
 [project.scripts]

src/blueapi/__init__.py

@@ -3,4 +3,5 @@
 __version__ = version("blueapi")
 del version
 
+
 __all__ = ["__version__"]

src/blueapi/cli/cli.py

@@ -1,16 +1,17 @@
 import json
 import logging
 from pathlib import Path
+from pprint import pprint
 from typing import Optional
 
 import click
+import requests
+from requests.exceptions import ConnectionError
 
 from blueapi import __version__
 from blueapi.config import ApplicationConfig, ConfigLoader
-from blueapi.messaging import StompMessagingTemplate
-
-from .amq import AmqClient
-from .updates import CliEventRenderer
+from blueapi.service.handler import setup_handler
+from blueapi.service.main import app
 
 
 @click.group(invoke_without_command=True)
@@ -19,24 +20,30 @@
 @click.pass_context
 def main(ctx, config: Optional[Path]) -> None:
     # if no command is supplied, run with the options passed
+
     config_loader = ConfigLoader(ApplicationConfig)
     if config is not None:
         config_loader.use_values_from_yaml(config)
 
     ctx.ensure_object(dict)
-    ctx.obj["config"] = config_loader.load()
+    ctx.obj["config_loader"] = config_loader
 
     if ctx.invoked_subcommand is None:
         print("Please invoke subcommand!")
 
 
-@main.command(name="worker")
+@click.version_option(version=__version__)
+@main.command(name="run")
 @click.pass_obj
-def start_worker(obj: dict) -> None:
-    from blueapi.service import start
+def start_application(obj: dict):
+    import uvicorn
+
+    config_loader: ConfigLoader[ApplicationConfig] = obj["config_loader"]
+    config = config_loader.load()
+
+    setup_handler(config_loader)
 
-    config: ApplicationConfig = obj["config"]
-    start(config)
+    uvicorn.run(app, host=config.api.host, port=config.api.port)
 
 
 @main.group()
@@ -47,41 +54,55 @@ def controller(ctx) -> None:
         return
 
     ctx.ensure_object(dict)
-    config: ApplicationConfig = ctx.obj["config"]
+    config_loader: ConfigLoader[ApplicationConfig] = ctx.obj["config_loader"]
+    config: ApplicationConfig = config_loader.load()
     logging.basicConfig(level=config.logging.level)
-    client = AmqClient(StompMessagingTemplate.autoconfigured(config.stomp))
-    ctx.obj["client"] = client
-    client.app.connect()
+
+
+def check_connection(func):
+    def wrapper(*args, **kwargs):
+        try:
+            func(*args, **kwargs)
+        except ConnectionError:
+            print("Failed to establish connection to FastAPI server.")
+
+    return wrapper
 
 
 @controller.command(name="plans")
-@click.pass_context
-def get_plans(ctx) -> None:
-    client: AmqClient = ctx.obj["client"]
-    plans = client.get_plans()
-    print("PLANS")
-    for plan in plans.plans:
-        print("\t" + plan.name)
+@check_connection
+@click.pass_obj
+def get_plans(obj: dict) -> None:
+    config_loader: ConfigLoader[ApplicationConfig] = obj["config_loader"]
+    config: ApplicationConfig = config_loader.load()
+
+    resp = requests.get(f"http://{config.api.host}:{config.api.port}/plans")
+    print(f"Response returned with {resp.status_code}: ")
+    pprint(resp.json())
 
 
 @controller.command(name="devices")
-@click.pass_context
-def get_devices(ctx) -> None:
-    client: AmqClient = ctx.obj["client"]
-    print(client.get_devices().devices)
+@check_connection
+@click.pass_obj
+def get_devices(obj: dict) -> None:
+    config_loader: ConfigLoader[ApplicationConfig] = obj["config_loader"]
+    config: ApplicationConfig = config_loader.load()
+    resp = requests.get(f"http://{config.api.host}:{config.api.port}/devices")
+    print(f"Response returned with {resp.status_code}: ")
+    pprint(resp.json())
 
 
 @controller.command(name="run")
 @click.argument("name", type=str)
 @click.option("-p", "--parameters", type=str, help="Parameters as valid JSON")
-@click.pass_context
-def run_plan(ctx, name: str, parameters: str) -> None:
-    client: AmqClient = ctx.obj["client"]
-    renderer = CliEventRenderer()
-    client.run_plan(
-        name,
-        json.loads(parameters),
-        renderer.on_worker_event,
-        renderer.on_progress_event,
-        timeout=120.0,
+@check_connection
+@click.pass_obj
+def run_plan(obj: dict, name: str, parameters: str) -> None:
+    config_loader: ConfigLoader[ApplicationConfig] = obj["config_loader"]
+    config: ApplicationConfig = config_loader.load()
+    resp = requests.put(
+        f"http://{config.api.host}:{config.api.port}/task/{name}",
+        json=json.loads(parameters),
     )
+    print(f"Response returned with {resp.status_code}: ")
+    pprint(resp.json())

src/blueapi/config.py

@@ -35,6 +35,11 @@ class LoggingConfig(BlueapiBaseModel):
     level: LogLevel = "INFO"
 
 
+class FastApiConfig(BlueapiBaseModel):
+    host: str = "localhost"
+    port: int = 8000
+
+
 class ApplicationConfig(BlueapiBaseModel):
     """
     Config for the worker application as a whole. Root of
@@ -44,13 +49,15 @@ class ApplicationConfig(BlueapiBaseModel):
     stomp: StompConfig = Field(default_factory=StompConfig)
     env: EnvironmentConfig = Field(default_factory=EnvironmentConfig)
     logging: LoggingConfig = Field(default_factory=LoggingConfig)
+    api: FastApiConfig = Field(default_factory=FastApiConfig)
 
     def __eq__(self, other: object) -> bool:
         if isinstance(other, ApplicationConfig):
             return (
                 (self.stomp == other.stomp)
                 & (self.env == other.env)
                 & (self.logging == other.logging)
+                & (self.api == other.api)
             )
         return False
 

src/blueapi/service/__init__.py

@@ -1,4 +1,3 @@
-from .app import start
 from .model import DeviceModel, PlanModel
 
-__all__ = ["start", "PlanModel", "DeviceModel"]
+__all__ = ["PlanModel", "DeviceModel"]