mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-22 12:54:53 -05:00
84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
|
// Copyright 2024 The Forgejo Authors. All rights reserved.
|
||
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package user
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"code.gitea.io/gitea/models/db"
|
||
|
"code.gitea.io/gitea/modules/optional"
|
||
|
"code.gitea.io/gitea/modules/validation"
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
db.RegisterModel(new(FederatedUser))
|
||
|
}
|
||
|
|
||
|
func CreateFederatedUser(ctx context.Context, user *User, federatedUser *FederatedUser) error {
|
||
|
if res, err := validation.IsValid(user); !res {
|
||
|
return err
|
||
|
}
|
||
|
overwrite := CreateUserOverwriteOptions{
|
||
|
IsActive: optional.Some(false),
|
||
|
IsRestricted: optional.Some(false),
|
||
|
}
|
||
|
|
||
|
// Begin transaction
|
||
|
ctx, committer, err := db.TxContext((ctx))
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer committer.Close()
|
||
|
|
||
|
if err := CreateUser(ctx, user, &overwrite); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
federatedUser.UserID = user.ID
|
||
|
if res, err := validation.IsValid(federatedUser); !res {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
_, err = db.GetEngine(ctx).Insert(federatedUser)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// Commit transaction
|
||
|
return committer.Commit()
|
||
|
}
|
||
|
|
||
|
func FindFederatedUser(ctx context.Context, externalID string,
|
||
|
federationHostID int64,
|
||
|
) (*User, *FederatedUser, error) {
|
||
|
federatedUser := new(FederatedUser)
|
||
|
user := new(User)
|
||
|
has, err := db.GetEngine(ctx).Where("external_id=? and federation_host_id=?", externalID, federationHostID).Get(federatedUser)
|
||
|
if err != nil {
|
||
|
return nil, nil, err
|
||
|
} else if !has {
|
||
|
return nil, nil, nil
|
||
|
}
|
||
|
has, err = db.GetEngine(ctx).ID(federatedUser.UserID).Get(user)
|
||
|
if err != nil {
|
||
|
return nil, nil, err
|
||
|
} else if !has {
|
||
|
return nil, nil, fmt.Errorf("User %v for federated user is missing", federatedUser.UserID)
|
||
|
}
|
||
|
|
||
|
if res, err := validation.IsValid(*user); !res {
|
||
|
return nil, nil, err
|
||
|
}
|
||
|
if res, err := validation.IsValid(*federatedUser); !res {
|
||
|
return nil, nil, err
|
||
|
}
|
||
|
return user, federatedUser, nil
|
||
|
}
|
||
|
|
||
|
func DeleteFederatedUser(ctx context.Context, userID int64) error {
|
||
|
_, err := db.GetEngine(ctx).Delete(&FederatedUser{UserID: userID})
|
||
|
return err
|
||
|
}
|