aboutsummaryrefslogtreecommitdiff
path: root/models
diff options
context:
space:
mode:
authorMeaglith Ma <genedna@gmail.com>2014-04-06 17:28:33 +0800
committerMeaglith Ma <genedna@gmail.com>2014-04-06 17:28:33 +0800
commit2e8d5c2eb36b63468dc78561566975c72eeaaefa (patch)
tree92659a09ec784005c8d9a0540e5197a75b0a3ea9 /models
parenta044c24c3db4b18b20406045b40e55ce15f941a0 (diff)
parentef6b9784962d3152d3ec46833303bad72915af57 (diff)
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'models')
-rw-r--r--models/access.go8
-rw-r--r--models/action.go13
-rw-r--r--models/git.go19
-rw-r--r--models/models.go12
-rw-r--r--models/oauth2.go18
-rw-r--r--models/publickey.go4
-rw-r--r--models/repo.go123
-rw-r--r--models/user.go55
8 files changed, 220 insertions, 32 deletions
diff --git a/models/access.go b/models/access.go
index 42fccae0..83261575 100644
--- a/models/access.go
+++ b/models/access.go
@@ -32,6 +32,14 @@ func AddAccess(access *Access) error {
return err
}
+// UpdateAccess updates access information.
+func UpdateAccess(access *Access) error {
+ access.UserName = strings.ToLower(access.UserName)
+ access.RepoName = strings.ToLower(access.RepoName)
+ _, err := orm.Id(access.Id).Update(access)
+ return err
+}
+
// HasAccess returns true if someone can read or write to given repository.
func HasAccess(userName, repoName string, mode int) (bool, error) {
return orm.Get(&Access{
diff --git a/models/action.go b/models/action.go
index 1e55df85..a642a82c 100644
--- a/models/action.go
+++ b/models/action.go
@@ -21,6 +21,7 @@ const (
OP_COMMIT_REPO
OP_CREATE_ISSUE
OP_PULL_REQUEST
+ OP_TRANSFER_REPO
)
// Action represents user operation type and other information to repository.,
@@ -108,6 +109,18 @@ func NewRepoAction(user *User, repo *Repository) (err error) {
return err
}
+// TransferRepoAction adds new action for transfering repository.
+func TransferRepoAction(user, newUser *User, repo *Repository) (err error) {
+ if err = NotifyWatchers(&Action{ActUserId: user.Id, ActUserName: user.Name, ActEmail: user.Email,
+ OpType: OP_TRANSFER_REPO, RepoId: repo.Id, RepoName: repo.Name, Content: newUser.Name}); err != nil {
+ log.Error("action.TransferRepoAction(notify watchers): %d/%s", user.Id, repo.Name)
+ return err
+ }
+
+ log.Trace("action.TransferRepoAction: %s/%s", user.LowerName, repo.LowerName)
+ return err
+}
+
// GetFeeds returns action list of given user in given context.
func GetFeeds(userid, offset int64, isProfile bool) ([]Action, error) {
actions := make([]Action, 0, 20)
diff --git a/models/git.go b/models/git.go
index d3bad6e0..46345d0f 100644
--- a/models/git.go
+++ b/models/git.go
@@ -56,6 +56,25 @@ func GetBranches(userName, repoName string) ([]string, error) {
return brs, nil
}
+// GetTags returns all tags of given repository.
+func GetTags(userName, repoName string) ([]string, error) {
+ repo, err := git.OpenRepository(RepoPath(userName, repoName))
+ if err != nil {
+ return nil, err
+ }
+
+ refs, err := repo.AllTags()
+ if err != nil {
+ return nil, err
+ }
+
+ tags := make([]string, len(refs))
+ for i, ref := range refs {
+ tags[i] = ref.Name
+ }
+ return tags, nil
+}
+
func IsBranchExist(userName, repoName, branchName string) bool {
repo, err := git.OpenRepository(RepoPath(userName, repoName))
if err != nil {
diff --git a/models/models.go b/models/models.go
index 384f1fc4..0ad86337 100644
--- a/models/models.go
+++ b/models/models.go
@@ -53,10 +53,10 @@ func NewTestEngine(x *xorm.Engine) (err error) {
// os.MkdirAll(path.Dir(DbCfg.Path), os.ModePerm)
// x, err = xorm.NewEngine("sqlite3", DbCfg.Path)
default:
- return fmt.Errorf("Unknown database type: %s\n", DbCfg.Type)
+ return fmt.Errorf("Unknown database type: %s", DbCfg.Type)
}
if err != nil {
- return fmt.Errorf("models.init(fail to conntect database): %v\n", err)
+ return fmt.Errorf("models.init(fail to conntect database): %v", err)
}
return x.Sync(new(User), new(PublicKey), new(Repository), new(Watch),
@@ -75,10 +75,10 @@ func SetEngine() (err error) {
os.MkdirAll(path.Dir(DbCfg.Path), os.ModePerm)
orm, err = xorm.NewEngine("sqlite3", DbCfg.Path)
default:
- return fmt.Errorf("Unknown database type: %s\n", DbCfg.Type)
+ return fmt.Errorf("Unknown database type: %s", DbCfg.Type)
}
if err != nil {
- return fmt.Errorf("models.init(fail to conntect database): %v\n", err)
+ return fmt.Errorf("models.init(fail to conntect database): %v", err)
}
// WARNNING: for serv command, MUST remove the output to os.stdout,
@@ -89,7 +89,7 @@ func SetEngine() (err error) {
f, err := os.Create(logPath)
if err != nil {
- return fmt.Errorf("models.init(fail to create xorm.log): %v\n", err)
+ return fmt.Errorf("models.init(fail to create xorm.log): %v", err)
}
orm.Logger = f
@@ -104,7 +104,7 @@ func NewEngine() (err error) {
return err
} else if err = orm.Sync(new(User), new(PublicKey), new(Repository), new(Watch),
new(Action), new(Access), new(Issue), new(Comment)); err != nil {
- return fmt.Errorf("sync database struct error: %v\n", err)
+ return fmt.Errorf("sync database struct error: %v", err)
}
return nil
}
diff --git a/models/oauth2.go b/models/oauth2.go
new file mode 100644
index 00000000..70dcd510
--- /dev/null
+++ b/models/oauth2.go
@@ -0,0 +1,18 @@
+package models
+
+import "time"
+
+// OT: Oauth2 Type
+const (
+ OT_GITHUB = iota + 1
+ OT_GOOGLE
+ OT_TWITTER
+)
+
+type Oauth2 struct {
+ Uid int64 `xorm:"pk"` // userId
+ Type int `xorm:"pk unique(oauth)"` // twitter,github,google...
+ Identity string `xorm:"pk unique(oauth)"` // id..
+ Token string `xorm:"VARCHAR(200) not null"`
+ RefreshTime time.Time `xorm:"created"`
+}
diff --git a/models/publickey.go b/models/publickey.go
index 3f2fcabd..42d2523b 100644
--- a/models/publickey.go
+++ b/models/publickey.go
@@ -77,8 +77,8 @@ func init() {
// PublicKey represents a SSH key of user.
type PublicKey struct {
Id int64
- OwnerId int64 `xorm:"index"`
- Name string `xorm:"unique not null"`
+ OwnerId int64 `xorm:" index not null"`
+ Name string `xorm:" not null"` //UNIQUE(s)
Fingerprint string
Content string `xorm:"TEXT not null"`
Created time.Time `xorm:"created"`
diff --git a/models/repo.go b/models/repo.go
index 0c808f18..e8ebce92 100644
--- a/models/repo.go
+++ b/models/repo.go
@@ -74,6 +74,7 @@ type Repository struct {
NumStars int
NumForks int
NumIssues int
+ NumReleases int `xorm:"NOT NULL"`
NumClosedIssues int
NumOpenIssues int `xorm:"-"`
IsPrivate bool
@@ -142,17 +143,17 @@ func CreateRepository(user *User, repoName, desc, repoLang, license string, priv
if err = initRepository(repoPath, user, repo, initReadme, repoLang, license); err != nil {
return nil, err
}
- session := orm.NewSession()
- defer session.Close()
- session.Begin()
+ sess := orm.NewSession()
+ defer sess.Close()
+ sess.Begin()
- if _, err = session.Insert(repo); err != nil {
+ if _, err = sess.Insert(repo); err != nil {
if err2 := os.RemoveAll(repoPath); err2 != nil {
log.Error("repo.CreateRepository(repo): %v", err)
return nil, errors.New(fmt.Sprintf(
"delete repo directory %s/%s failed(1): %v", user.Name, repoName, err2))
}
- session.Rollback()
+ sess.Rollback()
return nil, err
}
@@ -161,8 +162,8 @@ func CreateRepository(user *User, repoName, desc, repoLang, license string, priv
RepoName: strings.ToLower(path.Join(user.Name, repo.Name)),
Mode: AU_WRITABLE,
}
- if _, err = session.Insert(&access); err != nil {
- session.Rollback()
+ if _, err = sess.Insert(&access); err != nil {
+ sess.Rollback()
if err2 := os.RemoveAll(repoPath); err2 != nil {
log.Error("repo.CreateRepository(access): %v", err)
return nil, errors.New(fmt.Sprintf(
@@ -172,8 +173,8 @@ func CreateRepository(user *User, repoName, desc, repoLang, license string, priv
}
rawSql := "UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?"
- if _, err = session.Exec(rawSql, user.Id); err != nil {
- session.Rollback()
+ if _, err = sess.Exec(rawSql, user.Id); err != nil {
+ sess.Rollback()
if err2 := os.RemoveAll(repoPath); err2 != nil {
log.Error("repo.CreateRepository(repo count): %v", err)
return nil, errors.New(fmt.Sprintf(
@@ -182,8 +183,8 @@ func CreateRepository(user *User, repoName, desc, repoLang, license string, priv
return nil, err
}
- if err = session.Commit(); err != nil {
- session.Rollback()
+ if err = sess.Commit(); err != nil {
+ sess.Rollback()
if err2 := os.RemoveAll(repoPath); err2 != nil {
log.Error("repo.CreateRepository(commit): %v", err)
return nil, errors.New(fmt.Sprintf(
@@ -368,14 +369,86 @@ func RepoPath(userName, repoName string) string {
return filepath.Join(UserPath(userName), strings.ToLower(repoName)+".git")
}
+// TransferOwnership transfers all corresponding setting from old user to new one.
+func TransferOwnership(user *User, newOwner string, repo *Repository) (err error) {
+ newUser, err := GetUserByName(newOwner)
+ if err != nil {
+ return err
+ }
+
+ // Update accesses.
+ accesses := make([]Access, 0, 10)
+ if err = orm.Find(&accesses, &Access{RepoName: user.LowerName + "/" + repo.LowerName}); err != nil {
+ return err
+ }
+ for i := range accesses {
+ accesses[i].RepoName = newUser.LowerName + "/" + repo.LowerName
+ if accesses[i].UserName == user.LowerName {
+ accesses[i].UserName = newUser.LowerName
+ }
+ if err = UpdateAccess(&accesses[i]); err != nil {
+ return err
+ }
+ }
+
+ // Update repository.
+ repo.OwnerId = newUser.Id
+ if _, err := orm.Id(repo.Id).Update(repo); err != nil {
+ return err
+ }
+
+ // Update user repository number.
+ rawSql := "UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?"
+ if _, err = orm.Exec(rawSql, newUser.Id); err != nil {
+ return err
+ }
+ rawSql = "UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?"
+ if _, err = orm.Exec(rawSql, user.Id); err != nil {
+ return err
+ }
+
+ // Add watch of new owner to repository.
+ if !IsWatching(newUser.Id, repo.Id) {
+ if err = WatchRepo(newUser.Id, repo.Id, true); err != nil {
+ return err
+ }
+ }
+
+ if err = TransferRepoAction(user, newUser, repo); err != nil {
+ return err
+ }
+
+ // Change repository directory name.
+ return os.Rename(RepoPath(user.Name, repo.Name), RepoPath(newUser.Name, repo.Name))
+}
+
+// ChangeRepositoryName changes all corresponding setting from old repository name to new one.
+func ChangeRepositoryName(userName, oldRepoName, newRepoName string) (err error) {
+ // Update accesses.
+ accesses := make([]Access, 0, 10)
+ if err = orm.Find(&accesses, &Access{RepoName: strings.ToLower(userName + "/" + oldRepoName)}); err != nil {
+ return err
+ }
+ for i := range accesses {
+ accesses[i].RepoName = userName + "/" + newRepoName
+ if err = UpdateAccess(&accesses[i]); err != nil {
+ return err
+ }
+ }
+
+ // Change repository directory name.
+ return os.Rename(RepoPath(userName, oldRepoName), RepoPath(userName, newRepoName))
+}
+
func UpdateRepository(repo *Repository) error {
+ repo.LowerName = strings.ToLower(repo.Name)
+
if len(repo.Description) > 255 {
repo.Description = repo.Description[:255]
}
if len(repo.Website) > 255 {
repo.Website = repo.Website[:255]
}
-
_, err := orm.Id(repo.Id).AllCols().Update(repo)
return err
}
@@ -390,29 +463,30 @@ func DeleteRepository(userId, repoId int64, userName string) (err error) {
return ErrRepoNotExist
}
- session := orm.NewSession()
- if err = session.Begin(); err != nil {
+ sess := orm.NewSession()
+ defer sess.Close()
+ if err = sess.Begin(); err != nil {
return err
}
- if _, err = session.Delete(&Repository{Id: repoId}); err != nil {
- session.Rollback()
+ if _, err = sess.Delete(&Repository{Id: repoId}); err != nil {
+ sess.Rollback()
return err
}
- if _, err := session.Delete(&Access{RepoName: strings.ToLower(path.Join(userName, repo.Name))}); err != nil {
- session.Rollback()
+ if _, err := sess.Delete(&Access{RepoName: strings.ToLower(path.Join(userName, repo.Name))}); err != nil {
+ sess.Rollback()
return err
}
rawSql := "UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?"
- if _, err = session.Exec(rawSql, userId); err != nil {
- session.Rollback()
+ if _, err = sess.Exec(rawSql, userId); err != nil {
+ sess.Rollback()
return err
}
- if _, err = session.Delete(&Watch{RepoId: repoId}); err != nil {
- session.Rollback()
+ if _, err = sess.Delete(&Watch{RepoId: repoId}); err != nil {
+ sess.Rollback()
return err
}
- if err = session.Commit(); err != nil {
- session.Rollback()
+ if err = sess.Commit(); err != nil {
+ sess.Rollback()
return err
}
if err = os.RemoveAll(RepoPath(userName, repo.Name)); err != nil {
@@ -513,6 +587,7 @@ func NotifyWatchers(act *Action) error {
continue
}
+ act.Id = 0
act.UserId = watches[i].UserId
if _, err = orm.InsertOne(act); err != nil {
return errors.New("repo.NotifyWatchers(create action): " + err.Error())
diff --git a/models/user.go b/models/user.go
index 4908552f..2641a15f 100644
--- a/models/user.go
+++ b/models/user.go
@@ -105,11 +105,17 @@ type Member struct {
// IsUserExist checks if given user name exist,
// the user name should be noncased unique.
func IsUserExist(name string) (bool, error) {
+ if len(name) == 0 {
+ return false, nil
+ }
return orm.Get(&User{LowerName: strings.ToLower(name)})
}
// IsEmailUsed returns true if the e-mail has been used.
func IsEmailUsed(email string) (bool, error) {
+ if len(email) == 0 {
+ return false, nil
+ }
return orm.Get(&User{Email: email})
}
@@ -203,8 +209,52 @@ func VerifyUserActiveCode(code string) (user *User) {
return nil
}
+// ChangeUserName changes all corresponding setting from old user name to new one.
+func ChangeUserName(user *User, newUserName string) (err error) {
+ newUserName = strings.ToLower(newUserName)
+
+ // Update accesses of user.
+ accesses := make([]Access, 0, 10)
+ if err = orm.Find(&accesses, &Access{UserName: user.LowerName}); err != nil {
+ return err
+ }
+ for i := range accesses {
+ accesses[i].UserName = newUserName
+ if strings.HasPrefix(accesses[i].RepoName, user.LowerName+"/") {
+ accesses[i].RepoName = strings.Replace(accesses[i].RepoName, user.LowerName, newUserName, 1)
+ if err = UpdateAccess(&accesses[i]); err != nil {
+ return err
+ }
+ }
+ }
+
+ repos, err := GetRepositories(user)
+ if err != nil {
+ return err
+ }
+ for i := range repos {
+ accesses = make([]Access, 0, 10)
+ // Update accesses of user repository.
+ if err = orm.Find(&accesses, &Access{RepoName: user.LowerName + "/" + repos[i].LowerName}); err != nil {
+ return err
+ }
+
+ for j := range accesses {
+ accesses[j].RepoName = newUserName + "/" + repos[i].LowerName
+ if err = UpdateAccess(&accesses[j]); err != nil {
+ return err
+ }
+ }
+ }
+
+ // Change user directory name.
+ return os.Rename(UserPath(user.LowerName), UserPath(newUserName))
+}
+
// UpdateUser updates user's information.
func UpdateUser(user *User) (err error) {
+ user.LowerName = strings.ToLower(user.Name)
+
if len(user.Location) > 255 {
user.Location = user.Location[:255]
}
@@ -233,6 +283,11 @@ func DeleteUser(user *User) error {
return err
}
+ // Delete all accesses.
+ if _, err = orm.Delete(&Access{UserName: user.LowerName}); err != nil {
+ return err
+ }
+
// Delete all SSH keys.
keys := make([]PublicKey, 0, 10)
if err = orm.Find(&keys, &PublicKey{OwnerId: user.Id}); err != nil {