repo.go 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415
  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 repo
  5. import (
  6. "bytes"
  7. "encoding/base64"
  8. "errors"
  9. "fmt"
  10. "io/ioutil"
  11. "path"
  12. "path/filepath"
  13. "strings"
  14. "github.com/go-martini/martini"
  15. "github.com/gogits/git"
  16. "github.com/gogits/gogs/models"
  17. "github.com/gogits/gogs/modules/auth"
  18. "github.com/gogits/gogs/modules/base"
  19. "github.com/gogits/gogs/modules/log"
  20. "github.com/gogits/gogs/modules/middleware"
  21. )
  22. const (
  23. CREATE base.TplName = "repo/create"
  24. MIGRATE base.TplName = "repo/migrate"
  25. SINGLE base.TplName = "repo/single"
  26. )
  27. func Create(ctx *middleware.Context) {
  28. ctx.Data["Title"] = "Create repository"
  29. ctx.Data["PageIsNewRepo"] = true
  30. ctx.Data["LanguageIgns"] = models.LanguageIgns
  31. ctx.Data["Licenses"] = models.Licenses
  32. if err := ctx.User.GetOrganizations(); err != nil {
  33. ctx.Handle(500, "home.Dashboard(GetOrganizations)", err)
  34. return
  35. }
  36. ctx.Data["Orgs"] = ctx.User.Orgs
  37. ctx.HTML(200, CREATE)
  38. }
  39. func CreatePost(ctx *middleware.Context, form auth.CreateRepoForm) {
  40. ctx.Data["Title"] = "Create repository"
  41. ctx.Data["PageIsNewRepo"] = true
  42. ctx.Data["LanguageIgns"] = models.LanguageIgns
  43. ctx.Data["Licenses"] = models.Licenses
  44. if err := ctx.User.GetOrganizations(); err != nil {
  45. ctx.Handle(500, "home.CreatePost(GetOrganizations)", err)
  46. return
  47. }
  48. ctx.Data["Orgs"] = ctx.User.Orgs
  49. if ctx.HasError() {
  50. ctx.HTML(200, CREATE)
  51. return
  52. }
  53. u := ctx.User
  54. // Not equal means current user is an organization.
  55. if u.Id != form.Uid {
  56. var err error
  57. u, err = models.GetUserById(form.Uid)
  58. if err != nil {
  59. if err == models.ErrUserNotExist {
  60. ctx.Handle(404, "home.CreatePost(GetUserById)", err)
  61. } else {
  62. ctx.Handle(500, "home.CreatePost(GetUserById)", err)
  63. }
  64. return
  65. }
  66. }
  67. repo, err := models.CreateRepository(u, form.RepoName, form.Description,
  68. form.Language, form.License, form.Private, false, form.InitReadme)
  69. if err == nil {
  70. log.Trace("%s Repository created: %s/%s", ctx.Req.RequestURI, u.LowerName, form.RepoName)
  71. ctx.Redirect("/" + u.Name + "/" + form.RepoName)
  72. return
  73. } else if err == models.ErrRepoAlreadyExist {
  74. ctx.RenderWithErr("Repository name has already been used", CREATE, &form)
  75. return
  76. } else if err == models.ErrRepoNameIllegal {
  77. ctx.RenderWithErr(models.ErrRepoNameIllegal.Error(), CREATE, &form)
  78. return
  79. }
  80. if repo != nil {
  81. if errDelete := models.DeleteRepository(u.Id, repo.Id, u.Name); errDelete != nil {
  82. log.Error("repo.CreatePost(DeleteRepository): %v", errDelete)
  83. }
  84. }
  85. ctx.Handle(500, "repo.CreatePost(CreateRepository)", err)
  86. }
  87. func Migrate(ctx *middleware.Context) {
  88. ctx.Data["Title"] = "Migrate repository"
  89. ctx.Data["PageIsNewRepo"] = true
  90. if err := ctx.User.GetOrganizations(); err != nil {
  91. ctx.Handle(500, "home.Migrate(GetOrganizations)", err)
  92. return
  93. }
  94. ctx.Data["Orgs"] = ctx.User.Orgs
  95. ctx.HTML(200, MIGRATE)
  96. }
  97. func MigratePost(ctx *middleware.Context, form auth.MigrateRepoForm) {
  98. ctx.Data["Title"] = "Migrate repository"
  99. ctx.Data["PageIsNewRepo"] = true
  100. if err := ctx.User.GetOrganizations(); err != nil {
  101. ctx.Handle(500, "home.MigratePost(GetOrganizations)", err)
  102. return
  103. }
  104. ctx.Data["Orgs"] = ctx.User.Orgs
  105. if ctx.HasError() {
  106. ctx.HTML(200, MIGRATE)
  107. return
  108. }
  109. u := ctx.User
  110. // Not equal means current user is an organization.
  111. if u.Id != form.Uid {
  112. var err error
  113. u, err = models.GetUserById(form.Uid)
  114. if err != nil {
  115. if err == models.ErrUserNotExist {
  116. ctx.Handle(404, "home.MigratePost(GetUserById)", err)
  117. } else {
  118. ctx.Handle(500, "home.MigratePost(GetUserById)", err)
  119. }
  120. return
  121. }
  122. }
  123. authStr := strings.Replace(fmt.Sprintf("://%s:%s",
  124. form.AuthUserName, form.AuthPasswd), "@", "%40", -1)
  125. url := strings.Replace(form.Url, "://", authStr+"@", 1)
  126. repo, err := models.MigrateRepository(u, form.RepoName, form.Description, form.Private,
  127. form.Mirror, url)
  128. if err == nil {
  129. log.Trace("%s Repository migrated: %s/%s", ctx.Req.RequestURI, u.LowerName, form.RepoName)
  130. ctx.Redirect("/" + u.Name + "/" + form.RepoName)
  131. return
  132. } else if err == models.ErrRepoAlreadyExist {
  133. ctx.RenderWithErr("Repository name has already been used", MIGRATE, &form)
  134. return
  135. } else if err == models.ErrRepoNameIllegal {
  136. ctx.RenderWithErr(models.ErrRepoNameIllegal.Error(), MIGRATE, &form)
  137. return
  138. }
  139. if repo != nil {
  140. if errDelete := models.DeleteRepository(u.Id, repo.Id, u.Name); errDelete != nil {
  141. log.Error("repo.MigratePost(DeleteRepository): %v", errDelete)
  142. }
  143. }
  144. if strings.Contains(err.Error(), "Authentication failed") {
  145. ctx.RenderWithErr(err.Error(), MIGRATE, &form)
  146. return
  147. }
  148. ctx.Handle(500, "repo.Migrate(MigrateRepository)", err)
  149. }
  150. func Single(ctx *middleware.Context, params martini.Params) {
  151. branchName := ctx.Repo.BranchName
  152. userName := ctx.Repo.Owner.Name
  153. repoName := ctx.Repo.Repository.Name
  154. repoLink := ctx.Repo.RepoLink
  155. branchLink := ctx.Repo.RepoLink + "/src/" + branchName
  156. rawLink := ctx.Repo.RepoLink + "/raw/" + branchName
  157. // Get tree path
  158. treename := params["_1"]
  159. if len(treename) > 0 && treename[len(treename)-1] == '/' {
  160. ctx.Redirect(repoLink + "/src/" + branchName + "/" + treename[:len(treename)-1])
  161. return
  162. }
  163. ctx.Data["IsRepoToolbarSource"] = true
  164. isViewBranch := ctx.Repo.IsBranch
  165. ctx.Data["IsViewBranch"] = isViewBranch
  166. treePath := treename
  167. if len(treePath) != 0 {
  168. treePath = treePath + "/"
  169. }
  170. entry, err := ctx.Repo.Commit.GetTreeEntryByPath(treename)
  171. if err != nil && err != git.ErrNotExist {
  172. ctx.Handle(404, "repo.Single(GetTreeEntryByPath)", err)
  173. return
  174. }
  175. if len(treename) != 0 && entry == nil {
  176. ctx.Handle(404, "repo.Single", nil)
  177. return
  178. }
  179. if entry != nil && !entry.IsDir() {
  180. blob := entry.Blob()
  181. if dataRc, err := blob.Data(); err != nil {
  182. ctx.Handle(404, "repo.Single(blob.Data)", err)
  183. } else {
  184. ctx.Data["FileSize"] = blob.Size()
  185. ctx.Data["IsFile"] = true
  186. ctx.Data["FileName"] = blob.Name()
  187. ext := path.Ext(blob.Name())
  188. if len(ext) > 0 {
  189. ext = ext[1:]
  190. }
  191. ctx.Data["FileExt"] = ext
  192. ctx.Data["FileLink"] = rawLink + "/" + treename
  193. buf := make([]byte, 1024)
  194. n, _ := dataRc.Read(buf)
  195. if n > 0 {
  196. buf = buf[:n]
  197. }
  198. defer func() {
  199. dataRc.Close()
  200. }()
  201. _, isTextFile := base.IsTextFile(buf)
  202. _, isImageFile := base.IsImageFile(buf)
  203. ctx.Data["FileIsText"] = isTextFile
  204. switch {
  205. case isImageFile:
  206. ctx.Data["IsImageFile"] = true
  207. case isTextFile:
  208. d, _ := ioutil.ReadAll(dataRc)
  209. buf = append(buf, d...)
  210. readmeExist := base.IsMarkdownFile(blob.Name()) || base.IsReadmeFile(blob.Name())
  211. ctx.Data["ReadmeExist"] = readmeExist
  212. if readmeExist {
  213. ctx.Data["FileContent"] = string(base.RenderMarkdown(buf, ""))
  214. } else {
  215. ctx.Data["FileContent"] = string(buf)
  216. }
  217. }
  218. }
  219. } else {
  220. // Directory and file list.
  221. tree, err := ctx.Repo.Commit.SubTree(treename)
  222. if err != nil {
  223. ctx.Handle(404, "repo.Single(SubTree)", err)
  224. return
  225. }
  226. entries := tree.ListEntries()
  227. entries.Sort()
  228. files := make([][]interface{}, 0, len(entries))
  229. for _, te := range entries {
  230. c, err := ctx.Repo.Commit.GetCommitOfRelPath(filepath.Join(treePath, te.Name()))
  231. if err != nil {
  232. ctx.Handle(404, "repo.Single(SubTree)", err)
  233. return
  234. }
  235. files = append(files, []interface{}{te, c})
  236. }
  237. ctx.Data["Files"] = files
  238. var readmeFile *git.Blob
  239. for _, f := range entries {
  240. if f.IsDir() || !base.IsReadmeFile(f.Name()) {
  241. continue
  242. } else {
  243. readmeFile = f.Blob()
  244. break
  245. }
  246. }
  247. if readmeFile != nil {
  248. ctx.Data["ReadmeInSingle"] = true
  249. ctx.Data["ReadmeExist"] = true
  250. if dataRc, err := readmeFile.Data(); err != nil {
  251. ctx.Handle(404, "repo.Single(readmeFile.LookupBlob)", err)
  252. return
  253. } else {
  254. buf := make([]byte, 1024)
  255. n, _ := dataRc.Read(buf)
  256. if n > 0 {
  257. buf = buf[:n]
  258. }
  259. defer func() {
  260. dataRc.Close()
  261. }()
  262. ctx.Data["FileSize"] = readmeFile.Size
  263. ctx.Data["FileLink"] = rawLink + "/" + treename
  264. _, isTextFile := base.IsTextFile(buf)
  265. ctx.Data["FileIsText"] = isTextFile
  266. ctx.Data["FileName"] = readmeFile.Name()
  267. if isTextFile {
  268. d, _ := ioutil.ReadAll(dataRc)
  269. buf = append(buf, d...)
  270. switch {
  271. case base.IsMarkdownFile(readmeFile.Name()):
  272. buf = base.RenderMarkdown(buf, branchLink)
  273. default:
  274. buf = bytes.Replace(buf, []byte("\n"), []byte(`<br>`), -1)
  275. }
  276. ctx.Data["FileContent"] = string(buf)
  277. }
  278. }
  279. }
  280. }
  281. ctx.Data["Username"] = userName
  282. ctx.Data["Reponame"] = repoName
  283. var treenames []string
  284. Paths := make([]string, 0)
  285. if len(treename) > 0 {
  286. treenames = strings.Split(treename, "/")
  287. for i, _ := range treenames {
  288. Paths = append(Paths, strings.Join(treenames[0:i+1], "/"))
  289. }
  290. ctx.Data["HasParentPath"] = true
  291. if len(Paths)-2 >= 0 {
  292. ctx.Data["ParentPath"] = "/" + Paths[len(Paths)-2]
  293. }
  294. }
  295. ctx.Data["LastCommit"] = ctx.Repo.Commit
  296. ctx.Data["Paths"] = Paths
  297. ctx.Data["TreeName"] = treename
  298. ctx.Data["Treenames"] = treenames
  299. ctx.Data["TreePath"] = treePath
  300. ctx.Data["BranchLink"] = branchLink
  301. ctx.HTML(200, SINGLE)
  302. }
  303. func basicEncode(username, password string) string {
  304. auth := username + ":" + password
  305. return base64.StdEncoding.EncodeToString([]byte(auth))
  306. }
  307. func basicDecode(encoded string) (user string, name string, err error) {
  308. var s []byte
  309. s, err = base64.StdEncoding.DecodeString(encoded)
  310. if err != nil {
  311. return
  312. }
  313. a := strings.Split(string(s), ":")
  314. if len(a) == 2 {
  315. user, name = a[0], a[1]
  316. } else {
  317. err = errors.New("decode failed")
  318. }
  319. return
  320. }
  321. func authRequired(ctx *middleware.Context) {
  322. ctx.ResponseWriter.Header().Set("WWW-Authenticate", "Basic realm=\".\"")
  323. ctx.Data["ErrorMsg"] = "no basic auth and digit auth"
  324. ctx.HTML(401, base.TplName("status/401"))
  325. }
  326. func Action(ctx *middleware.Context, params martini.Params) {
  327. var err error
  328. switch params["action"] {
  329. case "watch":
  330. err = models.WatchRepo(ctx.User.Id, ctx.Repo.Repository.Id, true)
  331. case "unwatch":
  332. err = models.WatchRepo(ctx.User.Id, ctx.Repo.Repository.Id, false)
  333. case "desc":
  334. if !ctx.Repo.IsOwner {
  335. ctx.Error(404)
  336. return
  337. }
  338. ctx.Repo.Repository.Description = ctx.Query("desc")
  339. ctx.Repo.Repository.Website = ctx.Query("site")
  340. err = models.UpdateRepository(ctx.Repo.Repository)
  341. }
  342. if err != nil {
  343. log.Error("repo.Action(%s): %v", params["action"], err)
  344. ctx.JSON(200, map[string]interface{}{
  345. "ok": false,
  346. "err": err.Error(),
  347. })
  348. return
  349. }
  350. ctx.JSON(200, map[string]interface{}{
  351. "ok": true,
  352. })
  353. }