From 9a6a6e52eafcdf48fc4b3efd6f5858ff056a6b66 Mon Sep 17 00:00:00 2001
From: nonamethanks <hellafrickingepic@gmail.com>
Date: Fri, 10 Sep 2021 13:09:53 +0200
Subject: [PATCH] Lofter: raise timeout for file download

---
 app/logical/sources/strategies/base.rb   | 7 ++++++-
 app/logical/sources/strategies/lofter.rb | 4 ++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/app/logical/sources/strategies/base.rb b/app/logical/sources/strategies/base.rb
index 4b0e50690..89ec4124f 100644
--- a/app/logical/sources/strategies/base.rb
+++ b/app/logical/sources/strategies/base.rb
@@ -235,9 +235,14 @@ module Sources
       end
       memoize :http
 
+      # The http timeout to download a file. Overrideable by each site
+      def download_timeout
+        30
+      end
+
       # A http client for downloading files.
       def http_downloader
-        http.timeout(30).max_size(Danbooru.config.max_file_size).use(:spoof_referrer).use(:unpolish_cloudflare)
+        http.timeout(download_timeout).max_size(Danbooru.config.max_file_size).use(:spoof_referrer).use(:unpolish_cloudflare)
       end
       memoize :http_downloader
 
diff --git a/app/logical/sources/strategies/lofter.rb b/app/logical/sources/strategies/lofter.rb
index cd8839291..c91e45850 100644
--- a/app/logical/sources/strategies/lofter.rb
+++ b/app/logical/sources/strategies/lofter.rb
@@ -96,6 +96,10 @@ module Sources
       def artist_name
         urls.map { |u| u[PROFILE_URL, :artist_name] || u[PAGE_URL, :artist_name] }.compact.first
       end
+
+      def download_timeout
+        60
+      end
     end
   end
 end