Commit c409a75

mo khan <mo@mokhan.ca>
2015-06-16 02:38:31
extract referrer domain configuration.
1 parent 7d64670
Changed files (3)
app/models/google_drive.rb
@@ -1,8 +1,9 @@
 class GoogleDrive
   attr_reader :user
 
-  def initialize(user)
+  def initialize(user, referrer_domain: ENV['GOOGLE_REFERRER_DOMAIN'])
     @user = user
+    @referrer_domain = referrer_domain
   end
 
   def download(params)
@@ -24,8 +25,8 @@ class GoogleDrive
 curl '#{download_url}' \
 -o '#{download_path}' \
 -H 'Authorization: Bearer #{access_token}' \
--H 'Referer: http://stronglifters.dev/dashboard' \
--H 'Origin: http://stronglifters.dev' \
+-H 'Referer: #{@referrer_domain}/dashboard' \
+-H 'Origin: #{@referrer_domain}' \
 -H 'User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/43.0.2357.81 Safari/537.36' \
 --compressed
     COMMAND
spec/models/google_drive_spec.rb
@@ -3,6 +3,7 @@ require "rails_helper"
 describe GoogleDrive do
   subject { GoogleDrive.new(user) }
   let(:user) { build(:user) }
+  let(:referrer_domain) { 'https://www.stronglifters.com' }
 
   describe "#download" do
     let(:params) do
.env.example
@@ -4,3 +4,4 @@ SMTP_PASSWORD=''
 SMTP_PORT=''
 SMTP_USERNAME=''
 DEFAULT_FROM_ADDRESS=''
+GOOGLE_REFERRER_DOMAIN=http://stronglifters.dev