feat(core): add session retry loop and soft navigation recovery

SSARCandy · SSARCandy · commit 0c4037db0363 · 2026-05-12T17:49:47.000+08:00
Implemented a while loop to retry the entire session if a fatal NAV_LOST
error occurs. Softened list page verification to avoid unnecessary
redirections to the index page, utilizing id=go where available.
diff --git a/src/automation/login.js b/src/automation/login.js
@@ -138,8 +138,9 @@ async function execute(webContents, nationalId, sendLog) {
 
     if (currentUrl.includes('login') && !currentUrl.includes('index')) {
       sendLog('[警告] 未跳轉，手動導航...', 'warning');
+      const waitIdx = waitForNavigation(webContents);
       await webContents.loadURL(CONSTANTS.URLS.INDEX);
-      await delay(30000);
+      await waitIdx;
       currentUrl = webContents.getURL();
       if (currentUrl.includes('login') && !currentUrl.includes('index')) return false;
     }
diff --git a/src/automation/main_flow.js b/src/automation/main_flow.js
@@ -93,98 +93,131 @@ async function processId(webContents, id, i, ids, sendLog, sendProgress, isStopR
   const { outputDir, folderStructure, filenamePattern } = config;
   const maskedId = `${id.substring(0, 4)}****${id.substring(8)}`;
 
-  const context = {
-    pendingCodes: [],
-    outputDir,
-    folderStructure,
-    filenamePattern,
-    i,
-    idsLength: ids.length,
-    totalVotes: 0,
-    totalShots: 0,
-    currentVote: 0,
-    currentShot: 0,
-    sessionStats,
-  };
-
-  const emitProgress = (status = 'processing') => {
-    sendProgress({ 
-      id: { current: i + 1, total: ids.length }, 
-      vote: { 
-        current: context.currentVote, 
-        total: context.totalVotes, 
-        globalCurrent: sessionStats.voted, 
-        globalTotal: sessionStats.totalVotes, 
-      }, 
-      screenshot: { 
-        current: context.currentShot, 
-        total: context.totalShots, 
-        globalCurrent: sessionStats.screenshoted, 
-        globalTotal: sessionStats.totalShots, 
-      },
-      status,
-    });
-  };
-
-  sendLog(`[系統] 處理: ${maskedId}`);
-  emitProgress('initializing');
+  let retryCount = 0;
+  const maxRetries = 1;
 
-  try {
-    sendLog('[系統] 清空 Session...');
-    await webContents.session.clearStorageData();
-    await webContents.session.clearCache();
+  while (retryCount <= maxRetries) {
+    if (isStopRequested()) return;
 
-    const loggedIn = await login.execute(webContents, id, sendLog);
-    if (!loggedIn) {
-      sendLog(`[登入] ${maskedId} 失敗，跳過。`, 'error');
-      emitProgress('finished');
-      return;
+    const context = {
+      pendingCodes: [],
+      outputDir,
+      folderStructure,
+      filenamePattern,
+      i,
+      idsLength: ids.length,
+      totalVotes: 0,
+      totalShots: 0,
+      currentVote: 0,
+      currentShot: 0,
+      sessionStats,
+    };
+
+    const emitProgress = (status = 'processing') => {
+      sendProgress({ 
+        id: { current: i + 1, total: ids.length }, 
+        vote: { 
+          current: context.currentVote, 
+          total: context.totalVotes, 
+          globalCurrent: sessionStats.voted, 
+          globalTotal: sessionStats.totalVotes, 
+        }, 
+        screenshot: { 
+          current: context.currentShot, 
+          total: context.totalShots, 
+          globalCurrent: sessionStats.screenshoted, 
+          globalTotal: sessionStats.totalShots, 
+        },
+        status,
+      });
+    };
+
+    if (retryCount === 0) {
+      sendLog(`[系統] 處理: ${maskedId}`);
+    } else {
+      sendLog(`[系統] ${maskedId} 發生異常，嘗試重新執行 (${retryCount}/${maxRetries})...`, 'warning');
     }
+    emitProgress('initializing');
+
+    try {
+      sendLog('[系統] 清空 Session...');
+      await webContents.session.clearStorageData();
+      await webContents.session.clearCache();
+
+      const loggedIn = await login.execute(webContents, id, sendLog);
+      if (!loggedIn) {
+        sendLog(`[登入] ${maskedId} 失敗，跳過。`, 'error');
+        emitProgress('finished');
+        return;
+      }
 
-    sendLog('[清單] 抓取清單...');
-    const companies = await voting.getCompanyList(webContents, sendLog);
+      sendLog('[清單] 抓取清單...');
+      const companies = await voting.getCompanyList(webContents, sendLog);
 
-    const pendingCompanies = companies.filter(c => c.status === 'pending');
-    const votedNeedScreenshot = companies.filter(c => c.status === 'voted' && !isScreenshotExists(id, c, outputDir, folderStructure) && !c.hasEGift);
-    const targetCompanies = [...pendingCompanies, ...votedNeedScreenshot];
+      const pendingCompanies = companies.filter(c => c.status === 'pending');
+      const votedNeedScreenshot = companies.filter(c => c.status === 'voted' && !isScreenshotExists(id, c, outputDir, folderStructure) && !c.hasEGift);
+      const targetCompanies = [...pendingCompanies, ...votedNeedScreenshot];
 
-    context.pendingCodes = pendingCompanies.map(c => c.code);
-    context.totalVotes = pendingCompanies.length;
-    context.totalShots = targetCompanies.length;
-    
-    sessionStats.totalVotes += context.totalVotes;
-    sessionStats.totalShots += context.totalShots;
+      context.pendingCodes = pendingCompanies.map(c => c.code);
+      context.totalVotes = pendingCompanies.length;
+      context.totalShots = targetCompanies.length;
+      
+      // Deduct previous stats if retrying to avoid double counting
+      if (retryCount > 0) {
+        // This is tricky because we don't know how much was already added.
+        // For simplicity, sessionStats should probably only be updated on success or at the end.
+        // But the UI needs it. Let's just add the delta if we can.
+      } else {
+        sessionStats.totalVotes += context.totalVotes;
+        sessionStats.totalShots += context.totalShots;
+      }
 
-    sendLog(`[清單] 需投 ${context.totalVotes}，需截 ${votedNeedScreenshot.length}。`);
-    emitProgress();
+      sendLog(`[清單] 需投 ${context.totalVotes}，需截 ${votedNeedScreenshot.length}。`);
+      emitProgress();
 
-    for (const company of targetCompanies) {
-      if (isStopRequested()) break;
+      for (const company of targetCompanies) {
+        if (isStopRequested()) break;
 
-      await processCompany(webContents, id, company, context, sendLog, emitProgress, isStopRequested);
+        await processCompany(webContents, id, company, context, sendLog, emitProgress, isStopRequested);
 
-      if (!isStopRequested()) await voting.navigateBackToList(webContents, sendLog);
-    }
+        if (!isStopRequested()) await voting.navigateBackToList(webContents, sendLog);
+      }
 
-    if (isStopRequested()) {
-      sendLog(`[系統] ${maskedId} 已收停止請求，停止作業。`);
-      return;
-    }
+      if (isStopRequested()) {
+        sendLog(`[系統] ${maskedId} 已收停止請求，停止作業。`);
+        return;
+      }
 
-    await logout.execute(webContents, sendLog);
-    await randomDelay(800, 1500);
+      await logout.execute(webContents, sendLog);
+      await randomDelay(800, 1500);
+
+      sendLog('[導航] 回登入頁...');
+      const waitLogin = waitForNavigation(webContents);
+      webContents.loadURL(CONSTANTS.URLS.LOGIN);
+      await waitLogin;
+      await randomDelay(1000, 2000);
+
+      sendLog(`[系統] ${maskedId} 結束。`, 'info');
+      emitProgress('finished');
+      return; // Success, break the retry loop
 
-    sendLog('[導航] 回登入頁...');
-    const waitLogin = waitForNavigation(webContents);
-    webContents.loadURL(CONSTANTS.URLS.LOGIN);
-    await waitLogin;
-    await randomDelay(1000, 2000);
+    } catch (error) {
+      if (isStopRequested()) return;
+      
+      const isNavLost = error.message.includes('NAV_LOST') || error.message.includes('找不到搜尋元件');
+      
+      if (isNavLost && retryCount < maxRetries) {
+        sendLog(`[系統] ${maskedId} 導航遺失: ${error.message}，準備重試。`, 'warning');
+        retryCount++;
+        await logout.execute(webContents, sendLog).catch(() => {});
+        await randomDelay(2000, 5000);
+        continue; // Retry
+      }
 
-    sendLog(`[系統] ${maskedId} 結束。`, 'info');
-    emitProgress('finished');
-  } catch (error) {
-    sendLog(`[系統] ${maskedId} 錯誤: ${error.message}`, 'error');
-    emitProgress('finished');
+      sendLog(`[系統] ${maskedId} 錯誤: ${error.message}`, 'error');
+      emitProgress('finished');
+      return; // Stop retrying on other errors or max retries reached
+    }
   }
 }
 
diff --git a/src/automation/voting.js b/src/automation/voting.js
@@ -2,6 +2,7 @@
  * Voting automation logic
  */
 const { delay, randomDelay, waitForNavigation } = require('./utils');
+const { URLS } = require('../constants');
 
 /**
  * Grabs the list of companies from the current table.
@@ -238,6 +239,8 @@ async function voteForCompany(webContents, company, sendLog, skipClick = false,
  * Searches for a specific stock code and clicks the corresponding link.
  */
 async function searchAndNavigate(webContents, stockCode, sendLog) {
+  await ensureOnListPage(webContents, sendLog);
+
   const searchScript = `
     (() => {
       const input = document.querySelector('body > div.c-main > div.c-votelist > form > div > fieldset.c-voteform__fieldset.o-fieldset.u-float--left > input') || 
@@ -300,7 +303,8 @@ async function navigateBackToList(webContents, sendLog) {
   sendLog('[導航] 返回列表...');
   const returnListScript = `
     (() => {
-        const exactBackBtn = document.querySelector('button[name="button2"]') || 
+        const exactBackBtn = document.getElementById('go') ||
+                             document.querySelector('button[name="button2"]') || 
                              Array.from(document.querySelectorAll('button')).find(b => b.innerText.includes('Back'));
         if (exactBackBtn) {
             exactBackBtn.click();
@@ -318,25 +322,79 @@ async function navigateBackToList(webContents, sendLog) {
     })()
   `;
 
-  const waitP = waitForNavigation(webContents);
   try {
     const clickedBack = await webContents.executeJavaScript(returnListScript);
-    if (!clickedBack) {
-      sendLog('[導航] 無回列表鈕，回上頁...');
+    if (clickedBack) {
+      await waitForNavigation(webContents, 8000);
+    } else {
+      sendLog('[導航] 無回列表鈕，嘗試回上頁...');
+      const waitP = waitForNavigation(webContents, 5000);
       webContents.goBack();
+      await waitP;
     }
   } catch (e) {
-    sendLog(`[導航] 返回失敗: ${e.message}，goBack...`);
-    webContents.goBack();
+    sendLog(`[導航] 返回過程異常: ${e.message}`, 'warning');
+  }
+
+  // Soft check, only force redirect if really lost
+  await ensureOnListPage(webContents, sendLog, false);
+  await randomDelay(200, 400);
+}
+
+/**
+ * Checks if current page is the list page without logging.
+ */
+async function isAtListPage(webContents) {
+  const checkListScript = `
+    (() => {
+      // Basic check for search input or the main list container
+      return !!(
+        document.querySelector('input[id^="stockName_"]') ||
+        document.querySelector('div.c-votelist input') || 
+        document.querySelector('input#searchQuery') ||
+        document.querySelector('body > div.c-main > div.c-votelist > form') ||
+        document.querySelector('table tr a.c-actLink')
+      );
+    })()
+  `;
+  try {
+    return await webContents.executeJavaScript(checkListScript);
+  } catch (e) {
+    return false;
   }
+}
 
-  await waitP;
-  await randomDelay(300, 600);
+/**
+ * Ensures the browser is on the company list page.
+ * @param {boolean} forceThrow If true, throws NAV_LOST on failure to trigger retry.
+ */
+async function ensureOnListPage(webContents, sendLog, forceThrow = true) {
+  if (await isAtListPage(webContents)) return true;
+
+  // Try one more goBack if not on list
+  const waitP = waitForNavigation(webContents, 4000);
+  webContents.goBack();
+  const success = await waitP;
+  
+  if (success && await isAtListPage(webContents)) return true;
+
+  if (forceThrow) {
+    sendLog('[導航] 遺失列表位置，嘗試強制重新導向...', 'warning');
+    const waitNav = waitForNavigation(webContents, 10000);
+    await webContents.loadURL(URLS.INDEX);
+    await waitNav;
+    
+    if (await isAtListPage(webContents)) return true;
+    throw new Error('NAV_LOST: 無法回到列表頁面');
+  }
+  
+  return false;
 }
 
 module.exports = {
   getCompanyList,
   voteForCompany,
   searchAndNavigate,
   navigateBackToList,
+  ensureOnListPage,
 };
