tt-help-cli-ycl 1.3.34 → 1.3.36

package/src/cli/config.js

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