Use Html2Text to generate plaintext for keyword mutes. #236.
This allows us to match URLs inside link hrefs.
This commit is contained in:
		@@ -1,16 +1,12 @@
 | 
			
		||||
require 'htmlentities'
 | 
			
		||||
require 'html2text'
 | 
			
		||||
 | 
			
		||||
class Glitch::KeywordMuteHelper
 | 
			
		||||
  include ActionView::Helpers::SanitizeHelper
 | 
			
		||||
 | 
			
		||||
  attr_reader :text_matcher
 | 
			
		||||
  attr_reader :tag_matcher
 | 
			
		||||
  attr_reader :entity_decoder
 | 
			
		||||
 | 
			
		||||
  def initialize(receiver_id)
 | 
			
		||||
    @text_matcher   = Glitch::KeywordMute.text_matcher_for(receiver_id)
 | 
			
		||||
    @tag_matcher    = Glitch::KeywordMute.tag_matcher_for(receiver_id)
 | 
			
		||||
    @entity_decoder = HTMLEntities.new
 | 
			
		||||
  end
 | 
			
		||||
 | 
			
		||||
  def matches?(status)
 | 
			
		||||
@@ -26,6 +22,6 @@ class Glitch::KeywordMuteHelper
 | 
			
		||||
  end
 | 
			
		||||
 | 
			
		||||
  def prepare_text(text)
 | 
			
		||||
    entity_decoder.decode(strip_tags(text)).tap { |x| puts x }
 | 
			
		||||
    Html2Text.convert(text)
 | 
			
		||||
  end
 | 
			
		||||
end
 | 
			
		||||
 
 | 
			
		||||
@@ -39,5 +39,12 @@ RSpec.describe Glitch::KeywordMuteHelper do
 | 
			
		||||
 | 
			
		||||
      expect(helper.matches?(status)).to be true
 | 
			
		||||
    end
 | 
			
		||||
 | 
			
		||||
    it 'matches link hrefs in HTML text' do
 | 
			
		||||
      status = Fabricate(:status, text: '<p><a href="https://example.com/it-was-milk">yep</a></p>')
 | 
			
		||||
      Glitch::KeywordMute.create!(account: alice, keyword: 'milk')
 | 
			
		||||
 | 
			
		||||
      expect(helper.matches?(status)).to be true
 | 
			
		||||
    end
 | 
			
		||||
  end
 | 
			
		||||
end
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user