From 04d6640792577bfcb4e654a8cf71ac2bb6f6c6de Mon Sep 17 00:00:00 2001
From: Toks <tttoksss@gmail.com>
Date: Wed, 3 Jun 2015 20:12:12 -0400
Subject: [PATCH] Fix upload page fetching source for wrong page from twitter

It would always fetch from the referer, regardless of what page that
was. It didn't even have to be a page on twitter.
---
 app/logical/sources/strategies/twitter.rb | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/app/logical/sources/strategies/twitter.rb b/app/logical/sources/strategies/twitter.rb
index 613829770..b9fd1dda3 100644
--- a/app/logical/sources/strategies/twitter.rb
+++ b/app/logical/sources/strategies/twitter.rb
@@ -1,9 +1,17 @@
 module Sources::Strategies
   class Twitter < Base
     def self.url_match?(url)
-      url =~ %r!https?://(?:mobile\.)?twitter\.com/\w+/status/\d+!
+      url =~ %r!https?://(?:mobile\.)?twitter\.com/\w+/status/\d+! || url =~ %r{https?://pbs\.twimg\.com/media/}
     end
 
+      def referer_url(template)
+        if template.params[:ref] =~ %r!https?://(?:mobile\.)?twitter\.com/\w+/status/\d+! && template.params[:url] =~ %r{https?://pbs\.twimg\.com/media/}
+          template.params[:ref]
+        else
+          template.params[:url]
+        end
+      end
+
     def tags
       []
     end