61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package services
|
|
|
|
import (
|
|
"code.smartsheep.studio/hydrogen/identity/pkg/grpc/proto"
|
|
"code.smartsheep.studio/hydrogen/interactive/pkg/database"
|
|
"code.smartsheep.studio/hydrogen/interactive/pkg/grpc"
|
|
"code.smartsheep.studio/hydrogen/interactive/pkg/models"
|
|
"context"
|
|
"errors"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
func LinkAccount(userinfo *proto.Userinfo) (models.Account, error) {
|
|
var account models.Account
|
|
if err := database.C.Where(&models.Account{
|
|
ExternalID: uint(userinfo.Id),
|
|
}).First(&account).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
account = models.Account{
|
|
Name: userinfo.Name,
|
|
Nick: userinfo.Nick,
|
|
Avatar: userinfo.Avatar,
|
|
EmailAddress: userinfo.Email,
|
|
PowerLevel: 0,
|
|
ExternalID: uint(userinfo.Id),
|
|
}
|
|
return account, database.C.Save(&account).Error
|
|
}
|
|
return account, err
|
|
}
|
|
|
|
account.Name = userinfo.Name
|
|
account.Nick = userinfo.Nick
|
|
account.Avatar = userinfo.Avatar
|
|
account.EmailAddress = userinfo.Email
|
|
|
|
err := database.C.Save(&account).Error
|
|
|
|
return account, err
|
|
}
|
|
|
|
func Authenticate(atk, rtk string) (models.Account, string, string, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
defer cancel()
|
|
|
|
var err error
|
|
var user models.Account
|
|
reply, err := grpc.Auth.Authenticate(ctx, &proto.AuthRequest{
|
|
AccessToken: atk,
|
|
RefreshToken: &rtk,
|
|
})
|
|
if err != nil {
|
|
return user, reply.GetAccessToken(), reply.GetRefreshToken(), err
|
|
}
|
|
|
|
user, err = LinkAccount(reply.Userinfo)
|
|
|
|
return user, reply.GetAccessToken(), reply.GetRefreshToken(), err
|
|
}
|