vibecodingmachine-cli 2026.1.3-2209 → 2026.1.23-1010

package/src/utils/auth.js

@@ -50,7 +50,7 @@ class CLIAuth {
 
         // Save new tokens
         await sharedAuth.saveToken(newTokens);
-        
+
         // Update user activity in database
         await this._updateUserActivity();
         return true;
@@ -72,11 +72,16 @@ class CLIAuth {
 
   /**
    * Get auth token string for API requests
+   * Ensures the token is valid and refreshed if necessary
    */
   async getAuthToken() {
+    // Check if authenticated and refresh if needed
+    const isAuth = await this.isAuthenticated();
+    if (!isAuth) return null;
+
     const token = await sharedAuth.getToken();
     if (!token) return null;
-    
+
     // Handle both string token and object with id_token
     return typeof token === 'string' ? token : token.id_token;
   }
@@ -718,10 +723,10 @@ class CLIAuth {
     try {
       // Decode JWT to get user info (without verification since we already validated)
       const payload = JSON.parse(Buffer.from(idToken.split('.')[1], 'base64').toString());
-      
+
       const UserDatabase = require('vibecodingmachine-core/src/database/user-schema');
       const userDb = new UserDatabase();
-      
+
       const userInfo = {
         email: payload.email,
         name: payload.name || payload.email.split('@')[0],
@@ -730,7 +735,7 @@ class CLIAuth {
 
       // Register/update user
       const user = await userDb.registerUser(userInfo);
-      
+
       // Register computer
       await userDb.registerComputer(user.userId, {
         interface: 'cli'
@@ -763,12 +768,12 @@ class CLIAuth {
 
       // Decode JWT to get user info
       const payload = JSON.parse(Buffer.from(token.id_token.split('.')[1], 'base64').toString());
-      
+
       const UserDatabase = require('vibecodingmachine-core/src/database/user-schema');
       const userDb = new UserDatabase();
-      
+
       const userId = userDb.generateUserId(payload.email);
-      
+
       // Update last activity
       await userDb.updateUserActivity(userId, {
         lastActivity: Date.now()
@@ -793,12 +798,12 @@ class CLIAuth {
       if (!idToken) return;
 
       const payload = JSON.parse(Buffer.from(idToken.split('.')[1], 'base64').toString());
-      
+
       const UserDatabase = require('vibecodingmachine-core/src/database/user-schema');
       const userDb = new UserDatabase();
-      
+
       const userId = userDb.generateUserId(payload.email);
-      
+
       await userDb.trackActivity(userId, {
         interface: 'cli',
         action,
@@ -828,12 +833,31 @@ class CLIAuth {
       if (!idToken) return null;
 
       const payload = JSON.parse(Buffer.from(idToken.split('.')[1], 'base64').toString());
-      
+
       const UserDatabase = require('vibecodingmachine-core/src/database/user-schema');
       const userDb = new UserDatabase();
+
+      // Get user from API to get the correct userId format
+      const userDbClient = userDb.apiClient;
+      userDbClient.setAuthToken(idToken);
       
+      try {
+        const apiUser = await userDbClient.getUser();
+        if (apiUser && apiUser.userId) {
+          return {
+            userId: apiUser.userId, // Use API server's userId format
+            email: payload.email,
+            name: payload.name || payload.email.split('@')[0],
+            cognitoId: payload.sub
+          };
+        }
+      } catch (error) {
+        console.warn('Could not fetch user from API:', error.message);
+      }
+
+      // Fallback to generated userId if API call fails
       const userId = userDb.generateUserId(payload.email);
-      
+
       return {
         userId,
         email: payload.email,

package/src/utils/compliance-check.js

@@ -38,8 +38,18 @@ async function checkCompliance() {
       return true
     }
 
+    // Set auth token for database operations
+    const token = await auth.getAuthToken()
+    if (token) {
+      const UserDatabase = require('vibecodingmachine-core/src/database/user-schema')
+      const userDb = new UserDatabase()
+      userDb.setAuthToken(token)
+    }
+
     // Check and prompt for compliance
     const compliancePrompt = new CompliancePrompt()
+    // Set auth token for compliance manager
+    compliancePrompt.complianceManager.userDb.setAuthToken(token)
     const status = await compliancePrompt.complianceManager.checkComplianceStatus(user.userId)
     
     if (!status.needsAcknowledgment) {

package/src/utils/config.js

@@ -86,6 +86,30 @@ async function setComputerFilter(computerName) {
   await writeConfig(cfg);
 }
 
+async function getProviderCache() {
+  const cfg = await readConfig();
+  return cfg.providerCache || {};
+}
+
+async function setProviderCache(cacheData) {
+  const cfg = await readConfig();
+  // Merge with existing cache
+  cfg.providerCache = { ...(cfg.providerCache || {}), ...cacheData };
+  await writeConfig(cfg);
+}
+
+async function getAutoTimeout() {
+  const cfg = await readConfig();
+  // Default to 5 minutes (300000 ms) if not set
+  return cfg.autoTimeout !== undefined ? cfg.autoTimeout : 5 * 60 * 1000;
+}
+
+async function setAutoTimeout(timeoutMs) {
+  const cfg = await readConfig();
+  cfg.autoTimeout = timeoutMs;
+  await writeConfig(cfg);
+}
+
 module.exports = {
   getRepoPath,
   setRepoPath,
@@ -97,7 +121,11 @@ module.exports = {
   setStages,
   DEFAULT_STAGES,
   getComputerFilter,
-  setComputerFilter
+  setComputerFilter,
+  getProviderCache,
+  setProviderCache,
+  getAutoTimeout,
+  setAutoTimeout
 };
 
 

package/src/utils/date-formatter.js

@@ -0,0 +1,44 @@
+/**
+ * Format a date into a rate limit reset label
+ * @param {string|number|Date} resetsAt - The reset time
+ * @returns {string|null} - Formatted label (e.g., "Resets at 4:23 pm CST on January 17, 2026")
+ */
+function formatResetsAtLabel(resetsAt) {
+    if (!resetsAt) return null;
+    const resetDate = new Date(resetsAt);
+    if (Number.isNaN(resetDate.getTime())) return null;
+
+    const msUntilReset = resetDate.getTime() - Date.now();
+    // If the reset minute has started (i.e., we're at or past the beginning of the stated minute),
+    // consider the quota reset to have occurred and don't show a label.
+    const resetMinuteStart = new Date(resetDate);
+    resetMinuteStart.setSeconds(0, 0);
+    if (resetMinuteStart.getTime() <= Date.now()) return null;
+
+    // Format time part: "4:23 pm"
+    const timePart = resetDate.toLocaleTimeString('en-US', {
+        hour: 'numeric',
+        minute: '2-digit',
+        hour12: true
+    }).toLowerCase();
+
+    // Format timezone part: "mst"
+    // Note: toLocaleString with timeZoneName: 'short' often returns 'MST', 'PDT', etc.
+    const tzPart = resetDate.toLocaleTimeString('en-US', {
+        timeZoneName: 'short'
+    }).split(' ').pop().toLowerCase();
+
+    const months = [
+        'January', 'February', 'March', 'April', 'May', 'June',
+        'July', 'August', 'September', 'October', 'November', 'December'
+    ];
+
+    const monthName = months[resetDate.getMonth()];
+    const day = resetDate.getDate();
+    const year = resetDate.getFullYear();
+
+    // "Resets at 4:23 pm mst on January 17, 2026"
+    return `Resets at ${timePart} ${tzPart} on ${monthName} ${day}, ${year}`;
+}
+
+module.exports = { formatResetsAtLabel };