mirror of
https://github.com/bunny-lab-io/Borealis.git
synced 2025-10-26 21:21:57 -06:00
Add TLS bootstrap support for Engine runtime
This commit is contained in:
@@ -2,25 +2,8 @@
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from .auth import (
|
||||
DeviceAuthService,
|
||||
DeviceRecord,
|
||||
RefreshTokenRecord,
|
||||
TokenRefreshError,
|
||||
TokenRefreshErrorCode,
|
||||
TokenService,
|
||||
)
|
||||
from .enrollment import (
|
||||
EnrollmentRequestResult,
|
||||
EnrollmentService,
|
||||
EnrollmentStatus,
|
||||
EnrollmentTokenBundle,
|
||||
PollingResult,
|
||||
)
|
||||
from Data.Engine.domain.device_enrollment import EnrollmentValidationError
|
||||
from .jobs.scheduler_service import SchedulerService
|
||||
from .github import GitHubService, GitHubTokenPayload
|
||||
from .realtime import AgentRealtimeService, AgentRecord
|
||||
from importlib import import_module
|
||||
from typing import Any, Dict, Tuple
|
||||
|
||||
__all__ = [
|
||||
"DeviceAuthService",
|
||||
@@ -41,3 +24,39 @@ __all__ = [
|
||||
"GitHubService",
|
||||
"GitHubTokenPayload",
|
||||
]
|
||||
|
||||
_LAZY_TARGETS: Dict[str, Tuple[str, str]] = {
|
||||
"DeviceAuthService": ("Data.Engine.services.auth.device_auth_service", "DeviceAuthService"),
|
||||
"DeviceRecord": ("Data.Engine.services.auth.device_auth_service", "DeviceRecord"),
|
||||
"RefreshTokenRecord": ("Data.Engine.services.auth.device_auth_service", "RefreshTokenRecord"),
|
||||
"TokenService": ("Data.Engine.services.auth.token_service", "TokenService"),
|
||||
"TokenRefreshError": ("Data.Engine.services.auth.token_service", "TokenRefreshError"),
|
||||
"TokenRefreshErrorCode": ("Data.Engine.services.auth.token_service", "TokenRefreshErrorCode"),
|
||||
"EnrollmentService": ("Data.Engine.services.enrollment.enrollment_service", "EnrollmentService"),
|
||||
"EnrollmentRequestResult": ("Data.Engine.services.enrollment.enrollment_service", "EnrollmentRequestResult"),
|
||||
"EnrollmentStatus": ("Data.Engine.services.enrollment.enrollment_service", "EnrollmentStatus"),
|
||||
"EnrollmentTokenBundle": ("Data.Engine.services.enrollment.enrollment_service", "EnrollmentTokenBundle"),
|
||||
"PollingResult": ("Data.Engine.services.enrollment.enrollment_service", "PollingResult"),
|
||||
"EnrollmentValidationError": ("Data.Engine.domain.device_enrollment", "EnrollmentValidationError"),
|
||||
"AgentRealtimeService": ("Data.Engine.services.realtime.agent_registry", "AgentRealtimeService"),
|
||||
"AgentRecord": ("Data.Engine.services.realtime.agent_registry", "AgentRecord"),
|
||||
"SchedulerService": ("Data.Engine.services.jobs.scheduler_service", "SchedulerService"),
|
||||
"GitHubService": ("Data.Engine.services.github.github_service", "GitHubService"),
|
||||
"GitHubTokenPayload": ("Data.Engine.services.github.github_service", "GitHubTokenPayload"),
|
||||
}
|
||||
|
||||
|
||||
def __getattr__(name: str) -> Any:
|
||||
try:
|
||||
module_name, attribute = _LAZY_TARGETS[name]
|
||||
except KeyError as exc:
|
||||
raise AttributeError(name) from exc
|
||||
|
||||
module = import_module(module_name)
|
||||
value = getattr(module, attribute)
|
||||
globals()[name] = value
|
||||
return value
|
||||
|
||||
|
||||
def __dir__() -> Any: # pragma: no cover - interactive helper
|
||||
return sorted(set(__all__))
|
||||
|
||||
Reference in New Issue
Block a user