mirror of https://github.com/go-gitea/gitea.git
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
![]() |
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a MIT-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package externalaccount
|
||
|
|
||
|
import (
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
|
||
|
"code.gitea.io/gitea/models"
|
||
|
"code.gitea.io/gitea/modules/structs"
|
||
|
|
||
|
"github.com/markbates/goth"
|
||
|
)
|
||
|
|
||
|
// LinkAccountToUser link the gothUser to the user
|
||
|
func LinkAccountToUser(user *models.User, gothUser goth.User) error {
|
||
|
loginSource, err := models.GetActiveOAuth2LoginSourceByName(gothUser.Provider)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
externalLoginUser := &models.ExternalLoginUser{
|
||
|
ExternalID: gothUser.UserID,
|
||
|
UserID: user.ID,
|
||
|
LoginSourceID: loginSource.ID,
|
||
|
RawData: gothUser.RawData,
|
||
|
Provider: gothUser.Provider,
|
||
|
Email: gothUser.Email,
|
||
|
Name: gothUser.Name,
|
||
|
FirstName: gothUser.FirstName,
|
||
|
LastName: gothUser.LastName,
|
||
|
NickName: gothUser.NickName,
|
||
|
Description: gothUser.Description,
|
||
|
AvatarURL: gothUser.AvatarURL,
|
||
|
Location: gothUser.Location,
|
||
|
AccessToken: gothUser.AccessToken,
|
||
|
AccessTokenSecret: gothUser.AccessTokenSecret,
|
||
|
RefreshToken: gothUser.RefreshToken,
|
||
|
ExpiresAt: gothUser.ExpiresAt,
|
||
|
}
|
||
|
|
||
|
if err := models.LinkExternalToUser(user, externalLoginUser); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
externalID, err := strconv.ParseInt(externalLoginUser.ExternalID, 10, 64)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
var tp structs.GitServiceType
|
||
|
for _, s := range structs.SupportedFullGitService {
|
||
|
if strings.EqualFold(s.Name(), gothUser.Provider) {
|
||
|
tp = s
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if tp.Name() != "" {
|
||
|
return models.UpdateMigrationsByType(tp, externalID, user.ID)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|