coffeeinabit 0.0.23 → 0.0.26

package/linkedin_automation.js

@@ -1,4 +1,4 @@
-import { firefox } from 'playwright';
+import { chromium } from 'playwright';
 import path from 'path';
 import fs from 'fs';
 import { getContextDirectory } from './tools/context_paths.js';
@@ -369,9 +369,23 @@ export class LinkedInAutomation {
               this.emitStatus();
               
               if (this.originalHeadless && this.needsManualLogin) {
-                console.log('[LinkedInAutomation] Manual login completed. Switching back to headless mode...');
+                console.log('[LinkedInAutomation] Manual login completed. Waiting for session to be saved...');
                 clearInterval(pollInterval);
                 
+                await this.waitRandom(5000, 7000);
+                
+                const storageStatePath = this.getStorageStatePath();
+                try {
+                  await this.context.storageState({ path: storageStatePath });
+                  const cookies = await this.context.cookies();
+                  console.log(`[LinkedInAutomation] Saved storage state with ${cookies.length} cookies to: ${storageStatePath}`);
+                } catch (error) {
+                  console.error('[LinkedInAutomation] Error saving storage state:', error.message);
+                }
+                
+                await this.waitRandom(3000, 5000);
+                
+                console.log('[LinkedInAutomation] Switching back to headless mode...');
                 this.status = 'switching_to_headless';
                 this.emitStatus();
                 
@@ -423,7 +437,24 @@ export class LinkedInAutomation {
       }
 
       if (this.context) {
-        await this.context.close();
+        try {
+          const storageStatePath = this.getStorageStatePath();
+          const cookies = await this.context.cookies();
+          if (cookies.length > 0) {
+            await this.context.storageState({ path: storageStatePath });
+            console.log(`[LinkedInAutomation] Saved storage state with ${cookies.length} cookies to: ${storageStatePath}`);
+          }
+          await this.context.close();
+          await this.waitRandom(3000, 5000);
+        } catch (error) {
+          console.error('[LinkedInAutomation] Error saving storage state before close:', error.message);
+          try {
+            await this.context.close();
+            await this.waitRandom(3000, 5000);
+          } catch (closeError) {
+            console.error('[LinkedInAutomation] Error closing context:', closeError.message);
+          }
+        }
         this.context = null;
         this.browser = null;
       }
@@ -910,12 +941,18 @@ export class LinkedInAutomation {
     return this._currentAccessToken || null;
   }
 
+  getStorageStatePath() {
+    const sanitizedEmail = this.userEmail?.replace(/[^a-zA-Z0-9]/g, '_') || 'default_user';
+    return path.join(this.linkedInSessionDir, `storage_state_${sanitizedEmail}.json`);
+  }
+
   async launchBrowserAndLogin() {
-    console.log('[LinkedInAutomation] Launching Firefox with persistent context for LinkedIn login...');
+    console.log('[LinkedInAutomation] Launching Chromium with persistent context for LinkedIn login...');
     
     this.userEmail = this._currentUserEmail || 'default_user';
     const sanitizedEmail = this.userEmail.replace(/[^a-zA-Z0-9]/g, '_');
     const userDataPath = path.join(this.linkedInSessionDir, `linkedin_auth_${sanitizedEmail}`);
+    const storageStatePath = this.getStorageStatePath();
     
     console.log('[LinkedInAutomation] User data path:', userDataPath);
 
@@ -927,10 +964,16 @@ export class LinkedInAutomation {
     const launchOptions = {
       headless: this.headless || false,
       viewport: null,
-      ignoreHTTPSErrors: true
+      ignoreHTTPSErrors: true,
+      args: [
+        '--disable-blink-features=AutomationControlled',
+        '--disable-dev-shm-usage',
+        '--no-sandbox',
+        '--disable-setuid-sandbox'
+      ]
     };
 
-    this.context = await firefox.launchPersistentContext(userDataPath, launchOptions);
+    this.context = await chromium.launchPersistentContext(userDataPath, launchOptions);
 
     console.log('[LinkedInAutomation] Persistent context launched successfully');
 
@@ -938,6 +981,23 @@ export class LinkedInAutomation {
     
     await this.enablePerformanceMode();
     await this.preparePrimaryPage();
+    
+    if (fs.existsSync(storageStatePath)) {
+      try {
+        console.log('[LinkedInAutomation] Loading cookies from storage state file:', storageStatePath);
+        const storageState = JSON.parse(fs.readFileSync(storageStatePath, 'utf8'));
+        if (storageState.cookies && storageState.cookies.length > 0) {
+          await this.context.addCookies(storageState.cookies);
+          console.log(`[LinkedInAutomation] Added ${storageState.cookies.length} cookies from storage state file`);
+        }
+      } catch (error) {
+        console.error('[LinkedInAutomation] Error loading storage state:', error.message);
+      }
+    }
+    
+    const cookiesAfterLaunch = await this.context.cookies();
+    console.log(`[LinkedInAutomation] Total ${cookiesAfterLaunch.length} cookies in context`);
+
     // await this.installCursorHighlight();
 
     console.log('[LinkedInAutomation] Checking if user is already authenticated...');
@@ -1001,15 +1061,45 @@ export class LinkedInAutomation {
       const sanitizedEmail = this.userEmail.replace(/[^a-zA-Z0-9]/g, '_');
       const userDataPath = path.join(this.linkedInSessionDir, `linkedin_auth_${sanitizedEmail}`);
       
+      const storageStatePath = this.getStorageStatePath();
+      try {
+        await this.context.storageState({ path: storageStatePath });
+        const cookies = await this.context.cookies();
+        console.log(`[LinkedInAutomation] Saved storage state with ${cookies.length} cookies before mode switch`);
+      } catch (error) {
+        console.error('[LinkedInAutomation] Error saving storage state:', error);
+      }
+      
       await this.context.close();
+      await this.waitRandom(3000, 5000);
       
       this.headless = newHeadless;
       
-      this.context = await firefox.launchPersistentContext(userDataPath, {
+      const launchOptions = {
         headless: this.headless,
         viewport: null,
-        ignoreHTTPSErrors: true
-      });
+        ignoreHTTPSErrors: true,
+        args: [
+          '--disable-blink-features=AutomationControlled',
+          '--disable-dev-shm-usage',
+          '--no-sandbox',
+          '--disable-setuid-sandbox'
+        ]
+      };
+      
+      this.context = await chromium.launchPersistentContext(userDataPath, launchOptions);
+      
+      if (fs.existsSync(storageStatePath)) {
+        try {
+          const storageState = JSON.parse(fs.readFileSync(storageStatePath, 'utf8'));
+          if (storageState.cookies && storageState.cookies.length > 0) {
+            await this.context.addCookies(storageState.cookies);
+            console.log(`[LinkedInAutomation] Added ${storageState.cookies.length} cookies from storage state for mode switch`);
+          }
+        } catch (error) {
+          console.error('[LinkedInAutomation] Error loading storage state for mode switch:', error.message);
+        }
+      }
 
       this.browser = this.context;
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "coffeeinabit",
-  "version": "0.0.23",
+  "version": "0.0.26",
   "description": "CoffeeInABit App",
   "main": "server.js",
   "type": "module",
@@ -17,7 +17,8 @@
   },
   "scripts": {
     "start": "node server.js",
-    "dev": "nodemon server.js"
+    "dev": "nodemon server.js",
+    "postinstall": "npx playwright install chromium"
   },
   "dependencies": {
     "axios": "^1.6.0",

package/server.js

@@ -17,13 +17,6 @@ dotenv.config();
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 
-try {
-  execSync('npx playwright install firefox', { stdio: 'inherit' });
-  console.log('[Server] Playwright Firefox browser installed/verified');
-} catch (error) {
-  console.warn('[Server] Warning: Could not install Playwright Firefox browser:', error.message);
-}
-
 const FileStoreSession = FileStore(session);
 
 const app = express();

package/tools/get_messages.js

@@ -298,6 +298,10 @@ export async function executeGetMessages(page, action) {
   }
 
   console.log('[get_messages] Successfully extracted messages');
+
+  console.log('[get_messages] Closing all conversation bubbles');
+  await closeAllConversationBubbles(page);
+
   return {
     action: 'get_messages',
     result: { status: 'success', message: messageData.content, items: messageData.items },