tronclass-auto 1.6.0 → 1.7.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tronclass-auto",
-  "version": "1.6.0",
+  "version": "1.7.1",
   "description": "自動觀看 eclass/TronClass 影片、填寫表單的 CLI 工具",
   "main": "src/index.js",
   "bin": {

package/src/index.js

@@ -28,6 +28,8 @@ async function getUncompleted(page, courseId) {
       try { completeness = scope.getActivityCompleteness(a); } catch (e) {}
       if (completeness === 'full') return;
 
+      if (a.completion_criterion_value === '' || a.completion_criterion_value === undefined || a.completion_criterion_value === null) return;
+
       let isUpcoming = false;
       let isExpired = false;
       try { isUpcoming = scope.activityUpcoming(a); } catch (e) {}
@@ -211,6 +213,50 @@ async function processCourse(page, courseId, stats) {
   return { done, skip, fail };
 }
 
+async function getAllVideos(page, courseId) {
+  try {
+    await page.goto(`${BASE_URL}/course/${courseId}/content#/`, {
+      waitUntil: 'domcontentloaded', timeout: 20000
+    });
+  } catch (e) {}
+  await page.waitForTimeout(8000);
+
+  return page.evaluate(() => {
+    const el = document.querySelector('.learning-activities');
+    if (!el) return [];
+    const scope = angular.element(el).scope();
+    if (!scope) return [];
+    const activities = document.querySelectorAll('.learning-activity.sortable');
+    let result = [];
+    activities.forEach((actEl) => {
+      const actScope = angular.element(actEl).scope();
+      if (!actScope || !actScope.activity) return;
+      const a = actScope.activity;
+      let completeness = '';
+      try { completeness = scope.getActivityCompleteness(a); } catch(e) {}
+      let isUpcoming = false;
+      try { isUpcoming = scope.activityUpcoming(a); } catch(e) {}
+      let startTime = '';
+      try {
+        const txt = actEl.textContent || '';
+        const tm = txt.match(/(\d{4}-\d{2}-\d{2})\s+\d{2}:\d{2}/);
+        if (tm) startTime = tm[1];
+      } catch(e) {}
+      const now = new Date();
+      const startDate = startTime ? new Date(startTime) : null;
+      const isFuture = startDate && startDate > now;
+      result.push({
+        id: a.id,
+        type: a.type || '',
+        completeness: completeness,
+        isUpcoming: isUpcoming,
+        isFuture: isFuture
+      });
+    });
+    return result;
+  });
+}
+
 async function showStatus() {
   const config = loadConfig();
   const cookies = loadCookies();
@@ -227,16 +273,17 @@ async function showStatus() {
   const browser = await chromium.launch(launchOpts);
   const context = await browser.newContext();
   await context.addCookies(cookies);
-  const page = await context.newPage();
 
-  await page.goto(BASE_URL, { waitUntil: 'domcontentloaded', timeout: 15000 });
-  await page.waitForTimeout(2000);
+  const checkPage = await context.newPage();
+  await checkPage.goto(BASE_URL, { waitUntil: 'domcontentloaded', timeout: 15000 });
+  await checkPage.waitForTimeout(2000);
 
-  if (page.url().includes('login')) {
+  if (checkPage.url().includes('login')) {
     console.log(chalk.red('[ERROR] Cookie 過期'));
     await browser.close();
     return;
   }
+  await checkPage.close();
 
   const courses = config.courses || [];
   console.log(chalk.cyan('\n┌' + '─'.repeat(48) + '┐'));
@@ -247,27 +294,41 @@ async function showStatus() {
   let totalRemaining = 0;
   let totalExams = 0;
 
-  for (const courseUrl of courses) {
+  const courseResults = await Promise.all(courses.map(async (courseUrl) => {
     const match = courseUrl.match(/\/course\/(\d+)\//);
     const courseId = match ? match[1] : '?';
     try {
-      const uncompleted = await getUncompleted(page, courseId);
-      const videos = uncompleted.filter(u => u.type === 'online_video');
-      const remaining = videos.filter(v => !isDone(courseId, v.id));
-      const exams = uncompleted.filter(u => u.type === 'exam').length;
+      const pg = await context.newPage();
+      const allVideos = await getAllVideos(pg, courseId);
+      await pg.close();
+      return { courseId, allVideos };
+    } catch (e) {
+      return { courseId, allVideos: [] };
+    }
+  }));
+
+  for (const { courseId, allVideos } of courseResults) {
+    try {
+      const videos = allVideos.filter(u => u.type === 'online_video');
+      const completed = videos.filter(v => v.completeness === 'full').length;
+      const future = videos.filter(v => v.isFuture && v.completeness !== 'full').length;
+      const remaining = videos.filter(v => v.completeness !== 'full' && !v.isFuture).length;
+      const exams = allVideos.filter(u => u.type === 'exam' && u.completeness !== 'full').length;
       totalVideos += videos.length;
-      totalRemaining += remaining.length;
+      totalRemaining += remaining;
       totalExams += exams;
 
-      const pct = videos.length > 0 ? Math.floor(((videos.length - remaining.length) / videos.length) * 100) : 100;
+      const pct = videos.length > 0 ? Math.floor((completed / videos.length) * 100) : 100;
       const filled = Math.floor(50 * pct / 100);
       const empty = 50 - filled;
       const bar = pct === 100
         ? chalk.green('█'.repeat(50))
         : chalk.green('█'.repeat(filled)) + chalk.gray('░'.repeat(empty));
 
-      const doneCount = videos.length - remaining.length;
-      console.log(chalk.cyan('│') + `  [${courseId}] ${doneCount}/${videos.length} videos  ${exams} exams`.padEnd(50) + chalk.cyan('│'));
+      const doneCount = completed;
+      const pending = remaining;
+      const info = future > 0 ? ` (${future} 未開放)` : '';
+      console.log(chalk.cyan('│') + `  [${courseId}] ${doneCount}/${videos.length} videos  ${exams} exams${info}`.padEnd(50) + chalk.cyan('│'));
       console.log(chalk.cyan('│') + `    ${bar} ${pct}%`.padEnd(52) + chalk.cyan('│'));
     } catch (e) {
       console.log(chalk.cyan('│') + `  [${courseId}] ERROR`.padEnd(50) + chalk.cyan('│'));

package/src/tui.js

@@ -1,16 +1,141 @@
-const inquirer = require('inquirer');
+const readline = require('readline');
 const chalk = require('chalk');
 const { loadConfig, saveConfig, BASE_URL } = require('./config');
 const { loadCookies } = require('./auth');
 const pkg = require('../package.json');
 
 function box(lines, width = 52) {
-  const top = '╔' + '═'.repeat(width - 2) + '╗';
-  const bot = '╚' + '═'.repeat(width - 2) + '╝';
-  const pad = (s) => '║ ' + s.padEnd(width - 4) + ' ║';
-  console.log(chalk.cyan(top));
-  lines.forEach(l => console.log(chalk.cyan(pad(l))));
-  console.log(chalk.cyan(bot));
+  console.log(chalk.cyan('╔' + '═'.repeat(width - 2) + '╗'));
+  lines.forEach(l => console.log(chalk.cyan('║ ') + l.padEnd(width - 4) + chalk.cyan(' ║')));
+  console.log(chalk.cyan('╚' + '═'.repeat(width - 2) + '╝'));
+}
+
+function promptList(message, choices, header) {
+  return new Promise((resolve) => {
+    let selected = 0;
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+
+    function render() {
+      const lines = [];
+      lines.push('');
+      choices.forEach((c, i) => {
+        const prefix = i === selected ? chalk.green('>') : ' ';
+        if (typeof c === 'string' && c === '__sep__') {
+          lines.push(chalk.gray('  ─────────────'));
+        } else {
+          const label = typeof c === 'string' ? c : c.name;
+          lines.push(`  ${prefix} ${label}`);
+        }
+      });
+      lines.push('');
+      lines.push(chalk.gray('  ↑↓ 移動  Enter 確認  ESC 返回'));
+
+      const headerLines = header || [];
+      const totalLines = headerLines.length + lines.length;
+
+      process.stdout.write('\x1B[2J\x1B[H');
+      headerLines.forEach(l => process.stdout.write(l + '\n'));
+      lines.forEach(l => process.stdout.write(l + '\n'));
+    }
+
+    render();
+
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+
+    function onData(data) {
+      const key = data.toString();
+
+      if (key === '\u001b[A') {
+        selected = Math.max(0, selected - 1);
+        while (selected < choices.length && (typeof choices[selected] === 'string' || choices[selected].value === '__sep__')) {
+          selected = Math.max(0, selected - 1);
+        }
+        render();
+      } else if (key === '\u001b[B') {
+        selected = Math.min(choices.length - 1, selected + 1);
+        while (selected < choices.length && (typeof choices[selected] === 'string' || choices[selected].value === '__sep__')) {
+          selected = Math.min(choices.length - 1, selected + 1);
+        }
+        render();
+      } else if (key === '\r' || key === '\n') {
+        cleanup();
+        const choice = choices[selected];
+        resolve(typeof choice === 'string' ? choice : choice.value);
+      } else if (key === '\u001b') {
+        cleanup();
+        resolve('__esc__');
+      }
+    }
+
+    function cleanup() {
+      process.stdin.removeListener('data', onData);
+      process.stdin.setRawMode(false);
+      process.stdin.pause();
+      rl.close();
+    }
+
+    process.stdin.on('data', onData);
+  });
+}
+
+function promptConfirm(message, defaultVal = true) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    process.stdout.write(`  ${message} (${defaultVal ? 'Y/n' : 'y/N'}) `);
+
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+
+    function onData(data) {
+      const key = data.toString();
+      if (key === '\r' || key === '\n') {
+        cleanup();
+        resolve(defaultVal);
+      } else if (key === '\u001b') {
+        cleanup();
+        resolve(null);
+      } else if (key.toLowerCase() === 'y') {
+        cleanup();
+        resolve(true);
+      } else if (key.toLowerCase() === 'n') {
+        cleanup();
+        resolve(false);
+      }
+    }
+
+    function cleanup() {
+      process.stdin.removeListener('data', onData);
+      process.stdin.setRawMode(false);
+      process.stdin.pause();
+      rl.close();
+    }
+
+    process.stdin.on('data', onData);
+  });
+}
+
+function promptInput(message) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    rl.question(`  ${message}: `, (answer) => {
+      rl.close();
+      resolve(answer);
+    });
+  });
+}
+
+function pressAnyKey() {
+  return new Promise((resolve) => {
+    process.stdout.write(chalk.gray('\n  按任意鍵返回...'));
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+    process.stdin.once('data', () => {
+      process.stdin.setRawMode(false);
+      process.stdin.pause();
+      resolve();
+    });
+  });
 }
 
 async function mainMenu() {
@@ -22,31 +147,33 @@ async function mainMenu() {
 
   let running = true;
   while (running) {
-    console.clear();
     const config = loadConfig();
 
+    console.clear();
     box([
       chalk.bold.white(`tronclass-auto  v${pkg.version}`),
       chalk.gray('自動觀看 eclass/TronClass 影片'),
       '',
-      `課程: ${chalk.cyan(config.courses.length + ' 個')}  |  Cookie: ${chalk.green('✓')}  |  倍速: ${chalk.yellow(config.playbackRate || 2)}x`,
+      `課程: ${chalk.cyan(config.courses.length)} 個  |  Cookie: ${chalk.green('✓')}  |  倍速: ${chalk.yellow(config.playbackRate || 2)}x`,
     ]);
-    console.log('');
-
-    const { action } = await inquirer.prompt([{
-      type: 'list',
-      name: 'action',
-      message: '選擇操作:',
-      choices: [
-        { name: chalk.green('▶  開始自動觀看'), value: 'run' },
-        { name: chalk.cyan('📋 管理課程列表'), value: 'courses' },
-        { name: chalk.yellow('📊 查看進度統計'), value: 'status' },
-        { name: chalk.magenta('⚙  設定'), value: 'settings' },
-        new inquirer.Separator(),
-        { name: chalk.gray('🚪 離開'), value: 'exit' }
-      ],
-      pageSize: 10
-    }]);
+
+    const action = await promptList('選擇操作:', [
+      { name: chalk.green('▶  開始自動觀看'), value: 'run' },
+      { name: chalk.cyan('📋 管理課程列表'), value: 'courses' },
+      { name: chalk.yellow('📊 查看進度統計'), value: 'status' },
+      { name: chalk.magenta('⚙  設定'), value: 'settings' },
+      '__sep__',
+      { name: chalk.gray('🚪 離開'), value: 'exit' }
+    ], [
+      '',
+      chalk.white('╔' + '═'.repeat(50) + '╗'),
+      chalk.white('║') + chalk.bold.white(`  tronclass-auto  v${pkg.version}`.padEnd(50)) + chalk.white('║'),
+      chalk.white('║') + chalk.gray('  自動觀看 eclass/TronClass 影片'.padEnd(50)) + chalk.white('║'),
+      chalk.white('║') + `  課程: ${chalk.cyan(config.courses.length)} 個  |  Cookie: ${chalk.green('✓')}  |  倍速: ${chalk.yellow(config.playbackRate || 2)}x`.padEnd(50) + chalk.white('║'),
+      chalk.white('╚' + '═'.repeat(50) + '╝'),
+    ]);
+
+    if (action === '__esc__') { running = false; continue; }
 
     switch (action) {
       case 'run': await runMenu(config); break;
@@ -71,20 +198,11 @@ async function runMenu(config) {
     return;
   }
 
-  const { confirm } = await inquirer.prompt([{
-    type: 'confirm',
-    name: 'confirm',
-    message: `開始觀看全部 ${config.courses.length} 個課程?`,
-    default: true
-  }]);
-  if (!confirm) return;
-
-  const { headless } = await inquirer.prompt([{
-    type: 'confirm',
-    name: 'headless',
-    message: '無頭模式（背景執行）?',
-    default: false
-  }]);
+  const confirm = await promptConfirm(`開始觀看全部 ${config.courses.length} 個課程?`);
+  if (confirm === null) return;
+
+  const headless = await promptConfirm('無頭模式（背景執行）?', false);
+  if (headless === null) return;
 
   console.log('');
   box([
@@ -107,72 +225,99 @@ async function courseMenu() {
     const courses = config.courses || [];
 
     console.clear();
-    box([
-      chalk.bold.white('📋 課程列表'),
-      ...(courses.length === 0
-        ? [chalk.gray('  還沒有課程')]
-        : courses.map((url, i) => {
+
+    let courseNames = {};
+    if (courses.length > 0) {
+      try {
+        const { chromium } = require('playwright');
+        const cookies = loadCookies();
+        if (cookies.length > 0) {
+          const browser = await chromium.launch({ headless: true, slowMo: 50 });
+          const ctx = await browser.newContext();
+          await ctx.addCookies(cookies);
+
+          const results = await Promise.all(courses.map(async (url) => {
             const match = url.match(/\/course\/(\d+)\//);
-            const id = match ? match[1] : '?';
-            return `  ${chalk.white(i + 1 + '.')} [${chalk.cyan(id)}] ${chalk.gray(url.substring(0, 45))}`;
-          })),
+            if (!match) return { id: '?', name: '' };
+            const pg = await ctx.newPage();
+            try {
+              await pg.goto(`${BASE_URL}/course/${match[1]}/content#/`, { waitUntil: 'domcontentloaded', timeout: 8000 });
+              await pg.waitForTimeout(2000);
+              const name = await pg.evaluate(() => {
+                const el = document.querySelector('.learning-activities');
+                if (!el) return '';
+                const scope = angular.element(el).scope();
+                return (scope && scope.course) ? (scope.course.name || '') : '';
+              });
+              return { id: match[1], name: name.substring(0, 30) };
+            } catch (e) {
+              return { id: match[1], name: '' };
+            } finally {
+              await pg.close();
+            }
+          }));
+
+          results.forEach(r => { if (r.name) courseNames[r.id] = r.name; });
+          await browser.close();
+        }
+      } catch (e) {}
+    }
+
+    const choices = [
+      { name: chalk.green('➕ 新增課程'), value: 'add' },
+      '__sep__',
+    ];
+    courses.forEach((url, i) => {
+      const match = url.match(/\/course\/(\d+)\//);
+      const id = match ? match[1] : '?';
+      const name = courseNames[id] || id;
+      choices.push({ name: `${chalk.red('✕')} 移除 ${name} [${id}]`, value: `remove_${i}` });
+    });
+    if (courses.length > 0) {
+      choices.push({ name: chalk.red('🗑  清空所有'), value: 'clear' });
+    }
+    choices.push('__sep__');
+    choices.push({ name: chalk.gray('⬅ 返回'), value: 'back' });
+
+    const action = await promptList(`課程管理 (${courses.length} 個):`, choices, [
       '',
-      chalk.gray('  新增: tronclass course --add <ID或URL>'),
-      chalk.gray('  移除: tronclass course --remove <編號>'),
+      chalk.white('┌' + '─'.repeat(50) + '┐'),
+      chalk.white('│') + chalk.bold.white('  📋 課程列表'.padEnd(50)) + chalk.white('│'),
+      chalk.white('├' + '─'.repeat(50) + '┤'),
+      ...courses.map((url, i) => {
+        const match = url.match(/\/course\/(\d+)\//);
+        const id = match ? match[1] : '?';
+        const name = courseNames[id] || id;
+        return chalk.white('│') + `  ${chalk.white(i + 1 + '.')} ${chalk.cyan(name)} [${id}]`.padEnd(51) + chalk.white('│');
+      }),
+      chalk.white('└' + '─'.repeat(50) + '┘'),
     ]);
-    console.log('');
-
-    const { action } = await inquirer.prompt([{
-      type: 'list',
-      name: 'action',
-      message: `課程管理 (${courses.length} 個)`,
-      choices: [
-        { name: chalk.green('➕ 新增課程（互動式）'), value: 'add' },
-        ...(courses.length > 0 ? [
-          ...courses.map((url, i) => {
-            const match = url.match(/\/course\/(\d+)\//);
-            const id = match ? match[1] : '?';
-            return { name: `${chalk.red('✕')} [${id}] 移除`, value: `remove_${i}` };
-          }),
-          new inquirer.Separator(),
-          { name: chalk.red('🗑  清空所有'), value: 'clear' }
-        ] : []),
-        new inquirer.Separator(),
-        { name: chalk.gray('⬅ 返回'), value: 'back' }
-      ],
-      pageSize: 20
-    }]);
-
-    if (action === 'back') {
+
+    if (action === '__esc__' || action === 'back') {
       editing = false;
     } else if (action === 'add') {
-      const { input } = await inquirer.prompt([{
-        type: 'input',
-        name: 'input',
-        message: '輸入課程 ID（如 127331）或完整 URL:',
-        validate: (v) => v.trim().length > 0 ? true : '不可為空'
-      }]);
-      const config2 = loadConfig();
-      config2.courses = config2.courses || [];
-      config2.courses.push(normalizeUrl(input.trim()));
-      saveConfig(config2);
-      console.log(chalk.green('  已新增！'));
+      const input = await promptInput('輸入課程 ID 或 URL');
+      if (input && input.trim()) {
+        config.courses = config.courses || [];
+        config.courses.push(normalizeUrl(input.trim()));
+        saveConfig(config);
+        console.log(chalk.green('  已新增！'));
+        await pressAnyKey();
+      }
     } else if (action === 'clear') {
-      const { confirm } = await inquirer.prompt([{
-        type: 'confirm', name: 'confirm', message: '確定要清空所有課程?', default: false
-      }]);
+      const confirm = await promptConfirm('確定要清空所有課程?', false);
       if (confirm) {
-        const config2 = loadConfig();
-        config2.courses = [];
-        saveConfig(config2);
+        config.courses = [];
+        saveConfig(config);
         console.log(chalk.green('  已清空'));
+        await pressAnyKey();
       }
-    } else if (action.startsWith('remove_')) {
+    } else if (typeof action === 'string' && action.startsWith('remove_')) {
       const idx = parseInt(action.split('_')[1]);
-      const config2 = loadConfig();
-      const removed = config2.courses.splice(idx, 1)[0];
-      saveConfig(config2);
+      const removed = config.courses.splice(idx, 1)[0];
+      saveConfig(config);
       console.log(chalk.green(`  已移除: ${removed}`));
+      await pressAnyKey();
     }
   }
 }
@@ -180,48 +325,46 @@ async function courseMenu() {
 async function settingsMenu() {
   const config = loadConfig();
 
-  const { setting } = await inquirer.prompt([{
-    type: 'list',
-    name: 'setting',
-    message: '設定',
-    choices: [
-      { name: `倍速: ${chalk.cyan(config.playbackRate || 2)}x`, value: 'speed' },
-      { name: `無頭模式: ${chalk.cyan(config.headless ? '是' : '否')}`, value: 'headless' },
-      { name: `SlowMo: ${chalk.cyan(config.slowMo || 50)}ms`, value: 'slowmo' },
-      new inquirer.Separator(),
-      { name: chalk.gray('⬅ 返回'), value: 'back' }
-    ]
-  }]);
-
-  if (setting === 'back') return;
+  const setting = await promptList('設定:', [
+    { name: `倍速: ${chalk.cyan(config.playbackRate || 2)}x`, value: 'speed' },
+    { name: `無頭模式: ${chalk.cyan(config.headless ? '是' : '否')}`, value: 'headless' },
+    { name: `SlowMo: ${chalk.cyan(config.slowMo || 50)}ms`, value: 'slowmo' },
+    '__sep__',
+    { name: chalk.gray('⬅ 返回'), value: 'back' }
+  ], [
+    '',
+    chalk.white('╔' + '═'.repeat(50) + '╗'),
+    chalk.white('║') + chalk.bold.white('  ⚙  設定'.padEnd(50)) + chalk.white('║'),
+    chalk.white('╚' + '═'.repeat(50) + '╝'),
+  ]);
+
+  if (setting === '__esc__' || setting === 'back') return;
 
   if (setting === 'speed') {
-    const { speed } = await inquirer.prompt([{
-      type: 'list', name: 'speed', message: '選擇播放倍速:',
-      choices: [
-        { name: '1x（正常）', value: 1 },
-        { name: '1.5x', value: 1.5 },
-        { name: '2x（推薦）', value: 2 },
-        { name: '4x', value: 4 },
-        { name: '8x', value: 8 }
-      ],
-      default: config.playbackRate || 2
-    }]);
-    config.playbackRate = speed;
-    saveConfig(config);
-    console.log(chalk.green(`  已設定為 ${speed}x`));
+    const speed = await promptList('選擇播放倍速:', [
+      { name: '2x', value: 2 },
+      { name: '1.5x', value: 1.5 },
+      { name: '1.25x', value: 1.25 },
+      { name: '1x', value: 1 },
+      { name: '0.75x', value: 0.75 },
+    ]);
+    if (speed !== '__esc__') {
+      config.playbackRate = speed;
+      saveConfig(config);
+      console.log(chalk.green(`  已設定為 ${speed}x`));
+    }
   } else if (setting === 'headless') {
-    const { hl } = await inquirer.prompt([{
-      type: 'confirm', name: 'hl', message: '啟用無頭模式?', default: config.headless || false
-    }]);
-    config.headless = hl;
-    saveConfig(config);
+    const hl = await promptConfirm('啟用無頭模式?', config.headless || false);
+    if (hl !== null) {
+      config.headless = hl;
+      saveConfig(config);
+    }
   } else if (setting === 'slowmo') {
-    const { ms } = await inquirer.prompt([{
-      type: 'number', name: 'ms', message: 'SlowMo 毫秒數:', default: config.slowMo || 50
-    }]);
-    config.slowMo = ms;
-    saveConfig(config);
+    const ms = await promptInput('SlowMo 毫秒數');
+    if (ms && !isNaN(parseInt(ms))) {
+      config.slowMo = parseInt(ms);
+      saveConfig(config);
+    }
   }
 }
 
@@ -235,8 +378,4 @@ function normalizeUrl(input) {
   return input;
 }
 
-async function pressAnyKey() {
-  await inquirer.prompt([{ type: 'input', name: '_', message: chalk.gray('按 Enter 返回...') }]);
-}
-
 module.exports = { mainMenu };

package/src/video.js

@@ -119,17 +119,30 @@ async function watchYouTube(page, stats) {
 
   await page.waitForTimeout(2000);
 
-  const ytDuration = await page.evaluate(() => {
-    const actText = document.querySelector('.activity-attribute, [class*="attribute"]');
-    if (actText) {
-      const m = actText.textContent.match(/(\d{2}):(\d{2}):(\d{2})/);
-      if (m) return parseInt(m[1]) * 3600 + parseInt(m[2]) * 60 + parseInt(m[3]);
-    }
-    return 180;
-  });
+  let ytDuration = 0;
+  for (let i = 0; i < 10; i++) {
+    try {
+      const freshFrame = page.frame({ url: /youtube|youtu\.be/ });
+      if (freshFrame) {
+        ytDuration = await freshFrame.evaluate(() => document.querySelector('#movie_player').getDuration());
+        if (ytDuration > 0) break;
+      }
+    } catch (e) {}
+    await page.waitForTimeout(2000);
+  }
+  if (!ytDuration || ytDuration <= 0) {
+    ytDuration = await page.evaluate(() => {
+      const actText = document.querySelector('.activity-attribute, [class*="attribute"]');
+      if (actText) {
+        const m = actText.textContent.match(/(\d{2}):(\d{2}):(\d{2})/);
+        if (m) return parseInt(m[1]) * 3600 + parseInt(m[2]) * 60 + parseInt(m[3]);
+      }
+      return 180;
+    });
+  }
 
   const waitTime = (rate >= 2 ? Math.ceil(ytDuration / 2) : ytDuration) + 30;
-  log(chalk.blue(`  ~${formatTime(ytDuration)} video → ~${formatTime(waitTime)} (${rate}x)`));
+  log(chalk.blue(`  ${formatTime(ytDuration)} video → ~${formatTime(waitTime)} (${rate}x)`));
 
   if (stats) {
     stats.currentVideoDuration = ytDuration;