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