tigrbl-kms 0.0.1.dev1__py3-none-any.whl → 0.3.0.dev2__py3-none-any.whl

tigrbl_kms/__init__.py

@@ -1,14 +0,0 @@
-from .ExampleAgent import ExampleAgent as ExampleAgent
-
-__version__ = "0.6.0.dev26"
-__long_desc__ = """
-
-# Swarmauri Example Plugin
-
-This repository includes an example of a Swarmauri Plugin.
-
-Visit us at: https://swarmauri.com
-Follow us at: https://github.com/swarmauri
-Star us at: https://github.com/swarmauri/swarmauri-sdk
-
-"""

tigrbl_kms/__main__.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+import os
+import uvicorn
+
+
+def main() -> None:
+    """Run the tigrbl_kms server using uvicorn."""
+    host = os.getenv("HOST", "0.0.0.0")
+    port = int(os.getenv("PORT", "8000"))
+    log_level = os.getenv("TIGRBL_KMS_LOG_LEVEL", "debug").lower()
+    uvicorn.run(
+        "tigrbl_kms.app:app",
+        host=host,
+        port=port,
+        log_level=log_level,
+        proxy_headers=True,
+        forwarded_allow_ips="*",
+    )
+
+
+if __name__ == "__main__":
+    main()

tigrbl_kms/app.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+import os
+
+from tigrbl import TigrblApp
+from tigrbl.engine import engine as build_engine
+from swarmauri_crypto_paramiko import ParamikoCrypto
+from swarmauri_standard.key_providers import InMemoryKeyProvider
+from .orm import Key, KeyVersion
+
+DB_URL = os.getenv("KMS_DATABASE_URL", "sqlite+aiosqlite:///./kms.db")
+ENGINE = build_engine(DB_URL)
+
+
+# API-level hooks (v3): stash shared services into ctx before any handler runs
+async def _stash_ctx(ctx):
+    global CRYPTO, KEY_PROVIDER
+    try:
+        CRYPTO
+    except NameError:
+        CRYPTO = ParamikoCrypto()
+    try:
+        KEY_PROVIDER
+    except NameError:
+        KEY_PROVIDER = InMemoryKeyProvider()
+    ctx["crypto"] = CRYPTO
+    ctx["key_provider"] = KEY_PROVIDER
+
+
+app = TigrblApp(
+    title="Tigrbl-KMS",
+    version="0.1.0",
+    openapi_url="/openapi.json",
+    docs_url="/docs",
+    engine=ENGINE,
+    api_hooks={"*": {"PRE_TX_BEGIN": [_stash_ctx]}},
+)
+
+
+# Custom ops return raw dicts so no finalize hook needed
+app.include_models([Key, KeyVersion], base_prefix="/kms")
+app.mount_jsonrpc(prefix="/kms/rpc")
+app.attach_diagnostics(prefix="/system")
+
+
+# Initialize database tables on startup
+@app.on_event("startup")
+async def startup_event():
+    await app.initialize()

tigrbl_kms/cli.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+"""Entry point for the TigrblKMS command-line interface."""
+
+from __future__ import annotations
+
+import typer
+import uvicorn
+
+
+app = typer.Typer(help="CLI for TigrblKMS service")
+
+
+@app.command()
+def serve(
+    host: str = typer.Option("127.0.0.1", help="Bind host"),
+    port: int = typer.Option(8000, help="Bind port"),
+    reload: bool = typer.Option(True, help="Auto-reload on changes"),
+):
+    """Start the TigrblKMS FastAPI service via uvicorn."""
+    uvicorn.run(
+        "tigrbl_kms.app:app",
+        host=host,
+        port=port,
+        reload=reload,
+        factory=False,
+    )
+
+
+if __name__ == "__main__":
+    app()

tigrbl_kms/orm/__init__.py

@@ -0,0 +1,4 @@
+from .key import Key
+from .key_version import KeyVersion
+
+__all__ = ["Key", "KeyVersion"]