opencode-studio-server 1.9.5 → 1.9.8

package/index.js

@@ -17,10 +17,24 @@ const atomicWriteFileSync = (filePath, data, options = 'utf8') => {
     const tempPath = path.join(dir, `.${path.basename(filePath)}.${crypto.randomBytes(6).toString('hex')}.tmp`);
     try {
         fs.writeFileSync(tempPath, data, options);
-        fs.renameSync(tempPath, filePath);
+        // Retry rename for Windows file locking issues
+        let retries = 5;
+        while (retries > 0) {
+            try {
+                fs.renameSync(tempPath, filePath);
+                break;
+            } catch (e) {
+                if (retries === 1) throw e;
+                retries--;
+                // Synchronous delay
+                const start = Date.now();
+                while (Date.now() - start < 50) {} 
+            }
+        }
     } catch (err) {
-        // Clean up temp file if rename fails
-        try { fs.unlinkSync(tempPath); } catch {}
+        if (fs.existsSync(tempPath)) {
+            try { fs.unlinkSync(tempPath); } catch (e) {}
+        }
         throw err;
     }
 };
@@ -276,6 +290,10 @@ function loadStudioConfig() {
         activeGooglePlugin: 'gemini',
         availableGooglePlugins: [],
         presets: [],
+        cooldownRules: [
+            { name: "Opus 4.5 (Antigravity)", duration: 4 * 60 * 60 * 1000 }, // 4 hours
+            { name: "Gemini 3 Pro (Antigravity)", duration: 24 * 60 * 60 * 1000 } // 24 hours
+        ],
         pluginModels: {
             gemini: {
                 "gemini-3-pro-preview": {
@@ -632,6 +650,36 @@ app.post('/api/restore', (req, res) => {
     }
 });
 
+app.get('/api/cooldowns', (req, res) => {
+    const studio = loadStudioConfig();
+    res.json(studio.cooldownRules || []);
+});
+
+app.post('/api/cooldowns', (req, res) => {
+    const { name, duration } = req.body;
+    if (!name || typeof duration !== 'number') return res.status(400).json({ error: 'Invalid name or duration' });
+    const studio = loadStudioConfig();
+    if (!studio.cooldownRules) studio.cooldownRules = [];
+    
+    // Update if exists, else push
+    const idx = studio.cooldownRules.findIndex(r => r.name === name);
+    if (idx >= 0) studio.cooldownRules[idx] = { name, duration };
+    else studio.cooldownRules.push({ name, duration });
+    
+    saveStudioConfig(studio);
+    res.json(studio.cooldownRules);
+});
+
+app.delete('/api/cooldowns/:name', (req, res) => {
+    const { name } = req.params;
+    const studio = loadStudioConfig();
+    if (studio.cooldownRules) {
+        studio.cooldownRules = studio.cooldownRules.filter(r => r.name !== name);
+        saveStudioConfig(studio);
+    }
+    res.json(studio.cooldownRules || []);
+});
+
 const DROPBOX_CLIENT_ID = 'your-dropbox-app-key';
 const GDRIVE_CLIENT_ID = 'your-google-client-id';
 
@@ -1342,6 +1390,7 @@ app.get('/api/auth/providers', (req, res) => {
 });
 
 app.get('/api/auth', (req, res) => {
+    importCurrentGoogleAuthToPool();
     syncAntigravityPool();
     const authCfg = loadAuthConfig() || {};
     const studio = loadStudioConfig();
@@ -1814,6 +1863,55 @@ function listAntigravityAccounts() {
     }));
 }
 
+function importCurrentGoogleAuthToPool() {
+    const studio = loadStudioConfig();
+    // Only applies if Antigravity is active
+    if (studio.activeGooglePlugin !== 'antigravity') return;
+
+    const authCfg = loadAuthConfig();
+    if (!authCfg || !authCfg.google || !authCfg.google.email) return;
+
+    const namespace = 'google.antigravity';
+    const profileDir = path.join(AUTH_PROFILES_DIR, namespace);
+    if (!fs.existsSync(profileDir)) fs.mkdirSync(profileDir, { recursive: true });
+
+    const email = authCfg.google.email;
+    const profilePath = path.join(profileDir, `${email}.json`);
+
+    // Check if we need to sync (new account or updated tokens)
+    let shouldSync = true;
+    if (fs.existsSync(profilePath)) {
+        try {
+            const current = JSON.parse(fs.readFileSync(profilePath, 'utf8'));
+            if (current.access_token === authCfg.google.access_token) {
+                shouldSync = false;
+            }
+        } catch {
+            // Corrupt file, overwrite
+        }
+    }
+
+    if (shouldSync) {
+        console.log(`[Auth] Syncing Google login for ${email} to Antigravity pool.`);
+        atomicWriteFileSync(profilePath, JSON.stringify(authCfg.google, null, 2));
+
+        const metadata = loadPoolMetadata();
+        if (!metadata[namespace]) metadata[namespace] = {};
+        
+        // Only update metadata if it doesn't exist
+        if (!metadata[namespace][email]) {
+            metadata[namespace][email] = {
+                email: email,
+                createdAt: Date.now(),
+                lastUsed: Date.now(),
+                usageCount: 0,
+                imported: true
+            };
+            savePoolMetadata(metadata);
+        }
+    }
+}
+
 function syncAntigravityPool() {
     const accounts = listAntigravityAccounts();
     const namespace = 'google.antigravity';
@@ -2159,7 +2257,15 @@ app.post('/api/auth/pool/limit', (req, res) => {
 // PUT /api/auth/pool/:name/cooldown - Mark account as in cooldown
 app.put('/api/auth/pool/:name/cooldown', (req, res) => {
     const { name } = req.params;
-    const { duration = 3600000, provider = 'google' } = req.body; // default 1 hour
+    let { duration, provider = 'google', rule } = req.body;
+    
+    if (rule) {
+        const studio = loadStudioConfig();
+        const r = (studio.cooldownRules || []).find(cr => cr.name === rule);
+        if (r) duration = r.duration;
+    }
+
+    if (!duration) duration = 3600000; // default 1 hour
     
     const activePlugin = getActiveGooglePlugin();
     const namespace = provider === 'google'
@@ -2886,7 +2992,7 @@ app.post('/api/presets/:id/apply', (req, res) => {
 
 // Start watcher on server start
 function startServer() {
-    setupLogWatcher();
+    // setupLogWatcher(); // Disabled as per user request (manual switching only)
     importExistingAuth();
     app.listen(PORT, () => console.log(`Server running at http://localhost:${PORT}`));
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-studio-server",
-  "version": "1.9.5",
+  "version": "1.9.8",
   "description": "Backend server for OpenCode Studio - manages opencode configurations",
   "main": "index.js",
   "bin": {