From 866347a9f7498d82ba253ee1a8a6c44d253ad461 Mon Sep 17 00:00:00 2001
From: r888888888 <r888888888@gmail.com>
Date: Thu, 11 Jun 2015 21:38:41 -0700
Subject: [PATCH] fix errors

---
 app/logical/sources/site.rb            | 6 +++++-
 app/logical/sources/strategies/base.rb | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/app/logical/sources/site.rb b/app/logical/sources/site.rb
index df304e8c0..04162c5c5 100644
--- a/app/logical/sources/site.rb
+++ b/app/logical/sources/site.rb
@@ -7,7 +7,7 @@ module Sources
       :profile_url, :image_url, :tags, :artist_record, :unique_id, 
       :page_count, :file_url, :ugoira_frame_data, :image_urls, 
       :has_artist_commentary?, :artist_commentary_title,
-      :artist_commentary_desc, :referer_url, :to => :strategy
+      :artist_commentary_desc, :to => :strategy
 
     def self.strategies
       [Strategies::Pixiv, Strategies::NicoSeiga, Strategies::DeviantArt, Strategies::Nijie, Strategies::Twitter]
@@ -24,6 +24,10 @@ module Sources
       end
     end
 
+    def referer_url
+      strategy.try(:referer_url)
+    end
+
     def normalized_for_artist_finder?
       available? && strategy.normalized_for_artist_finder?
     end
diff --git a/app/logical/sources/strategies/base.rb b/app/logical/sources/strategies/base.rb
index b9f0f9a5d..4d1dcda3a 100644
--- a/app/logical/sources/strategies/base.rb
+++ b/app/logical/sources/strategies/base.rb
@@ -9,7 +9,7 @@ module Sources
         false
       end
 
-      def initialize(url, referer_url)
+      def initialize(url, referer_url = nil)
         @url = url
         @referer_url = referer_url
         @page_count = 1