vibex-sh 0.2.0 → 0.2.1

package/index.js

@@ -377,10 +377,13 @@ async function main() {
   
   // Auto-claim session if token is available and fetch auth code
   let authCode = null;
-  if (token && !options.sessionId) {
-    // Only auto-claim new sessions (not when reusing existing session)
+  if (token) {
+    // Try to claim session (works for both new and existing sessions)
+    // For new sessions, this will create and claim
+    // For existing sessions, this will return the auth code if user owns it
     authCode = await claimSession(sessionId, token, webUrl);
-    if (authCode) {
+    if (authCode && !options.sessionId) {
+      // Only show claim message for new sessions
       console.log('  ✓ Session automatically claimed to your account\n');
     }
   }
@@ -446,11 +449,15 @@ async function main() {
 
   // Listen for auth code from socket.io (for unclaimed sessions)
   socket.on('session-auth-code', (data) => {
-    if (data.sessionId === sessionId && data.authCode && !receivedAuthCode) {
-      receivedAuthCode = data.authCode;
-      // Display auth code when received (for both new and existing sessions)
-      console.log(`  🔑 Auth Code: ${receivedAuthCode}`);
-      console.log(`  📋 Dashboard: ${webUrl}/${sessionId}?auth=${receivedAuthCode}\n`);
+    if (data.sessionId === sessionId && data.authCode) {
+      // Always update and display auth code if received from socket
+      // This ensures we show it even if we didn't get it from claimSession
+      if (!receivedAuthCode || receivedAuthCode !== data.authCode) {
+        receivedAuthCode = data.authCode;
+        // Display auth code when received (for both new and existing sessions)
+        console.log(`  🔑 Auth Code: ${receivedAuthCode}`);
+        console.log(`  📋 Dashboard: ${webUrl}/${sessionId}?auth=${receivedAuthCode}\n`);
+      }
     }
   });
 
@@ -502,6 +509,48 @@ async function main() {
     }
   });
 
+  // Handle rate limit errors from server
+  socket.on('rate-limit-exceeded', (data) => {
+    console.error('\n  ⚠️  Rate Limit Exceeded');
+    console.error(`  ${data.message || 'Too many requests. Please try again later.'}`);
+    if (data.rateLimit) {
+      const { limit, remaining, resetAt, windowSeconds } = data.rateLimit;
+      if (limit !== undefined) {
+        console.error(`  Limit: ${limit} requests`);
+      }
+      if (remaining !== undefined) {
+        console.error(`  Remaining: ${remaining} requests`);
+      }
+      if (resetAt) {
+        const resetDate = new Date(resetAt);
+        const now = new Date();
+        const secondsUntilReset = Math.ceil((resetDate - now) / 1000);
+        if (secondsUntilReset > 0) {
+          console.error(`  Resets in: ${secondsUntilReset} seconds`);
+        }
+      }
+    }
+    console.error('');
+    // Don't exit - let user decide, but clear the queue
+    logQueue.length = 0;
+  });
+
+  // Handle general errors from server
+  socket.on('error', (data) => {
+    console.error('\n  ✗ Server Error');
+    if (typeof data === 'string') {
+      console.error(`  ${data}`);
+    } else if (data && data.message) {
+      console.error(`  ${data.message}`);
+      if (data.error) {
+        console.error(`  Error: ${data.error}`);
+      }
+    } else {
+      console.error('  An unexpected error occurred');
+    }
+    console.error('');
+  });
+
   const rl = readline.createInterface({
     input: process.stdin,
     output: process.stdout,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vibex-sh",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Zero-config observability CLI - pipe logs and visualize instantly",
   "type": "module",
   "bin": {