Browse Source
Tab on user profile to show starred repos (#519)
* Tab on user profile to show starred repos * Make golint happy and use transactions on StarRepo function * x -> sess * Use sess.Close() instead of sess.Rollback() * Add copyright * Fix lintfor-closed-social
Andrey Nering
7 years ago
committed by
Lunny Xiao
Lunny Xiao
7 changed files with 115 additions and 76 deletions
Split View
Diff Options
-
+0 -1cmd/web.go
-
+0 -60models/repo.go
-
+87 -0models/star.go
-
+1 -0options/locale/locale_en-US.ini
-
+8 -5routers/user/profile.go
-
+4 -0templates/base/head.tmpl
-
+15 -10templates/user/profile.tmpl
+ 0
- 1
cmd/web.go
View File
+ 0
- 60
models/repo.go
View File
+ 87
- 0
models/star.go
View File
| @ -0,0 +1,87 @@ | |||
| // Copyright 2016 The Gitea Authors. All rights reserved. | |||
| // Use of this source code is governed by a MIT-style | |||
| // license that can be found in the LICENSE file. | |||
| package models | |||
| // Star represents a starred repo by an user. | |||
| type Star struct { | |||
| ID int64 `xorm:"pk autoincr"` | |||
| UID int64 `xorm:"UNIQUE(s)"` | |||
| RepoID int64 `xorm:"UNIQUE(s)"` | |||
| } | |||
| // StarRepo or unstar repository. | |||
| func StarRepo(userID, repoID int64, star bool) error { | |||
| sess := x.NewSession() | |||
| defer sess.Close() | |||
| if err := sess.Begin(); err != nil { | |||
| return err | |||
| } | |||
| if star { | |||
| if IsStaring(userID, repoID) { | |||
| return nil | |||
| } | |||
| if _, err := sess.Insert(&Star{UID: userID, RepoID: repoID}); err != nil { | |||
| return err | |||
| } | |||
| if _, err := sess.Exec("UPDATE `repository` SET num_stars = num_stars + 1 WHERE id = ?", repoID); err != nil { | |||
| return err | |||
| } | |||
| if _, err := sess.Exec("UPDATE `user` SET num_stars = num_stars + 1 WHERE id = ?", userID); err != nil { | |||
| return err | |||
| } | |||
| } else { | |||
| if !IsStaring(userID, repoID) { | |||
| return nil | |||
| } | |||
| if _, err := sess.Delete(&Star{0, userID, repoID}); err != nil { | |||
| return err | |||
| } | |||
| if _, err := sess.Exec("UPDATE `repository` SET num_stars = num_stars - 1 WHERE id = ?", repoID); err != nil { | |||
| return err | |||
| } | |||
| if _, err := sess.Exec("UPDATE `user` SET num_stars = num_stars - 1 WHERE id = ?", userID); err != nil { | |||
| return err | |||
| } | |||
| } | |||
| return sess.Commit() | |||
| } | |||
| // IsStaring checks if user has starred given repository. | |||
| func IsStaring(userID, repoID int64) bool { | |||
| has, _ := x.Get(&Star{0, userID, repoID}) | |||
| return has | |||
| } | |||
| // GetStargazers returns the users that starred the repo. | |||
| func (repo *Repository) GetStargazers(page int) ([]*User, error) { | |||
| users := make([]*User, 0, ItemsPerPage) | |||
| err := x. | |||
| Limit(ItemsPerPage, (page-1)*ItemsPerPage). | |||
| Where("star.repo_id = ?", repo.ID). | |||
| Join("LEFT", "star", "`user`.id = star.uid"). | |||
| Find(&users) | |||
| return users, err | |||
| } | |||
| // GetStarredRepos returns the repos the user starred. | |||
| func (u *User) GetStarredRepos(private bool) (repos []*Repository, err error) { | |||
| sess := x. | |||
| Join("INNER", "star", "star.repo_id = repository.id"). | |||
| Where("star.uid = ?", u.ID) | |||
| if !private { | |||
| sess = sess.And("is_private = ?", false) | |||
| } | |||
| err = sess. | |||
| Find(&repos) | |||
| return | |||
| } | |||