elect 1.1.0 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4f7352a40c19964176a0d00add7f035879262fd8188804c1da6e1bd43b85b9d6
-  data.tar.gz: 46a8cce817616178a6a4c016975e6f0a494e38136c84100bc7fb0c73b2eb638a
+  metadata.gz: 8f2c7494c8ded06589b2e6776d3adb086665072796c0bc29cc531416be1c74e2
+  data.tar.gz: 373fea6a80865af29e1790e1161980ffbda34eb6ba4c170e38c81e91aa9fc297
 SHA512:
-  metadata.gz: 20c1005f70e2e588f547cf32adee6fbe67ddb07c7b9e802ef32051cadda62f1c3f8825b3f40d6e4df10f89209f188bd05c6dbc781fcd6cafa7653881d40c2a50
-  data.tar.gz: 6a677ae493ac9819e0797df3bde53761d14818ab61f0fc4c182a4ed730c4d1d217965d5ad233245af2ebe99fa15dba125c8924be38fa59342d85e02881e5951d
+  metadata.gz: 6e4ba49802f8d14a3e081e430d304c67b1aafe16e1b9a5b73bb8b663b0210fa7f50a65725eb96fe927f620b3ea15324b5d26914a94ec6807ec9e77f6e80576a7
+  data.tar.gz: a57120d7a330804e1bdd62e8b26e7da115416437a9027c538612f3d4ad23261fb07b299f84e490df84ebc30d91463545bac67453192ac5e172c16df985c6144d

data/lib/elect/cleaner.rb

@@ -4,46 +4,52 @@
 require_relative "pattern"
 
 module Elect
-  # 清理器：接收文本，移除 AIGC 水印块。
+  # 清理器：接收文本，移除 AIGC 水印块与 `> AI生成` 提示行。
   #
   module Cleaner
-    # 从文本中移除所有水印块。
+    # 从文本中移除所有水印块与 AI 生成提示行。
     #
     # 策略:
-    #   1. 用 Pattern::REGEX 全局替换为空串
-    #   2. 清理因移除产生的连续空行（>2 连续空行压缩为 2）
-    #   3. 清理文件首尾多余空白
+    #   1. 用 Pattern::REGEX 全局替换为空串（水印块）
+    #   2. 用 Pattern::AI_GEN_LINE 全局替换为空串（> AI生成 行）
+    #   3. 清理因移除产生的连续空行（>2 连续空行压缩为 2）
+    #   4. 清理文件首尾多余空白
     #
     # 返回 [清理后文本, 移除数量]
     #
     def self.clean(text)
-      n = Pattern.count(text)
+      n = detect(text)
       return [text, 0] if n.zero?
 
+      # 探测原始换行符类型，压缩空行与补尾换行时保持一致，
+      # 避免 CRLF 文件被混入 LF（反之亦然）。
+      nl = text.include?("\r\n") ? "\r\n" : "\n"
+
       cleaned = text.gsub(Pattern::REGEX, "")
+      cleaned = cleaned.gsub(Pattern::AI_GEN_LINE, "")
 
       # 压缩连续空行（保留最多 2 换行 = 1 空行），兼容 CRLF
-      cleaned.gsub!(/(?:\r\n|\n){3,}/, "\n\n")
+      cleaned.gsub!(/(?:\r\n|\n){3,}/, nl + nl)
 
       # 移除首尾多余空白
       cleaned.strip!
 
       # 确保文件结尾有且仅有一个换行
-      cleaned << "\n" unless cleaned.empty?
+      cleaned << nl unless cleaned.empty?
 
       [cleaned, n]
     end
 
-    # 仅检测不修改，返回水印数量
+    # 仅检测不修改，返回水印块 + AI 生成行总数
     #
     def self.detect(text)
-      Pattern.count(text)
+      Pattern.count(text) + Pattern.ai_gen_count(text)
     end
 
-    # 判断文本是否含水印
+    # 判断文本是否含水印或 AI 生成提示行
     #
     def self.watermarked?(text)
-      Pattern.watermarked?(text)
+      Pattern.watermarked?(text) || Pattern.ai_gen_count(text).positive?
     end
   end
 end

data/lib/elect/cli.rb

@@ -70,7 +70,7 @@ module Elect
             elect [选项] <文件或目录...>
 
           示例:
-            elect ./src                  # 清理 src 目录下所有水印
+            elect ./src                  # 清理 src 目录下所有水印与 AI 标记
             elect -d ./docs              # 仅检测不修改（dry-run）
             elect -v main.py             # 清理单个文件，详细输出
             elect -e md,txt ./docs       # 只扫描 md 和 txt

data/lib/elect/pattern.rb

@@ -47,6 +47,16 @@ module Elect
       )
     /x.freeze
 
+    # 以 `> AI生成` 开头的行（Markdown 块引用格式的 AI 生成提示标记）。
+    #
+    # 形如:
+    #   > AI生成
+    #   > AI生成：本节内容由 AI 生成
+    #
+    # 匹配整行并连同换行符一起删除，兼容 LF / CRLF。
+    #
+    AI_GEN_LINE = /^[ \t]*>[ \t]*AI生成[^\r\n]*\r?\n/.freeze
+
     # 匹配水印块，返回 MatchData 或 nil
     #
     def self.match(text)
@@ -68,6 +78,13 @@ module Elect
       s.scan(REGEX).size
     end
 
+    # 统计以 `> AI生成` 开头的行数
+    #
+    def self.ai_gen_count(text)
+      s = sanitize(text)
+      s.scan(AI_GEN_LINE).size
+    end
+
     # 提取所有水印块的文本（供调试用）
     #
     def self.extract(text)

data/lib/elect/processor.rb

@@ -68,15 +68,17 @@ module Elect
       if dry_run
         result.removed = n
         result.status = :found
-        puts "  [DRY] #{p} — 发现 #{n} 处水印"
+        puts "  [DRY] #{p} — 发现 #{n} 处水印/AI标记"
         return result
       end
 
       cleaned, removed = Cleaner.clean(raw)
-      p.write(cleaned, encoding: "utf-8")
+      # 用二进制模式写回，避免 Windows 文本模式把 \r\n 里的 \n
+      # 二次转成 \r\n，产生 \r\r\n 导致每行后多出一个空行。
+      File.binwrite(p.to_s, cleaned)
       result.removed = removed
       result.status = :cleaned
-      puts "  [OK]  #{p} — 移除 #{removed} 处水印" if verbose
+      puts "  [OK]  #{p} — 移除 #{removed} 处水印/AI标记" if verbose
       result
     end
 
@@ -129,9 +131,9 @@ module Elect
       puts "  扫描完毕"
       puts "=" * 50
       puts "  扫描文件 : #{total}"
-      puts "  发现水印 : #{found + cleaned} 个文件"
-      puts "  已清理   : #{cleaned} 个文件 (共移除 #{total_rm} 处水印)"
-      puts "  无水印   : #{clean} 个文件"
+      puts "  发现标记 : #{found + cleaned} 个文件"
+      puts "  已清理   : #{cleaned} 个文件 (共移除 #{total_rm} 处水印/AI标记)"
+      puts "  无标记   : #{clean} 个文件"
       puts "  跳过     : #{skipped}"
       puts "  错误     : #{errors}" if errors.positive?
       puts "=" * 50

data/lib/elect/version.rb

@@ -2,5 +2,5 @@
 # frozen_string_literal: true
 
 module Elect
-  VERSION = "1.1.0"
+  VERSION = "1.3.0"
 end

data/lib/elect.rb

@@ -9,7 +9,8 @@ require_relative "elect/cli"
 
 # elect — AIGC 水印干掉器
 #
-# 从代码/文档文件中移除天翼星辰超级智能体等工具注入的 AIGC 水印块。
+# 从代码/文档文件中移除智能体工具注入的 AIGC 水印块
+# 以及以 `> AI生成` 开头的 AI 提示标记行。
 #
 # 快速使用:
 #

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: elect
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.3.0
 platform: ruby
 authors:
 - arc-v