2019-10-14 09:10:42 +03:00
// Copyright 2019 The Gitea Authors. All rights reserved.
2022-11-27 21:20:29 +03:00
// SPDX-License-Identifier: MIT
2019-10-14 09:10:42 +03:00
package externalaccount
import (
"strings"
"code.gitea.io/gitea/models"
2022-01-02 16:12:35 +03:00
"code.gitea.io/gitea/models/auth"
2021-11-24 12:49:20 +03:00
user_model "code.gitea.io/gitea/models/user"
2019-10-14 09:10:42 +03:00
"code.gitea.io/gitea/modules/structs"
"github.com/markbates/goth"
)
2021-12-14 11:37:11 +03:00
func toExternalLoginUser ( user * user_model . User , gothUser goth . User ) ( * user_model . ExternalLoginUser , error ) {
2022-01-02 16:12:35 +03:00
authSource , err := auth . GetActiveOAuth2SourceByName ( gothUser . Provider )
2019-10-14 09:10:42 +03:00
if err != nil {
2021-12-14 11:37:11 +03:00
return nil , err
2019-10-14 09:10:42 +03:00
}
2021-12-14 11:37:11 +03:00
return & user_model . ExternalLoginUser {
2019-10-14 09:10:42 +03:00
ExternalID : gothUser . UserID ,
UserID : user . ID ,
2022-01-02 16:12:35 +03:00
LoginSourceID : authSource . ID ,
2019-10-14 09:10:42 +03:00
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 ,
2021-12-14 11:37:11 +03:00
} , nil
}
// LinkAccountToUser link the gothUser to the user
func LinkAccountToUser ( user * user_model . User , gothUser goth . User ) error {
externalLoginUser , err := toExternalLoginUser ( user , gothUser )
if err != nil {
return err
2019-10-14 09:10:42 +03:00
}
2021-11-28 17:11:58 +03:00
if err := user_model . LinkExternalToUser ( user , externalLoginUser ) ; err != nil {
2019-10-14 09:10:42 +03:00
return err
}
2019-10-17 05:06:28 +03:00
externalID := externalLoginUser . ExternalID
2019-10-14 09:10:42 +03:00
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
}
2021-12-14 11:37:11 +03:00
// UpdateExternalUser updates external user's information
func UpdateExternalUser ( user * user_model . User , gothUser goth . User ) error {
externalLoginUser , err := toExternalLoginUser ( user , gothUser )
if err != nil {
return err
}
return user_model . UpdateExternalUserByExternalID ( externalLoginUser )
}