From c75a512a1a769d63b4a05a80efe1ecf7a728b17b Mon Sep 17 00:00:00 2001
From: Mark Dodwell <markdodwell@gmail.com>
Date: Sat, 24 Oct 2015 18:00:41 -0700
Subject: [PATCH] implement omniauth skip_info option

---
 lib/omniauth/strategies/slack.rb | 48 +++++++++++++++++++++-----------
 test/test.rb                     | 15 ++++++++++
 2 files changed, 46 insertions(+), 17 deletions(-)

diff --git a/lib/omniauth/strategies/slack.rb b/lib/omniauth/strategies/slack.rb
index 5482e75..56cd0d8 100644
--- a/lib/omniauth/strategies/slack.rb
+++ b/lib/omniauth/strategies/slack.rb
@@ -20,34 +20,48 @@ class Slack < OmniAuth::Strategies::OAuth2
       uid { raw_info['user_id'] }
 
       info do
-        {
-          name: user_info['user'].to_h['profile'].to_h['real_name_normalized'],
-          email: user_info['user'].to_h['profile'].to_h['email'],
+        hash = {
           nickname: raw_info['user'],
-          first_name: user_info['user'].to_h['profile'].to_h['first_name'],
-          last_name: user_info['user'].to_h['profile'].to_h['last_name'],
-          description: user_info['user'].to_h['profile'].to_h['title'],
-          image_24: user_info['user'].to_h['profile'].to_h['image_24'],
-          image_48: user_info['user'].to_h['profile'].to_h['image_48'],
-          image: user_info['user'].to_h['profile'].to_h['image_192'],
           team: raw_info['team'],
           user: raw_info['user'],
           team_id: raw_info['team_id'],
-          team_domain: team_info['team'].to_h['domain'],
-          user_id: raw_info['user_id'],
-          is_admin: user_info['user'].to_h['is_admin'],
-          is_owner: user_info['user'].to_h['is_owner'],
-          time_zone: user_info['user'].to_h['tz']
+          user_id: raw_info['user_id']
         }
+
+        unless skip_info?
+          hash.merge!(
+            name: user_info['user'].to_h['profile'].to_h['real_name_normalized'],
+            email: user_info['user'].to_h['profile'].to_h['email'],
+            first_name: user_info['user'].to_h['profile'].to_h['first_name'],
+            last_name: user_info['user'].to_h['profile'].to_h['last_name'],
+            description: user_info['user'].to_h['profile'].to_h['title'],
+            image_24: user_info['user'].to_h['profile'].to_h['image_24'],
+            image_48: user_info['user'].to_h['profile'].to_h['image_48'],
+            image: user_info['user'].to_h['profile'].to_h['image_192'],
+            team_domain: team_info['team'].to_h['domain'],
+            is_admin: user_info['user'].to_h['is_admin'],
+            is_owner: user_info['user'].to_h['is_owner'],
+            time_zone: user_info['user'].to_h['tz']
+          )
+        end
+
+        hash
       end
 
       extra do
-        {
+        hash = {
           raw_info: raw_info,
-          user_info: user_info,
-          team_info: team_info,
           web_hook_info: web_hook_info
         }
+
+        unless skip_info?
+          hash.merge!(
+            user_info: user_info,
+            team_info: team_info
+          )
+        end
+
+        hash
       end
 
       def raw_info
diff --git a/test/test.rb b/test/test.rb
index f24f1f9..cc545c6 100644
--- a/test/test.rb
+++ b/test/test.rb
@@ -98,3 +98,18 @@ def setup
     refute_has_key "refresh_token", strategy.credentials
   end
 end
+
+class UserInfoTest < StrategyTestCase
+  test 'info should not include extended info when skip_info is specified' do
+    @options = { skip_info: true }
+    strategy.stubs(:raw_info).returns({})
+    assert_equal %w[nickname team user team_id user_id], strategy.info.keys.map(&:to_s)
+  end
+
+  test 'extra should not include extended info when skip_info is specified' do
+    @options = { skip_info: true }
+    strategy.stubs(:raw_info).returns({})
+    strategy.stubs(:webhook_info).returns({})
+    assert_equal %w[raw_info web_hook_info], strategy.extra.keys.map(&:to_s)
+  end
+end