mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package repo
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
user_model "github.com/hanzoai/git/models/user"
|
|
"github.com/hanzoai/git/modules/timeutil"
|
|
|
|
"github.com/hanzoai/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})
|
|
}
|