@q78kg/koishi-plugin-text-censor 1.0.4-beta.1 → 1.0.4-beta.11

package/lib/index.d.ts

@@ -1,9 +1,9 @@
 import { Context, Schema } from 'koishi';
 export declare const name = "text-censor";
 export interface Config {
-    filename: string;
+    textDatabase: [string][];
     removeWords: boolean;
-    transformToUpper: boolean;
+    caseStrategy: 'capital' | 'none' | 'lower';
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -38,27 +38,50 @@ 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),
+    caseStrategy: import_koishi.Schema.union(["none", "lower", "capital"]).description("敏感词处理时的大小写策略。").default("none")
+    // 默认不处理大小写
+  })
+  // 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"
-    // 这里我们将值设置为可能的类型
+    transform: config.caseStrategy
   };
   const filter = new import_mint_filter.default(words, mintOptions);
   ctx.plugin(import_censor.default);
@@ -69,15 +92,7 @@ function apply(ctx, config) {
       if (typeof result.text !== "string")
         return [];
       if (config.removeWords) {
-        const filteredText = result.text;
-        const removedCharacters = Array.from(originalText).filter(
-          (char, index) => filteredText[index] !== char
-          // 如果对应位置的字符不相同，则为被过滤的字符
-        ).join("");
-        const cleanedText = originalText.split("").filter(
-          (char) => !removedCharacters.includes(char)
-          // 过滤掉被移除的字符
-        ).join("");
+        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.4-beta.1",
+  "version": "1.0.4-beta.11",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -33,8 +33,7 @@
       ]
     },
     "description": {
-      "en": "Filter out sensitive words in text messages",
-      "zh": "文本敏感词过滤"
+      "zh": "文本敏感词过滤服务，支持多敏感词库"
     }
   },
   "peerDependencies": {