@neon/tools 0.0.0 → 0.2.0

package/dist/mcp.js

@@ -0,0 +1,60 @@
+//#region src/mcp.ts
+const errorMessage = (error) => error instanceof Error ? error.message : String(error);
+const isRecord = (value) => typeof value === "object" && value !== null;
+const authInfoFrom = (context) => {
+	if (!isRecord(context)) return;
+	if (isRecord(context.http) && "authInfo" in context.http) return context.http.authInfo;
+	if ("authInfo" in context) return context.authInfo;
+};
+const bearerCredentialFromMcpContext = (context) => {
+	const authInfo = authInfoFrom(context);
+	if (authInfo === void 0) return;
+	if (!isRecord(authInfo) || typeof authInfo.token !== "string" || authInfo.token.length === 0) throw new TypeError("A Neon API key or OAuth access token is required");
+	return authInfo.token;
+};
+const signalFrom = (context) => {
+	if (!isRecord(context)) return;
+	if (context.signal instanceof AbortSignal) return context.signal;
+	if (isRecord(context.mcpReq) && context.mcpReq.signal instanceof AbortSignal) return context.mcpReq.signal;
+};
+const registerNeonTools = (server, tools) => {
+	if (typeof server.registerTool !== "function") throw new TypeError("Expected an MCP server with registerTool().");
+	for (const tool of Object.values(tools)) Reflect.apply(server.registerTool, server, [
+		tool.id,
+		{
+			title: tool.title,
+			description: tool.description,
+			inputSchema: tool.inputSchema,
+			annotations: tool.annotations,
+			_meta: { "neon/requiresApproval": tool.requiresApproval }
+		},
+		async (input, context) => {
+			try {
+				const apiKey = bearerCredentialFromMcpContext(context);
+				const structuredContent = { data: (await tool.execute(input, {
+					signal: signalFrom(context),
+					...apiKey === void 0 ? {} : { apiKey }
+				})).data };
+				return {
+					content: [{
+						type: "text",
+						text: JSON.stringify(structuredContent)
+					}],
+					structuredContent
+				};
+			} catch (error) {
+				const structuredContent = { error: { message: errorMessage(error) } };
+				return {
+					content: [{
+						type: "text",
+						text: JSON.stringify(structuredContent)
+					}],
+					structuredContent,
+					isError: true
+				};
+			}
+		}
+	]);
+};
+//#endregion
+export { registerNeonTools };