mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2025-04-25 04:57:31 +00:00
f6a5b783d2
## Checklist The [contributor guide](https://forgejo.org/docs/next/contributor/) contains information that will be helpful to first time contributors. There also are a few [conditions for merging Pull Requests in Forgejo repositories](https://codeberg.org/forgejo/governance/src/branch/main/PullRequestsAgreement.md). You are also welcome to join the [Forgejo development chatroom](https://matrix.to/#/#forgejo-development:matrix.org). ### Tests - I added test coverage for Go changes... - [x] in their respective `*_test.go` for unit tests. - [ ] in the `tests/integration` directory if it involves interactions with a live Forgejo server. ### Documentation - [ ] I created a pull request [to the documentation](https://codeberg.org/forgejo/docs) to explain to Forgejo users how to use this change. - [ ] I did not document these changes and I do not expect someone else to do it. ### Release notes - [ ] I do not want this change to show in the release notes. - [ ] I want the title to show in the release notes with a link to this pull request. - [ ] I want the content of the `release-notes/<pull request number>.md` to be be used for the release notes instead of the title. Co-authored-by: Michael Jerger <michael.jerger@meissa-gmbh.de> Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/7203 Reviewed-by: Earl Warren <earl-warren@noreply.codeberg.org> Co-authored-by: zam <mirco.zachmann@meissa.de> Co-committed-by: zam <mirco.zachmann@meissa.de>
107 lines
2.6 KiB
Go
107 lines
2.6 KiB
Go
// Copyright 2024 The Forgejo Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package user
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"forgejo.org/models/db"
|
|
"forgejo.org/modules/optional"
|
|
"forgejo.org/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 FindFederatedUserByKeyID(ctx context.Context, keyID string) (*User, *FederatedUser, error) {
|
|
federatedUser := new(FederatedUser)
|
|
user := new(User)
|
|
has, err := db.GetEngine(ctx).Where("key_id=?", keyID).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
|
|
}
|