You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

repo.go 22 kB

11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816
  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. "os/exec"
  11. "path"
  12. "path/filepath"
  13. "strings"
  14. "time"
  15. "unicode/utf8"
  16. "github.com/Unknwon/cae/zip"
  17. "github.com/Unknwon/com"
  18. "github.com/gogits/git"
  19. "github.com/gogits/gogs/modules/base"
  20. "github.com/gogits/gogs/modules/log"
  21. )
  22. var (
  23. ErrRepoAlreadyExist = errors.New("Repository already exist")
  24. ErrRepoNotExist = errors.New("Repository does not exist")
  25. ErrRepoFileNotExist = errors.New("Target Repo file does not exist")
  26. ErrRepoNameIllegal = errors.New("Repository name contains illegal characters")
  27. ErrRepoFileNotLoaded = errors.New("repo file not loaded")
  28. ErrMirrorNotExist = errors.New("Mirror does not exist")
  29. )
  30. var (
  31. LanguageIgns, Licenses []string
  32. )
  33. func LoadRepoConfig() {
  34. LanguageIgns = strings.Split(base.Cfg.MustValue("repository", "LANG_IGNS"), "|")
  35. Licenses = strings.Split(base.Cfg.MustValue("repository", "LICENSES"), "|")
  36. }
  37. func NewRepoContext() {
  38. zip.Verbose = false
  39. // Check if server has basic git setting.
  40. stdout, stderr, err := com.ExecCmd("git", "config", "--get", "user.name")
  41. if strings.Contains(stderr, "fatal:") {
  42. fmt.Printf("repo.NewRepoContext(fail to get git user.name): %s", stderr)
  43. os.Exit(2)
  44. } else if err != nil || len(strings.TrimSpace(stdout)) == 0 {
  45. if _, stderr, err = com.ExecCmd("git", "config", "--global", "user.email", "gogitservice@gmail.com"); err != nil {
  46. fmt.Printf("repo.NewRepoContext(fail to set git user.email): %s", stderr)
  47. os.Exit(2)
  48. } else if _, stderr, err = com.ExecCmd("git", "config", "--global", "user.name", "Gogs"); err != nil {
  49. fmt.Printf("repo.NewRepoContext(fail to set git user.name): %s", stderr)
  50. os.Exit(2)
  51. }
  52. }
  53. }
  54. // Repository represents a git repository.
  55. type Repository struct {
  56. Id int64
  57. OwnerId int64 `xorm:"unique(s)"`
  58. Owner *User `xorm:"-"`
  59. ForkId int64
  60. LowerName string `xorm:"unique(s) index not null"`
  61. Name string `xorm:"index not null"`
  62. Description string
  63. Website string
  64. NumWatches int
  65. NumStars int
  66. NumForks int
  67. NumIssues int
  68. NumClosedIssues int
  69. NumOpenIssues int `xorm:"-"`
  70. NumTags int `xorm:"-"`
  71. IsPrivate bool
  72. IsMirror bool
  73. IsBare bool
  74. IsGoget bool
  75. DefaultBranch string
  76. Created time.Time `xorm:"created"`
  77. Updated time.Time `xorm:"updated"`
  78. }
  79. func (repo *Repository) GetOwner() (err error) {
  80. repo.Owner, err = GetUserById(repo.OwnerId)
  81. return err
  82. }
  83. // IsRepositoryExist returns true if the repository with given name under user has already existed.
  84. func IsRepositoryExist(user *User, repoName string) (bool, error) {
  85. repo := Repository{OwnerId: user.Id}
  86. has, err := orm.Where("lower_name = ?", strings.ToLower(repoName)).Get(&repo)
  87. if err != nil {
  88. return has, err
  89. } else if !has {
  90. return false, nil
  91. }
  92. return com.IsDir(RepoPath(user.Name, repoName)), nil
  93. }
  94. var (
  95. illegalEquals = []string{"raw", "install", "api", "avatar", "user", "help", "stars", "issues", "pulls", "commits", "repo", "template", "admin"}
  96. illegalSuffixs = []string{".git"}
  97. )
  98. // IsLegalName returns false if name contains illegal characters.
  99. func IsLegalName(repoName string) bool {
  100. repoName = strings.ToLower(repoName)
  101. for _, char := range illegalEquals {
  102. if repoName == char {
  103. return false
  104. }
  105. }
  106. for _, char := range illegalSuffixs {
  107. if strings.HasSuffix(repoName, char) {
  108. return false
  109. }
  110. }
  111. return true
  112. }
  113. // Mirror represents a mirror information of repository.
  114. type Mirror struct {
  115. Id int64
  116. RepoId int64
  117. RepoName string // <user name>/<repo name>
  118. Interval int // Hour.
  119. Updated time.Time `xorm:"UPDATED"`
  120. NextUpdate time.Time
  121. }
  122. func GetMirror(repoId int64) (*Mirror, error) {
  123. m := &Mirror{RepoId: repoId}
  124. has, err := orm.Get(m)
  125. if err != nil {
  126. return nil, err
  127. } else if !has {
  128. return nil, ErrMirrorNotExist
  129. }
  130. return m, nil
  131. }
  132. func UpdateMirror(m *Mirror) error {
  133. _, err := orm.Id(m.Id).Update(m)
  134. return err
  135. }
  136. // MirrorUpdate checks and updates mirror repositories.
  137. func MirrorUpdate() {
  138. if err := orm.Iterate(new(Mirror), func(idx int, bean interface{}) error {
  139. m := bean.(*Mirror)
  140. if m.NextUpdate.After(time.Now()) {
  141. return nil
  142. }
  143. repoPath := filepath.Join(base.RepoRootPath, m.RepoName+".git")
  144. _, stderr, err := com.ExecCmdDir(repoPath, "git", "remote", "update")
  145. if err != nil {
  146. return errors.New("git remote update: " + stderr)
  147. } else if err = git.UnpackRefs(repoPath); err != nil {
  148. return err
  149. }
  150. m.NextUpdate = time.Now().Add(time.Duration(m.Interval) * time.Hour)
  151. return UpdateMirror(m)
  152. }); err != nil {
  153. log.Error("repo.MirrorUpdate: %v", err)
  154. }
  155. }
  156. // MirrorRepository creates a mirror repository from source.
  157. func MirrorRepository(repoId int64, userName, repoName, repoPath, url string) error {
  158. _, stderr, err := com.ExecCmd("git", "clone", "--mirror", url, repoPath)
  159. if err != nil {
  160. return errors.New("git clone --mirror: " + stderr)
  161. }
  162. if _, err = orm.InsertOne(&Mirror{
  163. RepoId: repoId,
  164. RepoName: strings.ToLower(userName + "/" + repoName),
  165. Interval: 24,
  166. NextUpdate: time.Now().Add(24 * time.Hour),
  167. }); err != nil {
  168. return err
  169. }
  170. return git.UnpackRefs(repoPath)
  171. }
  172. // MigrateRepository migrates a existing repository from other project hosting.
  173. func MigrateRepository(user *User, name, desc string, private, mirror bool, url string) (*Repository, error) {
  174. repo, err := CreateRepository(user, name, desc, "", "", private, mirror, false)
  175. if err != nil {
  176. return nil, err
  177. }
  178. // Clone to temprory path and do the init commit.
  179. tmpDir := filepath.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().Nanosecond()))
  180. os.MkdirAll(tmpDir, os.ModePerm)
  181. repoPath := RepoPath(user.Name, name)
  182. repo.IsBare = false
  183. if mirror {
  184. if err = MirrorRepository(repo.Id, user.Name, repo.Name, repoPath, url); err != nil {
  185. return repo, err
  186. }
  187. repo.IsMirror = true
  188. return repo, UpdateRepository(repo)
  189. }
  190. // Clone from local repository.
  191. _, stderr, err := com.ExecCmd("git", "clone", repoPath, tmpDir)
  192. if err != nil {
  193. return repo, errors.New("git clone: " + stderr)
  194. }
  195. // Pull data from source.
  196. _, stderr, err = com.ExecCmdDir(tmpDir, "git", "pull", url)
  197. if err != nil {
  198. return repo, errors.New("git pull: " + stderr)
  199. }
  200. // Push data to local repository.
  201. if _, stderr, err = com.ExecCmdDir(tmpDir, "git", "push", "origin", "master"); err != nil {
  202. return repo, errors.New("git push: " + stderr)
  203. }
  204. return repo, UpdateRepository(repo)
  205. }
  206. // CreateRepository creates a repository for given user or orgnaziation.
  207. func CreateRepository(user *User, name, desc, lang, license string, private, mirror, initReadme bool) (*Repository, error) {
  208. if !IsLegalName(name) {
  209. return nil, ErrRepoNameIllegal
  210. }
  211. isExist, err := IsRepositoryExist(user, name)
  212. if err != nil {
  213. return nil, err
  214. } else if isExist {
  215. return nil, ErrRepoAlreadyExist
  216. }
  217. repo := &Repository{
  218. OwnerId: user.Id,
  219. Name: name,
  220. LowerName: strings.ToLower(name),
  221. Description: desc,
  222. IsPrivate: private,
  223. IsBare: lang == "" && license == "" && !initReadme,
  224. }
  225. if !repo.IsBare {
  226. repo.DefaultBranch = "master"
  227. }
  228. repoPath := RepoPath(user.Name, repo.Name)
  229. sess := orm.NewSession()
  230. defer sess.Close()
  231. sess.Begin()
  232. if _, err = sess.Insert(repo); err != nil {
  233. if err2 := os.RemoveAll(repoPath); err2 != nil {
  234. log.Error("repo.CreateRepository(repo): %v", err)
  235. return nil, errors.New(fmt.Sprintf(
  236. "delete repo directory %s/%s failed(1): %v", user.Name, repo.Name, err2))
  237. }
  238. sess.Rollback()
  239. return nil, err
  240. }
  241. mode := AU_WRITABLE
  242. if mirror {
  243. mode = AU_READABLE
  244. }
  245. access := Access{
  246. UserName: user.LowerName,
  247. RepoName: strings.ToLower(path.Join(user.Name, repo.Name)),
  248. Mode: mode,
  249. }
  250. if _, err = sess.Insert(&access); err != nil {
  251. sess.Rollback()
  252. if err2 := os.RemoveAll(repoPath); err2 != nil {
  253. log.Error("repo.CreateRepository(access): %v", err)
  254. return nil, errors.New(fmt.Sprintf(
  255. "delete repo directory %s/%s failed(2): %v", user.Name, repo.Name, err2))
  256. }
  257. return nil, err
  258. }
  259. rawSql := "UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?"
  260. if _, err = sess.Exec(rawSql, user.Id); err != nil {
  261. sess.Rollback()
  262. if err2 := os.RemoveAll(repoPath); err2 != nil {
  263. log.Error("repo.CreateRepository(repo count): %v", err)
  264. return nil, errors.New(fmt.Sprintf(
  265. "delete repo directory %s/%s failed(3): %v", user.Name, repo.Name, err2))
  266. }
  267. return nil, err
  268. }
  269. if err = sess.Commit(); err != nil {
  270. sess.Rollback()
  271. if err2 := os.RemoveAll(repoPath); err2 != nil {
  272. log.Error("repo.CreateRepository(commit): %v", err)
  273. return nil, errors.New(fmt.Sprintf(
  274. "delete repo directory %s/%s failed(3): %v", user.Name, repo.Name, err2))
  275. }
  276. return nil, err
  277. }
  278. if err = WatchRepo(user.Id, repo.Id, true); err != nil {
  279. log.Error("repo.CreateRepository(WatchRepo): %v", err)
  280. }
  281. if err = NewRepoAction(user, repo); err != nil {
  282. log.Error("repo.CreateRepository(NewRepoAction): %v", err)
  283. }
  284. // No need for init for mirror.
  285. if mirror {
  286. return repo, nil
  287. }
  288. if err = initRepository(repoPath, user, repo, initReadme, lang, license); err != nil {
  289. return nil, err
  290. }
  291. c := exec.Command("git", "update-server-info")
  292. c.Dir = repoPath
  293. if err = c.Run(); err != nil {
  294. log.Error("repo.CreateRepository(exec update-server-info): %v", err)
  295. }
  296. return repo, nil
  297. }
  298. // extractGitBareZip extracts git-bare.zip to repository path.
  299. func extractGitBareZip(repoPath string) error {
  300. z, err := zip.Open("conf/content/git-bare.zip")
  301. if err != nil {
  302. return err
  303. }
  304. defer z.Close()
  305. return z.ExtractTo(repoPath)
  306. }
  307. // initRepoCommit temporarily changes with work directory.
  308. func initRepoCommit(tmpPath string, sig *git.Signature) (err error) {
  309. var stderr string
  310. if _, stderr, err = com.ExecCmdDir(tmpPath, "git", "add", "--all"); err != nil {
  311. return errors.New("git add: " + stderr)
  312. }
  313. if _, stderr, err = com.ExecCmdDir(tmpPath, "git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
  314. "-m", "Init commit"); err != nil {
  315. return errors.New("git commit: " + stderr)
  316. }
  317. if _, stderr, err = com.ExecCmdDir(tmpPath, "git", "push", "origin", "master"); err != nil {
  318. return errors.New("git push: " + stderr)
  319. }
  320. return nil
  321. }
  322. func createHookUpdate(hookPath, content string) error {
  323. pu, err := os.OpenFile(hookPath, os.O_CREATE|os.O_WRONLY, 0777)
  324. if err != nil {
  325. return err
  326. }
  327. defer pu.Close()
  328. _, err = pu.WriteString(content)
  329. return err
  330. }
  331. // SetRepoEnvs sets environment variables for command update.
  332. func SetRepoEnvs(userId int64, userName, repoName, repoUserName string) {
  333. os.Setenv("userId", base.ToStr(userId))
  334. os.Setenv("userName", userName)
  335. os.Setenv("repoName", repoName)
  336. os.Setenv("repoUserName", repoUserName)
  337. }
  338. // InitRepository initializes README and .gitignore if needed.
  339. func initRepository(f string, user *User, repo *Repository, initReadme bool, repoLang, license string) error {
  340. repoPath := RepoPath(user.Name, repo.Name)
  341. // Create bare new repository.
  342. if err := extractGitBareZip(repoPath); err != nil {
  343. return err
  344. }
  345. rp := strings.NewReplacer("\\", "/", " ", "\\ ")
  346. // hook/post-update
  347. if err := createHookUpdate(filepath.Join(repoPath, "hooks", "update"),
  348. fmt.Sprintf("#!/usr/bin/env %s\n%s update $1 $2 $3\n", base.ScriptType,
  349. rp.Replace(appPath))); err != nil {
  350. return err
  351. }
  352. // Initialize repository according to user's choice.
  353. fileName := map[string]string{}
  354. if initReadme {
  355. fileName["readme"] = "README.md"
  356. }
  357. if repoLang != "" {
  358. fileName["gitign"] = ".gitignore"
  359. }
  360. if license != "" {
  361. fileName["license"] = "LICENSE"
  362. }
  363. // Clone to temprory path and do the init commit.
  364. tmpDir := filepath.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().Nanosecond()))
  365. os.MkdirAll(tmpDir, os.ModePerm)
  366. _, stderr, err := com.ExecCmd("git", "clone", repoPath, tmpDir)
  367. if err != nil {
  368. return errors.New("git clone: " + stderr)
  369. }
  370. // README
  371. if initReadme {
  372. defaultReadme := repo.Name + "\n" + strings.Repeat("=",
  373. utf8.RuneCountInString(repo.Name)) + "\n\n" + repo.Description
  374. if err := ioutil.WriteFile(filepath.Join(tmpDir, fileName["readme"]),
  375. []byte(defaultReadme), 0644); err != nil {
  376. return err
  377. }
  378. }
  379. // .gitignore
  380. if repoLang != "" {
  381. filePath := "conf/gitignore/" + repoLang
  382. if com.IsFile(filePath) {
  383. if err := com.Copy(filePath,
  384. filepath.Join(tmpDir, fileName["gitign"])); err != nil {
  385. return err
  386. }
  387. }
  388. }
  389. // LICENSE
  390. if license != "" {
  391. filePath := "conf/license/" + license
  392. if com.IsFile(filePath) {
  393. if err := com.Copy(filePath,
  394. filepath.Join(tmpDir, fileName["license"])); err != nil {
  395. return err
  396. }
  397. }
  398. }
  399. if len(fileName) == 0 {
  400. return nil
  401. }
  402. SetRepoEnvs(user.Id, user.Name, repo.Name, user.Name)
  403. // Apply changes and commit.
  404. return initRepoCommit(tmpDir, user.NewGitSig())
  405. }
  406. // GetRepositoriesWithUsers returns given number of repository objects with offset.
  407. // It also auto-gets corresponding users.
  408. func GetRepositoriesWithUsers(num, offset int) ([]*Repository, error) {
  409. repos := make([]*Repository, 0, num)
  410. if err := orm.Limit(num, offset).Asc("id").Find(&repos); err != nil {
  411. return nil, err
  412. }
  413. for _, repo := range repos {
  414. repo.Owner = &User{Id: repo.OwnerId}
  415. has, err := orm.Get(repo.Owner)
  416. if err != nil {
  417. return nil, err
  418. } else if !has {
  419. return nil, ErrUserNotExist
  420. }
  421. }
  422. return repos, nil
  423. }
  424. // RepoPath returns repository path by given user and repository name.
  425. func RepoPath(userName, repoName string) string {
  426. return filepath.Join(UserPath(userName), strings.ToLower(repoName)+".git")
  427. }
  428. // TransferOwnership transfers all corresponding setting from old user to new one.
  429. func TransferOwnership(user *User, newOwner string, repo *Repository) (err error) {
  430. newUser, err := GetUserByName(newOwner)
  431. if err != nil {
  432. return err
  433. }
  434. // Update accesses.
  435. accesses := make([]Access, 0, 10)
  436. if err = orm.Find(&accesses, &Access{RepoName: user.LowerName + "/" + repo.LowerName}); err != nil {
  437. return err
  438. }
  439. sess := orm.NewSession()
  440. defer sess.Close()
  441. if err = sess.Begin(); err != nil {
  442. return err
  443. }
  444. for i := range accesses {
  445. accesses[i].RepoName = newUser.LowerName + "/" + repo.LowerName
  446. if accesses[i].UserName == user.LowerName {
  447. accesses[i].UserName = newUser.LowerName
  448. }
  449. if err = UpdateAccessWithSession(sess, &accesses[i]); err != nil {
  450. return err
  451. }
  452. }
  453. // Update repository.
  454. repo.OwnerId = newUser.Id
  455. if _, err := sess.Id(repo.Id).Update(repo); err != nil {
  456. sess.Rollback()
  457. return err
  458. }
  459. // Update user repository number.
  460. rawSql := "UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?"
  461. if _, err = sess.Exec(rawSql, newUser.Id); err != nil {
  462. sess.Rollback()
  463. return err
  464. }
  465. rawSql = "UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?"
  466. if _, err = sess.Exec(rawSql, user.Id); err != nil {
  467. sess.Rollback()
  468. return err
  469. }
  470. // Add watch of new owner to repository.
  471. if !IsWatching(newUser.Id, repo.Id) {
  472. if err = WatchRepo(newUser.Id, repo.Id, true); err != nil {
  473. sess.Rollback()
  474. return err
  475. }
  476. }
  477. if err = TransferRepoAction(user, newUser, repo); err != nil {
  478. sess.Rollback()
  479. return err
  480. }
  481. // Change repository directory name.
  482. if err = os.Rename(RepoPath(user.Name, repo.Name), RepoPath(newUser.Name, repo.Name)); err != nil {
  483. sess.Rollback()
  484. return err
  485. }
  486. return sess.Commit()
  487. }
  488. // ChangeRepositoryName changes all corresponding setting from old repository name to new one.
  489. func ChangeRepositoryName(userName, oldRepoName, newRepoName string) (err error) {
  490. // Update accesses.
  491. accesses := make([]Access, 0, 10)
  492. if err = orm.Find(&accesses, &Access{RepoName: strings.ToLower(userName + "/" + oldRepoName)}); err != nil {
  493. return err
  494. }
  495. sess := orm.NewSession()
  496. defer sess.Close()
  497. if err = sess.Begin(); err != nil {
  498. return err
  499. }
  500. for i := range accesses {
  501. accesses[i].RepoName = userName + "/" + newRepoName
  502. if err = UpdateAccessWithSession(sess, &accesses[i]); err != nil {
  503. return err
  504. }
  505. }
  506. // Change repository directory name.
  507. if err = os.Rename(RepoPath(userName, oldRepoName), RepoPath(userName, newRepoName)); err != nil {
  508. sess.Rollback()
  509. return err
  510. }
  511. return sess.Commit()
  512. }
  513. func UpdateRepository(repo *Repository) error {
  514. repo.LowerName = strings.ToLower(repo.Name)
  515. if len(repo.Description) > 255 {
  516. repo.Description = repo.Description[:255]
  517. }
  518. if len(repo.Website) > 255 {
  519. repo.Website = repo.Website[:255]
  520. }
  521. _, err := orm.Id(repo.Id).AllCols().Update(repo)
  522. return err
  523. }
  524. // DeleteRepository deletes a repository for a user or orgnaztion.
  525. func DeleteRepository(userId, repoId int64, userName string) (err error) {
  526. repo := &Repository{Id: repoId, OwnerId: userId}
  527. has, err := orm.Get(repo)
  528. if err != nil {
  529. return err
  530. } else if !has {
  531. return ErrRepoNotExist
  532. }
  533. sess := orm.NewSession()
  534. defer sess.Close()
  535. if err = sess.Begin(); err != nil {
  536. return err
  537. }
  538. if _, err = sess.Delete(&Repository{Id: repoId}); err != nil {
  539. sess.Rollback()
  540. return err
  541. }
  542. if _, err := sess.Delete(&Access{RepoName: strings.ToLower(path.Join(userName, repo.Name))}); err != nil {
  543. sess.Rollback()
  544. return err
  545. }
  546. if _, err := sess.Delete(&Action{RepoId: repo.Id}); err != nil {
  547. sess.Rollback()
  548. return err
  549. }
  550. if _, err = sess.Delete(&Watch{RepoId: repoId}); err != nil {
  551. sess.Rollback()
  552. return err
  553. }
  554. if _, err = sess.Delete(&Mirror{RepoId: repoId}); err != nil {
  555. sess.Rollback()
  556. return err
  557. }
  558. rawSql := "UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?"
  559. if _, err = sess.Exec(rawSql, userId); err != nil {
  560. sess.Rollback()
  561. return err
  562. }
  563. if err = sess.Commit(); err != nil {
  564. sess.Rollback()
  565. return err
  566. }
  567. if err = os.RemoveAll(RepoPath(userName, repo.Name)); err != nil {
  568. // TODO: log and delete manully
  569. log.Error("delete repo %s/%s failed: %v", userName, repo.Name, err)
  570. return err
  571. }
  572. return nil
  573. }
  574. // GetRepositoryByName returns the repository by given name under user if exists.
  575. func GetRepositoryByName(userId int64, repoName string) (*Repository, error) {
  576. repo := &Repository{
  577. OwnerId: userId,
  578. LowerName: strings.ToLower(repoName),
  579. }
  580. has, err := orm.Get(repo)
  581. if err != nil {
  582. return nil, err
  583. } else if !has {
  584. return nil, ErrRepoNotExist
  585. }
  586. return repo, err
  587. }
  588. // GetRepositoryById returns the repository by given id if exists.
  589. func GetRepositoryById(id int64) (*Repository, error) {
  590. repo := &Repository{}
  591. has, err := orm.Id(id).Get(repo)
  592. if err != nil {
  593. return nil, err
  594. } else if !has {
  595. return nil, ErrRepoNotExist
  596. }
  597. return repo, nil
  598. }
  599. // GetRepositories returns the list of repositories of given user.
  600. func GetRepositories(user *User, private bool) ([]*Repository, error) {
  601. repos := make([]*Repository, 0, 10)
  602. sess := orm.Desc("updated")
  603. if !private {
  604. sess.Where("is_private=?", false)
  605. }
  606. err := sess.Find(&repos, &Repository{OwnerId: user.Id})
  607. return repos, err
  608. }
  609. // GetRecentUpdatedRepositories returns the list of repositories that are recently updated.
  610. func GetRecentUpdatedRepositories() (repos []*Repository, err error) {
  611. err = orm.Where("is_private=?", false).Limit(5).Desc("updated").Find(&repos)
  612. return repos, err
  613. }
  614. // GetRepositoryCount returns the total number of repositories of user.
  615. func GetRepositoryCount(user *User) (int64, error) {
  616. return orm.Count(&Repository{OwnerId: user.Id})
  617. }
  618. // GetCollaboratorNames returns a list of user name of repository's collaborators.
  619. func GetCollaboratorNames(repoName string) ([]string, error) {
  620. accesses := make([]*Access, 0, 10)
  621. if err := orm.Find(&accesses, &Access{RepoName: strings.ToLower(repoName)}); err != nil {
  622. return nil, err
  623. }
  624. names := make([]string, len(accesses))
  625. for i := range accesses {
  626. names[i] = accesses[i].UserName
  627. }
  628. return names, nil
  629. }
  630. // GetCollaborators returns a list of users of repository's collaborators.
  631. func GetCollaborators(repoName string) (us []*User, err error) {
  632. accesses := make([]*Access, 0, 10)
  633. if err = orm.Find(&accesses, &Access{RepoName: strings.ToLower(repoName)}); err != nil {
  634. return nil, err
  635. }
  636. us = make([]*User, len(accesses))
  637. for i := range accesses {
  638. us[i], err = GetUserByName(accesses[i].UserName)
  639. if err != nil {
  640. return nil, err
  641. }
  642. }
  643. return us, nil
  644. }
  645. // Watch is connection request for receiving repository notifycation.
  646. type Watch struct {
  647. Id int64
  648. UserId int64 `xorm:"UNIQUE(watch)"`
  649. RepoId int64 `xorm:"UNIQUE(watch)"`
  650. }
  651. // Watch or unwatch repository.
  652. func WatchRepo(uid, rid int64, watch bool) (err error) {
  653. if watch {
  654. if _, err = orm.Insert(&Watch{RepoId: rid, UserId: uid}); err != nil {
  655. return err
  656. }
  657. rawSql := "UPDATE `repository` SET num_watches = num_watches + 1 WHERE id = ?"
  658. _, err = orm.Exec(rawSql, rid)
  659. } else {
  660. if _, err = orm.Delete(&Watch{0, uid, rid}); err != nil {
  661. return err
  662. }
  663. rawSql := "UPDATE `repository` SET num_watches = num_watches - 1 WHERE id = ?"
  664. _, err = orm.Exec(rawSql, rid)
  665. }
  666. return err
  667. }
  668. // GetWatchers returns all watchers of given repository.
  669. func GetWatchers(rid int64) ([]*Watch, error) {
  670. watches := make([]*Watch, 0, 10)
  671. err := orm.Find(&watches, &Watch{RepoId: rid})
  672. return watches, err
  673. }
  674. // NotifyWatchers creates batch of actions for every watcher.
  675. func NotifyWatchers(act *Action) error {
  676. // Add feeds for user self and all watchers.
  677. watches, err := GetWatchers(act.RepoId)
  678. if err != nil {
  679. return errors.New("repo.NotifyWatchers(get watches): " + err.Error())
  680. }
  681. // Add feed for actioner.
  682. act.UserId = act.ActUserId
  683. if _, err = orm.InsertOne(act); err != nil {
  684. return errors.New("repo.NotifyWatchers(create action): " + err.Error())
  685. }
  686. for i := range watches {
  687. if act.ActUserId == watches[i].UserId {
  688. continue
  689. }
  690. act.Id = 0
  691. act.UserId = watches[i].UserId
  692. if _, err = orm.InsertOne(act); err != nil {
  693. return errors.New("repo.NotifyWatchers(create action): " + err.Error())
  694. }
  695. }
  696. return nil
  697. }
  698. // IsWatching checks if user has watched given repository.
  699. func IsWatching(uid, rid int64) bool {
  700. has, _ := orm.Get(&Watch{0, uid, rid})
  701. return has
  702. }
  703. func ForkRepository(repoName string, uid int64) {
  704. }