60 lines
2.2 KiB
TypeScript
60 lines
2.2 KiB
TypeScript
|
/**
|
|||
|
|
* 能力目录（架构设计 2 / 4：registry 模块）。
|
|||
|
|
*
|
|||
|
|
* 保存已注册 Provider 与它们声明的能力；上层按身份取「能力清单」（A4 起按策略过滤，
|
|||
|
|
* 当前只做注册与查找）。能力目录是唯一来源——引擎的工具列表是它的投影。
|
|||
|
|
*/
|
|||
|
|
import type { Capability, Provider, ProviderManifest } from '@aps/provider-sdk';
|
|||
|
|
|
|||
|
|
export interface RegisteredProvider {
|
|||
|
|
manifest: ProviderManifest;
|
|||
|
|
provider: Provider;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export interface ResolvedCapability {
|
|||
|
|
capability: Capability;
|
|||
|
|
provider: Provider;
|
|||
|
|
providerId: string;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export class CapabilityRegistry {
|
|||
|
|
private readonly providers = new Map<string, RegisteredProvider>();
|
|||
|
|
|
|||
|
|
async register(provider: Provider): Promise<ProviderManifest> {
|
|||
|
|
const manifest = await provider.describe();
|
|||
|
|
if (manifest.platformVersion !== 1) {
|
|||
|
|
throw new Error(`Provider ${manifest.id} 声明兼容平台版本 ${manifest.platformVersion}，当前平台只支持 1`);
|
|||
|
|
}
|
|||
|
|
this.providers.set(manifest.id, { manifest, provider });
|
|||
|
|
return manifest;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
list(): Capability[] {
|
|||
|
|
return [...this.providers.values()].flatMap((entry) => entry.manifest.capabilities);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resolve(capabilityId: string): ResolvedCapability | undefined {
|
|||
|
|
for (const entry of this.providers.values()) {
|
|||
|
|
const capability = entry.manifest.capabilities.find((item) => item.id === capabilityId);
|
|||
|
|
if (capability) return { capability, provider: entry.provider, providerId: entry.manifest.id };
|
|||
|
|
}
|
|||
|
|
return undefined;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
providerIds(): string[] {
|
|||
|
|
return [...this.providers.keys()];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async health(): Promise<Array<{ providerId: string; status: string; detail?: string }>> {
|
|||
|
|
const results: Array<{ providerId: string; status: string; detail?: string }> = [];
|
|||
|
|
for (const [providerId, entry] of this.providers) {
|
|||
|
|
try {
|
|||
|
|
const health = await entry.provider.health();
|
|||
|
|
results.push(health.detail === undefined ? { providerId, status: health.status } : { providerId, status: health.status, detail: health.detail });
|
|||
|
|
} catch (error) {
|
|||
|
|
results.push({ providerId, status: 'down', detail: error instanceof Error ? error.message : '健康探测失败' });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return results;
|
|||
|
|
}
|
|||
|
|
}
|