mirror of
https://github.com/discourse/discourse.git
synced 2024-12-19 00:43:44 +08:00
56 lines
1.4 KiB
Ruby
56 lines
1.4 KiB
Ruby
|
class Auth::OAuth2Authenticator < Auth::Authenticator
|
||
|
|
||
|
def name
|
||
|
@name
|
||
|
end
|
||
|
|
||
|
# only option at the moment is :trusted
|
||
|
def initialize(name, opts={})
|
||
|
@name = name
|
||
|
@opts = opts
|
||
|
end
|
||
|
|
||
|
def after_authenticate(auth_token)
|
||
|
|
||
|
result = Auth::Result.new
|
||
|
|
||
|
oauth2_provider = auth_token[:provider]
|
||
|
oauth2_uid = auth_token[:uid]
|
||
|
data = auth_token[:info]
|
||
|
result.email = email = data[:email]
|
||
|
result.name = name = data[:name]
|
||
|
|
||
|
oauth2_user_info = Oauth2UserInfo.where(uid: oauth2_uid, provider: oauth2_provider).first
|
||
|
|
||
|
if !oauth2_user_info && @opts[:trusted] && user = User.find_by_email(email)
|
||
|
oauth2_user_info = Oauth2UserInfo.create(uid: oauth2_uid,
|
||
|
provider: oauth2_provider,
|
||
|
name: name,
|
||
|
email: email,
|
||
|
user: user)
|
||
|
end
|
||
|
|
||
|
result.user = oauth2_user_info.try(:user)
|
||
|
result.email_valid = @opts[:trusted]
|
||
|
|
||
|
result.extra_data = {
|
||
|
uid: oauth2_uid,
|
||
|
provider: oauth2_provider
|
||
|
}
|
||
|
|
||
|
result
|
||
|
end
|
||
|
|
||
|
def after_create_account(user, auth)
|
||
|
data = auth[:extra_data]
|
||
|
Oauth2UserInfo.create(
|
||
|
uid: data[:uid],
|
||
|
provider: data[:provider],
|
||
|
name: auth[:name],
|
||
|
email: auth[:email],
|
||
|
user_id: user.id
|
||
|
)
|
||
|
end
|
||
|
|
||
|
end
|