88 lines
2.1 KiB
JavaScript
88 lines
2.1 KiB
JavaScript
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
|
|
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
|
import { CallToolRequestSchema, ListResourcesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema } from "@modelcontextprotocol/sdk/types.js";
|
|
import pg from "pg";
|
|
|
|
const args = process.argv.slice(2);
|
|
console.log("argv", args);
|
|
if (args.length === 0) {
|
|
console.error("missing DSN");
|
|
process.exit(1);
|
|
}
|
|
const databaseUrl = args[0];
|
|
console.log("dsn", databaseUrl);
|
|
|
|
const server = new Server({
|
|
name: "debug",
|
|
version: "0.0.0",
|
|
}, {
|
|
capabilities: {
|
|
resources: {},
|
|
tools: {},
|
|
},
|
|
});
|
|
|
|
const pool = new pg.Pool({ connectionString: databaseUrl });
|
|
|
|
const SCHEMA_PATH = "schema";
|
|
|
|
server.onerror = (err) => {
|
|
console.error("server transport error", err);
|
|
};
|
|
|
|
server.setRequestHandler(ListResourcesRequestSchema, async () => {
|
|
const client = await pool.connect();
|
|
try {
|
|
const result = await client.query("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'");
|
|
return {
|
|
resources: result.rows.map((row) => ({
|
|
uri: row.table_name,
|
|
mimeType: "application/json",
|
|
name: row.table_name,
|
|
})),
|
|
};
|
|
} finally {
|
|
client.release();
|
|
}
|
|
});
|
|
|
|
server.setRequestHandler(ReadResourceRequestSchema, async () => ({
|
|
contents: [],
|
|
}));
|
|
|
|
server.setRequestHandler(ListToolsRequestSchema, async () => ({
|
|
tools: [],
|
|
}));
|
|
|
|
server.setRequestHandler(CallToolRequestSchema, async () => ({
|
|
content: [],
|
|
isError: false,
|
|
}));
|
|
|
|
process.on("exit", (code) => {
|
|
console.log("process exit", code);
|
|
});
|
|
|
|
process.on("beforeExit", (code) => {
|
|
console.log("beforeExit", code);
|
|
});
|
|
|
|
process.on("uncaughtException", (err) => {
|
|
console.error("uncaughtException", err);
|
|
});
|
|
|
|
process.on("unhandledRejection", (reason) => {
|
|
console.error("unhandledRejection", reason);
|
|
});
|
|
|
|
async function main() {
|
|
const transport = new StdioServerTransport();
|
|
await server.connect(transport);
|
|
console.log("server connected");
|
|
}
|
|
|
|
main().catch((err) => {
|
|
console.error("main error", err);
|
|
process.exit(1);
|
|
});
|