mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
oliver
4a23b715a2
本次迁移将网关/通道/工具/技能/脚本与协议资源集中到新结构，统一路径常量与脚本转发机制，减少顶层噪音并保证运行与测试行为一致。 Made-with: Cursor
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from .shared_types import GatewayRequestHandlers
|
|
from .validation import error_shape
|
|
|
|
|
|
def _ok(respond, payload: Any) -> None:
|
|
if callable(respond):
|
|
respond(True, payload, None, None)
|
|
|
|
|
|
def _bad(respond, message: str) -> None:
|
|
if callable(respond):
|
|
respond(False, None, error_shape("INVALID_REQUEST", message), None)
|
|
|
|
|
|
def _unavailable(respond, message: str) -> None:
|
|
if callable(respond):
|
|
respond(False, None, error_shape("UNAVAILABLE", message), None)
|
|
|
|
|
|
def _norm_str(v: Any) -> str | None:
|
|
if isinstance(v, str):
|
|
s = v.strip()
|
|
return s or None
|
|
return None
|
|
|
|
|
|
def _resolve_provider(context: Any) -> dict[str, Any] | None:
|
|
hook = context.get("resolve_web_login_provider") if isinstance(context, dict) else None
|
|
if callable(hook):
|
|
out = hook()
|
|
if isinstance(out, dict):
|
|
return out
|
|
return None
|
|
|
|
|
|
def _web_login_start_handler(opts: dict[str, Any]) -> None:
|
|
params = opts.get("params")
|
|
respond = opts.get("respond")
|
|
context = opts.get("context")
|
|
if params is not None and not isinstance(params, dict):
|
|
_bad(respond, "invalid web.login.start params")
|
|
return
|
|
p = dict(params or {})
|
|
provider = _resolve_provider(context)
|
|
if not provider:
|
|
_bad(respond, "web login provider is not available")
|
|
return
|
|
start_fn = provider.get("loginWithQrStart")
|
|
if not callable(start_fn):
|
|
_bad(respond, f"web login is not supported by provider {provider.get('id')}")
|
|
return
|
|
account_id = _norm_str(p.get("accountId"))
|
|
try:
|
|
stop_channel = context.get("stopChannel") if isinstance(context, dict) else None
|
|
if callable(stop_channel):
|
|
stop_channel(provider.get("id"), account_id)
|
|
result = start_fn(
|
|
{
|
|
"force": bool(p.get("force")),
|
|
"timeoutMs": p.get("timeoutMs") if isinstance(p.get("timeoutMs"), (int, float)) else None,
|
|
"verbose": bool(p.get("verbose")),
|
|
"accountId": account_id,
|
|
}
|
|
)
|
|
if not isinstance(result, dict):
|
|
result = {}
|
|
connected = bool(result.get("connected"))
|
|
if connected:
|
|
start_channel = context.get("startChannel") if isinstance(context, dict) else None
|
|
if callable(start_channel):
|
|
start_channel(provider.get("id"), account_id)
|
|
_ok(respond, result)
|
|
except Exception as exc:
|
|
_unavailable(respond, str(exc))
|
|
|
|
|
|
def _web_login_wait_handler(opts: dict[str, Any]) -> None:
|
|
params = opts.get("params")
|
|
respond = opts.get("respond")
|
|
context = opts.get("context")
|
|
if params is not None and not isinstance(params, dict):
|
|
_bad(respond, "invalid web.login.wait params")
|
|
return
|
|
p = dict(params or {})
|
|
provider = _resolve_provider(context)
|
|
if not provider:
|
|
_bad(respond, "web login provider is not available")
|
|
return
|
|
wait_fn = provider.get("loginWithQrWait")
|
|
if not callable(wait_fn):
|
|
_bad(respond, f"web login is not supported by provider {provider.get('id')}")
|
|
return
|
|
account_id = _norm_str(p.get("accountId"))
|
|
try:
|
|
result = wait_fn(
|
|
{
|
|
"timeoutMs": p.get("timeoutMs") if isinstance(p.get("timeoutMs"), (int, float)) else None,
|
|
"accountId": account_id,
|
|
}
|
|
)
|
|
if not isinstance(result, dict):
|
|
result = {}
|
|
if bool(result.get("connected")):
|
|
start_channel = context.get("startChannel") if isinstance(context, dict) else None
|
|
if callable(start_channel):
|
|
start_channel(provider.get("id"), account_id)
|
|
_ok(respond, result)
|
|
except Exception as exc:
|
|
_unavailable(respond, str(exc))
|
|
|
|
|
|
web_handlers: GatewayRequestHandlers = {
|
|
"web.login.start": _web_login_start_handler,
|
|
"web.login.wait": _web_login_wait_handler,
|
|
}
|
|
|