Fixes issue when auth_token_params is nil and adds possibility to setup :plan_id parameter

.gitignore

@@ -1,5 +1,6 @@
 *.gem
 .bundle
+.idea
 Gemfile.lock
 pkg/*
 coverage

lib/omniauth-square/version.rb

@@ -1,5 +1,5 @@
 module Omniauth
   module Square
-    VERSION = "1.0.2"
+    VERSION = "1.0.3"
   end
 end

lib/omniauth/strategies/square.rb

@@ -30,13 +30,24 @@ def raw_info
         @raw_info ||= access_token.get('/v1/me').parsed
       end
 
+      alias :old_request_phase :request_phase
+
+      def request_phase
+        if request.params['plan_id']
+          options[:authorize_params][:plan_id] = request.params['plan_id']
+        end
+
+        old_request_phase
+      end
+
       protected
 
       def build_access_token
         parsed_response = fetch_access_token
 
         parsed_response['expires_at'] = Time.parse(parsed_response['expires_at']).to_i
-        parsed_response.merge!(deep_symbolize(options.auth_token_params))
+        auth_token_params = options.auth_token_params || {}
+        parsed_response.merge!(deep_symbolize(auth_token_params))
 
         connect_client = client.dup
         connect_client.site = options.client_options.connect_site

omniauth-square.gemspec

@@ -24,4 +24,5 @@ Gem::Specification.new do |s|
   s.add_development_dependency 'rack-test'
   s.add_development_dependency 'simplecov'
   s.add_development_dependency 'webmock'
+  s.add_development_dependency 'pry'
 end

spec/omniauth/strategies/square_spec.rb

@@ -2,7 +2,7 @@
 
 describe OmniAuth::Strategies::Square do
   before :each do
-    @request = double('Request', :scheme => '', :url => '')
+    @request = double('Request', :scheme => '', :url => '', :params => {}, :cookies => {}, :env => {})
     @request.stub(:params) { {} }
   end
 
@@ -150,6 +150,18 @@
     end
   end
 
+  describe '#request_phase' do
+    before do
+      @request.stub(:params).and_return('plan_id' => 'test_plan_id')
+      subject.stub(:old_request_phase).and_return(:something)
+    end
+
+    it 'adds `plan_id` parameter to `authorize_params`' do
+      expect { subject.request_phase }.to change { subject.options.authorize_params.plan_id }.
+        from(nil).to('test_plan_id')
+    end
+  end
+
   describe '#access_token_request_payload' do
     before do
       @request.stub(:params).and_return('code' => '11111')

spec/spec_helper.rb

@@ -5,6 +5,7 @@
 require 'webmock/rspec'
 require 'omniauth'
 require 'omniauth-square'
+require 'pry'
 
 RSpec.configure do |config|
   config.include WebMock::API