dynamodb-toolbox
Version:
Lightweight and type-safe query builder for DynamoDB and TypeScript.
17 lines (16 loc) • 778 B
JavaScript
import { DatabaseAction } from '../../../database/index.js';
import { addAccessPatternTool } from './addAccessPatternTool.js';
import { addEntityTools } from './addEntityTools/index.js';
export class MCPToolkit extends DatabaseAction {
addTools(mcpServer, options = {}) {
for (const [dbTableKey, dbTable] of Object.entries(this.database.tables)) {
for (const entity of Object.values(dbTable.entities)) {
addEntityTools(mcpServer, entity, { ...options, dbTableKey });
}
for (const [dbAccessPatternKey, accessPattern] of Object.entries(dbTable.accessPatterns)) {
addAccessPatternTool(mcpServer, accessPattern, { dbTableKey, dbAccessPatternKey });
}
}
return this;
}
}