|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
# Copyright 2010 - 2017 RhodeCode GmbH and the AppEnlight project authors
|
|
|
#
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
# you may not use this file except in compliance with the License.
|
|
|
# You may obtain a copy of the License at
|
|
|
#
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
#
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
# See the License for the specific language governing permissions and
|
|
|
# limitations under the License.
|
|
|
|
|
|
from ziggurat_foundations.models.services.external_identity import \
|
|
|
ExternalIdentityService
|
|
|
from appenlight.models.external_identity import ExternalIdentity
|
|
|
|
|
|
|
|
|
def handle_social_data(request, user, social_data):
|
|
|
social_data = social_data
|
|
|
update_identity = False
|
|
|
|
|
|
extng_id = ExternalIdentityService.by_external_id_and_provider(
|
|
|
social_data['user']['id'],
|
|
|
social_data['credentials'].provider_name
|
|
|
)
|
|
|
|
|
|
# fix legacy accounts with wrong google ID
|
|
|
if not extng_id and social_data['credentials'].provider_name == 'google':
|
|
|
extng_id = ExternalIdentityService.by_external_id_and_provider(
|
|
|
social_data['user']['email'],
|
|
|
social_data['credentials'].provider_name
|
|
|
)
|
|
|
|
|
|
if extng_id:
|
|
|
extng_id.delete()
|
|
|
update_identity = True
|
|
|
|
|
|
if not social_data['user']['id']:
|
|
|
request.session.flash(
|
|
|
'No external user id found? Perhaps permissions for '
|
|
|
'authentication are set incorrectly', 'error')
|
|
|
return False
|
|
|
|
|
|
if not extng_id or update_identity:
|
|
|
if not update_identity:
|
|
|
request.session.flash('Your external identity is now '
|
|
|
'connected with your account')
|
|
|
ex_identity = ExternalIdentity()
|
|
|
ex_identity.external_id = social_data['user']['id']
|
|
|
ex_identity.external_user_name = social_data['user']['user_name']
|
|
|
ex_identity.provider_name = social_data['credentials'].provider_name
|
|
|
ex_identity.access_token = social_data['credentials'].token
|
|
|
ex_identity.token_secret = social_data['credentials'].token_secret
|
|
|
ex_identity.alt_token = social_data['credentials'].refresh_token
|
|
|
user.external_identities.append(ex_identity)
|
|
|
request.session.pop('zigg.social_auth', None)
|
|
|
|
