mirror of
https://github.com/go-gitea/gitea.git
synced 2026-07-28 11:36:30 +00:00
cbe1b703dc
This PR replaces a set of struct-based `Get` lookups with explicit `db.Get` / `db.Exist` conditions in places where zero-value fields can lead to ambiguous matches or incorrect records being returned. The main goal is to make read paths deterministic and avoid accidentally matching the wrong row when only part of a struct is populated. ### What changed - replace many `db.GetEngine(ctx).Get(bean)` calls with explicit `builder.Eq` conditions across models such as actions, admin tasks, issues, pull requests, repositories, users, packages, redirects, watches, stars, and follows - use quoted column names where needed for reserved fields like `index`, `type`, and `name` - add dedicated user lookup helpers for: - primary email - OAuth login source / login name - update sign-in and OAuth-related flows to use explicit individual-user lookups instead of partially populated `User` structs - tighten package property and Terraform lock lookups to avoid ambiguous reads and updates - keep existing fallback behavior where needed, while removing reliance on zero-value struct matching ### User-facing impact These changes primarily affect authentication and account lookup paths: - email/username sign-in now re-fetches users through explicit keys - OAuth2 auto-linking now resolves users by name or primary email explicitly - OAuth2 login/sync now looks up users by login source, login type, and login name explicitly - non-individual accounts are no longer implicitly matched through partial user lookups in these flows This should reduce the risk of incorrect account matches and make query behavior more predictable across the codebase. --------- Co-authored-by: bircni <bircni@icloud.com>
105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
// Copyright 2016 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package repo
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitea.dev/models/db"
|
|
user_model "gitea.dev/models/user"
|
|
"gitea.dev/modules/timeutil"
|
|
|
|
"xorm.io/builder"
|
|
)
|
|
|
|
// Star represents a starred repo by a user.
|
|
type Star struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
UID int64 `xorm:"UNIQUE(s)"`
|
|
RepoID int64 `xorm:"UNIQUE(s)"`
|
|
CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
|
|
}
|
|
|
|
func init() {
|
|
db.RegisterModel(new(Star))
|
|
}
|
|
|
|
// StarRepo or unstar repository.
|
|
func StarRepo(ctx context.Context, doer *user_model.User, repo *Repository, star bool) error {
|
|
return db.WithTx(ctx, func(ctx context.Context) error {
|
|
staring := IsStaring(ctx, doer.ID, repo.ID)
|
|
|
|
if star {
|
|
if user_model.IsUserBlockedBy(ctx, doer, repo.OwnerID) {
|
|
return user_model.ErrBlockedUser
|
|
}
|
|
|
|
if staring {
|
|
return nil
|
|
}
|
|
|
|
if err := db.Insert(ctx, &Star{UID: doer.ID, RepoID: repo.ID}); err != nil {
|
|
return err
|
|
}
|
|
if _, err := db.Exec(ctx, "UPDATE `repository` SET num_stars = num_stars + 1 WHERE id = ?", repo.ID); err != nil {
|
|
return err
|
|
}
|
|
if _, err := db.Exec(ctx, "UPDATE `user` SET num_stars = num_stars + 1 WHERE id = ?", doer.ID); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if !staring {
|
|
return nil
|
|
}
|
|
|
|
if _, err := db.DeleteByBean(ctx, &Star{UID: doer.ID, RepoID: repo.ID}); err != nil {
|
|
return err
|
|
}
|
|
if _, err := db.Exec(ctx, "UPDATE `repository` SET num_stars = num_stars - 1 WHERE id = ?", repo.ID); err != nil {
|
|
return err
|
|
}
|
|
if _, err := db.Exec(ctx, "UPDATE `user` SET num_stars = num_stars - 1 WHERE id = ?", doer.ID); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// IsStaring checks if user has starred given repository.
|
|
func IsStaring(ctx context.Context, userID, repoID int64) bool {
|
|
has, _ := db.Exist[Star](ctx, builder.Eq{"uid": userID, "repo_id": repoID})
|
|
return has
|
|
}
|
|
|
|
// GetStargazers returns the users that starred the repo.
|
|
func GetStargazers(ctx context.Context, repo *Repository, opts db.ListOptions) ([]*user_model.User, error) {
|
|
sess := db.GetEngine(ctx).Where("star.repo_id = ?", repo.ID).
|
|
Join("LEFT", "star", "`user`.id = star.uid")
|
|
if opts.Page > 0 {
|
|
db.SetSessionPagination(sess, &opts)
|
|
|
|
users := make([]*user_model.User, 0, opts.PageSize)
|
|
return users, sess.Find(&users)
|
|
}
|
|
|
|
users := make([]*user_model.User, 0, 8)
|
|
return users, sess.Find(&users)
|
|
}
|
|
|
|
// ClearRepoStars clears all stars for a repository and from the user that starred it.
|
|
// Used when a repository is set to private.
|
|
func ClearRepoStars(ctx context.Context, repoID int64) error {
|
|
if _, err := db.Exec(ctx, "UPDATE `user` SET num_stars=num_stars-1 WHERE id IN (SELECT `uid` FROM `star` WHERE repo_id = ?)", repoID); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := db.Exec(ctx, "UPDATE `repository` SET num_stars = 0 WHERE id = ?", repoID); err != nil {
|
|
return err
|
|
}
|
|
|
|
return db.DeleteBeans(ctx, Star{RepoID: repoID})
|
|
}
|