mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:20:44 +08:00
oliver
4a23b715a2
本次迁移将网关/通道/工具/技能/脚本与协议资源集中到新结构，统一路径常量与脚本转发机制，减少顶层噪音并保证运行与测试行为一致。 Made-with: Cursor
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import asdict
|
|
from typing import Any
|
|
|
|
from oclaw.platform.persistence.sqlite_store import SqliteStore
|
|
from oclaw.runtime.tools.mcp.manifest import McpServerManifest
|
|
|
|
|
|
class McpRegistry:
|
|
def __init__(self, store: SqliteStore):
|
|
self.store = store
|
|
|
|
def upsert_manifest(self, manifest: McpServerManifest) -> None:
|
|
self.store.upsert_mcp_server(
|
|
server_id=manifest.server_id,
|
|
source_type=manifest.source_type,
|
|
source_ref=manifest.source_ref,
|
|
version=manifest.version,
|
|
entry_command=manifest.entry_command,
|
|
entry_args=manifest.entry_args,
|
|
env_schema=manifest.env_schema,
|
|
required_permissions=manifest.permissions,
|
|
risk_level=manifest.risk_level,
|
|
timeout_s=manifest.timeout_s,
|
|
enabled=manifest.enabled,
|
|
)
|
|
|
|
def list_servers(self, *, enabled_only: bool = False) -> list[dict[str, Any]]:
|
|
return self.store.list_mcp_servers(enabled_only=enabled_only)
|
|
|
|
def snapshot(self) -> dict[str, Any]:
|
|
rows = self.list_servers(enabled_only=False)
|
|
return {"count": len(rows), "servers": rows}
|
|
|
|
@staticmethod
|
|
def manifest_to_dict(manifest: McpServerManifest) -> dict[str, Any]:
|
|
return asdict(manifest)
|
|
|
|
|
|
__all__ = ["McpRegistry"]
|
|
|