@q78kg/koishi-plugin-text-censor 1.0.2 → 1.0.4-beta.10

package/lib/index.d.ts

@@ -1,7 +1,7 @@
 import { Context, Schema } from 'koishi';
 export declare const name = "text-censor";
 export interface Config {
-    filename: string;
+    textDatabase: [string][];
     removeWords: boolean;
     transformToUpper: boolean;
 }

package/lib/index.js

@@ -38,41 +38,61 @@ module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 var import_node_fs = require("node:fs");
 var import_node_path = require("node:path");
-var import_censor = __toESM(require("@koishijs/censor"));
 var import_mint_filter = __toESM(require("mint-filter"));
+var import_censor = __toESM(require("@koishijs/censor"));
 var name = "text-censor";
-var Config = import_koishi.Schema.object({
-  filename: import_koishi.Schema.string().description("存储敏感词的文件路径。").default("data/censor.txt"),
-  removeWords: import_koishi.Schema.boolean().description("是否直接删除敏感词。").default(false),
-  // 默认不删除敏感词
-  transformToUpper: import_koishi.Schema.boolean().description("是否将字符转换为大写。").default(false)
-  // 默认不转换字符为大写
-});
+var Config = import_koishi.Schema.intersect([
+  import_koishi.Schema.object({
+    textDatabase: import_koishi.Schema.array(
+      import_koishi.Schema.tuple([
+        import_koishi.Schema.string().default("data/text-censor/censor.txt")
+      ])
+    ).description("敏感词库的文件路径。").default([["data/text-censor/censor.txt"]])
+  }),
+  import_koishi.Schema.object({
+    removeWords: import_koishi.Schema.boolean().description("是否直接删除敏感词。").default(false),
+    transformToUpper: import_koishi.Schema.boolean().description("是否将字符转换为大写。").default(false)
+  })
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+]);
 function apply(ctx, config) {
-  const filename = (0, import_node_path.resolve)(ctx.baseDir, config.filename);
-  if (!(0, import_node_fs.existsSync)(filename)) {
-    ctx.logger.warn("dictionary file not found");
+  let words = [];
+  for (const [file] of config.textDatabase) {
+    const filePath = (0, import_node_path.resolve)(ctx.baseDir, file);
+    if (!(0, import_node_fs.existsSync)(filePath)) {
+      ctx.logger.warn(
+        `dictionary file not found: ${filePath}, creating a new one.`
+      );
+      const dirPath = (0, import_node_path.dirname)(filePath);
+      if (!(0, import_node_fs.existsSync)(dirPath)) {
+        (0, import_node_fs.mkdirSync)(dirPath, { recursive: true });
+      }
+      (0, import_node_fs.writeFileSync)(filePath, "");
+    }
+    const source = (0, import_node_fs.readFileSync)(filePath, "utf8");
+    const fileWords = source.split("\n").map((word) => word.trim()).filter(
+      (word) => word && !word.startsWith("//") && !word.startsWith("#")
+    );
+    words = words.concat(fileWords);
+  }
+  if (words.length === 0) {
+    ctx.logger.warn("no sensitive words found");
     return;
   }
-  const source = (0, import_node_fs.readFileSync)(filename, "utf8");
-  const words = source.split("\n").map((word) => word.trim()).filter((word) => word && !word.startsWith("//") && !word.startsWith("#"));
   const mintOptions = {
     transform: config.transformToUpper ? "capital" : "none"
-    // 这里我们将值设置为可能的类型
   };
   const filter = new import_mint_filter.default(words, mintOptions);
   ctx.plugin(import_censor.default);
   ctx.get("censor").intercept({
     async text(attrs) {
-      const result = await filter.filter(attrs.content);
+      const originalText = attrs.content;
+      const result = await filter.filter(originalText);
       if (typeof result.text !== "string")
         return [];
       if (config.removeWords) {
-        const filteredText = words.reduce((text, word) => {
-          const regex = new RegExp(`\\b${word}\\b`, "gi");
-          return text.replace(regex, "");
-        }, result.text);
-        return [filteredText.trim()];
+        const cleanedText = result.text.replace(/\*/g, "");
+        return [cleanedText.trim()];
       } else {
         return [result.text];
       }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@q78kg/koishi-plugin-text-censor",
   "description": "A text censor provider for Koishi",
-  "version": "1.0.2",
+  "version": "1.0.4-beta.10",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [