Limit to mention and lightbox anchor tags

Author: nattsw

app/services/discourse_translator/base.rb

@@ -75,7 +75,8 @@ def self.language_supported?(detected_lang)
     def self.strip_tags_for_detection(detection_text)
       html_doc = Nokogiri::HTML::DocumentFragment.parse(detection_text)
       html_doc.css("img").remove
-      html_doc.css("a").remove
+      html_doc.css("a.mention").remove
+      html_doc.css("a.lightbox").remove
       html_doc.to_html
     end
 

spec/services/base_spec.rb

@@ -41,11 +41,27 @@ class EmptyTranslator < DiscourseTranslator::Base
       expect(DiscourseTranslator::Base.text_for_detection(post)).to eq("")
     end
 
-    it "strips anchor tags" do
-      post.cooked = "<a href='http://cat.com/image.png' />"
+    it "strips @ mention anchor tags" do
+      post.cooked = "<a class='mention' href='/u/cat' >cat</a>"
       expect(DiscourseTranslator::Base.text_for_detection(post)).to eq("")
     end
 
+    it "strips lightbox anchor tags" do
+      post.cooked = "<a class='lightbox' href='http://cloudfront.net/image.png' />"
+      expect(DiscourseTranslator::Base.text_for_detection(post)).to eq("")
+    end
+
+    it "leaves other anchor tags alone" do
+      cooked = <<~HTML
+        <p>
+          <a href="http://cat.com/image.png"></a>
+          <a class="derp" href="http://cat.com/image.png"></a>
+        </p>
+        HTML
+      post.cooked = cooked
+      expect(DiscourseTranslator::Base.text_for_detection(post)).to eq(cooked)
+    end
+
     it "truncates to DETECTION_CHAR_LIMIT of 1000" do
       post.cooked = "a" * 1001
       expect(DiscourseTranslator::Base.text_for_detection(post).length).to eq(1000)