tt-help-cli-ycl 1.0.7 → 1.1.0

package/src/lib/args.js

@@ -1,288 +1,398 @@
-import { readFileSync } from 'fs';
-import { proxy } from './constants.js';
-
-const PRESETS = ['fast', 'normal', 'slow', 'stealth'];
-
-function parseScrapeArgs(args) {
-  let scrapeUrl = null;
-  let scrapePreset = null;
-  let scrapeMaxVideos = 20;
-  let scrapeMaxComments = 999;
-  let scrapeSwitchDelay = null;
-  let scrapeCommentDelay = null;
-  let outputFile = null;
-
-  const positional = [];
-
-  for (let i = 0; i < args.length; i++) {
-    const arg = args[i];
-    if (arg === '-o' || arg === '--output') {
-      outputFile = args[++i];
-    } else if (arg === '--switch-delay') {
-      scrapeSwitchDelay = parseInt(args[++i]) || null;
-    } else if (arg === '--comment-delay') {
-      scrapeCommentDelay = parseInt(args[++i]) || null;
-    } else {
-      positional.push(arg);
-    }
-  }
-
-  scrapeUrl = positional[0] || null;
-
-  if (positional[1]) {
-    if (PRESETS.includes(positional[1].toLowerCase())) {
-      scrapePreset = positional[1].toLowerCase();
-      scrapeMaxVideos = parseInt(positional[2]) || 20;
-      scrapeMaxComments = parseInt(positional[3]) || 999;
-    } else {
-      scrapeMaxVideos = parseInt(positional[1]) || 20;
-      scrapeMaxComments = parseInt(positional[2]) || 999;
-    }
-  }
-
-  return {
-    subcommand: 'scrape',
-    scrapeUrl,
-    scrapePreset,
-    scrapeMaxVideos,
-    scrapeMaxComments,
-    scrapeSwitchDelay,
-    scrapeCommentDelay,
-    outputFile,
-    urls: [],
-    outputFormat: 'json',
-    exploreCount: 0,
-    showConfig: false,
-    showHelp: false,
-    customProxy: null,
-    configAction: null,
-    configValue: null,
-    pipeMode: false,
-    filterStr: null,
-  };
-}
-
-function parseAutoArgs(args) {
-  let autoCollectMax = 1;
-  let autoScrapeDepth = 50;
-  let autoMaxComments = 200;
-  let autoPreset = 'fast';
-  let autoSwitchDelay = null;
-  let autoCommentDelay = null;
-  let outputFile = null;
-  let autoWatch = false;
-  let autoWatchPort = 3000;
-
-  const positional = [];
-  const PRESETS = ['fast', 'normal', 'slow', 'stealth'];
-
-  for (let i = 0; i < args.length; i++) {
-    const arg = args[i];
-    if (arg === '-o' || arg === '--output') {
-      outputFile = args[++i];
-    } else if (arg === '--switch-delay') {
-      autoSwitchDelay = parseInt(args[++i]) || null;
-    } else if (arg === '--comment-delay') {
-      autoCommentDelay = parseInt(args[++i]) || null;
-    } else if (arg === '--watch') {
-      autoWatch = true;
-      if (args[i + 1] === '-p') {
-        autoWatchPort = parseInt(args[i + 2]) || 3000;
-        i += 2;
-      }
-    } else if (arg === '-p' && autoWatch) {
-      autoWatchPort = parseInt(args[++i]) || 3000;
-    } else {
-      positional.push(arg);
-    }
-  }
-
-  // 收集用户名（非 preset、非数字的都是用户名）
-  const usernames = [];
-  let j = 0;
-  while (j < positional.length && !PRESETS.includes(positional[j]?.toLowerCase()) && isNaN(positional[j])) {
-    usernames.push(positional[j].replace('@', ''));
-    j++;
-  }
-
-  // preset
-  if (j < positional.length && PRESETS.includes(positional[j].toLowerCase())) {
-    autoPreset = positional[j].toLowerCase();
-    j++;
-    autoCollectMax = parseInt(positional[j]) || 1; j++;
-    autoScrapeDepth = parseInt(positional[j]) || 50; j++;
-    autoMaxComments = parseInt(positional[j]) || 200; j++;
-  } else if (usernames.length > 0) {
-    autoCollectMax = parseInt(positional[j]) || 1; j++;
-    autoScrapeDepth = parseInt(positional[j]) || 50; j++;
-    autoMaxComments = parseInt(positional[j]) || 200;
-  }
-
-  return {
-    subcommand: 'auto',
-    autoUsernames: usernames,
-    autoCollectMax,
-    autoScrapeDepth,
-    autoMaxComments,
-    autoPreset,
-    autoSwitchDelay,
-    autoCommentDelay,
-    outputFile,
-    autoWatch,
-    autoWatchPort,
-    urls: [],
-    outputFormat: 'json',
-    exploreCount: 0,
-    showConfig: false,
-    showHelp: false,
-    customProxy: null,
-    configAction: null,
-    configValue: null,
-    pipeMode: false,
-    filterStr: null,
-  };
-}
-
-function parseVideosArgs(args) {
-  let videosUsername = null;
-  let videosMax = 5;
-  let outputFile = null;
-
-  const positional = [];
-
-  for (let i = 0; i < args.length; i++) {
-    const arg = args[i];
-    if (arg === '-o' || arg === '--output') {
-      outputFile = args[++i];
-    } else {
-      positional.push(arg);
-    }
-  }
-
-  videosUsername = positional[0] ? positional[0].replace('@', '') : null;
-  videosMax = parseInt(positional[1]) || 5;
-
-  return {
-    subcommand: 'videos',
-    videosUsername,
-    videosMax,
-    outputFile,
-    urls: [],
-    outputFormat: 'json',
-    exploreCount: 0,
-    showConfig: false,
-    showHelp: false,
-    customProxy: null,
-    configAction: null,
-    configValue: null,
-    pipeMode: false,
-    filterStr: null,
-  };
-}
-
-function parseWatchArgs(args) {
-  let outputFile = null;
-  let watchPort = 3000;
-
-  for (let i = 0; i < args.length; i++) {
-    const arg = args[i];
-    if (arg === '-o' || arg === '--output') {
-      outputFile = args[++i];
-    } else if (arg === '-p') {
-      watchPort = parseInt(args[++i]) || 3000;
-    }
-  }
-
-  return {
-    subcommand: 'watch',
-    outputFile,
-    watchPort,
-    urls: [],
-    outputFormat: 'json',
-    exploreCount: 0,
-    showConfig: false,
-    showHelp: false,
-    customProxy: null,
-    configAction: null,
-    configValue: null,
-    pipeMode: false,
-    filterStr: null,
-  };
-}
-
-export function parseArgs() {
-  const args = process.argv.slice(2);
-
-  if (args.length > 0 && args[0] === 'scrape') {
-    return parseScrapeArgs(args.slice(1));
-  }
-
-  if (args.length > 0 && args[0] === 'videos') {
-    return parseVideosArgs(args.slice(1));
-  }
-
-  if (args.length > 0 && args[0] === 'auto') {
-    return parseAutoArgs(args.slice(1));
-  }
-
-  if (args.length > 0 && args[0] === 'watch') {
-    return parseWatchArgs(args.slice(1));
-  }
-
-  const urls = [];
-  let inputFile = null;
-  let outputFile = null;
-  let outputFormat = 'json';
-  let exploreCount = 0;
-  let showConfig = false;
-  let showHelp = false;
-  let customProxy = null;
-  let configAction = null;
-  let configValue = null;
-  let pipeMode = false;
-  let filterStr = null;
-
-  for (let i = 0; i < args.length; i++) {
-    const arg = args[i];
-
-    if (arg === '--explore') {
-      exploreCount = args[i + 1] && !args[i + 1].startsWith('http')
-        ? parseInt(args[++i], 10)
-        : 100;
-    } else if (arg === '--proxy') {
-      customProxy = args[++i];
-    } else if (arg === '--filter') {
-      filterStr = args[++i];
-    } else if (arg === 'config') {
-      configAction = args[i + 1];
-      if (configAction === 'set' || configAction === 'set-proxy' || configAction === 'set-browser') {
-        configValue = args[i + 2];
-        i += 2;
-      } else {
-        i++;
-      }
-    } else if (arg === '--pipe') {
-      pipeMode = true;
-    } else if (arg === '-i' || arg === '--input') {
-      inputFile = args[++i];
-    } else if (arg === '-o' || arg === '--output') {
-      outputFile = args[++i];
-    } else if (arg === '-f' || arg === '--format') {
-      outputFormat = args[++i];
-    } else if (arg === '-c' || arg === '--config') {
-      showConfig = true;
-    } else if (arg === '-h' || arg === '--help') {
-      showHelp = true;
-    } else if (arg.startsWith('http')) {
-      urls.push(arg);
-    }
-  }
-
-  if (inputFile) {
-    const content = readFileSync(inputFile, 'utf-8');
-    const lines = content.split(/\r?\n/).map(l => l.trim()).filter(l => l.startsWith('http'));
-    urls.push(...lines);
-  }
-
-  return { urls, outputFile, outputFormat, exploreCount, showConfig, showHelp, customProxy, configAction, configValue, pipeMode, filterStr };
+import { readFileSync } from 'fs';
+import { proxy } from './constants.js';
+
+const PRESETS = ['fast', 'normal', 'slow', 'stealth'];
+
+function parseScrapeArgs(args) {
+  let scrapeUrl = null;
+  let scrapePreset = null;
+  let scrapeMaxVideos = 20;
+  let scrapeMaxComments = 999;
+  let scrapeMaxGuess = 10;
+  let scrapeSwitchDelay = null;
+  let scrapeCommentDelay = null;
+  let outputFile = null;
+
+  const positional = [];
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '-o' || arg === '--output') {
+      outputFile = args[++i];
+    } else if (arg === '--switch-delay') {
+      scrapeSwitchDelay = parseInt(args[++i]) || null;
+    } else if (arg === '--comment-delay') {
+      scrapeCommentDelay = parseInt(args[++i]) || null;
+    } else {
+      positional.push(arg);
+    }
+  }
+
+  scrapeUrl = positional[0] || null;
+
+  if (positional[1]) {
+    if (PRESETS.includes(positional[1].toLowerCase())) {
+      scrapePreset = positional[1].toLowerCase();
+      scrapeMaxVideos = parseInt(positional[2]) || 20;
+      scrapeMaxComments = parseInt(positional[3]) || 999;
+      scrapeMaxGuess = parseInt(positional[4]) || 10;
+    } else {
+      scrapeMaxVideos = parseInt(positional[1]) || 20;
+      scrapeMaxComments = parseInt(positional[2]) || 999;
+      scrapeMaxGuess = parseInt(positional[3]) || 10;
+    }
+  }
+
+  return {
+    subcommand: 'scrape',
+    scrapeUrl,
+    scrapePreset,
+    scrapeMaxVideos,
+    scrapeMaxComments,
+    scrapeMaxGuess,
+    scrapeSwitchDelay,
+    scrapeCommentDelay,
+    outputFile,
+    urls: [],
+    outputFormat: 'json',
+    exploreCount: 0,
+    showConfig: false,
+    showHelp: false,
+    customProxy: null,
+    configAction: null,
+    configValue: null,
+    pipeMode: false,
+    filterStr: null,
+  };
+}
+
+function parseAutoArgs(args) {
+  let autoCollectMax = 1;
+  let autoScrapeDepth = 50;
+  let autoMaxComments = 200;
+  let autoMaxGuess = 10;
+  let autoPreset = 'fast';
+  let autoSwitchDelay = null;
+  let autoCommentDelay = null;
+  let outputFile = null;
+  let autoWatch = false;
+  let autoWatchPort = 3000;
+  let autoEnableFollow = false;
+  let autoMaxFollowing = 200;
+  let autoMaxFollowers = 200;
+
+  const positional = [];
+  const PRESETS = ['fast', 'normal', 'slow', 'stealth'];
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '-o' || arg === '--output') {
+      outputFile = args[++i];
+    } else if (arg === '--switch-delay') {
+      autoSwitchDelay = parseInt(args[++i]) || null;
+    } else if (arg === '--comment-delay') {
+      autoCommentDelay = parseInt(args[++i]) || null;
+    } else if (arg === '--watch') {
+      autoWatch = true;
+      if (args[i + 1] === '-p') {
+        autoWatchPort = parseInt(args[i + 2]) || 3000;
+        i += 2;
+      }
+    } else if (arg === '-p' && autoWatch) {
+      autoWatchPort = parseInt(args[++i]) || 3000;
+    } else if (arg === '--enable-follow') {
+      autoEnableFollow = true;
+    } else if (arg === '--max-following') {
+      autoMaxFollowing = parseInt(args[++i]) || 200;
+    } else if (arg === '--max-followers') {
+      autoMaxFollowers = parseInt(args[++i]) || 200;
+    } else {
+      positional.push(arg);
+    }
+  }
+
+  // 收集用户名（非 preset、非数字的都是用户名）
+  const usernames = [];
+  let j = 0;
+  while (j < positional.length && !PRESETS.includes(positional[j]?.toLowerCase()) && isNaN(positional[j])) {
+    usernames.push(positional[j].replace('@', ''));
+    j++;
+  }
+
+  // preset
+  if (j < positional.length && PRESETS.includes(positional[j].toLowerCase())) {
+    autoPreset = positional[j].toLowerCase();
+    j++;
+    autoCollectMax = parseInt(positional[j]) || 1; j++;
+    autoScrapeDepth = parseInt(positional[j]) || 50; j++;
+    autoMaxComments = parseInt(positional[j]) || 200; j++;
+    autoMaxGuess = parseInt(positional[j]) || 10; j++;
+  } else if (usernames.length > 0) {
+    autoCollectMax = parseInt(positional[j]) || 1; j++;
+    autoScrapeDepth = parseInt(positional[j]) || 50; j++;
+    autoMaxComments = parseInt(positional[j]) || 200; j++;
+    autoMaxGuess = parseInt(positional[j]) || 10;
+  }
+
+  return {
+    subcommand: 'auto',
+    autoUsernames: usernames,
+    autoCollectMax,
+    autoScrapeDepth,
+    autoMaxComments,
+    autoMaxGuess,
+    autoPreset,
+    autoSwitchDelay,
+    autoCommentDelay,
+    outputFile,
+    autoWatch,
+    autoWatchPort,
+    autoEnableFollow,
+    autoMaxFollowing,
+    autoMaxFollowers,
+    urls: [],
+    outputFormat: 'json',
+    exploreCount: 0,
+    showConfig: false,
+    showHelp: false,
+    customProxy: null,
+    configAction: null,
+    configValue: null,
+    pipeMode: false,
+    filterStr: null,
+  };
+}
+
+function parseExploreArgs(args) {
+  let outputFile = null;
+  let explorePreset = 'normal';
+  let exploreMaxComments = 100;
+  let exploreMaxGuess = 0;
+  let exploreEnableFollow = true;
+  let exploreMaxFollowing = 200;
+  let exploreMaxFollowers = 200;
+  let exploreLocation = 'ES';
+  let exploreWatch = false;
+  let exploreWatchPort = 3000;
+  let exploreMaxUsers = 0;
+
+  const positional = [];
+  const PRESETS = ['fast', 'normal', 'slow', 'stealth'];
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '-o' || arg === '--output') {
+      outputFile = args[++i];
+    } else if (arg === '--max-comments') {
+      exploreMaxComments = parseInt(args[++i]) || 0;
+    } else if (arg === '--max-guess') {
+      exploreMaxGuess = parseInt(args[++i]) || 0;
+    } else if (arg === '--location') {
+      exploreLocation = args[++i];
+    } else if (arg === '--enable-follow') {
+      exploreEnableFollow = true;
+    } else if (arg === '--disable-follow') {
+      exploreEnableFollow = false;
+    } else if (arg === '--max-following') {
+      exploreMaxFollowing = parseInt(args[++i]) || 200;
+    } else if (arg === '--max-followers') {
+      exploreMaxFollowers = parseInt(args[++i]) || 200;
+    } else if (arg === '--max-users') {
+      exploreMaxUsers = parseInt(args[++i]) || 0;
+    } else if (arg === '--watch') {
+      exploreWatch = true;
+      if (args[i + 1] === '-p') {
+        exploreWatchPort = parseInt(args[i + 2]) || 3000;
+        i += 2;
+      }
+    } else {
+      positional.push(arg);
+    }
+  }
+
+  const usernames = [];
+  let j = 0;
+  while (j < positional.length && !PRESETS.includes(positional[j]?.toLowerCase()) && isNaN(positional[j])) {
+    usernames.push(positional[j].replace('@', ''));
+    j++;
+  }
+
+  if (j < positional.length && PRESETS.includes(positional[j].toLowerCase())) {
+    explorePreset = positional[j].toLowerCase();
+    j++;
+  }
+
+  return {
+    subcommand: 'explore',
+    exploreUsernames: usernames,
+    explorePreset,
+    exploreMaxComments,
+    exploreMaxGuess,
+    exploreEnableFollow,
+    exploreMaxFollowing,
+    exploreMaxFollowers,
+    exploreLocation,
+    exploreWatch,
+    exploreWatchPort,
+    exploreMaxUsers,
+    outputFile,
+    urls: [],
+    outputFormat: 'json',
+    exploreCount: 0,
+    showConfig: false,
+    showHelp: false,
+    customProxy: null,
+    configAction: null,
+    configValue: null,
+    pipeMode: false,
+    filterStr: null,
+  };
+}
+
+function parseVideosArgs(args) {
+  let videosUsername = null;
+  let videosMax = 5;
+  let outputFile = null;
+
+  const positional = [];
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '-o' || arg === '--output') {
+      outputFile = args[++i];
+    } else {
+      positional.push(arg);
+    }
+  }
+
+  videosUsername = positional[0] ? positional[0].replace('@', '') : null;
+  videosMax = parseInt(positional[1]) || 5;
+
+  return {
+    subcommand: 'videos',
+    videosUsername,
+    videosMax,
+    outputFile,
+    urls: [],
+    outputFormat: 'json',
+    exploreCount: 0,
+    showConfig: false,
+    showHelp: false,
+    customProxy: null,
+    configAction: null,
+    configValue: null,
+    pipeMode: false,
+    filterStr: null,
+  };
+}
+
+function parseWatchArgs(args) {
+  let outputFile = null;
+  let watchPort = 3000;
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '-o' || arg === '--output') {
+      outputFile = args[++i];
+    } else if (arg === '-p') {
+      watchPort = parseInt(args[++i]) || 3000;
+    }
+  }
+
+  return {
+    subcommand: 'watch',
+    outputFile,
+    watchPort,
+    urls: [],
+    outputFormat: 'json',
+    exploreCount: 0,
+    showConfig: false,
+    showHelp: false,
+    customProxy: null,
+    configAction: null,
+    configValue: null,
+    pipeMode: false,
+    filterStr: null,
+  };
+}
+
+export function parseArgs() {
+  const args = process.argv.slice(2);
+
+  if (args.length > 0 && args[0] === 'scrape') {
+    return parseScrapeArgs(args.slice(1));
+  }
+
+  if (args.length > 0 && args[0] === 'videos') {
+    return parseVideosArgs(args.slice(1));
+  }
+
+  if (args.length > 0 && args[0] === 'auto') {
+    return parseAutoArgs(args.slice(1));
+  }
+
+  if (args.length > 0 && args[0] === 'explore') {
+    return parseExploreArgs(args.slice(1));
+  }
+
+  if (args.length > 0 && args[0] === 'watch') {
+    return parseWatchArgs(args.slice(1));
+  }
+
+  const urls = [];
+  let inputFile = null;
+  let outputFile = null;
+  let outputFormat = 'json';
+  let exploreCount = 0;
+  let showConfig = false;
+  let showHelp = false;
+  let customProxy = null;
+  let configAction = null;
+  let configValue = null;
+  let pipeMode = false;
+  let filterStr = null;
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+
+    if (arg === '--explore') {
+      exploreCount = args[i + 1] && !args[i + 1].startsWith('http')
+        ? parseInt(args[++i], 10)
+        : 100;
+    } else if (arg === '--proxy') {
+      customProxy = args[++i];
+    } else if (arg === '--filter') {
+      filterStr = args[++i];
+    } else if (arg === 'config') {
+      configAction = args[i + 1];
+      if (configAction === 'set' || configAction === 'set-proxy' || configAction === 'set-browser') {
+        configValue = args[i + 2];
+        i += 2;
+      } else {
+        i++;
+      }
+    } else if (arg === '--pipe') {
+      pipeMode = true;
+    } else if (arg === '-i' || arg === '--input') {
+      inputFile = args[++i];
+    } else if (arg === '-o' || arg === '--output') {
+      outputFile = args[++i];
+    } else if (arg === '-f' || arg === '--format') {
+      outputFormat = args[++i];
+    } else if (arg === '-c' || arg === '--config') {
+      showConfig = true;
+    } else if (arg === '-h' || arg === '--help') {
+      showHelp = true;
+    } else if (arg.startsWith('http')) {
+      urls.push(arg);
+    }
+  }
+
+  if (inputFile) {
+    const content = readFileSync(inputFile, 'utf-8');
+    const lines = content.split(/\r?\n/).map(l => l.trim()).filter(l => l.startsWith('http'));
+    urls.push(...lines);
+  }
+
+  return { urls, outputFile, outputFormat, exploreCount, showConfig, showHelp, customProxy, configAction, configValue, pipeMode, filterStr };
 }