tt-help-cli-ycl 1.3.35 → 1.3.37

package/src/cli/config.js

@@ -1,11 +1,21 @@
-import { HELP_TEXT, configPath, saveBrowser, saveUserId, saveMaxFollowing, saveMaxFollowers, saveMaxVideos, saveMaxComments, getConfigText } from '../lib/constants.js';
-import { readFileSync, writeFileSync, existsSync } from 'fs';
-import { fileURLToPath } from 'url';
-import { dirname, join } from 'path';
+import {
+  HELP_TEXT,
+  configPath,
+  saveBrowser,
+  saveUserId,
+  saveMaxFollowing,
+  saveMaxFollowers,
+  saveMaxVideos,
+  saveMaxComments,
+  getConfigText,
+} from "../lib/constants.js";
+import { readFileSync, writeFileSync, existsSync } from "fs";
+import { fileURLToPath } from "url";
+import { dirname, join } from "path";
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
-const pkgPath = join(__dirname, '..', '..', 'package.json');
-const { version } = JSON.parse(readFileSync(pkgPath, 'utf-8'));
+const pkgPath = join(__dirname, "..", "..", "package.json");
+const { version } = JSON.parse(readFileSync(pkgPath, "utf-8"));
 
 function showConfig(urls, outputFile) {
   const configLines = getConfigText();
@@ -13,101 +23,103 @@ function showConfig(urls, outputFile) {
     configLines.push(`\n  URL数量: ${urls.length}`);
   }
   if (outputFile) {
-    configLines.push(`  输出文件: ${outputFile}`);
+    configLines.push(`  输出/锚点路径: ${outputFile}`);
   }
-  console.error(configLines.join('\n'));
+  console.error(configLines.join("\n"));
 }
 
-function showUsage() {
-  console.error(HELP_TEXT.join('\n'));
+function showUsage(helpText = HELP_TEXT) {
+  console.error(helpText.join("\n"));
   process.exit(0);
 }
 
 function handleConfig(action, key, value) {
   switch (action) {
-    case 'show': {
+    case "show": {
       const configLines = getConfigText();
-      console.error(configLines.join('\n'));
+      console.error(configLines.join("\n"));
       break;
     }
 
-    case 'set': {
+    case "set": {
       if (!key) {
-        console.error('用法: tt-help config set <key> <value>');
-        console.error('  可用 key: proxy, server, browser, userId, maxFollowing, maxFollowers, maxVideos, maxComments');
+        console.error("用法: tt-help config set <key> <value>");
+        console.error(
+          "  可用 key: proxy, server, browser, userId, maxFollowing, maxFollowers, maxVideos, maxComments",
+        );
         return;
       }
 
       switch (key) {
-        case 'proxy':
+        case "proxy":
           if (!value) {
-            console.error('请提供 proxy 的值');
-            console.error('用法: tt-help config set proxy <代理地址>');
+            console.error("请提供 proxy 的值");
+            console.error("用法: tt-help config set proxy <代理地址>");
             return;
           }
           saveProxy(value);
           console.error(`代理已更新: ${value}`);
           break;
 
-        case 'server':
+        case "server":
           if (!value) {
-            console.error('请提供 server 的值');
+            console.error("请提供 server 的值");
             return;
           }
           saveServer(value);
           console.error(`服务器已更新: ${value}`);
           break;
 
-        case 'browser':
+        case "browser":
           if (!value) {
-            console.error('请提供 browser 的值');
-            console.error('用法: tt-help config set browser <浏览器路径>');
-            console.error('  或: tt-help config set-browser <浏览器路径>');
+            console.error("请提供 browser 的值");
+            console.error("用法: tt-help config set browser <浏览器路径>");
+            console.error("  或: tt-help config set-browser <浏览器路径>");
             return;
           }
           saveBrowser(value);
           console.error(`浏览器路径已更新: ${value}`);
           break;
 
-        case 'userId':
+        case "userId":
           if (!value) {
-            console.error('请提供 userId 的值');
+            console.error("请提供 userId 的值");
             return;
           }
           saveUserId(value);
           console.error(`用户号已更新: ${value}`);
           break;
 
-        case 'maxFollowing':
+        case "maxFollowing":
           if (!value) {
-            console.error('请提供 maxFollowing 的值');
+            console.error("请提供 maxFollowing 的值");
             return;
           }
           saveMaxFollowing(value);
           console.error(`商家关注采集数已更新: ${value}`);
           break;
 
-        case 'maxFollowers':
+        case "maxFollowers":
           if (!value) {
-            console.error('请提供 maxFollowers 的值');
+            console.error("请提供 maxFollowers 的值");
             return;
           }
           saveMaxFollowers(value);
           console.error(`粉丝采集数已更新: ${value}`);
           break;
 
-        case 'maxVideos':
+        case "maxVideos":
           if (!value) {
-            console.error('请提供 maxVideos 的值');
+            console.error("请提供 maxVideos 的值");
             return;
           }
           saveMaxVideos(value);
           console.error(`视频采集数已更新: ${value}`);
           break;
 
-        case 'maxComments':
+        case "maxComments":
           if (!value) {
-            console.error('请提供 maxComments 的值');
+            console.error("请提供 maxComments 的值");
             return;
           }
           saveMaxComments(value);
@@ -116,37 +128,43 @@ function handleConfig(action, key, value) {
 
         default:
           console.error(`未知配置项: ${key}`);
-          console.error('  可用 key: proxy, server, browser, userId, maxFollowing, maxFollowers, maxVideos, maxComments');
+          console.error(
+            "  可用 key: proxy, server, browser, userId, maxFollowing, maxFollowers, maxVideos, maxComments",
+          );
       }
       break;
     }
 
-    case 'reset': {
+    case "reset": {
       if (existsSync(configPath)) {
-        writeFileSync(configPath, '{}', 'utf-8');
-        console.error('配置已重置为默认');
+        writeFileSync(configPath, "{}", "utf-8");
+        console.error("配置已重置为默认");
       } else {
-        console.error('配置文件不存在或已是默认状态');
+        console.error("配置文件不存在或已是默认状态");
       }
       break;
     }
 
     default:
       console.error(`未知配置命令: ${action}`);
-      console.error('用法: tt-help config [show|set|reset]');
+      console.error("用法: tt-help config [show|set|reset]");
   }
 }
 
 function saveProxy(newProxy) {
-  const cfg = existsSync(configPath) ? JSON.parse(readFileSync(configPath, 'utf-8')) : {};
+  const cfg = existsSync(configPath)
+    ? JSON.parse(readFileSync(configPath, "utf-8"))
+    : {};
   cfg.proxy = newProxy;
-  writeFileSync(configPath, JSON.stringify(cfg, null, 2), 'utf-8');
+  writeFileSync(configPath, JSON.stringify(cfg, null, 2), "utf-8");
 }
 
 function saveServer(newServer) {
-  const cfg = existsSync(configPath) ? JSON.parse(readFileSync(configPath, 'utf-8')) : {};
+  const cfg = existsSync(configPath)
+    ? JSON.parse(readFileSync(configPath, "utf-8"))
+    : {};
   cfg.server = newServer;
-  writeFileSync(configPath, JSON.stringify(cfg, null, 2), 'utf-8');
+  writeFileSync(configPath, JSON.stringify(cfg, null, 2), "utf-8");
 }
 
 export { handleConfig, showConfig, showUsage, version };

package/src/cli/db-import.js

@@ -0,0 +1,51 @@
+import { closeStoreDb, importLegacyJsonToDb } from "../watch/data-store.js";
+
+export async function handleDbImport(options) {
+  const { dbPath, usersFilePath, doneFilePath, videosFilePath, showHelp } =
+    options;
+
+  if (showHelp) {
+    console.error(
+      "用法: tt-help db-import --db <db路径> [--users users.json] [--done done.json] [--videos videos.json]",
+    );
+    console.error("");
+    console.error("示例:");
+    console.error(
+      "  tt-help db-import --db data/result.db --users data/result.json --done data/result-done.json --videos data/result-videos.json",
+    );
+    return;
+  }
+
+  if (!dbPath) {
+    console.error("请提供 --db <db路径>");
+    process.exit(1);
+  }
+
+  if (!usersFilePath && !doneFilePath && !videosFilePath) {
+    console.error("至少提供一个导入源：--users / --done / --videos");
+    process.exit(1);
+  }
+
+  try {
+    const result = importLegacyJsonToDb({
+      dbFilePath: dbPath,
+      usersFilePath,
+      doneFilePath,
+      videosFilePath,
+    });
+
+    console.error("[db-import] 导入完成");
+    console.error(`[db-import] 数据库: ${result.dbPath}`);
+    console.error(
+      `[db-import] 新增 users: ${result.usersImported}, 总数: ${result.totalUsers}`,
+    );
+    console.error(
+      `[db-import] 新增 jobs: ${result.jobsImported}, 总数: ${result.totalJobs}`,
+    );
+    console.error(
+      `[db-import] 新增 videos: ${result.videosImported}, 总数: ${result.totalVideos}`,
+    );
+  } finally {
+    closeStoreDb();
+  }
+}