omniauth_callbacks_controller.rb
2.08 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
class Users::OmniauthCallbacksController < Devise::OmniauthCallbacksController
def github
github_login = env["omniauth.auth"].extra.raw_info.login
github_token = env["omniauth.auth"].credentials.token
github_user = User.where(:github_login => github_login).first
if github_user.nil? && github_org_id = Errbit::Config.github_org_id
# See if they are a member of the organization that we have access for
# If they are, automatically create an account
client = Octokit::Client.new(access_token: github_token)
org_ids = client.organizations.map { |org| org.id.to_s }
if org_ids.include?(github_org_id)
github_user = User.create(name: env["omniauth.auth"].extra.raw_info.name, email: env["omniauth.auth"].extra.raw_info.email)
end
end
# If user is already signed in, link github details to their account
if current_user
# ... unless a user is already registered with same github login
if github_user && github_user != current_user
flash[:error] = "User already registered with GitHub login '#{github_login}'!"
else
# Add github details to current user
update_user_with_github_attributes(current_user, github_login, github_token)
flash[:success] = "Successfully linked GitHub account!"
end
# User must have clicked 'link account' from their user page, so redirect there.
redirect_to user_path(current_user)
elsif github_user
# Store OAuth token
update_user_with_github_attributes(github_user, github_login, github_token)
flash[:success] = I18n.t "devise.omniauth_callbacks.success", :kind => "GitHub"
sign_in_and_redirect github_user, :event => :authentication
else
flash[:error] = "There are no authorized users with GitHub login '#{github_login}'. Please ask an administrator to register your user account."
redirect_to new_user_session_path
end
end
private
def update_user_with_github_attributes(user, login, token)
user.update_attributes(
:github_login => login,
:github_oauth_token => token
)
end
end