Replace psql dependency with native pg client for connection testing

NikolayS · NikolayS · commit 2469a5e4bb9e · 2025-10-30T16:47:08.000-07:00
diff --git a/cli/bin/postgres-ai.ts b/cli/bin/postgres-ai.ts
@@ -645,13 +645,18 @@ targets
       
       console.log(`Testing connection to monitoring target '${name}'...`);
       
-      const { stdout, stderr } = await execFilePromise(
-        "psql",
-        [instance.conn_str, "-c", "SELECT version();", "--no-psqlrc"],
-        { timeout: 10000, env: { ...process.env, PAGER: 'cat' } }
-      );
-      console.log(`✓ Connection successful`);
-      console.log(stdout.trim());
+      // Use native pg client instead of requiring psql to be installed
+      const { Client } = require('pg');
+      const client = new Client({ connectionString: instance.conn_str });
+      
+      try {
+        await client.connect();
+        const result = await client.query('select version();');
+        console.log(`✓ Connection successful`);
+        console.log(result.rows[0].version);
+      } finally {
+        await client.end();
+      }
     } catch (error) {
       const message = error instanceof Error ? error.message : String(error);
       console.error(`✗ Connection failed: ${message}`);
diff --git a/cli/package-lock.json b/cli/package-lock.json
diff --git a/cli/package.json b/cli/package.json
@@ -28,13 +28,15 @@
     "dev": "tsc --watch"
   },
   "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.20.2",
     "commander": "^12.1.0",
     "js-yaml": "^4.1.0",
-    "@modelcontextprotocol/sdk": "^1.20.2"
+    "pg": "^8.16.3"
   },
   "devDependencies": {
     "@types/js-yaml": "^4.0.9",
     "@types/node": "^18.19.0",
+    "@types/pg": "^8.15.6",
     "typescript": "^5.3.3"
   },
   "publishConfig": {
