bb1d662ee1
Rebrand the Rust client crate (coven/ → hh/, package+binary "hack-house"), README, CLI strings, and branch (coven → hack-house). Gitea repo renamed cmd-chat → hack-house to match. Crypto/server logic unchanged; selftest + golden-vector test still green, binary is now `hack-house`. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
116 lines
3.1 KiB
Python
116 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any, Callable, Optional
|
|
|
|
from sanic.app import Sanic
|
|
from sanic.config import Config
|
|
|
|
from .constructor import Constructor
|
|
|
|
|
|
class InjectionRegistry:
|
|
def __init__(self):
|
|
self._registry: dict[type, Optional[Callable[..., Any]]] = {}
|
|
|
|
def __getitem__(self, key):
|
|
return self._registry[key]
|
|
|
|
def __str__(self) -> str:
|
|
return str(self._registry)
|
|
|
|
def __contains__(self, other: Any):
|
|
return other in self._registry
|
|
|
|
def get(self, key, default=None):
|
|
return self._registry.get(key, default)
|
|
|
|
def register(
|
|
self,
|
|
_type: type,
|
|
constructor: Optional[Callable[..., Any]],
|
|
request_arg: Optional[str] = None,
|
|
) -> None:
|
|
constructor = constructor or _type
|
|
constructor = Constructor(constructor, request_arg=request_arg)
|
|
self._registry[_type] = constructor
|
|
|
|
def finalize(
|
|
self, app: Sanic, constant_registry: ConstantRegistry, allowed_types
|
|
):
|
|
for constructor in self._registry.values():
|
|
if isinstance(constructor, Constructor):
|
|
constructor.prepare(
|
|
app, self, constant_registry, allowed_types
|
|
)
|
|
|
|
@property
|
|
def length(self):
|
|
return len(self._registry)
|
|
|
|
|
|
class SignatureRegistry:
|
|
def __init__(self):
|
|
self._registry: dict[
|
|
str,
|
|
tuple[
|
|
dict[str, tuple[type, Optional[Callable[..., Any]]]],
|
|
dict[str, Any],
|
|
],
|
|
] = {}
|
|
|
|
def __getitem__(self, key):
|
|
return self._registry[key]
|
|
|
|
def __str__(self) -> str:
|
|
return str(self._registry)
|
|
|
|
def get(self, key, default=None):
|
|
return self._registry.get(key, default)
|
|
|
|
def register(
|
|
self,
|
|
route_name: str,
|
|
dependencies: dict[str, tuple[type, Optional[Callable[..., Any]]]],
|
|
constants: Optional[dict[str, Any]] = None,
|
|
) -> None:
|
|
self._registry[route_name] = (dependencies, constants or {})
|
|
|
|
|
|
class ConstantRegistry:
|
|
def __init__(self, config: Config):
|
|
self._config = config
|
|
self._registry: set[str] = set()
|
|
|
|
def __str__(self) -> str:
|
|
return str(self._registry)
|
|
|
|
def __iter__(self):
|
|
return iter(self._registry)
|
|
|
|
def __getitem__(self, key):
|
|
return self._registry[key]
|
|
|
|
def __contains__(self, other: Any):
|
|
return other in self._registry
|
|
|
|
def register(self, key: str, value: Any, overwrite: bool):
|
|
attribute = key.upper()
|
|
if attribute in self._config and not overwrite:
|
|
raise ValueError(
|
|
f"A value for {attribute} has already been assigned"
|
|
)
|
|
key = key.lower()
|
|
setattr(self._config, attribute, value)
|
|
return self._registry.add(key)
|
|
|
|
def get(self, key: str):
|
|
key = key.lower()
|
|
if key not in self._registry:
|
|
raise ValueError
|
|
attribute = key.upper()
|
|
return getattr(self._config, attribute)
|
|
|
|
@property
|
|
def length(self):
|
|
return len(self._registry)
|