chore: initial commit - CloudSearch v0.0.2

2026-05-15 05:50:50 +08:00
commit d83225d736
102 changed files with 37926 additions and 0 deletions
@@ -0,0 +1,289 @@
+import { getSystemConfig } from "../../admin/system-config.service";
+import { getHeaders, makeQuery } from "./quark-api";
+import { listDir, listDirAllPages } from "./quark-api";
+import { humanDelay } from "./quark-api";
+
+/**
+ * 广告关键词清理模块。
+ * 在转存完成后执行：
+ * 1. 遍历转存的目录，删除文件名/文件夹名含广告关键词的内容
+ * 2. 在转存根目录下创建警示文件夹（置顶提醒）
+ */
+
+// ==================== 配置读取 ====================
+
+/** 从 DB 读取广告关键词列表 */
+export function getAdKeywords(): string[] {
+  const raw = getSystemConfig("quark_ad_keywords") || "";
+  return raw
+    .split("\n")
+    .map((s) => s.trim())
+    .filter(Boolean);
+}
+
+/** 从 DB 读取警示文件夹名称列表 */
+export function getWarningFolderNames(): string[] {
+  const raw = getSystemConfig("quark_warning_folder_names") || "";
+  return raw
+    .split("\n")
+    .map((s) => s.trim())
+    .filter(Boolean);
+}
+
+/** 从 DB 读取可疑文件后缀列表 */
+export function getSusExtensions(): string[] {
+  const raw = getSystemConfig("quark_sus_extensions") || "";
+  if (raw.trim()) {
+    return raw
+      .split("\n")
+      .map((s) => s.trim().toLowerCase().replace(/^\./, ""))
+      .filter(Boolean);
+  }
+  // 默认可疑后缀
+  return ["bat", "exe", "vbs", "scr", "cmd", "com", "pif", "js", "jar", "msi", "reg", "inf", "ps1"];
+}
+
+// ==================== 关键词检测 ====================
+
+/** 检查文件名是否包含任意广告关键词 */
+export function containsAdKeyword(
+  fileName: string,
+  keywords: string[],
+): boolean {
+  if (!keywords.length) return false;
+  const lower = fileName.toLowerCase();
+  return keywords.some((kw) => kw && lower.includes(kw.toLowerCase()));
+}
+
+// ==================== 删除操作 ====================
+
+/**
+ * 遍历指定目录（含子目录），删除匹配广告关键词的文件和文件夹。
+ * 返回删除的文件数。
+ */
+export async function deleteAdFiles(
+  cookie: string,
+  dirFid: string,
+  keywords: string[],
+): Promise<number> {
+  if (!keywords.length) return 0;
+
+  let deletedCount = 0;
+  const stack: string[] = [dirFid];
+  const visited = new Set<string>();
+
+  while (stack.length > 0) {
+    const fid = stack.pop()!;
+    if (visited.has(fid)) continue;
+    visited.add(fid);
+
+    await humanDelay();
+    const files = await listDir(cookie, fid);
+    if (!files || files.length === 0) continue;
+
+    // 先收集所有需要删除的 fid
+    const toDelete: string[] = [];
+    const toKeep: string[] = [];
+
+    const extensions = getSusExtensions();
+    for (const file of files) {
+      const ext = file.file_name.split(".").pop()?.toLowerCase() || "";
+      const isSusExt = extensions.includes(ext);
+      if (containsAdKeyword(file.file_name, keywords) || isSusExt) {
+        toDelete.push(file.fid);
+        console.log(
+          `[Quark-AdCleanup] 标记删除: "${file.file_name}" (fid: ${file.fid})${isSusExt ? " [可疑后缀]" : " [广告关键词]"}`,
+        );
+      } else {
+        toKeep.push(file.fid);
+        // 如果是目录且不删除，继续遍历子目录
+        if (file.dir) {
+          stack.push(file.fid);
+        }
+      }
+    }
+
+    // 批量删除
+    if (toDelete.length > 0) {
+      const deleteOk = await batchDeleteFiles(cookie, toDelete);
+      if (deleteOk) {
+        deletedCount += toDelete.length;
+        console.log(
+          `[Quark-AdCleanup] 已删除 ${toDelete.length} 个广告文件`,
+        );
+      }
+    }
+  }
+
+  return deletedCount;
+}
+
+/**
+ * 批量删除文件/文件夹（移入回收站）。
+ */
+async function batchDeleteFiles(
+  cookie: string,
+  fids: string[],
+): Promise<boolean> {
+  try {
+    const resp = await fetch(
+      `https://drive-pc.quark.cn/1/clouddrive/file/trash?${makeQuery()}`,
+      {
+        method: "POST",
+        headers: {
+          ...getHeaders(cookie),
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify({
+          action_type: 2, // 2 = 移入回收站
+          file_list: fids.map((fid) => ({ fid })),
+          exclude_fids: [],
+        }),
+        signal: AbortSignal.timeout(15000),
+      },
+    );
+    const data = (await resp.json()) as any;
+    if (data.status === 200) {
+      return true;
+    }
+    console.log(
+      `[Quark-AdCleanup] batchDelete 返回非200: status=${data.status} msg=${data.message}`,
+    );
+    return false;
+  } catch (err: any) {
+    console.log(`[Quark-AdCleanup] batchDelete 错误: ${err.message}`);
+    return false;
+  }
+}
+
+// ==================== 警示文件夹创建 ====================
+
+/**
+ * 在转存根目录下创建警示文件夹。
+ * 文件夹名前加 ⚠️ 和空格，让其按字母排序置顶。
+ * 已存在的则跳过。
+ */
+export async function createWarningDirectories(
+  cookie: string,
+  dirNames: string[],
+): Promise<void> {
+  if (!dirNames.length) return;
+
+  // 先获取根目录下所有文件夹，避免重复创建
+  await humanDelay();
+  const rootFiles = await listDirAllPages(cookie, "0");
+  const existingDirs = new Set(
+    rootFiles.filter((f) => f.dir).map((f) => f.file_name),
+  );
+
+  for (const name of dirNames) {
+    // 格式化名称：确保以 ⚠️ 开头
+    let formattedName = name;
+    if (!formattedName.startsWith("⚠️") && !formattedName.startsWith("⚠")) {
+      formattedName = `⚠️ ${formattedName}`;
+    }
+    // 去掉多余空格
+    formattedName = formattedName.replace(/\s+/g, " ").trim();
+
+    if (existingDirs.has(formattedName)) {
+      console.log(
+        `[Quark-AdCleanup] 警示文件夹已存在，跳过: "${formattedName}"`,
+      );
+      continue;
+    }
+
+    await createSingleDir(cookie, formattedName);
+    // 加入已存在集合，防止同名重试
+    existingDirs.add(formattedName);
+  }
+}
+
+/**
+ * 创建单个文件夹。
+ */
+async function createSingleDir(
+  cookie: string,
+  dirName: string,
+): Promise<boolean> {
+  try {
+    const resp = await fetch(
+      `https://drive-pc.quark.cn/1/clouddrive/file?${makeQuery()}`,
+      {
+        method: "POST",
+        headers: {
+          ...getHeaders(cookie),
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify({
+          pdir_fid: "0",
+          file_name: dirName,
+          dir: true,
+          dir_path: "",
+        }),
+        signal: AbortSignal.timeout(10000),
+      },
+    );
+    const data = (await resp.json()) as any;
+    if (data.status === 200 && data.data?.fid) {
+      console.log(
+        `[Quark-AdCleanup] 已创建警示文件夹: "${dirName}" (fid: ${data.data.fid})`,
+      );
+      return true;
+    }
+    console.log(
+      `[Quark-AdCleanup] 创建文件夹失败: status=${data.status} msg=${data.message}`,
+    );
+    return false;
+  } catch (err: any) {
+    console.log(
+      `[Quark-AdCleanup] 创建文件夹错误: "${dirName}" — ${err.message}`,
+    );
+    return false;
+  }
+}
+
+// ==================== 主入口 ====================
+
+/**
+ * 执行广告清理 + 创建警示文件夹。
+ * 在转存重命名后调用。
+ */
+export async function runAdCleanup(
+  cookie: string,
+  savedDirFid: string,
+): Promise<{ adDeleted: number; warningDirs: number }> {
+  const keywords = getAdKeywords();
+  const warningNames = getWarningFolderNames();
+
+  let adDeleted = 0;
+  let warningDirs = 0;
+
+  // 1. 广告关键词清理
+  if (keywords.length > 0) {
+    console.log(
+      `[Quark-AdCleanup] 开始广告关键词清理: ${keywords.length} 个关键词`,
+    );
+    adDeleted = await deleteAdFiles(cookie, savedDirFid, keywords);
+    console.log(
+      `[Quark-AdCleanup] 广告清理完成，共删除 ${adDeleted} 个文件/文件夹`,
+    );
+  } else {
+    console.log("[Quark-AdCleanup] 无广告关键词配置，跳过清理");
+  }
+
+  // 2. 创建警示文件夹
+  if (warningNames.length > 0) {
+    console.log(
+      `[Quark-AdCleanup] 开始创建警示文件夹: ${warningNames.length} 个`,
+    );
+    await createWarningDirectories(cookie, warningNames);
+    warningDirs = warningNames.length;
+    console.log(
+      `[Quark-AdCleanup] 警示文件夹创建完成（共 ${warningDirs} 个）`,
+    );
+  } else {
+    console.log("[Quark-AdCleanup] 无警示文件夹配置，跳过创建");
+  }
+
+  return { adDeleted, warningDirs };
+}