wiki.go 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181
  1. // Copyright 2015 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. "fmt"
  7. "io/ioutil"
  8. "net/url"
  9. "os"
  10. "path"
  11. "path/filepath"
  12. "strings"
  13. "github.com/Unknwon/com"
  14. "github.com/gogits/git-module"
  15. "github.com/gogits/gogs/modules/setting"
  16. "github.com/gogits/gogs/modules/sync"
  17. )
  18. var wikiWorkingPool = sync.NewExclusivePool()
  19. // ToWikiPageURL formats a string to corresponding wiki URL name.
  20. func ToWikiPageURL(name string) string {
  21. return url.QueryEscape(strings.Replace(name, " ", "-", -1))
  22. }
  23. // ToWikiPageName formats a URL back to corresponding wiki page name,
  24. // and removes leading characters './' to prevent changing files
  25. // that are not belong to wiki repository.
  26. func ToWikiPageName(urlString string) string {
  27. name, _ := url.QueryUnescape(strings.Replace(urlString, "-", " ", -1))
  28. return strings.Replace(strings.TrimLeft(name, "./"), "/", " ", -1)
  29. }
  30. // WikiCloneLink returns clone URLs of repository wiki.
  31. func (repo *Repository) WikiCloneLink() (cl *CloneLink) {
  32. return repo.cloneLink(true)
  33. }
  34. // WikiPath returns wiki data path by given user and repository name.
  35. func WikiPath(userName, repoName string) string {
  36. return filepath.Join(UserPath(userName), strings.ToLower(repoName)+".wiki.git")
  37. }
  38. func (repo *Repository) WikiPath() string {
  39. return WikiPath(repo.MustOwner().Name, repo.Name)
  40. }
  41. // HasWiki returns true if repository has wiki.
  42. func (repo *Repository) HasWiki() bool {
  43. return com.IsDir(repo.WikiPath())
  44. }
  45. // InitWiki initializes a wiki for repository,
  46. // it does nothing when repository already has wiki.
  47. func (repo *Repository) InitWiki() error {
  48. if repo.HasWiki() {
  49. return nil
  50. }
  51. if err := git.InitRepository(repo.WikiPath(), true); err != nil {
  52. return fmt.Errorf("InitRepository: %v", err)
  53. } else if err = createUpdateHook(repo.WikiPath()); err != nil {
  54. return fmt.Errorf("createUpdateHook: %v", err)
  55. }
  56. return nil
  57. }
  58. func (repo *Repository) LocalWikiPath() string {
  59. return path.Join(setting.AppDataPath, "tmp/local-wiki", com.ToStr(repo.ID))
  60. }
  61. // UpdateLocalWiki makes sure the local copy of repository wiki is up-to-date.
  62. func (repo *Repository) UpdateLocalWiki() error {
  63. // Don't pass branch name here because it fails to clone and
  64. // checkout to a specific branch when wiki is an empty repository.
  65. return UpdateLocalCopyBranch(repo.WikiPath(), repo.LocalWikiPath(), "")
  66. }
  67. func discardLocalWikiChanges(localPath string) error {
  68. return discardLocalRepoBranchChanges(localPath, "master")
  69. }
  70. // updateWikiPage adds new page to repository wiki.
  71. func (repo *Repository) updateWikiPage(doer *User, oldTitle, title, content, message string, isNew bool) (err error) {
  72. wikiWorkingPool.CheckIn(com.ToStr(repo.ID))
  73. defer wikiWorkingPool.CheckOut(com.ToStr(repo.ID))
  74. if err = repo.InitWiki(); err != nil {
  75. return fmt.Errorf("InitWiki: %v", err)
  76. }
  77. localPath := repo.LocalWikiPath()
  78. if err = discardLocalWikiChanges(localPath); err != nil {
  79. return fmt.Errorf("discardLocalWikiChanges: %v", err)
  80. } else if err = repo.UpdateLocalWiki(); err != nil {
  81. return fmt.Errorf("UpdateLocalWiki: %v", err)
  82. }
  83. title = ToWikiPageName(title)
  84. filename := path.Join(localPath, title+".md")
  85. // If not a new file, show perform update not create.
  86. if isNew {
  87. if com.IsExist(filename) {
  88. return ErrWikiAlreadyExist{filename}
  89. }
  90. } else {
  91. os.Remove(path.Join(localPath, oldTitle+".md"))
  92. }
  93. // SECURITY: if new file is a symlink to non-exist critical file,
  94. // attack content can be written to the target file (e.g. authorized_keys2)
  95. // as a new page operation.
  96. // So we want to make sure the symlink is removed before write anything.
  97. // The new file we created will be in normal text format.
  98. os.Remove(filename)
  99. if err = ioutil.WriteFile(filename, []byte(content), 0666); err != nil {
  100. return fmt.Errorf("WriteFile: %v", err)
  101. }
  102. if len(message) == 0 {
  103. message = "Update page '" + title + "'"
  104. }
  105. if err = git.AddChanges(localPath, true); err != nil {
  106. return fmt.Errorf("AddChanges: %v", err)
  107. } else if err = git.CommitChanges(localPath, git.CommitChangesOptions{
  108. Committer: doer.NewGitSig(),
  109. Message: message,
  110. }); err != nil {
  111. return fmt.Errorf("CommitChanges: %v", err)
  112. } else if err = git.Push(localPath, "origin", "master"); err != nil {
  113. return fmt.Errorf("Push: %v", err)
  114. }
  115. return nil
  116. }
  117. func (repo *Repository) AddWikiPage(doer *User, title, content, message string) error {
  118. return repo.updateWikiPage(doer, "", title, content, message, true)
  119. }
  120. func (repo *Repository) EditWikiPage(doer *User, oldTitle, title, content, message string) error {
  121. return repo.updateWikiPage(doer, oldTitle, title, content, message, false)
  122. }
  123. func (repo *Repository) DeleteWikiPage(doer *User, title string) (err error) {
  124. wikiWorkingPool.CheckIn(com.ToStr(repo.ID))
  125. defer wikiWorkingPool.CheckOut(com.ToStr(repo.ID))
  126. localPath := repo.LocalWikiPath()
  127. if err = discardLocalWikiChanges(localPath); err != nil {
  128. return fmt.Errorf("discardLocalWikiChanges: %v", err)
  129. } else if err = repo.UpdateLocalWiki(); err != nil {
  130. return fmt.Errorf("UpdateLocalWiki: %v", err)
  131. }
  132. title = ToWikiPageName(title)
  133. filename := path.Join(localPath, title+".md")
  134. os.Remove(filename)
  135. message := "Delete page '" + title + "'"
  136. if err = git.AddChanges(localPath, true); err != nil {
  137. return fmt.Errorf("AddChanges: %v", err)
  138. } else if err = git.CommitChanges(localPath, git.CommitChangesOptions{
  139. Committer: doer.NewGitSig(),
  140. Message: message,
  141. }); err != nil {
  142. return fmt.Errorf("CommitChanges: %v", err)
  143. } else if err = git.Push(localPath, "origin", "master"); err != nil {
  144. return fmt.Errorf("Push: %v", err)
  145. }
  146. return nil
  147. }