@optima-chat/dev-skills 0.7.9 → 0.7.10

package/bin/helpers/query-db.ts

@@ -214,20 +214,42 @@ async function main() {
       { encoding: 'utf-8' }
     );
 
+    console.log('\n' + result);
+  } else if (environment === 'stage') {
+    // Stage 环境：直连 RDS（Stage RDS 在公有子网，可以本地直连）
+    const infisicalConfig = getInfisicalConfig();
+    console.log('✓ Loaded Infisical config from GitHub Variables');
+
+    const token = getInfisicalToken(infisicalConfig);
+    console.log('✓ Obtained Infisical access token');
+
+    const secrets = getInfisicalSecrets(infisicalConfig, token, 'staging');
+    console.log('✓ Retrieved database credentials from Infisical');
+
+    const { userKey, passwordKey, database } = serviceConfig as any;
+    const dbHost = RDS_HOSTS.stage;
+    const dbUser = secrets[userKey];
+    const dbPassword = secrets[passwordKey];
+
+    if (!dbUser || !dbPassword) {
+      throw new Error(`Database credentials not found in Infisical for ${service}. Keys: ${userKey}, ${passwordKey}`);
+    }
+
+    const result = queryDatabase(dbHost, 5432, dbUser, dbPassword, database, sql);
     console.log('\n' + result);
   } else {
-    // Stage/Prod 环境：通过 SSH 隧道访问 RDS
+    // Prod 环境：通过 SSH 隧道访问 RDS（Prod RDS 在私有子网）
     const infisicalConfig = getInfisicalConfig();
     console.log('✓ Loaded Infisical config from GitHub Variables');
 
     const token = getInfisicalToken(infisicalConfig);
     console.log('✓ Obtained Infisical access token');
 
-    const secrets = getInfisicalSecrets(infisicalConfig, token, environment === 'stage' ? 'staging' : 'prod');
+    const secrets = getInfisicalSecrets(infisicalConfig, token, 'prod');
     console.log('✓ Retrieved database credentials from Infisical');
 
     const { userKey, passwordKey, database } = serviceConfig as any;
-    const dbHost = RDS_HOSTS[environment as 'stage' | 'prod'];
+    const dbHost = RDS_HOSTS.prod;
     const dbUser = secrets[userKey];
     const dbPassword = secrets[passwordKey];
 
@@ -235,7 +257,7 @@ async function main() {
       throw new Error(`Database credentials not found in Infisical for ${service}. Keys: ${userKey}, ${passwordKey}`);
     }
 
-    const localPort = environment === 'stage' ? 15432 : 15433;
+    const localPort = 15433;
 
     setupSSHTunnel(EC2_HOST, dbHost, localPort);
 

package/dist/bin/helpers/query-db.js

@@ -195,22 +195,40 @@ async function main() {
         const result = (0, child_process_1.execSync)(`sshpass -p "${ciPassword}" ssh -o StrictHostKeyChecking=no ${ciUser}@${ciHost} "docker exec ${container} psql -U ${user} -d ${database} -c \\"${sql}\\""`, { encoding: 'utf-8' });
         console.log('\n' + result);
     }
+    else if (environment === 'stage') {
+        // Stage 环境：直连 RDS（Stage RDS 在公有子网，可以本地直连）
+        const infisicalConfig = getInfisicalConfig();
+        console.log('✓ Loaded Infisical config from GitHub Variables');
+        const token = getInfisicalToken(infisicalConfig);
+        console.log('✓ Obtained Infisical access token');
+        const secrets = getInfisicalSecrets(infisicalConfig, token, 'staging');
+        console.log('✓ Retrieved database credentials from Infisical');
+        const { userKey, passwordKey, database } = serviceConfig;
+        const dbHost = RDS_HOSTS.stage;
+        const dbUser = secrets[userKey];
+        const dbPassword = secrets[passwordKey];
+        if (!dbUser || !dbPassword) {
+            throw new Error(`Database credentials not found in Infisical for ${service}. Keys: ${userKey}, ${passwordKey}`);
+        }
+        const result = queryDatabase(dbHost, 5432, dbUser, dbPassword, database, sql);
+        console.log('\n' + result);
+    }
     else {
-        // Stage/Prod 环境：通过 SSH 隧道访问 RDS
+        // Prod 环境：通过 SSH 隧道访问 RDS（Prod RDS 在私有子网）
         const infisicalConfig = getInfisicalConfig();
         console.log('✓ Loaded Infisical config from GitHub Variables');
         const token = getInfisicalToken(infisicalConfig);
         console.log('✓ Obtained Infisical access token');
-        const secrets = getInfisicalSecrets(infisicalConfig, token, environment === 'stage' ? 'staging' : 'prod');
+        const secrets = getInfisicalSecrets(infisicalConfig, token, 'prod');
         console.log('✓ Retrieved database credentials from Infisical');
         const { userKey, passwordKey, database } = serviceConfig;
-        const dbHost = RDS_HOSTS[environment];
+        const dbHost = RDS_HOSTS.prod;
         const dbUser = secrets[userKey];
         const dbPassword = secrets[passwordKey];
         if (!dbUser || !dbPassword) {
             throw new Error(`Database credentials not found in Infisical for ${service}. Keys: ${userKey}, ${passwordKey}`);
         }
-        const localPort = environment === 'stage' ? 15432 : 15433;
+        const localPort = 15433;
         setupSSHTunnel(EC2_HOST, dbHost, localPort);
         // 等待隧道建立
         await new Promise(resolve => setTimeout(resolve, 1000));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@optima-chat/dev-skills",
-  "version": "0.7.9",
+  "version": "0.7.10",
   "description": "Claude Code Skills for Optima development team - cross-environment collaboration tools",
   "main": "index.js",
   "bin": {