bunny-lab/Borealis-Github-Replica
1
0
Fork 0
mirror of https://github.com/bunny-lab-io/Borealis.git synced 2025-12-16 06:45:48 -07:00
Code Issues Packages Projects Releases Wiki Activity

Fixed Several Bugs/Errors from the Engine Tunnel Implementation

Browse Source
This commit is contained in:
Nicole Rappe
2025-12-01 01:57:46 -07:00
parent db8dd423f6
commit 478237f487
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Data/Engine/services/WebSocket/Agent/ReverseTunnel.py
View File

@@ -36,7 +36,7 @@ except Exception: # pragma: no cover - dependency resolved at runtime
websockets = None websockets = None
ws_serve = None ws_serve = None
from .....server import EngineContext from ....server import EngineContext
TunnelState = str TunnelState = str

7
Data/Engine/services/WebSocket/__init__.py
View File

@@ -15,8 +15,8 @@ from dataclasses import dataclass, field
from pathlib import Path from pathlib import Path
from typing import Any, Callable, Dict, Optional from typing import Any, Callable, Dict, Optional
from flask_socketio import SocketIO
from flask import session, request from flask import session, request
from flask_socketio import SocketIO
from ...database import initialise_engine_database from ...database import initialise_engine_database
from ...server import EngineContext from ...server import EngineContext
@@ -26,7 +26,6 @@ from .Agent.ReverseTunnel import (
decode_frame, decode_frame,
TunnelFrame, TunnelFrame,
) )
from ..API import _make_db_conn_factory, _make_service_logger
def _now_ts() -> int: def _now_ts() -> int:
@@ -51,6 +50,8 @@ class EngineRealtimeAdapters:
service_log: Callable[[str, str, Optional[str]], None] = field(init=False) service_log: Callable[[str, str, Optional[str]], None] = field(init=False)
def __post_init__(self) -> None: def __post_init__(self) -> None:
from ..API import _make_db_conn_factory, _make_service_logger # Local import to avoid circular import at module load
initialise_engine_database(self.context.database_path, logger=self.context.logger) initialise_engine_database(self.context.database_path, logger=self.context.logger)
self.db_conn_factory = _make_db_conn_factory(self.context.database_path) self.db_conn_factory = _make_db_conn_factory(self.context.database_path)
@@ -69,6 +70,8 @@ class EngineRealtimeAdapters:
def register_realtime(socket_server: SocketIO, context: EngineContext) -> None: def register_realtime(socket_server: SocketIO, context: EngineContext) -> None:
"""Register Socket.IO event handlers for the Engine runtime.""" """Register Socket.IO event handlers for the Engine runtime."""
from ..API import _make_db_conn_factory, _make_service_logger # Local import to avoid circular import at module load
adapters = EngineRealtimeAdapters(context) adapters = EngineRealtimeAdapters(context)
logger = context.logger.getChild("realtime.quick_jobs") logger = context.logger.getChild("realtime.quick_jobs")
tunnel_service = getattr(context, "reverse_tunnel_service", None) tunnel_service = getattr(context, "reverse_tunnel_service", None)