From c2e47f58713373c3d10423bb5d07c51309f40f70 Mon Sep 17 00:00:00 2001
From: multiple creatures <dev@multiple-creature.party>
Date: Mon, 22 Jul 2019 21:37:11 -0500
Subject: [PATCH] autoreject check before fetching link preview cards & feeds

---
 app/services/fetch_atom_service.rb      | 6 ++++++
 app/services/fetch_link_card_service.rb | 6 ++++++
 app/services/fetch_oembed_service.rb    | 7 +++++++
 3 files changed, 19 insertions(+)

diff --git a/app/services/fetch_atom_service.rb b/app/services/fetch_atom_service.rb
index ce148bfb4..b6a33706f 100644
--- a/app/services/fetch_atom_service.rb
+++ b/app/services/fetch_atom_service.rb
@@ -2,9 +2,11 @@
 
 class FetchAtomService < BaseService
   include JsonLdHelper
+  include AutorejectHelper
 
   def call(url)
     return if url.blank?
+    return if autoreject?(url)
 
     result = process(url)
   rescue OpenSSL::SSL::SSLError => e
@@ -77,4 +79,8 @@ class FetchAtomService < BaseService
   def parse_link_header(response)
     LinkHeader.parse(response['Link'].is_a?(Array) ? response['Link'].first : response['Link'])
   end
+
+  def object_uri
+    nil
+  end
 end
diff --git a/app/services/fetch_link_card_service.rb b/app/services/fetch_link_card_service.rb
index 561daf6e1..b23090cd7 100644
--- a/app/services/fetch_link_card_service.rb
+++ b/app/services/fetch_link_card_service.rb
@@ -2,6 +2,7 @@
 
 class FetchLinkCardService < BaseService
   include UrlHelper
+  include AutorejectHelper
 
   URL_PATTERN = %r{
     (                                                                                                 #   $1 URL
@@ -21,6 +22,7 @@ class FetchLinkCardService < BaseService
 
     @url = sanitize_query_string(@url.to_s)
     return if @url.nil?
+    return if autoreject?(url)
 
     RedisLock.acquire(lock_options) do |lock|
       if lock.acquired?
@@ -174,4 +176,8 @@ class FetchLinkCardService < BaseService
   def lock_options
     { redis: Redis.current, key: "fetch:#{@url}" }
   end
+
+  def object_uri
+    nil
+  end
 end
diff --git a/app/services/fetch_oembed_service.rb b/app/services/fetch_oembed_service.rb
index 10176cfb9..528881911 100644
--- a/app/services/fetch_oembed_service.rb
+++ b/app/services/fetch_oembed_service.rb
@@ -1,12 +1,15 @@
 # frozen_string_literal: true
 
 class FetchOEmbedService
+  include AutorejectHelper
+
   attr_reader :url, :options, :format, :endpoint_url
 
   def call(url, options = {})
     @url     = url
     @options = options
 
+    return if autoreject?(url)
     discover_endpoint!
     fetch!
   end
@@ -68,4 +71,8 @@ class FetchOEmbedService
       res.code != 200 || res.mime_type != 'text/html' ? nil : res.body_with_limit
     end
   end
+
+  def object_uri
+    nil
+  end
 end