tt-help-cli-ycl 1.3.44 → 1.3.46

package/src/cli/utils.js

@@ -1,18 +1,18 @@
-import { writeFileSync } from 'fs';
-
-export function cleanError(msg) {
-  return msg
-    .replace(/\x1b\[[0-9;]*m/g, '')
-    .replace(/\s*- navigating to.*/s, '')
-    .replace(/\s*Call log:/s, '')
-    .trim();
-}
-
-export function writeJson(data, outputFile) {
-  const json = JSON.stringify(data, null, 2);
-  if (outputFile) {
-    writeFileSync(outputFile, json, 'utf-8');
-  } else {
-    process.stdout.write(json + '\n');
-  }
-}
+import { writeFileSync } from 'fs';
+
+export function cleanError(msg) {
+  return msg
+    .replace(/\x1b\[[0-9;]*m/g, '')
+    .replace(/\s*- navigating to.*/s, '')
+    .replace(/\s*Call log:/s, '')
+    .trim();
+}
+
+export function writeJson(data, outputFile) {
+  const json = JSON.stringify(data, null, 2);
+  if (outputFile) {
+    writeFileSync(outputFile, json, 'utf-8');
+  } else {
+    process.stdout.write(json + '\n');
+  }
+}

package/src/cli/videos.js

@@ -1,41 +1,41 @@
-import { writeFileSync } from 'fs';
-
-export async function handleVideos(options) {
-  const { videosUsername, videosMax, outputFile } = options;
-
-  if (!videosUsername) {
-    console.error('用法: tt-help videos <用户名> [最大视频数] [-o 输出路径]');
-    console.error('示例: tt-help videos bar.lar.lar.moeta 1000');
-    console.error('      tt-help videos username 50 -o videos.json');
-    console.error('');
-    console.error('选项: -o, --output <路径>  输出到文件（默认输出到 stdout）');
-    process.exit(1);
-  }
-
-  const { runGetUserVideos } = await import('../videos/core.js');
-
-  let browser;
-  try {
-    const { output, browser: b } = await runGetUserVideos({
-      username: videosUsername,
-      maxVideos: videosMax,
-      log: console.error,
-    });
-    browser = b;
-
-    const json = JSON.stringify(output, null, 2);
-    if (outputFile) {
-      writeFileSync(outputFile, json, 'utf-8');
-      console.error(`结果已写入: ${outputFile}`);
-    } else {
-      process.stdout.write(json + '\n');
-    }
-
-    console.error(`\n共 ${output.videos.length} 个视频, 用户: @${videosUsername}`);
-  } catch (err) {
-    console.error(`获取用户视频失败: ${err.message}`);
-    process.exit(1);
-  } finally {
-    if (browser) await browser.close().catch(() => {});
-  }
-}
+import { writeFileSync } from 'fs';
+
+export async function handleVideos(options) {
+  const { videosUsername, videosMax, outputFile } = options;
+
+  if (!videosUsername) {
+    console.error('用法: tt-help videos <用户名> [最大视频数] [-o 输出路径]');
+    console.error('示例: tt-help videos bar.lar.lar.moeta 1000');
+    console.error('      tt-help videos username 50 -o videos.json');
+    console.error('');
+    console.error('选项: -o, --output <路径>  输出到文件（默认输出到 stdout）');
+    process.exit(1);
+  }
+
+  const { runGetUserVideos } = await import('../videos/core.js');
+
+  let browser;
+  try {
+    const { output, browser: b } = await runGetUserVideos({
+      username: videosUsername,
+      maxVideos: videosMax,
+      log: console.error,
+    });
+    browser = b;
+
+    const json = JSON.stringify(output, null, 2);
+    if (outputFile) {
+      writeFileSync(outputFile, json, 'utf-8');
+      console.error(`结果已写入: ${outputFile}`);
+    } else {
+      process.stdout.write(json + '\n');
+    }
+
+    console.error(`\n共 ${output.videos.length} 个视频, 用户: @${videosUsername}`);
+  } catch (err) {
+    console.error(`获取用户视频失败: ${err.message}`);
+    process.exit(1);
+  } finally {
+    if (browser) await browser.close().catch(() => {});
+  }
+}

package/src/cli/videostats.js

@@ -1,196 +1,196 @@
-import path from "path";
-import Database from "better-sqlite3";
-import { TikTokScraper } from "../lib/tiktok-scraper.mjs";
-
-function resolveVideoSource(inputPath) {
-  if (!inputPath) return null;
-
-  const resolved = path.resolve(inputPath);
-  const ext = path.extname(resolved).toLowerCase();
-
-  if (ext === ".db") {
-    return { kind: "db", dbPath: resolved, label: resolved };
-  }
-
-  return null;
-}
-
-function ensureVideoStatsColumns(db) {
-  const existing = new Set(
-    db
-      .prepare("PRAGMA table_info(videos)")
-      .all()
-      .map((column) => column.name),
-  );
-  const required = {
-    play_count: "INTEGER",
-    digg_count: "INTEGER",
-    comment_count: "INTEGER",
-    share_count: "INTEGER",
-    collect_count: "INTEGER",
-    stats_updated_at: "INTEGER",
-  };
-  for (const [column, type] of Object.entries(required)) {
-    if (!existing.has(column)) {
-      db.exec(`ALTER TABLE videos ADD COLUMN ${column} ${type}`);
-    }
-  }
-}
-
-function loadVideosFromDb(dbPath) {
-  const db = new Database(dbPath);
-  ensureVideoStatsColumns(db);
-  const rows = db
-    .prepare("SELECT * FROM videos ORDER BY registered_at ASC")
-    .all();
-  const videos = rows.map((row) => ({
-    id: row.id,
-    href: row.href,
-    authorUniqueId: row.author_unique_id,
-    locationCreated: row.location_created,
-    ttSeller: !!row.tt_seller,
-    registeredAt: row.registered_at,
-    userUpdateCount: row.user_update_count,
-    stats:
-      row.play_count === null &&
-      row.digg_count === null &&
-      row.comment_count === null &&
-      row.share_count === null &&
-      row.collect_count === null
-        ? undefined
-        : {
-            playCount: row.play_count,
-            diggCount: row.digg_count,
-            commentCount: row.comment_count,
-            shareCount: row.share_count,
-            collectCount: row.collect_count,
-          },
-  }));
-  return { db, videos };
-}
-
-function writeVideosToDb(db, videos) {
-  const stmt = db.prepare(`
-    UPDATE videos
-    SET play_count = ?,
-        digg_count = ?,
-        comment_count = ?,
-        share_count = ?,
-        collect_count = ?,
-        stats_updated_at = ?
-    WHERE id = ?
-  `);
-  const now = Date.now();
-  const txn = db.transaction((items) => {
-    for (const video of items) {
-      if (!video.stats) continue;
-      stmt.run(
-        video.stats.playCount ?? null,
-        video.stats.diggCount ?? null,
-        video.stats.commentCount ?? null,
-        video.stats.shareCount ?? null,
-        video.stats.collectCount ?? null,
-        now,
-        video.id,
-      );
-    }
-  });
-  txn(videos);
-}
-
-export async function handleVideoStats(options) {
-  const statsSource = options.statsSource || options.statsFile;
-  const { statsParallel } = options;
-
-  if (!statsSource) {
-    console.error("用法: tt-help videostats <db路径> -p <并发数>");
-    console.error("      tt-help videostats data/result.db -p 3");
-    console.error("");
-    console.error("选项: -p, --parallel <N>  并发数（默认: 3）");
-    process.exit(1);
-  }
-
-  const source = resolveVideoSource(statsSource);
-  if (!source) {
-    console.error("[videostats] 仅支持 .db 路径");
-    process.exit(1);
-  }
-
-  let db = null;
-  let videos = [];
-  ({ db, videos } = loadVideosFromDb(source.dbPath));
-
-  if (!Array.isArray(videos) || videos.length === 0) {
-    console.error("[videostats] 视频源为空或格式错误");
-    db?.close();
-    process.exit(1);
-  }
-
-  const poolSize = Math.min(statsParallel, videos.length);
-  console.error(
-    `[videostats] 读取 ${source.label}: ${videos.length} 个视频，并发: ${poolSize}`,
-  );
-
-  // 初始化 scraper
-  const scraper = new TikTokScraper({ poolSize });
-  await scraper.init();
-  console.error("[videostats] 浏览器初始化完成，开始刷新...");
-
-  let success = 0,
-    failed = 0,
-    skipped = 0;
-
-  try {
-    for (let i = 0; i < videos.length; i += poolSize) {
-      const batch = videos.slice(i, i + poolSize);
-
-      const results = await Promise.allSettled(
-        batch.map(async (v) => {
-          if (!v.href) return { video: v, error: "no href" };
-          try {
-            const info = await scraper.getVideoInfo(v.href);
-            if (!info) return { video: v, error: "no info" };
-            return { video: v, info };
-          } catch (err) {
-            return { video: v, error: err.message };
-          }
-        }),
-      );
-
-      for (const result of results) {
-        if (result.status === "fulfilled") {
-          const { video, info, error } = result.value;
-          if (info && info.stats) {
-            video.stats = info.stats;
-            success++;
-          } else {
-            failed++;
-            const id = video.id || video.href || "?";
-            console.error(
-              `  [失败] ${id.substring(0, 30)} - ${error || "获取失败"}`,
-            );
-          }
-        } else {
-          failed++;
-        }
-      }
-
-      const processed = Math.min(i + poolSize, videos.length);
-      console.error(
-        `[videostats] 进度: ${processed}/${videos.length} (成功: ${success}, 失败: ${failed})`,
-      );
-    }
-
-    writeVideosToDb(db, videos);
-    console.error(``);
-    console.error(`[videostats] 完成: 成功 ${success}, 失败 ${failed}`);
-    console.error(`[videostats] 已更新 ${source.dbPath}`);
-  } catch (err) {
-    console.error(`[videostats] 异常: ${err.message}`);
-    db?.close();
-    process.exit(1);
-  } finally {
-    db?.close();
-    await scraper.close();
-  }
-}
+import path from "path";
+import Database from "better-sqlite3";
+import { TikTokScraper } from "../lib/tiktok-scraper.mjs";
+
+function resolveVideoSource(inputPath) {
+  if (!inputPath) return null;
+
+  const resolved = path.resolve(inputPath);
+  const ext = path.extname(resolved).toLowerCase();
+
+  if (ext === ".db") {
+    return { kind: "db", dbPath: resolved, label: resolved };
+  }
+
+  return null;
+}
+
+function ensureVideoStatsColumns(db) {
+  const existing = new Set(
+    db
+      .prepare("PRAGMA table_info(videos)")
+      .all()
+      .map((column) => column.name),
+  );
+  const required = {
+    play_count: "INTEGER",
+    digg_count: "INTEGER",
+    comment_count: "INTEGER",
+    share_count: "INTEGER",
+    collect_count: "INTEGER",
+    stats_updated_at: "INTEGER",
+  };
+  for (const [column, type] of Object.entries(required)) {
+    if (!existing.has(column)) {
+      db.exec(`ALTER TABLE videos ADD COLUMN ${column} ${type}`);
+    }
+  }
+}
+
+function loadVideosFromDb(dbPath) {
+  const db = new Database(dbPath);
+  ensureVideoStatsColumns(db);
+  const rows = db
+    .prepare("SELECT * FROM videos ORDER BY registered_at ASC")
+    .all();
+  const videos = rows.map((row) => ({
+    id: row.id,
+    href: row.href,
+    authorUniqueId: row.author_unique_id,
+    locationCreated: row.location_created,
+    ttSeller: !!row.tt_seller,
+    registeredAt: row.registered_at,
+    userUpdateCount: row.user_update_count,
+    stats:
+      row.play_count === null &&
+      row.digg_count === null &&
+      row.comment_count === null &&
+      row.share_count === null &&
+      row.collect_count === null
+        ? undefined
+        : {
+            playCount: row.play_count,
+            diggCount: row.digg_count,
+            commentCount: row.comment_count,
+            shareCount: row.share_count,
+            collectCount: row.collect_count,
+          },
+  }));
+  return { db, videos };
+}
+
+function writeVideosToDb(db, videos) {
+  const stmt = db.prepare(`
+    UPDATE videos
+    SET play_count = ?,
+        digg_count = ?,
+        comment_count = ?,
+        share_count = ?,
+        collect_count = ?,
+        stats_updated_at = ?
+    WHERE id = ?
+  `);
+  const now = Date.now();
+  const txn = db.transaction((items) => {
+    for (const video of items) {
+      if (!video.stats) continue;
+      stmt.run(
+        video.stats.playCount ?? null,
+        video.stats.diggCount ?? null,
+        video.stats.commentCount ?? null,
+        video.stats.shareCount ?? null,
+        video.stats.collectCount ?? null,
+        now,
+        video.id,
+      );
+    }
+  });
+  txn(videos);
+}
+
+export async function handleVideoStats(options) {
+  const statsSource = options.statsSource || options.statsFile;
+  const { statsParallel } = options;
+
+  if (!statsSource) {
+    console.error("用法: tt-help videostats <db路径> -p <并发数>");
+    console.error("      tt-help videostats data/result.db -p 3");
+    console.error("");
+    console.error("选项: -p, --parallel <N>  并发数（默认: 3）");
+    process.exit(1);
+  }
+
+  const source = resolveVideoSource(statsSource);
+  if (!source) {
+    console.error("[videostats] 仅支持 .db 路径");
+    process.exit(1);
+  }
+
+  let db = null;
+  let videos = [];
+  ({ db, videos } = loadVideosFromDb(source.dbPath));
+
+  if (!Array.isArray(videos) || videos.length === 0) {
+    console.error("[videostats] 视频源为空或格式错误");
+    db?.close();
+    process.exit(1);
+  }
+
+  const poolSize = Math.min(statsParallel, videos.length);
+  console.error(
+    `[videostats] 读取 ${source.label}: ${videos.length} 个视频，并发: ${poolSize}`,
+  );
+
+  // 初始化 scraper
+  const scraper = new TikTokScraper({ poolSize });
+  await scraper.init();
+  console.error("[videostats] 浏览器初始化完成，开始刷新...");
+
+  let success = 0,
+    failed = 0,
+    skipped = 0;
+
+  try {
+    for (let i = 0; i < videos.length; i += poolSize) {
+      const batch = videos.slice(i, i + poolSize);
+
+      const results = await Promise.allSettled(
+        batch.map(async (v) => {
+          if (!v.href) return { video: v, error: "no href" };
+          try {
+            const info = await scraper.getVideoInfo(v.href);
+            if (!info) return { video: v, error: "no info" };
+            return { video: v, info };
+          } catch (err) {
+            return { video: v, error: err.message };
+          }
+        }),
+      );
+
+      for (const result of results) {
+        if (result.status === "fulfilled") {
+          const { video, info, error } = result.value;
+          if (info && info.stats) {
+            video.stats = info.stats;
+            success++;
+          } else {
+            failed++;
+            const id = video.id || video.href || "?";
+            console.error(
+              `  [失败] ${id.substring(0, 30)} - ${error || "获取失败"}`,
+            );
+          }
+        } else {
+          failed++;
+        }
+      }
+
+      const processed = Math.min(i + poolSize, videos.length);
+      console.error(
+        `[videostats] 进度: ${processed}/${videos.length} (成功: ${success}, 失败: ${failed})`,
+      );
+    }
+
+    writeVideosToDb(db, videos);
+    console.error(``);
+    console.error(`[videostats] 完成: 成功 ${success}, 失败 ${failed}`);
+    console.error(`[videostats] 已更新 ${source.dbPath}`);
+  } catch (err) {
+    console.error(`[videostats] 异常: ${err.message}`);
+    db?.close();
+    process.exit(1);
+  } finally {
+    db?.close();
+    await scraper.close();
+  }
+}

package/src/cli/watch.js

@@ -1,30 +1,30 @@
-import { mkdirSync } from "fs";
-import path from "path";
-import { createStore } from "../watch/data-store.js";
-import { startWatchServer, openBrowser } from "../watch/server.js";
-
-export async function handleWatch(options) {
-  const dataAnchor = options.dataAnchor || options.outputFile;
-  const { watchPort } = options;
-
-  if (!dataAnchor) {
-    console.error("用法: tt-help watch -o <db路径> [-p 端口]");
-    console.error("      tt-help watch -o data/result.db");
-    console.error("      tt-help watch -o data/result.db -p 8080");
-    process.exit(1);
-  }
-
-  mkdirSync(path.dirname(path.resolve(dataAnchor)), { recursive: true });
-
-  const store = createStore(dataAnchor);
-  const { server, port } = await startWatchServer(dataAnchor, watchPort, store);
-  openBrowser(port);
-
-  process.once("SIGINT", () => {
-    store.stopBackup();
-    server.close();
-    process.exit(0);
-  });
-
-  console.error("按 Ctrl+C 停止监控服务");
-}
+import { mkdirSync } from "fs";
+import path from "path";
+import { createStore } from "../watch/data-store.js";
+import { startWatchServer, openBrowser } from "../watch/server.js";
+
+export async function handleWatch(options) {
+  const dataAnchor = options.dataAnchor || options.outputFile;
+  const { watchPort } = options;
+
+  if (!dataAnchor) {
+    console.error("用法: tt-help watch -o <db路径> [-p 端口]");
+    console.error("      tt-help watch -o data/result.db");
+    console.error("      tt-help watch -o data/result.db -p 8080");
+    process.exit(1);
+  }
+
+  mkdirSync(path.dirname(path.resolve(dataAnchor)), { recursive: true });
+
+  const store = createStore(dataAnchor);
+  const { server, port } = await startWatchServer(dataAnchor, watchPort, store);
+  openBrowser(port);
+
+  process.once("SIGINT", () => {
+    store.stopBackup();
+    server.close();
+    process.exit(0);
+  });
+
+  console.error("按 Ctrl+C 停止监控服务");
+}

package/src/cli/webserver.js

@@ -0,0 +1,19 @@
+import { startWebserver, closeScraper } from "../webserver/server.mjs";
+
+export async function handleWebserver(options) {
+  const { webserverPort } = options;
+
+  const { server, port } = await startWebserver(webserverPort);
+
+  process.once("SIGINT", async () => {
+    await closeScraper();
+    server.close();
+    process.exit(0);
+  });
+
+  process.once("SIGTERM", async () => {
+    await closeScraper();
+    server.close();
+    process.exit(0);
+  });
+}