censor_bear 0.1.1 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0cdef69a9d04136c6116ed8245122bda98d27849a91e313487282be5d0640c4d
-  data.tar.gz: 5088b2e4f868403f242b62344af06edbe47daf4e90bbbaa264848e7c982e574d
+  metadata.gz: afa762c57b39c4646ab5d236763e02f0f7cdb8f32a0e9b38058ccf342382e526
+  data.tar.gz: e82d0275c7be7525037a91451d852d54549686f3fed0ca4d73a142a64f2e7b34
 SHA512:
-  metadata.gz: 3b9138dc7b8af468e6fb59960f1bdc6801927c986758c90d709f49ce408b4a6127bf520e5480198c667717f24a2bfc9cd657938e385a5440c3c811862f47f682
-  data.tar.gz: 80966dea001f69dd7db9a1a64d6c687a67062a39fd01b2809a3a1c5d9b028b2066526e69594730805f76c975c49255da8974e6063e2446e7136aa2cf8bd61d89
+  metadata.gz: e7e2ca96a5d99afcfdbac547d67bfa7e4dbe21e321e2a260db2006cbf1943577cb8b9306150a01d308e23c25f6d51399af7f4cbbfb592be50d3a4a0d4d8fc91f
+  data.tar.gz: 95fe89608248a273f2fa5e2560afe8ec9510555bee25afe542057c93f04799cc841b34d23be8846f8fcc1eeccfd7846a70b2887d926062f0f1a4f5fded287def

data/app/controllers/censor_bear/stop_words_controller.rb

@@ -46,7 +46,7 @@ module CensorBear
 
     def load_stop_words
       builder ||= stop_word_scope
-      builder = builder.where("word ilike ?", "%#{params[:q]}%") if params[:q].present?
+      builder = builder.where("key ilike ?", "%#{params[:q]}%") if params[:q].present?
       builder = builder.order(id: :desc)
 
       @pagy, @stop_words = pagy(builder)

data/app/helpers/censor_bear/application_helper.rb

@@ -1,5 +1,9 @@
 module CensorBear
   module ApplicationHelper
     include Pagy::Frontend
+
+    def current_user
+      main_app.scope.request.env['warden'].user
+    end
   end
 end

data/app/views/censor_bear/logs/index.html.erb

@@ -17,6 +17,8 @@
     <% @logs.each do |log| %>
       <tr>
         <td><%= log.original_content %></td>
+        <td><%= log.scenario %></td>
+        <td><%= log.stage %></td>
         <td><%= log.action %></td>
         <td><%= log.filtered_content %></td>
         <td><%= log.mod_words %></td>

data/app/views/censor_bear/logs/show.html.erb

@@ -10,6 +10,16 @@
   <%= @log.action %>
 </p>
 
+<p>
+  <strong>Scenario:</strong>
+  <%= @log.scenario %>
+</p>
+
+<p>
+  <strong>Stage:</strong>
+  <%= @log.stage %>
+</p>
+
 <p>
   <strong>Filtered content:</strong>
   <%= @log.filtered_content %>

data/app/views/censor_bear/stop_words/_form.html.haml

@@ -10,19 +10,19 @@
       %div= form.text_field :key, class: "border p-1 w-full rounded-md"
     %div
       %div{class: "text-gray-500 text-xs"}= form.label :ugc, "UGC内容处理方式"
-      %div= form.select :ugc, options_for_select(CensorBear.action_options, params[:action_type]), class: "border p-1 w-full rounded-md"
+      %div= form.select :ugc, options_for_select(CensorBear.action_options, stop_word.ugc), class: "border p-1 w-full rounded-md"
     %div
       %div{class: "text-gray-500 text-xs"}= form.label :username, "用户名处理方式"
-      %div= form.select :username, options_for_select(CensorBear.action_options, params[:action_type]), class: "border p-1 w-full rounded-md"
+      %div= form.select :username, options_for_select(CensorBear.action_options, stop_word.username), class: "border p-1 w-full rounded-md"
     %div
       %div{class: "text-gray-500 text-xs"}= form.label :nickname, "用户昵称处理方式"
-      %div= form.select :nickname, options_for_select(CensorBear.action_options, params[:action_type]), class: "border p-1 w-full rounded-md"
+      %div= form.select :nickname, options_for_select(CensorBear.action_options, stop_word.nickname), class: "border p-1 w-full rounded-md"
     %div
       %div{class: "text-gray-500 text-xs"}= form.label :signature, "签名处理方式"
-      %div= form.select :signature, options_for_select(CensorBear.action_options, params[:action_type]), class: "border p-1 w-full rounded-md"
+      %div= form.select :signature, options_for_select(CensorBear.action_options, stop_word.signature), class: "border p-1 w-full rounded-md"
     %div
       %div{class: "text-gray-500 text-xs"}= form.label :dialog, "私信处理方式"
-      %div= form.select :dialog, options_for_select(CensorBear.action_options, params[:action_type]), class: "border p-1 w-full rounded-md"
+      %div= form.select :dialog, options_for_select(CensorBear.action_options, stop_word.dialog), class: "border p-1 w-full rounded-md"
     %div
       %div{class: "text-gray-500 text-xs"}= form.label :replacement, "过滤词替换"
       %div= form.text_field :replacement, class: "border p-1 w-full rounded-md"

data/db/migrate/20211129093508_create_censor_bear_logs.rb

@@ -2,7 +2,9 @@ class CreateCensorBearLogs < ActiveRecord::Migration[6.1]
   def change
     create_table :censor_bear_logs do |t|
       t.text :original_content
+      t.string :scenario
       t.string :action
+      t.string :stage
       t.text :filtered_content
       t.string :mod_words, array: true, default: []
 

data/lib/censor_bear/censor.rb

@@ -25,34 +25,39 @@ module CensorBear
       return Result.new(@content) unless CensorBear::StopWord::FIELDS.include?(@type)
 
       if @content.match(QQ_REG)
-        CensorBear::Log.create(
-          original_content: @content,
-          action: "#{@type}_regex_qq",
-          filtered_content: nil,
-          mod_words: @mod_words
-        )
+        CensorBear.info(@content, @type, 'BANNED', 'qq_regex')
         raise NotPassedException
       end
 
       if @content.match(WX_REG)
-        @is_mod = true
-        CensorBear::Log.create(
-          original_content: @content,
-          action: "#{@type}_regex_wx",
-          filtered_content: nil,
-          mod_words: @mod_words
-        )
+        CensorBear.info(@content, @type, 'BANNED', 'wx_regex')
         raise NotPassedException
       end
 
       local_check
 
-      # 用户相关类型直接封禁，速错
+      # 用户相关类型直接禁止创建
       raise NotPassedException if @is_mod && %w[username signature dialog nickname].include?(@type)
 
       Result.new(@content, @is_mod, @mod_words.uniq)
     end
 
+    # default type = :ugc
+    def check_search
+      result = false
+      stop_words = CensorBear::StopWord.where("#{@type} != 'IGNORE'")
+      stop_words.each do |word|
+        finder = Regexp.new(Regexp.escape(word.key))
+        action = word.send(@type.to_sym).upcase
+        if finder.match(@content)
+          result = true
+          CensorBear.info(@content, @type, action, 'check_search', mod_words: [word.key])
+          break
+        end
+      end
+      result
+    end
+
     def local_check
       original_content = @content.dup
       stop_words = CensorBear::StopWord.where("#{@type} != 'IGNORE'")
@@ -64,14 +69,26 @@ module CensorBear
         case action
         when 'REPLACE'
           @content = @content.gsub(finder, word.replacement)
-          CensorBear::Log.create(original_content: original_content, action: "#{@type}_#{action}", filtered_content: @content, mod_words: [word.key])
+          CensorBear.info(
+            original_content, @type, action, 'local_check',
+            filtered_content: @content,
+            mod_words: [word.key]
+          )
         when 'MOD'
           @is_mod = true
-          CensorBear::Log.create(original_content: original_content, action: "#{@type}_#{action}", filtered_content: nil, mod_words: [word.key])
+          CensorBear.info(
+            original_content, @type, action, 'local_check',
+            filtered_content: nil,
+            mod_words: [word.key]
+          )
         when 'BANNED'
           @is_mod = true
           # 禁止的直接抛出异常
-          CensorBear::Log.create(original_content: original_content, action: "#{@type}_#{action}", filtered_content: nil, mod_words: [word.key])
+          CensorBear.info(
+            original_content, @type, action, 'local_check',
+            filtered_content: nil,
+            mod_words: [word.key]
+          )
           raise NotPassedException
         end
       end

data/lib/censor_bear/version.rb

@@ -1,3 +1,3 @@
 module CensorBear
-  VERSION = '0.1.1'
+  VERSION = '0.1.5'.freeze
 end

data/lib/censor_bear.rb

@@ -1,9 +1,9 @@
-require "censor_bear/version"
-require "censor_bear/engine"
-require "censor_bear/censor"
-require "censor_bear/configuration"
+require 'censor_bear/version'
+require 'censor_bear/engine'
+require 'censor_bear/censor'
+require 'censor_bear/configuration'
 
-require "hamlit"
+require 'hamlit'
 require 'pagy'
 
 module CensorBear
@@ -31,6 +31,23 @@ module CensorBear
       Censor.new(content, type).check_text
     end
 
+    def check_search(content)
+      Censor.new(content, "ugc").check_search
+    end
+
+    def info(content, scenario, action, stage, options = {})
+      filtered_content = options[:filtered_content] || nil
+      mod_words = options[:mod_words] || []
+
+      CensorBear::Log.create(
+        original_content: content,
+        scenario: scenario,
+        action: action,
+        stage: stage,
+        filtered_content: filtered_content,
+        mod_words: mod_words
+      )
+    end
 
     def action_options
       I18n.t('censor_log.action').map { |k, v| [v, k.to_s] }

data/lib/tasks/censor_bear_tasks.rake

@@ -4,7 +4,7 @@ namespace :censor_bear do
     from_file = File.read(CensorBear.config.format_from_file_path)
     File.open(CensorBear.config.format_dest_file_path, 'w') do |f|
       from_file.each_line do |word|
-        line = "#{word.strip!}={BANNED}|{BANNED}|{BANNED}|{BANNED}\r\n"
+        line = "#{word.strip}={BANNED}|{BANNED}|{BANNED}|{BANNED}\r\n"
         f.write(line)
         f.flush
       end
@@ -35,7 +35,7 @@ namespace :censor_bear do
         dialog: dialog,
         nickname: nickname,
         replacement: replacement,
-        user_id: User.last&.id,
+        user_id: CensorBear.config.user_class.constantize.last&.id,
         created_at: Time.now,
         updated_at: Time.now
       }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: censor_bear
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.5
 platform: ruby
 authors:
 - 42up
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-12-01 00:00:00.000000000 Z
+date: 2021-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -16,20 +16,20 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 6.1.4
+        version: 6.0.4
     - - ">="
       - !ruby/object:Gem::Version
-        version: 6.1.4.1
+        version: 6.0.4.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 6.1.4
+        version: 6.0.4
     - - ">="
       - !ruby/object:Gem::Version
-        version: 6.1.4.1
+        version: 6.0.4.1
 - !ruby/object:Gem::Dependency
   name: hamlit
   requirement: !ruby/object:Gem::Requirement