repo.go 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package models
  5. import (
  6. "os"
  7. "path/filepath"
  8. "strings"
  9. "time"
  10. "github.com/gogits/gogs/utils/log"
  11. git "github.com/libgit2/git2go"
  12. )
  13. type Repo struct {
  14. Id int64
  15. OwnerId int64 `xorm:"unique(s)"`
  16. ForkId int64
  17. LowerName string `xorm:"unique(s) index not null"`
  18. Name string `xorm:"index not null"`
  19. NumWatchs int
  20. NumStars int
  21. NumForks int
  22. Created time.Time `xorm:"created"`
  23. Updated time.Time `xorm:"updated"`
  24. }
  25. // check if repository is exist
  26. func IsRepositoryExist(user *User, reposName string) (bool, error) {
  27. repo := Repo{OwnerId: user.Id}
  28. has, err := orm.Where("lower_name = ?", strings.ToLower(reposName)).Get(&repo)
  29. if err != nil {
  30. return has, err
  31. }
  32. s, err := os.Stat(RepoPath(user.Name, reposName))
  33. if err != nil {
  34. return false, nil
  35. }
  36. return s.IsDir(), nil
  37. }
  38. //
  39. // create a repository for a user or orgnaziation
  40. //
  41. func CreateRepository(user *User, reposName string) (*Repo, error) {
  42. f := RepoPath(user.Name, reposName)
  43. _, err := git.InitRepository(f, true)
  44. if err != nil {
  45. return nil, err
  46. }
  47. repo := Repo{OwnerId: user.Id, Name: reposName, LowerName: strings.ToLower(reposName)}
  48. session := orm.NewSession()
  49. defer session.Close()
  50. session.Begin()
  51. _, err = session.Insert(&repo)
  52. if err != nil {
  53. err2 := os.RemoveAll(f)
  54. if err2 != nil {
  55. log.Error("delete repo directory %s/%s failed", user.Name, reposName)
  56. }
  57. session.Rollback()
  58. return nil, err
  59. }
  60. access := Access{UserName: user.Name,
  61. RepoName: repo.Name,
  62. Mode: AU_WRITABLE,
  63. }
  64. _, err = session.Insert(&access)
  65. if err != nil {
  66. err2 := os.RemoveAll(f)
  67. if err2 != nil {
  68. log.Error("delete repo directory %s/%s failed", user.Name, reposName)
  69. }
  70. session.Rollback()
  71. return nil, err
  72. }
  73. _, err = session.Exec("update user set num_repos = num_repos + 1 where id = ?", user.Id)
  74. if err != nil {
  75. err2 := os.RemoveAll(f)
  76. if err2 != nil {
  77. log.Error("delete repo directory %s/%s failed", user.Name, reposName)
  78. }
  79. session.Rollback()
  80. return nil, err
  81. }
  82. err = session.Commit()
  83. if err != nil {
  84. err2 := os.RemoveAll(f)
  85. if err2 != nil {
  86. log.Error("delete repo directory %s/%s failed", user.Name, reposName)
  87. }
  88. session.Rollback()
  89. return nil, err
  90. }
  91. return &repo, nil
  92. }
  93. // GetRepositories returns the list of repositories of given user.
  94. func GetRepositories(user *User) ([]Repo, error) {
  95. repos := make([]Repo, 0)
  96. err := orm.Find(&repos, &Repo{OwnerId: user.Id})
  97. return repos, err
  98. }
  99. func StarReposiory(user *User, repoName string) error {
  100. return nil
  101. }
  102. func UnStarRepository() {
  103. }
  104. func WatchRepository() {
  105. }
  106. func UnWatchRepository() {
  107. }
  108. func RepoPath(userName, repoName string) string {
  109. return filepath.Join(UserPath(userName), repoName+".git")
  110. }
  111. // DeleteRepository deletes a repository for a user or orgnaztion.
  112. func DeleteRepository(user *User, reposName string) (err error) {
  113. session := orm.NewSession()
  114. if _, err = session.Delete(&Repo{OwnerId: user.Id, Name: reposName}); err != nil {
  115. session.Rollback()
  116. return err
  117. }
  118. if _, err = session.Exec("update user set num_repos = num_repos - 1 where id = ?", user.Id); err != nil {
  119. session.Rollback()
  120. return err
  121. }
  122. if err = session.Commit(); err != nil {
  123. session.Rollback()
  124. return err
  125. }
  126. if err = os.RemoveAll(RepoPath(user.Name, reposName)); err != nil {
  127. // TODO: log and delete manully
  128. log.Error("delete repo %s/%s failed", user.Name, reposName)
  129. return err
  130. }
  131. return nil
  132. }