This commit is contained in:
Unknwon 2015-08-17 17:05:37 +08:00
parent 875f917771
commit 407385db7e
9 changed files with 111 additions and 71 deletions

View File

@ -17,7 +17,7 @@ import (
"github.com/gogits/gogs/modules/setting" "github.com/gogits/gogs/modules/setting"
) )
const APP_VER = "0.6.5.0815 Beta" const APP_VER = "0.6.5.0817 Beta"
func init() { func init() {
runtime.GOMAXPROCS(runtime.NumCPU()) runtime.GOMAXPROCS(runtime.NumCPU())

View File

@ -1435,9 +1435,9 @@ func NotifyWatchers(act *Action) error {
// \/ \/ // \/ \/
type Star struct { type Star struct {
Id int64 ID int64 `xorm:"pk autoincr"`
Uid int64 `xorm:"UNIQUE(s)"` UID int64 `xorm:"uid UNIQUE(s)"`
RepoId int64 `xorm:"UNIQUE(s)"` RepoID int64 `xorm:"UNIQUE(s)"`
} }
// Star or unstar repository. // Star or unstar repository.
@ -1446,7 +1446,7 @@ func StarRepo(uid, repoId int64, star bool) (err error) {
if IsStaring(uid, repoId) { if IsStaring(uid, repoId) {
return nil return nil
} }
if _, err = x.Insert(&Star{Uid: uid, RepoId: repoId}); err != nil { if _, err = x.Insert(&Star{UID: uid, RepoID: repoId}); err != nil {
return err return err
} else if _, err = x.Exec("UPDATE `repository` SET num_stars = num_stars + 1 WHERE id = ?", repoId); err != nil { } else if _, err = x.Exec("UPDATE `repository` SET num_stars = num_stars + 1 WHERE id = ?", repoId); err != nil {
return err return err

View File

@ -18,8 +18,8 @@ var (
// AccessToken represents a personal access token. // AccessToken represents a personal access token.
type AccessToken struct { type AccessToken struct {
Id int64 ID int64 `xorm:"pk autoincr"`
Uid int64 UID int64 `xorm:"uid INDEX"`
Name string Name string
Sha1 string `xorm:"UNIQUE VARCHAR(40)"` Sha1 string `xorm:"UNIQUE VARCHAR(40)"`
Created time.Time `xorm:"CREATED"` Created time.Time `xorm:"CREATED"`

View File

@ -515,8 +515,12 @@ func DeleteBeans(e Engine, beans ...interface{}) (err error) {
} }
// FIXME: need some kind of mechanism to record failure. HINT: system notice // FIXME: need some kind of mechanism to record failure. HINT: system notice
// DeleteUser completely and permanently deletes everything of user. // DeleteUser completely and permanently deletes everything of a user,
// but issues/comments/pulls will be kept and shown as someone has been deleted.
func DeleteUser(u *User) error { func DeleteUser(u *User) error {
// Note: A user owns any repository or belongs to any organization
// cannot perform delete operation.
// Check ownership of repository. // Check ownership of repository.
count, err := GetRepositoryCount(u) count, err := GetRepositoryCount(u)
if err != nil { if err != nil {
@ -533,80 +537,116 @@ func DeleteUser(u *User) error {
return ErrUserHasOrgs{UID: u.Id} return ErrUserHasOrgs{UID: u.Id}
} }
// Get watches before session.
watches := make([]*Watch, 0, 10)
if err = x.Where("user_id=?", u.Id).Find(&watches); err != nil {
return fmt.Errorf("get all watches: %v", err)
}
repoIDs := make([]int64, 0, len(watches))
for i := range watches {
repoIDs = append(repoIDs, watches[i].RepoID)
}
// FIXME: check issues, other repos' commits
sess := x.NewSession() sess := x.NewSession()
defer sessionRelease(sess) defer sessionRelease(sess)
if err = sess.Begin(); err != nil { if err = sess.Begin(); err != nil {
return err return err
} }
if err = DeleteBeans(sess, // ***** START: Watch *****
&Follow{FollowID: u.Id}, watches := make([]*Watch, 0, 10)
&Oauth2{Uid: u.Id}, if err = x.Find(&watches, &Watch{UserID: u.Id}); err != nil {
&Action{UserID: u.Id}, return fmt.Errorf("get all watches: %v", err)
&Access{UserID: u.Id},
&Collaboration{UserID: u.Id},
&EmailAddress{Uid: u.Id},
&Watch{UserID: u.Id},
&IssueUser{UID: u.Id},
); err != nil {
return err
} }
for i := range watches {
// Decrease all watch numbers. if _, err = sess.Exec("UPDATE `repository` SET num_watches=num_watches-1 WHERE id=?", watches[i].RepoID); err != nil {
for i := range repoIDs { return fmt.Errorf("decrease repository watch number[%d]: %v", watches[i].RepoID, err)
if _, err = sess.Exec("UPDATE `repository` SET num_watches=num_watches-1 WHERE id=?", repoIDs[i]); err != nil {
return err
} }
} }
// ***** END: Watch *****
// Delete all SSH keys. // ***** START: Star *****
stars := make([]*Star, 0, 10)
if err = x.Find(&stars, &Star{UID: u.Id}); err != nil {
return fmt.Errorf("get all stars: %v", err)
}
for i := range stars {
if _, err = sess.Exec("UPDATE `repository` SET num_stars=num_stars-1 WHERE id=?", stars[i].RepoID); err != nil {
return fmt.Errorf("decrease repository star number[%d]: %v", stars[i].RepoID, err)
}
}
// ***** END: Star *****
// ***** START: Follow *****
followers := make([]*Follow, 0, 10)
if err = x.Find(&followers, &Follow{UserID: u.Id}); err != nil {
return fmt.Errorf("get all followers: %v", err)
}
for i := range followers {
if _, err = sess.Exec("UPDATE `user` SET num_followers=num_followers-1 WHERE id=?", followers[i].UserID); err != nil {
return fmt.Errorf("decrease user follower number[%d]: %v", followers[i].UserID, err)
}
}
// ***** END: Follow *****
if err = DeleteBeans(sess,
&Oauth2{Uid: u.Id},
&AccessToken{UID: u.Id},
&Collaboration{UserID: u.Id},
&Access{UserID: u.Id},
&Watch{UserID: u.Id},
&Star{UID: u.Id},
&Follow{FollowID: u.Id},
&Action{UserID: u.Id},
&IssueUser{UID: u.Id},
&EmailAddress{Uid: u.Id},
); err != nil {
return fmt.Errorf("DeleteBeans: %v", err)
}
// ***** START: PublicKey *****
keys := make([]*PublicKey, 0, 10) keys := make([]*PublicKey, 0, 10)
if err = sess.Find(&keys, &PublicKey{OwnerID: u.Id}); err != nil { if err = sess.Find(&keys, &PublicKey{OwnerID: u.Id}); err != nil {
return err return fmt.Errorf("get all public keys: %v", err)
} }
for _, key := range keys { for _, key := range keys {
if err = deletePublicKey(sess, key); err != nil { if err = deletePublicKey(sess, key); err != nil {
return fmt.Errorf("deletePublicKey: %v", err)
}
}
// ***** END: PublicKey *****
// Clear assignee.
if _, err = sess.Exec("UPDATE `issue` SET assignee_id=0 WHERE assignee_id=?", u.Id); err != nil {
return fmt.Errorf("clear assignee: %v", err)
}
if _, err = sess.Delete(u); err != nil {
return fmt.Errorf("Delete: %v", err)
}
if err = sess.Commit(); err != nil {
return fmt.Errorf("Commit: %v", err)
}
// FIXME: system notice
// Note: There are something just cannot be roll back,
// so just keep error logs of those operations.
RewriteAllPublicKeys()
os.RemoveAll(UserPath(u.Name))
os.Remove(u.CustomAvatarPath())
return nil
}
// DeleteInactivateUsers deletes all inactivate users and email addresses.
func DeleteInactivateUsers() (err error) {
users := make([]*User, 0, 10)
if err = x.Where("is_active=?", false).Find(&users); err != nil {
return fmt.Errorf("get all inactive users: %v", err)
}
for _, u := range users {
if err = DeleteUser(u); err != nil {
// Ignore users that were set inactive by admin.
if IsErrUserOwnRepos(err) || IsErrUserHasOrgs(err) {
continue
}
return err return err
} }
} }
// Clear assignee. _, err = x.Where("is_activated=?", false).Delete(new(EmailAddress))
if _, err = sess.Exec("UPDATE `issue` SET assignee_id=0 WHERE assignee_id=?", u.Id); err != nil {
return err
}
if _, err = sess.Delete(u); err != nil {
return err
}
// Delete user data.
if err = os.RemoveAll(UserPath(u.Name)); err != nil {
return err
}
// Delete avatar.
os.Remove(u.CustomAvatarPath())
return sess.Commit()
}
// DeleteInactivateUsers deletes all inactivate users and email addresses.
func DeleteInactivateUsers() error {
_, err := x.Where("is_active=?", false).Delete(new(User))
if err == nil {
_, err = x.Where("is_activated=?", false).Delete(new(EmailAddress))
}
return err return err
} }
@ -895,9 +935,9 @@ func SearchUserByName(opt SearchOption) (us []*User, err error) {
// Follow is connection request for receiving user notification. // Follow is connection request for receiving user notification.
type Follow struct { type Follow struct {
Id int64 ID int64 `xorm:"pk autoincr"`
UserID int64 `xorm:"unique(follow)"` UserID int64 `xorm:"UNIQUE(follow)"`
FollowID int64 `xorm:"unique(follow)"` FollowID int64 `xorm:"UNIQUE(follow)"`
} }
// FollowUser marks someone be another's follower. // FollowUser marks someone be another's follower.

View File

@ -44,7 +44,7 @@ func SignedInId(req *http.Request, sess session.Store) int64 {
} }
return 0 return 0
} }
return t.Uid return t.UID
} }
} }
} }

View File

@ -34,7 +34,7 @@ type CreateAccessTokenForm struct {
// POST /users/:username/tokens // POST /users/:username/tokens
func CreateAccessToken(ctx *middleware.Context, form CreateAccessTokenForm) { func CreateAccessToken(ctx *middleware.Context, form CreateAccessTokenForm) {
t := &models.AccessToken{ t := &models.AccessToken{
Uid: ctx.User.Id, UID: ctx.User.Id,
Name: form.Name, Name: form.Name,
} }
if err := models.NewAccessToken(t); err != nil { if err := models.NewAccessToken(t); err != nil {

View File

@ -122,7 +122,7 @@ func Http(ctx *middleware.Context) {
} }
return return
} }
authUser, err = models.GetUserByID(token.Uid) authUser, err = models.GetUserByID(token.UID)
if err != nil { if err != nil {
ctx.Handle(500, "GetUserById", err) ctx.Handle(500, "GetUserById", err)
return return

View File

@ -420,7 +420,7 @@ func SettingsApplicationsPost(ctx *middleware.Context, form auth.NewAccessTokenF
} }
t := &models.AccessToken{ t := &models.AccessToken{
Uid: ctx.User.Id, UID: ctx.User.Id,
Name: form.Name, Name: form.Name,
} }
if err := models.NewAccessToken(t); err != nil { if err := models.NewAccessToken(t); err != nil {

View File

@ -1 +1 @@
0.6.5.0815 Beta 0.6.5.0817 Beta