tt-help-cli-ycl 1.3.34 → 1.3.36

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,33 +1,144 @@
-import { readFileSync, writeFileSync } from 'fs';
-import { TikTokScraper } from '../lib/tiktok-scraper.mjs';
+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 { statsFile, statsParallel } = options;
+  const statsSource = options.statsSource || options.statsFile;
+  const { statsParallel } = options;
 
-  if (!statsFile) {
-    console.error('用法: tt-help videostats <文件路径> -p <并发数>');
-    console.error('示例: tt-help videostats data/result-videos.json -p 3');
-    console.error('');
-    console.error('选项: -p, --parallel <N>  并发数（默认: 3）');
+  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 videos = JSON.parse(readFileSync(statsFile, 'utf-8'));
+  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] 文件为空或格式错误');
+    console.error("[videostats] 视频源为空或格式错误");
+    db?.close();
     process.exit(1);
   }
 
   const poolSize = Math.min(statsParallel, videos.length);
-  console.error(`[videostats] 读取 ${statsFile}: ${videos.length} 个视频，并发: ${poolSize}`);
+  console.error(
+    `[videostats] 读取 ${source.label}: ${videos.length} 个视频，并发: ${poolSize}`,
+  );
 
   // 初始化 scraper
   const scraper = new TikTokScraper({ poolSize });
   await scraper.init();
-  console.error('[videostats] 浏览器初始化完成，开始刷新...');
+  console.error("[videostats] 浏览器初始化完成，开始刷新...");
 
-  let success = 0, failed = 0, skipped = 0;
+  let success = 0,
+    failed = 0,
+    skipped = 0;
 
   try {
     for (let i = 0; i < videos.length; i += poolSize) {
@@ -35,27 +146,29 @@ export async function handleVideoStats(options) {
 
       const results = await Promise.allSettled(
         batch.map(async (v) => {
-          if (!v.href) return { video: v, error: 'no href' };
+          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' };
+            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') {
+        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 || '获取失败'}`);
+            const id = video.id || video.href || "?";
+            console.error(
+              `  [失败] ${id.substring(0, 30)} - ${error || "获取失败"}`,
+            );
           }
         } else {
           failed++;
@@ -63,19 +176,21 @@ export async function handleVideoStats(options) {
       }
 
       const processed = Math.min(i + poolSize, videos.length);
-      console.error(`[videostats] 进度: ${processed}/${videos.length} (成功: ${success}, 失败: ${failed})`);
+      console.error(
+        `[videostats] 进度: ${processed}/${videos.length} (成功: ${success}, 失败: ${failed})`,
+      );
     }
 
-    // 写回文件
-    writeFileSync(statsFile, JSON.stringify(videos, null, 2), 'utf-8');
+    writeVideosToDb(db, videos);
     console.error(``);
     console.error(`[videostats] 完成: 成功 ${success}, 失败 ${failed}`);
-    console.error(`[videostats] 已更新 ${statsFile}`);
-
+    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,31 +1,30 @@
-import { existsSync } from 'fs';
-import { createStore } from '../watch/data-store.js';
-import { startWatchServer, openBrowser } from '../watch/server.js';
-
-export async function handleWatch(options) {
-  const { outputFile, watchPort } = options;
-
-  if (!outputFile) {
-    console.error('用法: tt-help watch -o <数据文件> [-p 端口]');
-    console.error('示例: tt-help watch -o data.json');
-    console.error('      tt-help watch -o data.json -p 8080');
-    process.exit(1);
-  }
-
-  if (!existsSync(outputFile)) {
-    console.error(`文件不存�? ${outputFile}`);
-    process.exit(1);
-  }
-
-  const store = createStore(outputFile);
-  const { server, port } = await startWatchServer(outputFile, 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 停止监控服务");
+}