repo.go 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271
  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. "errors"
  7. "fmt"
  8. "io/ioutil"
  9. "os"
  10. "path/filepath"
  11. "strings"
  12. "time"
  13. "unicode/utf8"
  14. "github.com/Unknwon/com"
  15. git "github.com/libgit2/git2go"
  16. "github.com/gogits/gogs/modules/base"
  17. "github.com/gogits/gogs/modules/log"
  18. )
  19. type Repository struct {
  20. Id int64
  21. OwnerId int64 `xorm:"unique(s)"`
  22. ForkId int64
  23. LowerName string `xorm:"unique(s) index not null"`
  24. Name string `xorm:"index not null"`
  25. Description string
  26. Private bool
  27. NumWatchs int
  28. NumStars int
  29. NumForks int
  30. Created time.Time `xorm:"created"`
  31. Updated time.Time `xorm:"updated"`
  32. }
  33. type Star struct {
  34. Id int64
  35. RepoId int64
  36. UserId int64
  37. Created time.Time `xorm:"created"`
  38. }
  39. var (
  40. LanguageIgns, Licenses []string
  41. )
  42. var (
  43. ErrRepoAlreadyExist = errors.New("Repository already exist")
  44. )
  45. func init() {
  46. LanguageIgns = strings.Split(base.Cfg.MustValue("repository", "LANG_IGNS"), "|")
  47. Licenses = strings.Split(base.Cfg.MustValue("repository", "LICENSES"), "|")
  48. }
  49. // check if repository is exist
  50. func IsRepositoryExist(user *User, repoName string) (bool, error) {
  51. repo := Repository{OwnerId: user.Id}
  52. has, err := orm.Where("lower_name = ?", strings.ToLower(repoName)).Get(&repo)
  53. if err != nil {
  54. return has, err
  55. }
  56. s, err := os.Stat(RepoPath(user.Name, repoName))
  57. if err != nil {
  58. return false, nil
  59. }
  60. return s.IsDir(), nil
  61. }
  62. // CreateRepository creates a repository for given user or orgnaziation.
  63. func CreateRepository(user *User, repoName, desc, repoLang, license string, private bool, initReadme bool) (*Repository, error) {
  64. isExist, err := IsRepositoryExist(user, repoName)
  65. if err != nil {
  66. return nil, err
  67. } else if isExist {
  68. return nil, ErrRepoAlreadyExist
  69. }
  70. repo := &Repository{
  71. OwnerId: user.Id,
  72. Name: repoName,
  73. LowerName: strings.ToLower(repoName),
  74. Description: desc,
  75. Private: private,
  76. }
  77. f := RepoPath(user.Name, repoName)
  78. if err = initRepository(f, user, repo, initReadme, repoLang, license); err != nil {
  79. return nil, err
  80. }
  81. session := orm.NewSession()
  82. defer session.Close()
  83. session.Begin()
  84. if _, err = session.Insert(repo); err != nil {
  85. if err2 := os.RemoveAll(f); err2 != nil {
  86. return nil, errors.New(fmt.Sprintf(
  87. "delete repo directory %s/%s failed", user.Name, repoName))
  88. }
  89. session.Rollback()
  90. return nil, err
  91. }
  92. // TODO: RemoveAll may fail due to not root access.
  93. access := Access{
  94. UserName: user.Name,
  95. RepoName: repo.Name,
  96. Mode: AU_WRITABLE,
  97. }
  98. if _, err = session.Insert(&access); err != nil {
  99. session.Rollback()
  100. if err2 := os.RemoveAll(f); err2 != nil {
  101. return nil, errors.New(fmt.Sprintf(
  102. "delete repo directory %s/%s failed", user.Name, repoName))
  103. }
  104. return nil, err
  105. }
  106. if _, err = session.Exec("update user set num_repos = num_repos + 1 where id = ?", user.Id); err != nil {
  107. session.Rollback()
  108. if err2 := os.RemoveAll(f); err2 != nil {
  109. return nil, errors.New(fmt.Sprintf(
  110. "delete repo directory %s/%s failed", user.Name, repoName))
  111. }
  112. return nil, err
  113. }
  114. if err = session.Commit(); err != nil {
  115. session.Rollback()
  116. if err2 := os.RemoveAll(f); err2 != nil {
  117. return nil, errors.New(fmt.Sprintf(
  118. "delete repo directory %s/%s failed", user.Name, repoName))
  119. }
  120. return nil, err
  121. }
  122. return repo, nil
  123. }
  124. // InitRepository initializes README and .gitignore if needed.
  125. func initRepository(f string, user *User, repo *Repository, initReadme bool, repoLang, license string) error {
  126. fileName := map[string]string{
  127. "readme": "README.md",
  128. "gitign": ".gitignore",
  129. "license": "LICENSE",
  130. }
  131. workdir := os.TempDir() + fmt.Sprintf("%d", time.Now().Nanosecond())
  132. os.MkdirAll(workdir, os.ModePerm)
  133. sig := &git.Signature{
  134. Name: user.Name,
  135. Email: user.Email,
  136. When: time.Now(),
  137. }
  138. // README
  139. defaultReadme := repo.Name + "\n" + strings.Repeat("=",
  140. utf8.RuneCountInString(repo.Name)) + "\n\n" + repo.Description
  141. if err := ioutil.WriteFile(filepath.Join(workdir, fileName["readme"]),
  142. []byte(defaultReadme), 0644); err != nil {
  143. return err
  144. }
  145. // .gitignore
  146. filePath := "conf/gitignore/" + repoLang
  147. if com.IsFile(filePath) {
  148. if _, err := com.Copy(filePath,
  149. filepath.Join(workdir, fileName["gitign"])); err != nil {
  150. return err
  151. }
  152. }
  153. // LICENSE
  154. filePath = "conf/license/" + license
  155. if com.IsFile(filePath) {
  156. if _, err := com.Copy(filePath,
  157. filepath.Join(workdir, fileName["license"])); err != nil {
  158. return err
  159. }
  160. }
  161. rp, err := git.InitRepository(f, true)
  162. if err != nil {
  163. return err
  164. }
  165. rp.SetWorkdir(workdir, false)
  166. idx, err := rp.Index()
  167. if err != nil {
  168. return err
  169. }
  170. for _, name := range fileName {
  171. if err = idx.AddByPath(name); err != nil {
  172. return err
  173. }
  174. }
  175. treeId, err := idx.WriteTree()
  176. if err != nil {
  177. return err
  178. }
  179. message := "Init commit"
  180. tree, err := rp.LookupTree(treeId)
  181. if err != nil {
  182. return err
  183. }
  184. if _, err = rp.CreateCommit("HEAD", sig, sig, message, tree); err != nil {
  185. return err
  186. }
  187. return nil
  188. }
  189. // GetRepositories returns the list of repositories of given user.
  190. func GetRepositories(user *User) ([]Repository, error) {
  191. repos := make([]Repository, 0, 10)
  192. err := orm.Find(&repos, &Repository{OwnerId: user.Id})
  193. return repos, err
  194. }
  195. func StarReposiory(user *User, repoName string) error {
  196. return nil
  197. }
  198. func UnStarRepository() {
  199. }
  200. func WatchRepository() {
  201. }
  202. func UnWatchRepository() {
  203. }
  204. func ForkRepository(reposName string, userId int64) {
  205. }
  206. func RepoPath(userName, repoName string) string {
  207. return filepath.Join(UserPath(userName), repoName+".git")
  208. }
  209. // DeleteRepository deletes a repository for a user or orgnaztion.
  210. func DeleteRepository(user *User, reposName string) (err error) {
  211. session := orm.NewSession()
  212. if _, err = session.Delete(&Repository{OwnerId: user.Id, Name: reposName}); err != nil {
  213. session.Rollback()
  214. return err
  215. }
  216. if _, err = session.Exec("update user set num_repos = num_repos - 1 where id = ?", user.Id); err != nil {
  217. session.Rollback()
  218. return err
  219. }
  220. if err = session.Commit(); err != nil {
  221. session.Rollback()
  222. return err
  223. }
  224. if err = os.RemoveAll(RepoPath(user.Name, reposName)); err != nil {
  225. // TODO: log and delete manully
  226. log.Error("delete repo %s/%s failed", user.Name, reposName)
  227. return err
  228. }
  229. return nil
  230. }