postgresai 0.11.0-alpha.12 → 0.11.0-alpha.13

package/lib/mcp-server.ts

@@ -0,0 +1,98 @@
+import * as pkg from "../package.json";
+import * as config from "./config";
+import { fetchIssues } from "./issues";
+import { resolveBaseUrls } from "./util";
+
+// MCP SDK imports
+import { Server } from "@modelcontextprotocol/sdk/server";
+import * as path from "path";
+// Types schemas will be loaded dynamically from the SDK's CJS bundle
+
+interface RootOptsLike {
+  apiKey?: string;
+  apiBaseUrl?: string;
+}
+
+export async function startMcpServer(rootOpts?: RootOptsLike, extra?: { debug?: boolean }): Promise<void> {
+  // Resolve stdio transport at runtime to avoid subpath export resolution issues
+  const serverEntry = require.resolve("@modelcontextprotocol/sdk/server");
+  const stdioPath = path.join(path.dirname(serverEntry), "stdio.js");
+  // eslint-disable-next-line @typescript-eslint/no-var-requires
+  const { StdioServerTransport } = require(stdioPath);
+  // Load schemas dynamically to avoid subpath export resolution issues
+  const typesPath = path.resolve(path.dirname(serverEntry), "../types.js");
+  // eslint-disable-next-line @typescript-eslint/no-var-requires
+  const { CallToolRequestSchema, ListToolsRequestSchema } = require(typesPath);
+
+  const server = new Server(
+    { name: "postgresai-mcp", version: pkg.version },
+    { capabilities: { tools: {} } }
+  );
+
+  server.setRequestHandler(ListToolsRequestSchema, async () => {
+    return {
+      tools: [
+        {
+          name: "list_issues",
+          description: "List issues from PostgresAI API (same as CLI 'issues list')",
+          inputSchema: {
+            type: "object",
+            properties: {
+              debug: { type: "boolean", description: "Enable verbose debug logs" },
+            },
+            additionalProperties: false,
+          },
+        },
+      ],
+    };
+  });
+
+  server.setRequestHandler(CallToolRequestSchema, async (req: any) => {
+    const toolName = req.params.name;
+    const args = (req.params.arguments as Record<string, unknown>) || {};
+
+    if (toolName !== "list_issues") {
+      throw new Error(`Unknown tool: ${toolName}`);
+    }
+
+    const cfg = config.readConfig();
+    const apiKey = (rootOpts?.apiKey || process.env.PGAI_API_KEY || cfg.apiKey || "").toString();
+    const { apiBaseUrl } = resolveBaseUrls(rootOpts, cfg);
+
+    const debug = Boolean(args.debug ?? extra?.debug);
+
+    if (!apiKey) {
+      return {
+        content: [
+          {
+            type: "text",
+            text: "API key is required. Run 'pgai auth' or set PGAI_API_KEY.",
+          },
+        ],
+        isError: true,
+      };
+    }
+
+    try {
+      const result = await fetchIssues({ apiKey, apiBaseUrl, debug });
+      return {
+        content: [
+          { type: "text", text: JSON.stringify(result, null, 2) },
+        ],
+      };
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      return {
+        content: [
+          { type: "text", text: message },
+        ],
+        isError: true,
+      };
+    }
+  });
+
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+}
+
+

package/lib/util.ts

@@ -0,0 +1,60 @@
+export function maskSecret(secret: string): string {
+  if (!secret) return "";
+  if (secret.length <= 8) return "****";
+  if (secret.length <= 16) return `${secret.slice(0, 4)}${"*".repeat(secret.length - 8)}${secret.slice(-4)}`;
+  return `${secret.slice(0, Math.min(12, secret.length - 8))}${"*".repeat(Math.max(4, secret.length - 16))}${secret.slice(-4)}`;
+}
+
+
+export interface RootOptsLike {
+  apiBaseUrl?: string;
+  uiBaseUrl?: string;
+}
+
+export interface ConfigLike {
+  baseUrl?: string | null;
+}
+
+export interface ResolvedBaseUrls {
+  apiBaseUrl: string;
+  uiBaseUrl: string;
+}
+
+/**
+ * Normalize a base URL by trimming a single trailing slash and validating.
+ * @throws Error if the URL is invalid
+ */
+export function normalizeBaseUrl(value: string): string {
+  const trimmed = (value || "").replace(/\/$/, "");
+  try {
+    // Validate
+    // eslint-disable-next-line no-new
+    new URL(trimmed);
+  } catch {
+    throw new Error(`Invalid base URL: ${value}`);
+  }
+  return trimmed;
+}
+
+/**
+ * Resolve API and UI base URLs using precedence and normalize them.
+ * Precedence (API): opts.apiBaseUrl → env.PGAI_API_BASE_URL → cfg.baseUrl → default
+ * Precedence (UI):  opts.uiBaseUrl  → env.PGAI_UI_BASE_URL  → default
+ */
+export function resolveBaseUrls(
+  opts?: RootOptsLike,
+  cfg?: ConfigLike,
+  defaults: { apiBaseUrl?: string; uiBaseUrl?: string } = {}
+): ResolvedBaseUrls {
+  const defApi = defaults.apiBaseUrl || "https://postgres.ai/api/general/";
+  const defUi = defaults.uiBaseUrl || "https://console.postgres.ai";
+
+  const apiCandidate = (opts?.apiBaseUrl || process.env.PGAI_API_BASE_URL || cfg?.baseUrl || defApi) as string;
+  const uiCandidate = (opts?.uiBaseUrl || process.env.PGAI_UI_BASE_URL || defUi) as string;
+
+  return {
+    apiBaseUrl: normalizeBaseUrl(apiCandidate),
+    uiBaseUrl: normalizeBaseUrl(uiCandidate),
+  };
+}
+

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "postgresai",
-  "version": "0.11.0-alpha.12",
+  "version": "0.11.0-alpha.13",
   "description": "postgres_ai CLI (Node.js)",
   "license": "Apache-2.0",
   "private": false,
@@ -29,7 +29,8 @@
   },
   "dependencies": {
     "commander": "^12.1.0",
-    "js-yaml": "^4.1.0"
+    "js-yaml": "^4.1.0",
+    "@modelcontextprotocol/sdk": "^1.20.2"
   },
   "devDependencies": {
     "@types/js-yaml": "^4.0.9",

package/tsconfig.json

@@ -1,7 +1,7 @@
 {
   "compilerOptions": {
     "target": "ES2020",
-    "module": "commonjs",
+    "module": "node16",
     "lib": ["ES2020"],
     "outDir": "./dist",
     "rootDir": "./",
@@ -13,7 +13,7 @@
     "declaration": true,
     "declarationMap": true,
     "sourceMap": true,
-    "moduleResolution": "node",
+    "moduleResolution": "node16",
     "types": ["node"]
   },
   "include": [