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

1229 lines
32 KiB

10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
  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. "html"
  9. "html/template"
  10. "io/ioutil"
  11. "os"
  12. "os/exec"
  13. "path"
  14. "path/filepath"
  15. "regexp"
  16. "sort"
  17. "strings"
  18. "time"
  19. "unicode/utf8"
  20. "github.com/Unknwon/cae/zip"
  21. "github.com/Unknwon/com"
  22. "github.com/gogits/gogs/modules/git"
  23. "github.com/gogits/gogs/modules/log"
  24. "github.com/gogits/gogs/modules/process"
  25. "github.com/gogits/gogs/modules/setting"
  26. )
  27. const (
  28. TPL_UPDATE_HOOK = "#!/usr/bin/env %s\n%s update $1 $2 $3\n"
  29. )
  30. var (
  31. ErrRepoAlreadyExist = errors.New("Repository already exist")
  32. ErrRepoNotExist = errors.New("Repository does not exist")
  33. ErrRepoFileNotExist = errors.New("Repository file does not exist")
  34. ErrRepoNameIllegal = errors.New("Repository name contains illegal characters")
  35. ErrRepoFileNotLoaded = errors.New("Repository file not loaded")
  36. ErrMirrorNotExist = errors.New("Mirror does not exist")
  37. ErrInvalidReference = errors.New("Invalid reference specified")
  38. )
  39. var (
  40. Gitignores, Licenses []string
  41. )
  42. var (
  43. DescriptionPattern = regexp.MustCompile(`https?://\S+`)
  44. )
  45. func LoadRepoConfig() {
  46. // Load .gitignore and license files.
  47. types := []string{"gitignore", "license"}
  48. typeFiles := make([][]string, 2)
  49. for i, t := range types {
  50. files, err := com.StatDir(path.Join("conf", t))
  51. if err != nil {
  52. log.Fatal(4, "Fail to get %s files: %v", t, err)
  53. }
  54. customPath := path.Join(setting.CustomPath, "conf", t)
  55. if com.IsDir(customPath) {
  56. customFiles, err := com.StatDir(customPath)
  57. if err != nil {
  58. log.Fatal(4, "Fail to get custom %s files: %v", t, err)
  59. }
  60. for _, f := range customFiles {
  61. if !com.IsSliceContainsStr(files, f) {
  62. files = append(files, f)
  63. }
  64. }
  65. }
  66. typeFiles[i] = files
  67. }
  68. Gitignores = typeFiles[0]
  69. Licenses = typeFiles[1]
  70. sort.Strings(Gitignores)
  71. sort.Strings(Licenses)
  72. }
  73. func NewRepoContext() {
  74. zip.Verbose = false
  75. // Check Git installation.
  76. if _, err := exec.LookPath("git"); err != nil {
  77. log.Fatal(4, "Fail to test 'git' command: %v (forgotten install?)", err)
  78. }
  79. // Check Git version.
  80. ver, err := git.GetVersion()
  81. if err != nil {
  82. log.Fatal(4, "Fail to get Git version: %v", err)
  83. }
  84. reqVer, err := git.ParseVersion("1.7.1")
  85. if err != nil {
  86. log.Fatal(4, "Fail to parse required Git version: %v", err)
  87. }
  88. if ver.LessThan(reqVer) {
  89. log.Fatal(4, "Gogs requires Git version greater or equal to 1.7.1")
  90. }
  91. // Check if server has basic git setting and set if not.
  92. if stdout, stderr, err := process.Exec("NewRepoContext(get setting)", "git", "config", "--get", "user.name"); err != nil || strings.TrimSpace(stdout) == "" {
  93. // ExitError indicates user.name is not set
  94. if _, ok := err.(*exec.ExitError); ok || strings.TrimSpace(stdout) == "" {
  95. stndrdUserName := "Gogs"
  96. stndrdUserEmail := "gogitservice@gmail.com"
  97. if _, stderr, gerr := process.Exec("NewRepoContext(set name)", "git", "config", "--global", "user.name", stndrdUserName); gerr != nil {
  98. log.Fatal(4, "Fail to set git user.name(%s): %s", gerr, stderr)
  99. }
  100. if _, stderr, gerr := process.Exec("NewRepoContext(set email)", "git", "config", "--global", "user.email", stndrdUserEmail); gerr != nil {
  101. log.Fatal(4, "Fail to set git user.email(%s): %s", gerr, stderr)
  102. }
  103. log.Info("Git user.name and user.email set to %s <%s>", stndrdUserName, stndrdUserEmail)
  104. } else {
  105. log.Fatal(4, "Fail to get git user.name(%s): %s", err, stderr)
  106. }
  107. }
  108. // Set git some configurations.
  109. if _, stderr, err := process.Exec("NewRepoContext(git config --global core.quotepath false)",
  110. "git", "config", "--global", "core.quotepath", "false"); err != nil {
  111. log.Fatal(4, "Fail to execute 'git config --global core.quotepath false': %s", stderr)
  112. }
  113. }
  114. // Repository represents a git repository.
  115. type Repository struct {
  116. Id int64
  117. OwnerId int64 `xorm:"UNIQUE(s)"`
  118. Owner *User `xorm:"-"`
  119. ForkId int64
  120. LowerName string `xorm:"UNIQUE(s) INDEX NOT NULL"`
  121. Name string `xorm:"INDEX NOT NULL"`
  122. Description string
  123. Website string
  124. NumWatches int
  125. NumStars int
  126. NumForks int
  127. NumIssues int
  128. NumClosedIssues int
  129. NumOpenIssues int `xorm:"-"`
  130. NumPulls int
  131. NumClosedPulls int
  132. NumOpenPulls int `xorm:"-"`
  133. NumMilestones int `xorm:"NOT NULL DEFAULT 0"`
  134. NumClosedMilestones int `xorm:"NOT NULL DEFAULT 0"`
  135. NumOpenMilestones int `xorm:"-"`
  136. NumTags int `xorm:"-"`
  137. IsPrivate bool
  138. IsMirror bool
  139. *Mirror `xorm:"-"`
  140. IsFork bool `xorm:"NOT NULL DEFAULT false"`
  141. IsBare bool
  142. IsGoget bool
  143. DefaultBranch string
  144. Created time.Time `xorm:"CREATED"`
  145. Updated time.Time `xorm:"UPDATED"`
  146. }
  147. func (repo *Repository) GetOwner() (err error) {
  148. repo.Owner, err = GetUserById(repo.OwnerId)
  149. return err
  150. }
  151. func (repo *Repository) GetMirror() (err error) {
  152. repo.Mirror, err = GetMirror(repo.Id)
  153. return err
  154. }
  155. // DescriptionHtml does special handles to description and return HTML string.
  156. func (repo *Repository) DescriptionHtml() template.HTML {
  157. sanitize := func(s string) string {
  158. // TODO(nuss-justin): Improve sanitization. Strip all tags?
  159. ss := html.EscapeString(s)
  160. return fmt.Sprintf(`<a href="%s" target="_blank">%s</a>`, ss, ss)
  161. }
  162. return template.HTML(DescriptionPattern.ReplaceAllStringFunc(repo.Description, sanitize))
  163. }
  164. // IsRepositoryExist returns true if the repository with given name under user has already existed.
  165. func IsRepositoryExist(u *User, repoName string) (bool, error) {
  166. repo := Repository{OwnerId: u.Id}
  167. has, err := x.Where("lower_name = ?", strings.ToLower(repoName)).Get(&repo)
  168. if err != nil {
  169. return has, err
  170. } else if !has {
  171. return false, nil
  172. }
  173. return com.IsDir(RepoPath(u.Name, repoName)), nil
  174. }
  175. var (
  176. illegalEquals = []string{"debug", "raw", "install", "api", "avatar", "user", "org", "help", "stars", "issues", "pulls", "commits", "repo", "template", "admin", "new"}
  177. illegalSuffixs = []string{".git"}
  178. )
  179. // IsLegalName returns false if name contains illegal characters.
  180. func IsLegalName(repoName string) bool {
  181. repoName = strings.ToLower(repoName)
  182. for _, char := range illegalEquals {
  183. if repoName == char {
  184. return false
  185. }
  186. }
  187. for _, char := range illegalSuffixs {
  188. if strings.HasSuffix(repoName, char) {
  189. return false
  190. }
  191. }
  192. return true
  193. }
  194. // Mirror represents a mirror information of repository.
  195. type Mirror struct {
  196. Id int64
  197. RepoId int64
  198. RepoName string // <user name>/<repo name>
  199. Interval int // Hour.
  200. Updated time.Time `xorm:"UPDATED"`
  201. NextUpdate time.Time
  202. }
  203. func GetMirror(repoId int64) (*Mirror, error) {
  204. m := &Mirror{RepoId: repoId}
  205. has, err := x.Get(m)
  206. if err != nil {
  207. return nil, err
  208. } else if !has {
  209. return nil, ErrMirrorNotExist
  210. }
  211. return m, nil
  212. }
  213. func UpdateMirror(m *Mirror) error {
  214. _, err := x.Id(m.Id).Update(m)
  215. return err
  216. }
  217. // MirrorRepository creates a mirror repository from source.
  218. func MirrorRepository(repoId int64, userName, repoName, repoPath, url string) error {
  219. _, stderr, err := process.ExecTimeout(10*time.Minute,
  220. fmt.Sprintf("MirrorRepository: %s/%s", userName, repoName),
  221. "git", "clone", "--mirror", url, repoPath)
  222. if err != nil {
  223. return errors.New("git clone --mirror: " + stderr)
  224. }
  225. if _, err = x.InsertOne(&Mirror{
  226. RepoId: repoId,
  227. RepoName: strings.ToLower(userName + "/" + repoName),
  228. Interval: 24,
  229. NextUpdate: time.Now().Add(24 * time.Hour),
  230. }); err != nil {
  231. return err
  232. }
  233. return nil
  234. }
  235. // MirrorUpdate checks and updates mirror repositories.
  236. func MirrorUpdate() {
  237. if err := x.Iterate(new(Mirror), func(idx int, bean interface{}) error {
  238. m := bean.(*Mirror)
  239. if m.NextUpdate.After(time.Now()) {
  240. return nil
  241. }
  242. repoPath := filepath.Join(setting.RepoRootPath, m.RepoName+".git")
  243. if _, stderr, err := process.ExecDir(10*time.Minute,
  244. repoPath, fmt.Sprintf("MirrorUpdate: %s", repoPath),
  245. "git", "remote", "update"); err != nil {
  246. return errors.New("git remote update: " + stderr)
  247. }
  248. m.NextUpdate = time.Now().Add(time.Duration(m.Interval) * time.Hour)
  249. return UpdateMirror(m)
  250. }); err != nil {
  251. log.Error(4, "repo.MirrorUpdate: %v", err)
  252. }
  253. }
  254. // MigrateRepository migrates a existing repository from other project hosting.
  255. func MigrateRepository(u *User, name, desc string, private, mirror bool, url string) (*Repository, error) {
  256. repo, err := CreateRepository(u, name, desc, "", "", private, mirror, false)
  257. if err != nil {
  258. return nil, err
  259. }
  260. // Clone to temprory path and do the init commit.
  261. tmpDir := filepath.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().Nanosecond()))
  262. os.MkdirAll(tmpDir, os.ModePerm)
  263. repoPath := RepoPath(u.Name, name)
  264. if u.IsOrganization() {
  265. t, err := u.GetOwnerTeam()
  266. if err != nil {
  267. return nil, err
  268. }
  269. repo.NumWatches = t.NumMembers
  270. } else {
  271. repo.NumWatches = 1
  272. }
  273. repo.IsBare = false
  274. if mirror {
  275. if err = MirrorRepository(repo.Id, u.Name, repo.Name, repoPath, url); err != nil {
  276. return repo, err
  277. }
  278. repo.IsMirror = true
  279. return repo, UpdateRepository(repo)
  280. } else {
  281. os.RemoveAll(repoPath)
  282. }
  283. // this command could for both migrate and mirror
  284. _, stderr, err := process.ExecTimeout(10*time.Minute,
  285. fmt.Sprintf("MigrateRepository: %s", repoPath),
  286. "git", "clone", "--mirror", "--bare", url, repoPath)
  287. if err != nil {
  288. return repo, errors.New("git clone: " + stderr)
  289. }
  290. return repo, UpdateRepository(repo)
  291. }
  292. // extractGitBareZip extracts git-bare.zip to repository path.
  293. func extractGitBareZip(repoPath string) error {
  294. z, err := zip.Open(path.Join(setting.ConfRootPath, "content/git-bare.zip"))
  295. if err != nil {
  296. return err
  297. }
  298. defer z.Close()
  299. return z.ExtractTo(repoPath)
  300. }
  301. // initRepoCommit temporarily changes with work directory.
  302. func initRepoCommit(tmpPath string, sig *git.Signature) (err error) {
  303. var stderr string
  304. if _, stderr, err = process.ExecDir(-1,
  305. tmpPath, fmt.Sprintf("initRepoCommit(git add): %s", tmpPath),
  306. "git", "add", "--all"); err != nil {
  307. return errors.New("git add: " + stderr)
  308. }
  309. if _, stderr, err = process.ExecDir(-1,
  310. tmpPath, fmt.Sprintf("initRepoCommit(git commit): %s", tmpPath),
  311. "git", "commit", fmt.Sprintf("--author='%s <%s>'", sig.Name, sig.Email),
  312. "-m", "Init commit"); err != nil {
  313. return errors.New("git commit: " + stderr)
  314. }
  315. if _, stderr, err = process.ExecDir(-1,
  316. tmpPath, fmt.Sprintf("initRepoCommit(git push): %s", tmpPath),
  317. "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. // InitRepository initializes README and .gitignore if needed.
  332. func initRepository(f string, u *User, repo *Repository, initReadme bool, repoLang, license string) error {
  333. repoPath := RepoPath(u.Name, repo.Name)
  334. // Create bare new repository.
  335. if err := extractGitBareZip(repoPath); err != nil {
  336. return err
  337. }
  338. // hook/post-update
  339. if err := createHookUpdate(filepath.Join(repoPath, "hooks", "update"),
  340. fmt.Sprintf(TPL_UPDATE_HOOK, setting.ScriptType, "\""+appPath+"\"")); err != nil {
  341. return err
  342. }
  343. // Initialize repository according to user's choice.
  344. fileName := map[string]string{}
  345. if initReadme {
  346. fileName["readme"] = "README.md"
  347. }
  348. if repoLang != "" {
  349. fileName["gitign"] = ".gitignore"
  350. }
  351. if license != "" {
  352. fileName["license"] = "LICENSE"
  353. }
  354. // Clone to temprory path and do the init commit.
  355. tmpDir := filepath.Join(os.TempDir(), com.ToStr(time.Now().Nanosecond()))
  356. os.MkdirAll(tmpDir, os.ModePerm)
  357. _, stderr, err := process.Exec(
  358. fmt.Sprintf("initRepository(git clone): %s", repoPath),
  359. "git", "clone", repoPath, tmpDir)
  360. if err != nil {
  361. return errors.New("initRepository(git clone): " + stderr)
  362. }
  363. // README
  364. if initReadme {
  365. defaultReadme := repo.Name + "\n" + strings.Repeat("=",
  366. utf8.RuneCountInString(repo.Name)) + "\n\n" + repo.Description
  367. if err := ioutil.WriteFile(filepath.Join(tmpDir, fileName["readme"]),
  368. []byte(defaultReadme), 0644); err != nil {
  369. return err
  370. }
  371. }
  372. // .gitignore
  373. filePath := "conf/gitignore/" + repoLang
  374. if com.IsFile(filePath) {
  375. targetPath := path.Join(tmpDir, fileName["gitign"])
  376. if com.IsFile(filePath) {
  377. if err = com.Copy(filePath, targetPath); err != nil {
  378. return err
  379. }
  380. } else {
  381. // Check custom files.
  382. filePath = path.Join(setting.CustomPath, "conf/gitignore", repoLang)
  383. if com.IsFile(filePath) {
  384. if err := com.Copy(filePath, targetPath); err != nil {
  385. return err
  386. }
  387. }
  388. }
  389. } else {
  390. delete(fileName, "gitign")
  391. }
  392. // LICENSE
  393. filePath = "conf/license/" + license
  394. if com.IsFile(filePath) {
  395. targetPath := path.Join(tmpDir, fileName["license"])
  396. if com.IsFile(filePath) {
  397. if err = com.Copy(filePath, targetPath); err != nil {
  398. return err
  399. }
  400. } else {
  401. // Check custom files.
  402. filePath = path.Join(setting.CustomPath, "conf/license", license)
  403. if com.IsFile(filePath) {
  404. if err := com.Copy(filePath, targetPath); err != nil {
  405. return err
  406. }
  407. }
  408. }
  409. } else {
  410. delete(fileName, "license")
  411. }
  412. if len(fileName) == 0 {
  413. repo.IsBare = true
  414. repo.DefaultBranch = "master"
  415. return UpdateRepository(repo)
  416. }
  417. // Apply changes and commit.
  418. return initRepoCommit(tmpDir, u.NewGitSig())
  419. }
  420. // CreateRepository creates a repository for given user or organization.
  421. func CreateRepository(u *User, name, desc, lang, license string, private, mirror, initReadme bool) (*Repository, error) {
  422. if !IsLegalName(name) {
  423. return nil, ErrRepoNameIllegal
  424. }
  425. isExist, err := IsRepositoryExist(u, name)
  426. if err != nil {
  427. return nil, err
  428. } else if isExist {
  429. return nil, ErrRepoAlreadyExist
  430. }
  431. sess := x.NewSession()
  432. defer sess.Close()
  433. if err = sess.Begin(); err != nil {
  434. return nil, err
  435. }
  436. repo := &Repository{
  437. OwnerId: u.Id,
  438. Owner: u,
  439. Name: name,
  440. LowerName: strings.ToLower(name),
  441. Description: desc,
  442. IsPrivate: private,
  443. }
  444. if _, err = sess.Insert(repo); err != nil {
  445. sess.Rollback()
  446. return nil, err
  447. }
  448. var t *Team // Owner team.
  449. mode := WRITABLE
  450. if mirror {
  451. mode = READABLE
  452. }
  453. access := &Access{
  454. UserName: u.LowerName,
  455. RepoName: path.Join(u.LowerName, repo.LowerName),
  456. Mode: mode,
  457. }
  458. // Give access to all members in owner team.
  459. if u.IsOrganization() {
  460. t, err = u.GetOwnerTeam()
  461. if err != nil {
  462. sess.Rollback()
  463. return nil, err
  464. }
  465. if err = t.GetMembers(); err != nil {
  466. sess.Rollback()
  467. return nil, err
  468. }
  469. for _, u := range t.Members {
  470. access.Id = 0
  471. access.UserName = u.LowerName
  472. if _, err = sess.Insert(access); err != nil {
  473. sess.Rollback()
  474. return nil, err
  475. }
  476. }
  477. } else {
  478. if _, err = sess.Insert(access); err != nil {
  479. sess.Rollback()
  480. return nil, err
  481. }
  482. }
  483. if _, err = sess.Exec(
  484. "UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?", u.Id); err != nil {
  485. sess.Rollback()
  486. return nil, err
  487. }
  488. // Update owner team info and count.
  489. if u.IsOrganization() {
  490. t.RepoIds += "$" + com.ToStr(repo.Id) + "|"
  491. t.NumRepos++
  492. if _, err = sess.Id(t.Id).AllCols().Update(t); err != nil {
  493. sess.Rollback()
  494. return nil, err
  495. }
  496. }
  497. if err = sess.Commit(); err != nil {
  498. return nil, err
  499. }
  500. if u.IsOrganization() {
  501. t, err := u.GetOwnerTeam()
  502. if err != nil {
  503. log.Error(4, "GetOwnerTeam: %v", err)
  504. } else {
  505. if err = t.GetMembers(); err != nil {
  506. log.Error(4, "GetMembers: %v", err)
  507. } else {
  508. for _, u := range t.Members {
  509. if err = WatchRepo(u.Id, repo.Id, true); err != nil {
  510. log.Error(4, "WatchRepo2: %v", err)
  511. }
  512. }
  513. }
  514. }
  515. } else {
  516. if err = WatchRepo(u.Id, repo.Id, true); err != nil {
  517. log.Error(4, "WatchRepo3: %v", err)
  518. }
  519. }
  520. if err = NewRepoAction(u, repo); err != nil {
  521. log.Error(4, "NewRepoAction: %v", err)
  522. }
  523. // No need for init mirror.
  524. if mirror {
  525. return repo, nil
  526. }
  527. repoPath := RepoPath(u.Name, repo.Name)
  528. if err = initRepository(repoPath, u, repo, initReadme, lang, license); err != nil {
  529. if err2 := os.RemoveAll(repoPath); err2 != nil {
  530. log.Error(4, "initRepository: %v", err)
  531. return nil, fmt.Errorf(
  532. "delete repo directory %s/%s failed(2): %v", u.Name, repo.Name, err2)
  533. }
  534. return nil, fmt.Errorf("initRepository: %v", err)
  535. }
  536. _, stderr, err := process.ExecDir(-1,
  537. repoPath, fmt.Sprintf("CreateRepository(git update-server-info): %s", repoPath),
  538. "git", "update-server-info")
  539. if err != nil {
  540. return nil, errors.New("CreateRepository(git update-server-info): " + stderr)
  541. }
  542. return repo, nil
  543. }
  544. // CountRepositories returns number of repositories.
  545. func CountRepositories() int64 {
  546. count, _ := x.Count(new(Repository))
  547. return count
  548. }
  549. // GetRepositoriesWithUsers returns given number of repository objects with offset.
  550. // It also auto-gets corresponding users.
  551. func GetRepositoriesWithUsers(num, offset int) ([]*Repository, error) {
  552. repos := make([]*Repository, 0, num)
  553. if err := x.Limit(num, offset).Asc("id").Find(&repos); err != nil {
  554. return nil, err
  555. }
  556. for _, repo := range repos {
  557. repo.Owner = &User{Id: repo.OwnerId}
  558. has, err := x.Get(repo.Owner)
  559. if err != nil {
  560. return nil, err
  561. } else if !has {
  562. return nil, ErrUserNotExist
  563. }
  564. }
  565. return repos, nil
  566. }
  567. // RepoPath returns repository path by given user and repository name.
  568. func RepoPath(userName, repoName string) string {
  569. return filepath.Join(UserPath(userName), strings.ToLower(repoName)+".git")
  570. }
  571. // TransferOwnership transfers all corresponding setting from old user to new one.
  572. func TransferOwnership(u *User, newOwner string, repo *Repository) error {
  573. newUser, err := GetUserByName(newOwner)
  574. if err != nil {
  575. return err
  576. }
  577. // Check if new owner has repository with same name.
  578. has, err := IsRepositoryExist(newUser, repo.Name)
  579. if err != nil {
  580. return err
  581. } else if has {
  582. return ErrRepoAlreadyExist
  583. }
  584. sess := x.NewSession()
  585. defer sess.Close()
  586. if err = sess.Begin(); err != nil {
  587. return err
  588. }
  589. owner := repo.Owner
  590. oldRepoLink := path.Join(owner.LowerName, repo.LowerName)
  591. // Delete all access first if current owner is an organization.
  592. if owner.IsOrganization() {
  593. if _, err = sess.Where("repo_name=?", oldRepoLink).Delete(new(Access)); err != nil {
  594. sess.Rollback()
  595. return fmt.Errorf("fail to delete current accesses: %v", err)
  596. }
  597. } else {
  598. // Delete current owner access.
  599. if _, err = sess.Where("repo_name=?", oldRepoLink).And("user_name=?", owner.LowerName).
  600. Delete(new(Access)); err != nil {
  601. sess.Rollback()
  602. return fmt.Errorf("fail to delete access(owner): %v", err)
  603. }
  604. // In case new owner has access.
  605. if _, err = sess.Where("repo_name=?", oldRepoLink).And("user_name=?", newUser.LowerName).
  606. Delete(new(Access)); err != nil {
  607. sess.Rollback()
  608. return fmt.Errorf("fail to delete access(new user): %v", err)
  609. }
  610. }
  611. // Change accesses to new repository path.
  612. if _, err = sess.Where("repo_name=?", oldRepoLink).
  613. Update(&Access{RepoName: path.Join(newUser.LowerName, repo.LowerName)}); err != nil {
  614. sess.Rollback()
  615. return fmt.Errorf("fail to update access(change reponame): %v", err)
  616. }
  617. // Update repository.
  618. repo.OwnerId = newUser.Id
  619. if _, err := sess.Id(repo.Id).Update(repo); err != nil {
  620. sess.Rollback()
  621. return err
  622. }
  623. // Update user repository number.
  624. if _, err = sess.Exec("UPDATE `user` SET num_repos = num_repos + 1 WHERE id = ?", newUser.Id); err != nil {
  625. sess.Rollback()
  626. return err
  627. }
  628. if _, err = sess.Exec("UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?", owner.Id); err != nil {
  629. sess.Rollback()
  630. return err
  631. }
  632. mode := WRITABLE
  633. if repo.IsMirror {
  634. mode = READABLE
  635. }
  636. // New owner is organization.
  637. if newUser.IsOrganization() {
  638. access := &Access{
  639. RepoName: path.Join(newUser.LowerName, repo.LowerName),
  640. Mode: mode,
  641. }
  642. // Give access to all members in owner team.
  643. t, err := newUser.GetOwnerTeam()
  644. if err != nil {
  645. sess.Rollback()
  646. return err
  647. }
  648. if err = t.GetMembers(); err != nil {
  649. sess.Rollback()
  650. return err
  651. }
  652. for _, u := range t.Members {
  653. access.Id = 0
  654. access.UserName = u.LowerName
  655. if _, err = sess.Insert(access); err != nil {
  656. sess.Rollback()
  657. return err
  658. }
  659. }
  660. // Update owner team info and count.
  661. t.RepoIds += "$" + com.ToStr(repo.Id) + "|"
  662. t.NumRepos++
  663. if _, err = sess.Id(t.Id).AllCols().Update(t); err != nil {
  664. sess.Rollback()
  665. return err
  666. }
  667. } else {
  668. access := &Access{
  669. RepoName: path.Join(newUser.LowerName, repo.LowerName),
  670. UserName: newUser.LowerName,
  671. Mode: mode,
  672. }
  673. if _, err = sess.Insert(access); err != nil {
  674. sess.Rollback()
  675. return fmt.Errorf("fail to insert access: %v", err)
  676. }
  677. }
  678. // Change repository directory name.
  679. if err = os.Rename(RepoPath(owner.Name, repo.Name), RepoPath(newUser.Name, repo.Name)); err != nil {
  680. sess.Rollback()
  681. return err
  682. }
  683. if err = sess.Commit(); err != nil {
  684. return err
  685. }
  686. if err = WatchRepo(newUser.Id, repo.Id, true); err != nil {
  687. log.Error(4, "WatchRepo", err)
  688. }
  689. if err = TransferRepoAction(u, newUser, repo); err != nil {
  690. return err
  691. }
  692. return nil
  693. }
  694. // ChangeRepositoryName changes all corresponding setting from old repository name to new one.
  695. func ChangeRepositoryName(userName, oldRepoName, newRepoName string) (err error) {
  696. if !IsLegalName(newRepoName) {
  697. return ErrRepoNameIllegal
  698. }
  699. // Update accesses.
  700. accesses := make([]Access, 0, 10)
  701. if err = x.Find(&accesses, &Access{RepoName: strings.ToLower(userName + "/" + oldRepoName)}); err != nil {
  702. return err
  703. }
  704. sess := x.NewSession()
  705. defer sess.Close()
  706. if err = sess.Begin(); err != nil {
  707. return err
  708. }
  709. for i := range accesses {
  710. accesses[i].RepoName = userName + "/" + newRepoName
  711. if err = UpdateAccessWithSession(sess, &accesses[i]); err != nil {
  712. return err
  713. }
  714. }
  715. // Change repository directory name.
  716. if err = os.Rename(RepoPath(userName, oldRepoName), RepoPath(userName, newRepoName)); err != nil {
  717. sess.Rollback()
  718. return err
  719. }
  720. return sess.Commit()
  721. }
  722. func UpdateRepository(repo *Repository) error {
  723. repo.LowerName = strings.ToLower(repo.Name)
  724. if len(repo.Description) > 255 {
  725. repo.Description = repo.Description[:255]
  726. }
  727. if len(repo.Website) > 255 {
  728. repo.Website = repo.Website[:255]
  729. }
  730. _, err := x.Id(repo.Id).AllCols().Update(repo)
  731. return err
  732. }
  733. // DeleteRepository deletes a repository for a user or orgnaztion.
  734. func DeleteRepository(uid, repoId int64, userName string) error {
  735. repo := &Repository{Id: repoId, OwnerId: uid}
  736. has, err := x.Get(repo)
  737. if err != nil {
  738. return err
  739. } else if !has {
  740. return ErrRepoNotExist
  741. }
  742. // In case is a organization.
  743. org, err := GetUserById(uid)
  744. if err != nil {
  745. return err
  746. }
  747. if org.IsOrganization() {
  748. if err = org.GetTeams(); err != nil {
  749. return err
  750. }
  751. }
  752. sess := x.NewSession()
  753. defer sess.Close()
  754. if err = sess.Begin(); err != nil {
  755. return err
  756. }
  757. if _, err = sess.Delete(&Repository{Id: repoId}); err != nil {
  758. sess.Rollback()
  759. return err
  760. }
  761. // Delete all access.
  762. if _, err := sess.Delete(&Access{RepoName: strings.ToLower(path.Join(userName, repo.Name))}); err != nil {
  763. sess.Rollback()
  764. return err
  765. }
  766. if org.IsOrganization() {
  767. idStr := "$" + com.ToStr(repoId) + "|"
  768. for _, t := range org.Teams {
  769. if !strings.Contains(t.RepoIds, idStr) {
  770. continue
  771. }
  772. t.NumRepos--
  773. t.RepoIds = strings.Replace(t.RepoIds, idStr, "", 1)
  774. if _, err = sess.Id(t.Id).AllCols().Update(t); err != nil {
  775. sess.Rollback()
  776. return err
  777. }
  778. }
  779. }
  780. if _, err := sess.Delete(&Action{RepoId: repo.Id}); err != nil {
  781. sess.Rollback()
  782. return err
  783. }
  784. if _, err = sess.Delete(&Watch{RepoId: repoId}); err != nil {
  785. sess.Rollback()
  786. return err
  787. }
  788. if _, err = sess.Delete(&Mirror{RepoId: repoId}); err != nil {
  789. sess.Rollback()
  790. return err
  791. }
  792. if _, err = sess.Delete(&IssueUser{RepoId: repoId}); err != nil {
  793. sess.Rollback()
  794. return err
  795. }
  796. if _, err = sess.Delete(&Milestone{RepoId: repoId}); err != nil {
  797. sess.Rollback()
  798. return err
  799. }
  800. if _, err = sess.Delete(&Release{RepoId: repoId}); err != nil {
  801. sess.Rollback()
  802. return err
  803. }
  804. // Delete comments.
  805. if err = x.Iterate(&Issue{RepoId: repoId}, func(idx int, bean interface{}) error {
  806. issue := bean.(*Issue)
  807. if _, err = sess.Delete(&Comment{IssueId: issue.Id}); err != nil {
  808. sess.Rollback()
  809. return err
  810. }
  811. return nil
  812. }); err != nil {
  813. sess.Rollback()
  814. return err
  815. }
  816. if _, err = sess.Delete(&Issue{RepoId: repoId}); err != nil {
  817. sess.Rollback()
  818. return err
  819. }
  820. if _, err = sess.Exec("UPDATE `user` SET num_repos = num_repos - 1 WHERE id = ?", uid); err != nil {
  821. sess.Rollback()
  822. return err
  823. }
  824. if err = os.RemoveAll(RepoPath(userName, repo.Name)); err != nil {
  825. sess.Rollback()
  826. return err
  827. }
  828. return sess.Commit()
  829. }
  830. // GetRepositoryByRef returns a Repository specified by a GFM reference.
  831. // See https://help.github.com/articles/writing-on-github#references for more information on the syntax.
  832. func GetRepositoryByRef(ref string) (*Repository, error) {
  833. n := strings.IndexByte(ref, byte('/'))
  834. if n < 2 {
  835. return nil, ErrInvalidReference
  836. }
  837. userName, repoName := ref[:n], ref[n+1:]
  838. user, err := GetUserByName(userName)
  839. if err != nil {
  840. return nil, err
  841. }
  842. return GetRepositoryByName(user.Id, repoName)
  843. }
  844. // GetRepositoryByName returns the repository by given name under user if exists.
  845. func GetRepositoryByName(uid int64, repoName string) (*Repository, error) {
  846. repo := &Repository{
  847. OwnerId: uid,
  848. LowerName: strings.ToLower(repoName),
  849. }
  850. has, err := x.Get(repo)
  851. if err != nil {
  852. return nil, err
  853. } else if !has {
  854. return nil, ErrRepoNotExist
  855. }
  856. return repo, err
  857. }
  858. // GetRepositoryById returns the repository by given id if exists.
  859. func GetRepositoryById(id int64) (*Repository, error) {
  860. repo := &Repository{}
  861. has, err := x.Id(id).Get(repo)
  862. if err != nil {
  863. return nil, err
  864. } else if !has {
  865. return nil, ErrRepoNotExist
  866. }
  867. return repo, nil
  868. }
  869. // GetRepositories returns a list of repositories of given user.
  870. func GetRepositories(uid int64, private bool) ([]*Repository, error) {
  871. repos := make([]*Repository, 0, 10)
  872. sess := x.Desc("updated")
  873. if !private {
  874. sess.Where("is_private=?", false)
  875. }
  876. err := sess.Find(&repos, &Repository{OwnerId: uid})
  877. return repos, err
  878. }
  879. // GetRecentUpdatedRepositories returns the list of repositories that are recently updated.
  880. func GetRecentUpdatedRepositories(num int) (repos []*Repository, err error) {
  881. err = x.Where("is_private=?", false).Limit(num).Desc("updated").Find(&repos)
  882. return repos, err
  883. }
  884. // GetRepositoryCount returns the total number of repositories of user.
  885. func GetRepositoryCount(user *User) (int64, error) {
  886. return x.Count(&Repository{OwnerId: user.Id})
  887. }
  888. // GetCollaboratorNames returns a list of user name of repository's collaborators.
  889. func GetCollaboratorNames(repoName string) ([]string, error) {
  890. accesses := make([]*Access, 0, 10)
  891. if err := x.Find(&accesses, &Access{RepoName: strings.ToLower(repoName)}); err != nil {
  892. return nil, err
  893. }
  894. names := make([]string, len(accesses))
  895. for i := range accesses {
  896. names[i] = accesses[i].UserName
  897. }
  898. return names, nil
  899. }
  900. // GetCollaborativeRepos returns a list of repositories that user is collaborator.
  901. func GetCollaborativeRepos(uname string) ([]*Repository, error) {
  902. uname = strings.ToLower(uname)
  903. accesses := make([]*Access, 0, 10)
  904. if err := x.Find(&accesses, &Access{UserName: uname}); err != nil {
  905. return nil, err
  906. }
  907. repos := make([]*Repository, 0, 10)
  908. for _, access := range accesses {
  909. infos := strings.Split(access.RepoName, "/")
  910. if infos[0] == uname {
  911. continue
  912. }
  913. u, err := GetUserByName(infos[0])
  914. if err != nil {
  915. return nil, err
  916. }
  917. repo, err := GetRepositoryByName(u.Id, infos[1])
  918. if err != nil {
  919. return nil, err
  920. }
  921. repo.Owner = u
  922. repos = append(repos, repo)
  923. }
  924. return repos, nil
  925. }
  926. // GetCollaborators returns a list of users of repository's collaborators.
  927. func GetCollaborators(repoName string) (us []*User, err error) {
  928. accesses := make([]*Access, 0, 10)
  929. if err = x.Find(&accesses, &Access{RepoName: strings.ToLower(repoName)}); err != nil {
  930. return nil, err
  931. }
  932. us = make([]*User, len(accesses))
  933. for i := range accesses {
  934. us[i], err = GetUserByName(accesses[i].UserName)
  935. if err != nil {
  936. return nil, err
  937. }
  938. }
  939. return us, nil
  940. }
  941. type SearchOption struct {
  942. Keyword string
  943. Uid int64
  944. Limit int
  945. }
  946. // SearchRepositoryByName returns given number of repositories whose name contains keyword.
  947. func SearchRepositoryByName(opt SearchOption) (repos []*Repository, err error) {
  948. // Prevent SQL inject.
  949. opt.Keyword = strings.TrimSpace(opt.Keyword)
  950. if len(opt.Keyword) == 0 {
  951. return repos, nil
  952. }
  953. opt.Keyword = strings.Split(opt.Keyword, " ")[0]
  954. if len(opt.Keyword) == 0 {
  955. return repos, nil
  956. }
  957. opt.Keyword = strings.ToLower(opt.Keyword)
  958. repos = make([]*Repository, 0, opt.Limit)
  959. // Append conditions.
  960. sess := x.Limit(opt.Limit)
  961. if opt.Uid > 0 {
  962. sess.Where("owner_id=?", opt.Uid)
  963. }
  964. sess.And("lower_name like '%" + opt.Keyword + "%'").Find(&repos)
  965. return repos, err
  966. }
  967. // __ __ __ .__
  968. // / \ / \_____ _/ |_ ____ | |__
  969. // \ \/\/ /\__ \\ __\/ ___\| | \
  970. // \ / / __ \| | \ \___| Y \
  971. // \__/\ / (____ /__| \___ >___| /
  972. // \/ \/ \/ \/
  973. // Watch is connection request for receiving repository notifycation.
  974. type Watch struct {
  975. Id int64
  976. UserId int64 `xorm:"UNIQUE(watch)"`
  977. RepoId int64 `xorm:"UNIQUE(watch)"`
  978. }
  979. // Watch or unwatch repository.
  980. func WatchRepo(uid, repoId int64, watch bool) (err error) {
  981. if watch {
  982. if IsWatching(uid, repoId) {
  983. return nil
  984. }
  985. if _, err = x.Insert(&Watch{RepoId: repoId, UserId: uid}); err != nil {
  986. return err
  987. }
  988. _, err = x.Exec("UPDATE `repository` SET num_watches = num_watches + 1 WHERE id = ?", repoId)
  989. } else {
  990. if !IsWatching(uid, repoId) {
  991. return nil
  992. }
  993. if _, err = x.Delete(&Watch{0, uid, repoId}); err != nil {
  994. return err
  995. }
  996. _, err = x.Exec("UPDATE `repository` SET num_watches = num_watches - 1 WHERE id = ?", repoId)
  997. }
  998. return err
  999. }
  1000. // IsWatching checks if user has watched given repository.
  1001. func IsWatching(uid, rid int64) bool {
  1002. has, _ := x.Get(&Watch{0, uid, rid})
  1003. return has
  1004. }
  1005. // GetWatchers returns all watchers of given repository.
  1006. func GetWatchers(rid int64) ([]*Watch, error) {
  1007. watches := make([]*Watch, 0, 10)
  1008. err := x.Find(&watches, &Watch{RepoId: rid})
  1009. return watches, err
  1010. }
  1011. // NotifyWatchers creates batch of actions for every watcher.
  1012. func NotifyWatchers(act *Action) error {
  1013. // Add feeds for user self and all watchers.
  1014. watches, err := GetWatchers(act.RepoId)
  1015. if err != nil {
  1016. return errors.New("repo.NotifyWatchers(get watches): " + err.Error())
  1017. }
  1018. // Add feed for actioner.
  1019. act.UserId = act.ActUserId
  1020. if _, err = x.InsertOne(act); err != nil {
  1021. return errors.New("repo.NotifyWatchers(create action): " + err.Error())
  1022. }
  1023. for i := range watches {
  1024. if act.ActUserId == watches[i].UserId {
  1025. continue
  1026. }
  1027. act.Id = 0
  1028. act.UserId = watches[i].UserId
  1029. if _, err = x.InsertOne(act); err != nil {
  1030. return errors.New("repo.NotifyWatchers(create action): " + err.Error())
  1031. }
  1032. }
  1033. return nil
  1034. }
  1035. // _________ __
  1036. // / _____// |______ _______
  1037. // \_____ \\ __\__ \\_ __ \
  1038. // / \| | / __ \| | \/
  1039. // /_______ /|__| (____ /__|
  1040. // \/ \/
  1041. type Star struct {
  1042. Id int64
  1043. Uid int64 `xorm:"UNIQUE(s)"`
  1044. RepoId int64 `xorm:"UNIQUE(s)"`
  1045. }
  1046. // Star or unstar repository.
  1047. func StarRepo(uid, repoId int64, star bool) (err error) {
  1048. if star {
  1049. if IsStaring(uid, repoId) {
  1050. return nil
  1051. }
  1052. if _, err = x.Insert(&Star{Uid: uid, RepoId: repoId}); err != nil {
  1053. return err
  1054. } else if _, err = x.Exec("UPDATE `repository` SET num_stars = num_stars + 1 WHERE id = ?", repoId); err != nil {
  1055. return err
  1056. }
  1057. _, err = x.Exec("UPDATE `user` SET num_stars = num_stars + 1 WHERE id = ?", uid)
  1058. } else {
  1059. if !IsStaring(uid, repoId) {
  1060. return nil
  1061. }
  1062. if _, err = x.Delete(&Star{0, uid, repoId}); err != nil {
  1063. return err
  1064. } else if _, err = x.Exec("UPDATE `repository` SET num_stars = num_stars - 1 WHERE id = ?", repoId); err != nil {
  1065. return err
  1066. }
  1067. _, err = x.Exec("UPDATE `user` SET num_stars = num_stars - 1 WHERE id = ?", uid)
  1068. }
  1069. return err
  1070. }
  1071. // IsStaring checks if user has starred given repository.
  1072. func IsStaring(uid, repoId int64) bool {
  1073. has, _ := x.Get(&Star{0, uid, repoId})
  1074. return has
  1075. }
  1076. func ForkRepository(repoName string, uid int64) {
  1077. }