mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 05:50:14 +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
|