Spaces:
Paused
Paused
File size: 2,662 Bytes
529090e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 | /**
* Simple Source Registry Implementation
*
* Manages available data sources and matches them to query intents
*/
import { QueryIntent, DataSource } from './autonomous/index.js';
export class SourceRegistryImpl {
private sources: Map<string, DataSource> = new Map();
/**
* Register a new data source
*/
registerSource(source: DataSource): void {
this.sources.set(source.name, source);
console.log(`📌 Registered source: ${source.name} (${source.type})`);
}
/**
* Get sources capable of handling a query intent
*/
getCapableSources(intent: QueryIntent): DataSource[] {
const capable: DataSource[] = [];
for (const source of this.sources.values()) {
if (this.canHandle(source, intent)) {
capable.push(source);
}
}
return capable;
}
/**
* Get all registered sources
*/
getAllSources(): DataSource[] {
return Array.from(this.sources.values());
}
/**
* Get source by name
*/
getSource(name: string): DataSource | undefined {
return this.sources.get(name);
}
/**
* Check if source can handle intent
*/
private canHandle(source: DataSource, intent: QueryIntent): boolean {
// Check if source has wildcard capability
if (source.capabilities.includes('*')) {
return true;
}
// Check for domain.operation match
const fullType = `${intent.domain}.${intent.operation}`;
if (source.capabilities.includes(fullType)) {
return true;
}
// Check for domain.* match
const domainWildcard = `${intent.domain}.*`;
if (source.capabilities.includes(domainWildcard)) {
return true;
}
// Check for simple type match
if (source.capabilities.includes(intent.type)) {
return true;
}
return false;
}
/**
* Get sources by type
*/
getSourcesByType(type: string): DataSource[] {
return Array.from(this.sources.values())
.filter(s => s.type === type);
}
/**
* Remove a source
*/
unregisterSource(name: string): boolean {
return this.sources.delete(name);
}
/**
* Clear all sources
*/
clear(): void {
this.sources.clear();
}
}
// Singleton instance
let registryInstance: SourceRegistryImpl | null = null;
export function getSourceRegistry(): SourceRegistryImpl {
if (!registryInstance) {
registryInstance = new SourceRegistryImpl();
}
return registryInstance;
}
|