From e1ee11e5be34471b0c0cb8cbd069645639e21645 Mon Sep 17 00:00:00 2001
From: Shreyas Sharma <shreyassharma9912@gmail.com>
Date: Wed, 23 Jul 2025 10:35:13 +0530
Subject: [PATCH] fix(playwright): add-logic-to-read-agent2-token

---
 playwright/global.setup.ts | 43 ++++++++++++++++++++++----------------
 1 file changed, 25 insertions(+), 18 deletions(-)

diff --git a/playwright/global.setup.ts b/playwright/global.setup.ts
index 30a29511c..ce1e99c08 100644
--- a/playwright/global.setup.ts
+++ b/playwright/global.setup.ts
@@ -4,25 +4,32 @@ const fs = require('fs');
 const path = require('path');
 
 setup('OAuth', async ({browser}) => {
-  const agentId = 'AGENT1'; // Configure which agent to set up
-  const page = await browser.newPage();
-  await oauthLogin(page, agentId);
+  const agents = ['AGENT1', 'AGENT2']; // Array of agents to set up
 
-  await page.getByRole('textbox').click();
-  const accessToken = await page.getByRole('textbox').inputValue();
+  for (const agentId of agents) {
+    const page = await browser.newPage();
+    await oauthLogin(page, agentId);
 
-  const envPath = path.resolve(__dirname, '../.env');
-  let envContent = '';
-  if (fs.existsSync(envPath)) {
-    envContent = fs.readFileSync(envPath, 'utf8');
-    // Remove any existing ACCESS_TOKEN line for this agent
-    const accessTokenPattern = new RegExp(`^PW_${agentId}_ACCESS_TOKEN=.*$`, 'm');
-    envContent = envContent.replace(accessTokenPattern, '');
-    // Also remove legacy ACCESS_TOKEN for backward compatibility
-    envContent = envContent.replace(/^ACCESS_TOKEN=.*$/m, '');
-    // Ensure trailing newline
-    if (!envContent.endsWith('\n')) envContent += '\n';
+    await page.getByRole('textbox').click();
+    const accessToken = await page.getByRole('textbox').inputValue();
+
+    const envPath = path.resolve(__dirname, '../.env');
+    let envContent = '';
+    if (fs.existsSync(envPath)) {
+      envContent = fs.readFileSync(envPath, 'utf8');
+      // Remove any existing ACCESS_TOKEN line for this agent
+      const accessTokenPattern = new RegExp(`^PW_${agentId}_ACCESS_TOKEN=.*$`, 'm');
+      envContent = envContent.replace(accessTokenPattern, '');
+      // Also remove legacy ACCESS_TOKEN for backward compatibility
+      if (agentId === 'AGENT1') {
+        envContent = envContent.replace(/^ACCESS_TOKEN=.*$/m, '');
+      }
+      // Ensure trailing newline
+      if (!envContent.endsWith('\n')) envContent += '\n';
+    }
+    envContent += `PW_${agentId}_ACCESS_TOKEN=${accessToken}\n`;
+    fs.writeFileSync(envPath, envContent, 'utf8');
+
+    await page.close();
   }
-  envContent += `PW_${agentId}_ACCESS_TOKEN=${accessToken}\n`;
-  fs.writeFileSync(envPath, envContent, 'utf8');
 });