Fork to maintain patches against the official gitea for https://code.ceondo.com https://github.com/go-gitea/gitea

repo.go 18KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Copyright 2017 The Gitea Authors. All rights reserved.
  3. // Use of this source code is governed by a MIT-style
  4. // license that can be found in the LICENSE file.
  5. package context
  6. import (
  7. "fmt"
  8. "io/ioutil"
  9. "path"
  10. "strings"
  11. "code.gitea.io/git"
  12. "code.gitea.io/gitea/models"
  13. "code.gitea.io/gitea/modules/setting"
  14. "github.com/Unknwon/com"
  15. "gopkg.in/editorconfig/editorconfig-core-go.v1"
  16. "gopkg.in/macaron.v1"
  17. )
  18. // PullRequest contains informations to make a pull request
  19. type PullRequest struct {
  20. BaseRepo *models.Repository
  21. Allowed bool
  22. SameRepo bool
  23. HeadInfo string // [<user>:]<branch>
  24. }
  25. // Repository contains information to operate a repository
  26. type Repository struct {
  27. AccessMode models.AccessMode
  28. IsWatching bool
  29. IsViewBranch bool
  30. IsViewTag bool
  31. IsViewCommit bool
  32. Repository *models.Repository
  33. Owner *models.User
  34. Commit *git.Commit
  35. Tag *git.Tag
  36. GitRepo *git.Repository
  37. BranchName string
  38. TagName string
  39. TreePath string
  40. CommitID string
  41. RepoLink string
  42. CloneLink models.CloneLink
  43. CommitsCount int64
  44. Mirror *models.Mirror
  45. PullRequest *PullRequest
  46. }
  47. // IsOwner returns true if current user is the owner of repository.
  48. func (r *Repository) IsOwner() bool {
  49. return r.AccessMode >= models.AccessModeOwner
  50. }
  51. // IsAdmin returns true if current user has admin or higher access of repository.
  52. func (r *Repository) IsAdmin() bool {
  53. return r.AccessMode >= models.AccessModeAdmin
  54. }
  55. // IsWriter returns true if current user has write or higher access of repository.
  56. func (r *Repository) IsWriter() bool {
  57. return r.AccessMode >= models.AccessModeWrite
  58. }
  59. // HasAccess returns true if the current user has at least read access for this repository
  60. func (r *Repository) HasAccess() bool {
  61. return r.AccessMode >= models.AccessModeRead
  62. }
  63. // CanEnableEditor returns true if repository is editable and user has proper access level.
  64. func (r *Repository) CanEnableEditor() bool {
  65. return r.Repository.CanEnableEditor() && r.IsViewBranch && r.IsWriter()
  66. }
  67. // CanCommitToBranch returns true if repository is editable and user has proper access level
  68. // and branch is not protected
  69. func (r *Repository) CanCommitToBranch(doer *models.User) (bool, error) {
  70. protectedBranch, err := r.Repository.IsProtectedBranch(r.BranchName, doer)
  71. if err != nil {
  72. return false, err
  73. }
  74. return r.CanEnableEditor() && !protectedBranch, nil
  75. }
  76. // CanUseTimetracker returns whether or not a user can use the timetracker.
  77. func (r *Repository) CanUseTimetracker(issue *models.Issue, user *models.User) bool {
  78. // Checking for following:
  79. // 1. Is timetracker enabled
  80. // 2. Is the user a contributor, admin, poster or assignee and do the repository policies require this?
  81. return r.Repository.IsTimetrackerEnabled() && (!r.Repository.AllowOnlyContributorsToTrackTime() ||
  82. r.IsWriter() || issue.IsPoster(user.ID) || issue.AssigneeID == user.ID)
  83. }
  84. // GetEditorconfig returns the .editorconfig definition if found in the
  85. // HEAD of the default repo branch.
  86. func (r *Repository) GetEditorconfig() (*editorconfig.Editorconfig, error) {
  87. commit, err := r.GitRepo.GetBranchCommit(r.Repository.DefaultBranch)
  88. if err != nil {
  89. return nil, err
  90. }
  91. treeEntry, err := commit.GetTreeEntryByPath(".editorconfig")
  92. if err != nil {
  93. return nil, err
  94. }
  95. reader, err := treeEntry.Blob().Data()
  96. if err != nil {
  97. return nil, err
  98. }
  99. data, err := ioutil.ReadAll(reader)
  100. if err != nil {
  101. return nil, err
  102. }
  103. return editorconfig.ParseBytes(data)
  104. }
  105. // RetrieveBaseRepo retrieves base repository
  106. func RetrieveBaseRepo(ctx *Context, repo *models.Repository) {
  107. // Non-fork repository will not return error in this method.
  108. if err := repo.GetBaseRepo(); err != nil {
  109. if models.IsErrRepoNotExist(err) {
  110. repo.IsFork = false
  111. repo.ForkID = 0
  112. return
  113. }
  114. ctx.Handle(500, "GetBaseRepo", err)
  115. return
  116. } else if err = repo.BaseRepo.GetOwner(); err != nil {
  117. ctx.Handle(500, "BaseRepo.GetOwner", err)
  118. return
  119. }
  120. }
  121. // ComposeGoGetImport returns go-get-import meta content.
  122. func ComposeGoGetImport(owner, repo string) string {
  123. return path.Join(setting.Domain, setting.AppSubURL, owner, repo)
  124. }
  125. // EarlyResponseForGoGetMeta responses appropriate go-get meta with status 200
  126. // if user does not have actual access to the requested repository,
  127. // or the owner or repository does not exist at all.
  128. // This is particular a workaround for "go get" command which does not respect
  129. // .netrc file.
  130. func EarlyResponseForGoGetMeta(ctx *Context) {
  131. username := ctx.Params(":username")
  132. reponame := ctx.Params(":reponame")
  133. ctx.PlainText(200, []byte(com.Expand(`<meta name="go-import" content="{GoGetImport} git {CloneLink}">`,
  134. map[string]string{
  135. "GoGetImport": ComposeGoGetImport(username, strings.TrimSuffix(reponame, ".git")),
  136. "CloneLink": models.ComposeHTTPSCloneURL(username, reponame),
  137. })))
  138. }
  139. // RedirectToRepo redirect to a differently-named repository
  140. func RedirectToRepo(ctx *Context, redirectRepoID int64) {
  141. ownerName := ctx.Params(":username")
  142. previousRepoName := ctx.Params(":reponame")
  143. repo, err := models.GetRepositoryByID(redirectRepoID)
  144. if err != nil {
  145. ctx.Handle(500, "GetRepositoryByID", err)
  146. return
  147. }
  148. redirectPath := strings.Replace(
  149. ctx.Req.URL.Path,
  150. fmt.Sprintf("%s/%s", ownerName, previousRepoName),
  151. fmt.Sprintf("%s/%s", ownerName, repo.Name),
  152. 1,
  153. )
  154. ctx.Redirect(redirectPath)
  155. }
  156. func repoAssignment(ctx *Context, repo *models.Repository) {
  157. // Admin has super access.
  158. if ctx.IsSigned && ctx.User.IsAdmin {
  159. ctx.Repo.AccessMode = models.AccessModeOwner
  160. } else {
  161. var userID int64
  162. if ctx.User != nil {
  163. userID = ctx.User.ID
  164. }
  165. mode, err := models.AccessLevel(userID, repo)
  166. if err != nil {
  167. ctx.Handle(500, "AccessLevel", err)
  168. return
  169. }
  170. ctx.Repo.AccessMode = mode
  171. }
  172. // Check access.
  173. if ctx.Repo.AccessMode == models.AccessModeNone {
  174. if ctx.Query("go-get") == "1" {
  175. EarlyResponseForGoGetMeta(ctx)
  176. return
  177. }
  178. ctx.Handle(404, "no access right", nil)
  179. return
  180. }
  181. ctx.Data["HasAccess"] = true
  182. if repo.IsMirror {
  183. var err error
  184. ctx.Repo.Mirror, err = models.GetMirrorByRepoID(repo.ID)
  185. if err != nil {
  186. ctx.Handle(500, "GetMirror", err)
  187. return
  188. }
  189. ctx.Data["MirrorEnablePrune"] = ctx.Repo.Mirror.EnablePrune
  190. ctx.Data["MirrorInterval"] = ctx.Repo.Mirror.Interval
  191. ctx.Data["Mirror"] = ctx.Repo.Mirror
  192. }
  193. ctx.Repo.Repository = repo
  194. ctx.Data["RepoName"] = ctx.Repo.Repository.Name
  195. ctx.Data["IsBareRepo"] = ctx.Repo.Repository.IsBare
  196. }
  197. // RepoIDAssignment returns a macaron handler which assigns the repo to the context.
  198. func RepoIDAssignment() macaron.Handler {
  199. return func(ctx *Context) {
  200. repoID := ctx.ParamsInt64(":repoid")
  201. // Get repository.
  202. repo, err := models.GetRepositoryByID(repoID)
  203. if err != nil {
  204. if models.IsErrRepoNotExist(err) {
  205. ctx.Handle(404, "GetRepositoryByID", nil)
  206. } else {
  207. ctx.Handle(500, "GetRepositoryByID", err)
  208. }
  209. return
  210. }
  211. if err = repo.GetOwner(); err != nil {
  212. ctx.Handle(500, "GetOwner", err)
  213. return
  214. }
  215. repoAssignment(ctx, repo)
  216. }
  217. }
  218. // RepoAssignment returns a macaron to handle repository assignment
  219. func RepoAssignment() macaron.Handler {
  220. return func(ctx *Context) {
  221. var (
  222. owner *models.User
  223. err error
  224. )
  225. userName := ctx.Params(":username")
  226. repoName := ctx.Params(":reponame")
  227. // Check if the user is the same as the repository owner
  228. if ctx.IsSigned && ctx.User.LowerName == strings.ToLower(userName) {
  229. owner = ctx.User
  230. } else {
  231. owner, err = models.GetUserByName(userName)
  232. if err != nil {
  233. if models.IsErrUserNotExist(err) {
  234. if ctx.Query("go-get") == "1" {
  235. EarlyResponseForGoGetMeta(ctx)
  236. return
  237. }
  238. ctx.Handle(404, "GetUserByName", nil)
  239. } else {
  240. ctx.Handle(500, "GetUserByName", err)
  241. }
  242. return
  243. }
  244. }
  245. ctx.Repo.Owner = owner
  246. ctx.Data["Username"] = ctx.Repo.Owner.Name
  247. // Get repository.
  248. repo, err := models.GetRepositoryByName(owner.ID, repoName)
  249. if err != nil {
  250. if models.IsErrRepoNotExist(err) {
  251. redirectRepoID, err := models.LookupRepoRedirect(owner.ID, repoName)
  252. if err == nil {
  253. RedirectToRepo(ctx, redirectRepoID)
  254. } else if models.IsErrRepoRedirectNotExist(err) {
  255. if ctx.Query("go-get") == "1" {
  256. EarlyResponseForGoGetMeta(ctx)
  257. return
  258. }
  259. ctx.Handle(404, "GetRepositoryByName", nil)
  260. } else {
  261. ctx.Handle(500, "LookupRepoRedirect", err)
  262. }
  263. } else {
  264. ctx.Handle(500, "GetRepositoryByName", err)
  265. }
  266. return
  267. }
  268. repo.Owner = owner
  269. repoAssignment(ctx, repo)
  270. if ctx.Written() {
  271. return
  272. }
  273. gitRepo, err := git.OpenRepository(models.RepoPath(userName, repoName))
  274. if err != nil {
  275. ctx.Handle(500, "RepoAssignment Invalid repo "+models.RepoPath(userName, repoName), err)
  276. return
  277. }
  278. ctx.Repo.GitRepo = gitRepo
  279. ctx.Repo.RepoLink = repo.Link()
  280. ctx.Data["RepoLink"] = ctx.Repo.RepoLink
  281. ctx.Data["RepoRelPath"] = ctx.Repo.Owner.Name + "/" + ctx.Repo.Repository.Name
  282. tags, err := ctx.Repo.GitRepo.GetTags()
  283. if err != nil {
  284. ctx.Handle(500, "GetTags", err)
  285. return
  286. }
  287. ctx.Data["Tags"] = tags
  288. count, err := models.GetReleaseCountByRepoID(ctx.Repo.Repository.ID, models.FindReleasesOptions{
  289. IncludeDrafts: false,
  290. IncludeTags: true,
  291. })
  292. if err != nil {
  293. ctx.Handle(500, "GetReleaseCountByRepoID", err)
  294. return
  295. }
  296. ctx.Repo.Repository.NumReleases = int(count)
  297. ctx.Data["Title"] = owner.Name + "/" + repo.Name
  298. ctx.Data["Repository"] = repo
  299. ctx.Data["Owner"] = ctx.Repo.Repository.Owner
  300. ctx.Data["IsRepositoryOwner"] = ctx.Repo.IsOwner()
  301. ctx.Data["IsRepositoryAdmin"] = ctx.Repo.IsAdmin()
  302. ctx.Data["IsRepositoryWriter"] = ctx.Repo.IsWriter()
  303. if ctx.Data["CanSignedUserFork"], err = ctx.Repo.Repository.CanUserFork(ctx.User); err != nil {
  304. ctx.Handle(500, "CanUserFork", err)
  305. return
  306. }
  307. ctx.Data["DisableSSH"] = setting.SSH.Disabled
  308. ctx.Data["ExposeAnonSSH"] = setting.SSH.ExposeAnonymous
  309. ctx.Data["DisableHTTP"] = setting.Repository.DisableHTTPGit
  310. ctx.Data["CloneLink"] = repo.CloneLink()
  311. ctx.Data["WikiCloneLink"] = repo.WikiCloneLink()
  312. if ctx.IsSigned {
  313. ctx.Data["IsWatchingRepo"] = models.IsWatching(ctx.User.ID, repo.ID)
  314. ctx.Data["IsStaringRepo"] = models.IsStaring(ctx.User.ID, repo.ID)
  315. }
  316. // repo is bare and display enable
  317. if ctx.Repo.Repository.IsBare {
  318. ctx.Data["BranchName"] = ctx.Repo.Repository.DefaultBranch
  319. return
  320. }
  321. ctx.Data["TagName"] = ctx.Repo.TagName
  322. brs, err := ctx.Repo.GitRepo.GetBranches()
  323. if err != nil {
  324. ctx.Handle(500, "GetBranches", err)
  325. return
  326. }
  327. ctx.Data["Branches"] = brs
  328. ctx.Data["BrancheCount"] = len(brs)
  329. // If not branch selected, try default one.
  330. // If default branch doesn't exists, fall back to some other branch.
  331. if len(ctx.Repo.BranchName) == 0 {
  332. if len(ctx.Repo.Repository.DefaultBranch) > 0 && gitRepo.IsBranchExist(ctx.Repo.Repository.DefaultBranch) {
  333. ctx.Repo.BranchName = ctx.Repo.Repository.DefaultBranch
  334. } else if len(brs) > 0 {
  335. ctx.Repo.BranchName = brs[0]
  336. }
  337. }
  338. ctx.Data["BranchName"] = ctx.Repo.BranchName
  339. ctx.Data["CommitID"] = ctx.Repo.CommitID
  340. if repo.IsFork {
  341. RetrieveBaseRepo(ctx, repo)
  342. if ctx.Written() {
  343. return
  344. }
  345. }
  346. // People who have push access or have forked repository can propose a new pull request.
  347. if ctx.Repo.IsWriter() || (ctx.IsSigned && ctx.User.HasForkedRepo(ctx.Repo.Repository.ID)) {
  348. // Pull request is allowed if this is a fork repository
  349. // and base repository accepts pull requests.
  350. if repo.BaseRepo != nil && repo.BaseRepo.AllowsPulls() {
  351. ctx.Data["BaseRepo"] = repo.BaseRepo
  352. ctx.Repo.PullRequest.BaseRepo = repo.BaseRepo
  353. ctx.Repo.PullRequest.Allowed = true
  354. ctx.Repo.PullRequest.HeadInfo = ctx.Repo.Owner.Name + ":" + ctx.Repo.BranchName
  355. } else {
  356. // Or, this is repository accepts pull requests between branches.
  357. if repo.AllowsPulls() {
  358. ctx.Data["BaseRepo"] = repo
  359. ctx.Repo.PullRequest.BaseRepo = repo
  360. ctx.Repo.PullRequest.Allowed = true
  361. ctx.Repo.PullRequest.SameRepo = true
  362. ctx.Repo.PullRequest.HeadInfo = ctx.Repo.BranchName
  363. }
  364. }
  365. // Reset repo units as otherwise user specific units wont be loaded later
  366. ctx.Repo.Repository.Units = nil
  367. }
  368. ctx.Data["PullRequestCtx"] = ctx.Repo.PullRequest
  369. if ctx.Query("go-get") == "1" {
  370. ctx.Data["GoGetImport"] = ComposeGoGetImport(owner.Name, repo.Name)
  371. prefix := setting.AppURL + path.Join(owner.Name, repo.Name, "src", ctx.Repo.BranchName)
  372. ctx.Data["GoDocDirectory"] = prefix + "{/dir}"
  373. ctx.Data["GoDocFile"] = prefix + "{/dir}/{file}#L{line}"
  374. }
  375. }
  376. }
  377. // RepoRef handles repository reference name including those contain `/`.
  378. func RepoRef() macaron.Handler {
  379. return func(ctx *Context) {
  380. // Empty repository does not have reference information.
  381. if ctx.Repo.Repository.IsBare {
  382. return
  383. }
  384. var (
  385. refName string
  386. err error
  387. )
  388. // For API calls.
  389. if ctx.Repo.GitRepo == nil {
  390. repoPath := models.RepoPath(ctx.Repo.Owner.Name, ctx.Repo.Repository.Name)
  391. ctx.Repo.GitRepo, err = git.OpenRepository(repoPath)
  392. if err != nil {
  393. ctx.Handle(500, "RepoRef Invalid repo "+repoPath, err)
  394. return
  395. }
  396. }
  397. // Get default branch.
  398. if len(ctx.Params("*")) == 0 {
  399. refName = ctx.Repo.Repository.DefaultBranch
  400. if !ctx.Repo.GitRepo.IsBranchExist(refName) {
  401. brs, err := ctx.Repo.GitRepo.GetBranches()
  402. if err != nil {
  403. ctx.Handle(500, "GetBranches", err)
  404. return
  405. } else if len(brs) == 0 {
  406. err = fmt.Errorf("No branches in non-bare repository %s",
  407. ctx.Repo.GitRepo.Path)
  408. ctx.Handle(500, "GetBranches", err)
  409. return
  410. }
  411. refName = brs[0]
  412. }
  413. ctx.Repo.Commit, err = ctx.Repo.GitRepo.GetBranchCommit(refName)
  414. if err != nil {
  415. ctx.Handle(500, "GetBranchCommit", err)
  416. return
  417. }
  418. ctx.Repo.CommitID = ctx.Repo.Commit.ID.String()
  419. ctx.Repo.IsViewBranch = true
  420. } else {
  421. hasMatched := false
  422. parts := strings.Split(ctx.Params("*"), "/")
  423. for i, part := range parts {
  424. refName = strings.TrimPrefix(refName+"/"+part, "/")
  425. if ctx.Repo.GitRepo.IsBranchExist(refName) ||
  426. ctx.Repo.GitRepo.IsTagExist(refName) {
  427. if i < len(parts)-1 {
  428. ctx.Repo.TreePath = strings.Join(parts[i+1:], "/")
  429. }
  430. hasMatched = true
  431. break
  432. }
  433. }
  434. if !hasMatched && len(parts[0]) == 40 {
  435. refName = parts[0]
  436. ctx.Repo.TreePath = strings.Join(parts[1:], "/")
  437. }
  438. if ctx.Repo.GitRepo.IsBranchExist(refName) {
  439. ctx.Repo.IsViewBranch = true
  440. ctx.Repo.Commit, err = ctx.Repo.GitRepo.GetBranchCommit(refName)
  441. if err != nil {
  442. ctx.Handle(500, "GetBranchCommit", err)
  443. return
  444. }
  445. ctx.Repo.CommitID = ctx.Repo.Commit.ID.String()
  446. } else if ctx.Repo.GitRepo.IsTagExist(refName) {
  447. ctx.Repo.IsViewTag = true
  448. ctx.Repo.Commit, err = ctx.Repo.GitRepo.GetTagCommit(refName)
  449. if err != nil {
  450. ctx.Handle(500, "GetTagCommit", err)
  451. return
  452. }
  453. ctx.Repo.CommitID = ctx.Repo.Commit.ID.String()
  454. } else if len(refName) == 40 {
  455. ctx.Repo.IsViewCommit = true
  456. ctx.Repo.CommitID = refName
  457. ctx.Repo.Commit, err = ctx.Repo.GitRepo.GetCommit(refName)
  458. if err != nil {
  459. ctx.Handle(404, "GetCommit", nil)
  460. return
  461. }
  462. } else {
  463. ctx.Handle(404, "RepoRef invalid repo", fmt.Errorf("branch or tag not exist: %s", refName))
  464. return
  465. }
  466. }
  467. ctx.Repo.BranchName = refName
  468. ctx.Data["BranchName"] = ctx.Repo.BranchName
  469. ctx.Data["CommitID"] = ctx.Repo.CommitID
  470. ctx.Data["TreePath"] = ctx.Repo.TreePath
  471. ctx.Data["IsViewBranch"] = ctx.Repo.IsViewBranch
  472. ctx.Data["IsViewTag"] = ctx.Repo.IsViewTag
  473. ctx.Data["IsViewCommit"] = ctx.Repo.IsViewCommit
  474. ctx.Repo.CommitsCount, err = ctx.Repo.Commit.CommitsCount()
  475. if err != nil {
  476. ctx.Handle(500, "CommitsCount", err)
  477. return
  478. }
  479. ctx.Data["CommitsCount"] = ctx.Repo.CommitsCount
  480. }
  481. }
  482. // RequireRepoAdmin returns a macaron middleware for requiring repository admin permission
  483. func RequireRepoAdmin() macaron.Handler {
  484. return func(ctx *Context) {
  485. if !ctx.IsSigned || (!ctx.Repo.IsAdmin() && !ctx.User.IsAdmin) {
  486. ctx.Handle(404, ctx.Req.RequestURI, nil)
  487. return
  488. }
  489. }
  490. }
  491. // RequireRepoWriter returns a macaron middleware for requiring repository write permission
  492. func RequireRepoWriter() macaron.Handler {
  493. return func(ctx *Context) {
  494. if !ctx.IsSigned || (!ctx.Repo.IsWriter() && !ctx.User.IsAdmin) {
  495. ctx.Handle(404, ctx.Req.RequestURI, nil)
  496. return
  497. }
  498. }
  499. }
  500. // LoadRepoUnits loads repsitory's units, it should be called after repository and user loaded
  501. func LoadRepoUnits() macaron.Handler {
  502. return func(ctx *Context) {
  503. var isAdmin bool
  504. if ctx.User != nil && ctx.User.IsAdmin {
  505. isAdmin = true
  506. }
  507. var userID int64
  508. if ctx.User != nil {
  509. userID = ctx.User.ID
  510. }
  511. err := ctx.Repo.Repository.LoadUnitsByUserID(userID, isAdmin)
  512. if err != nil {
  513. ctx.Handle(500, "LoadUnitsByUserID", err)
  514. return
  515. }
  516. }
  517. }
  518. // CheckUnit will check whether unit type is enabled
  519. func CheckUnit(unitType models.UnitType) macaron.Handler {
  520. return func(ctx *Context) {
  521. if !ctx.Repo.Repository.UnitEnabled(unitType) {
  522. ctx.Handle(404, "CheckUnit", fmt.Errorf("%s: %v", ctx.Tr("units.error.unit_not_allowed"), unitType))
  523. }
  524. }
  525. }
  526. // CheckAnyUnit will check whether any of the unit types are enabled
  527. func CheckAnyUnit(unitTypes ...models.UnitType) macaron.Handler {
  528. return func(ctx *Context) {
  529. if !ctx.Repo.Repository.AnyUnitEnabled(unitTypes...) {
  530. ctx.Handle(404, "CheckAnyUnit", fmt.Errorf("%s: %v", ctx.Tr("units.error.unit_not_allowed"), unitTypes))
  531. }
  532. }
  533. }
  534. // GitHookService checks if repository Git hooks service has been enabled.
  535. func GitHookService() macaron.Handler {
  536. return func(ctx *Context) {
  537. if !ctx.User.CanEditGitHook() {
  538. ctx.Handle(404, "GitHookService", nil)
  539. return
  540. }
  541. }
  542. }
  543. // UnitTypes returns a macaron middleware to set unit types to context variables.
  544. func UnitTypes() macaron.Handler {
  545. return func(ctx *Context) {
  546. ctx.Data["UnitTypeCode"] = models.UnitTypeCode
  547. ctx.Data["UnitTypeIssues"] = models.UnitTypeIssues
  548. ctx.Data["UnitTypePullRequests"] = models.UnitTypePullRequests
  549. ctx.Data["UnitTypeReleases"] = models.UnitTypeReleases
  550. ctx.Data["UnitTypeWiki"] = models.UnitTypeWiki
  551. ctx.Data["UnitTypeExternalWiki"] = models.UnitTypeExternalWiki
  552. ctx.Data["UnitTypeExternalTracker"] = models.UnitTypeExternalTracker
  553. }
  554. }