postgresai 0.11.0-alpha.4 → 0.11.0-alpha.6

package/bin/postgres-ai.js

@@ -123,7 +123,52 @@ program
     const code = await runCompose(args);
     if (code !== 0) process.exitCode = code;
   });
-program.command("health").description("health check").action(stub("health"));
+program
+  .command("health")
+  .description("health check")
+  .action(async () => {
+    const { exec } = require("child_process");
+    const util = require("util");
+    const execPromise = util.promisify(exec);
+    
+    console.log("Checking service health...\n");
+    
+    const services = [
+      { name: "Grafana", url: "http://localhost:3000/api/health" },
+      { name: "Prometheus", url: "http://localhost:59090/-/healthy" },
+      { name: "PGWatch (Postgres)", url: "http://localhost:58080/health" },
+      { name: "PGWatch (Prometheus)", url: "http://localhost:58089/health" },
+    ];
+    
+    let allHealthy = true;
+    
+    for (const service of services) {
+      try {
+        const { stdout, stderr } = await execPromise(
+          `curl -sf -o /dev/null -w "%{http_code}" ${service.url}`,
+          { timeout: 5000 }
+        );
+        const code = stdout.trim();
+        if (code === "200") {
+          console.log(`✓ ${service.name}: healthy`);
+        } else {
+          console.log(`✗ ${service.name}: unhealthy (HTTP ${code})`);
+          allHealthy = false;
+        }
+      } catch (error) {
+        console.log(`✗ ${service.name}: unreachable`);
+        allHealthy = false;
+      }
+    }
+    
+    console.log("");
+    if (allHealthy) {
+      console.log("All services are healthy");
+    } else {
+      console.log("Some services are unhealthy");
+      process.exitCode = 1;
+    }
+  });
 program
   .command("config")
   .description("show configuration")
@@ -151,7 +196,44 @@ program
   .command("reset [service]")
   .description("reset all or specific service")
   .action(stub("reset"));
-program.command("clean").description("cleanup artifacts").action(stub("clean"));
+program
+  .command("clean")
+  .description("cleanup artifacts")
+  .action(async () => {
+    const { exec } = require("child_process");
+    const util = require("util");
+    const execPromise = util.promisify(exec);
+    
+    console.log("Cleaning up Docker resources...\n");
+    
+    try {
+      // Remove stopped containers
+      const { stdout: containers } = await execPromise("docker ps -aq --filter 'status=exited'");
+      if (containers.trim()) {
+        await execPromise(`docker rm ${containers.trim().split('\n').join(' ')}`);
+        console.log("✓ Removed stopped containers");
+      } else {
+        console.log("✓ No stopped containers to remove");
+      }
+      
+      // Remove unused volumes
+      const { stdout: volumeOut } = await execPromise("docker volume prune -f");
+      console.log("✓ Removed unused volumes");
+      
+      // Remove unused networks
+      const { stdout: networkOut } = await execPromise("docker network prune -f");
+      console.log("✓ Removed unused networks");
+      
+      // Remove dangling images
+      const { stdout: imageOut } = await execPromise("docker image prune -f");
+      console.log("✓ Removed dangling images");
+      
+      console.log("\nCleanup completed");
+    } catch (error) {
+      console.error(`Error during cleanup: ${error.message}`);
+      process.exitCode = 1;
+    }
+  });
 program
   .command("shell <service>")
   .description("open service shell")
@@ -285,7 +367,69 @@ program
 program
   .command("test-instance <name>")
   .description("test instance connectivity")
-  .action(stub("test-instance"));
+  .action(async (name) => {
+    const fs = require("fs");
+    const path = require("path");
+    const { exec } = require("child_process");
+    const util = require("util");
+    const execPromise = util.promisify(exec);
+    
+    const instancesPath = path.resolve(process.cwd(), "instances.yml");
+    if (!fs.existsSync(instancesPath)) {
+      console.error("instances.yml not found");
+      process.exitCode = 1;
+      return;
+    }
+    
+    const content = fs.readFileSync(instancesPath, "utf8");
+    const lines = content.split(/\r?\n/);
+    let connStr = "";
+    let foundInstance = false;
+    
+    for (let i = 0; i < lines.length; i++) {
+      const nameLine = lines[i].match(/^-[\t ]*name:[\t ]*(.+)$/);
+      if (nameLine && nameLine[1].trim() === name) {
+        foundInstance = true;
+        // Look for conn_str in next lines
+        for (let j = i + 1; j < lines.length && j < i + 15; j++) {
+          const connLine = lines[j].match(/^[\t ]*conn_str:[\t ]*(.+)$/);
+          if (connLine) {
+            connStr = connLine[1].trim();
+            break;
+          }
+          // Stop at next instance
+          if (lines[j].match(/^-[\t ]*name:/)) break;
+        }
+        break;
+      }
+    }
+    
+    if (!foundInstance) {
+      console.error(`Instance '${name}' not found`);
+      process.exitCode = 1;
+      return;
+    }
+    
+    if (!connStr) {
+      console.error(`Connection string not found for instance '${name}'`);
+      process.exitCode = 1;
+      return;
+    }
+    
+    console.log(`Testing connection to '${name}'...`);
+    
+    try {
+      const { stdout, stderr } = await execPromise(
+        `psql "${connStr}" -c "SELECT version();" --no-psqlrc`,
+        { timeout: 10000, env: { ...process.env, PAGER: 'cat' } }
+      );
+      console.log(`✓ Connection successful`);
+      console.log(stdout.trim());
+    } catch (error) {
+      console.error(`✗ Connection failed: ${error.message}`);
+      process.exitCode = 1;
+    }
+  });
 
 // API key and grafana
 program
@@ -359,7 +503,36 @@ program
 program
   .command("show-grafana-credentials")
   .description("show Grafana credentials")
-  .action(stub("show-grafana-credentials"));
+  .action(async () => {
+    const fs = require("fs");
+    const path = require("path");
+    const cfgPath = path.resolve(process.cwd(), ".pgwatch-config");
+    if (!fs.existsSync(cfgPath)) {
+      console.error("Configuration file not found. Run 'quickstart' first.");
+      process.exitCode = 1;
+      return;
+    }
+    const content = fs.readFileSync(cfgPath, "utf8");
+    const lines = content.split(/\r?\n/);
+    let password = "";
+    for (const line of lines) {
+      const m = line.match(/^grafana_password=(.+)$/);
+      if (m) {
+        password = m[1].trim();
+        break;
+      }
+    }
+    if (!password) {
+      console.error("Grafana password not found in configuration");
+      process.exitCode = 1;
+      return;
+    }
+    console.log("\nGrafana credentials:");
+    console.log("  URL:      http://localhost:3000");
+    console.log("  Username: monitor");
+    console.log(`  Password: ${password}`);
+    console.log("");
+  });
 
 program.parseAsync(process.argv);
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "postgresai",
-  "version": "0.11.0-alpha.4",
+  "version": "0.11.0-alpha.6",
   "description": "PostgresAI CLI (Node.js)",
   "license": "MIT",
   "private": false,