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.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821
  1. // Copyright 2015 The Gogs Authors. All rights reserved.
  2. // Copyright 2019 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 models
  6. import (
  7. "fmt"
  8. "io"
  9. "strings"
  10. "code.gitea.io/gitea/modules/git"
  11. "code.gitea.io/gitea/modules/log"
  12. "code.gitea.io/gitea/modules/setting"
  13. "code.gitea.io/gitea/modules/timeutil"
  14. )
  15. // PullRequestType defines pull request type
  16. type PullRequestType int
  17. // Enumerate all the pull request types
  18. const (
  19. PullRequestGitea PullRequestType = iota
  20. PullRequestGit
  21. )
  22. // PullRequestStatus defines pull request status
  23. type PullRequestStatus int
  24. // Enumerate all the pull request status
  25. const (
  26. PullRequestStatusConflict PullRequestStatus = iota
  27. PullRequestStatusChecking
  28. PullRequestStatusMergeable
  29. PullRequestStatusManuallyMerged
  30. PullRequestStatusError
  31. )
  32. // PullRequest represents relation between pull request and repositories.
  33. type PullRequest struct {
  34. ID int64 `xorm:"pk autoincr"`
  35. Type PullRequestType
  36. Status PullRequestStatus
  37. ConflictedFiles []string `xorm:"TEXT JSON"`
  38. IssueID int64 `xorm:"INDEX"`
  39. Issue *Issue `xorm:"-"`
  40. Index int64
  41. HeadRepoID int64 `xorm:"INDEX"`
  42. HeadRepo *Repository `xorm:"-"`
  43. BaseRepoID int64 `xorm:"INDEX"`
  44. BaseRepo *Repository `xorm:"-"`
  45. HeadBranch string
  46. BaseBranch string
  47. ProtectedBranch *ProtectedBranch `xorm:"-"`
  48. MergeBase string `xorm:"VARCHAR(40)"`
  49. HasMerged bool `xorm:"INDEX"`
  50. MergedCommitID string `xorm:"VARCHAR(40)"`
  51. MergerID int64 `xorm:"INDEX"`
  52. Merger *User `xorm:"-"`
  53. MergedUnix timeutil.TimeStamp `xorm:"updated INDEX"`
  54. isHeadRepoLoaded bool `xorm:"-"`
  55. }
  56. // MustHeadUserName returns the HeadRepo's username if failed return blank
  57. func (pr *PullRequest) MustHeadUserName() string {
  58. if err := pr.LoadHeadRepo(); err != nil {
  59. if !IsErrRepoNotExist(err) {
  60. log.Error("LoadHeadRepo: %v", err)
  61. } else {
  62. log.Warn("LoadHeadRepo %d but repository does not exist: %v", pr.HeadRepoID, err)
  63. }
  64. return ""
  65. }
  66. if pr.HeadRepo == nil {
  67. return ""
  68. }
  69. return pr.HeadRepo.OwnerName
  70. }
  71. // Note: don't try to get Issue because will end up recursive querying.
  72. func (pr *PullRequest) loadAttributes(e Engine) (err error) {
  73. if pr.HasMerged && pr.Merger == nil {
  74. pr.Merger, err = getUserByID(e, pr.MergerID)
  75. if IsErrUserNotExist(err) {
  76. pr.MergerID = -1
  77. pr.Merger = NewGhostUser()
  78. } else if err != nil {
  79. return fmt.Errorf("getUserByID [%d]: %v", pr.MergerID, err)
  80. }
  81. }
  82. return nil
  83. }
  84. // LoadAttributes loads pull request attributes from database
  85. func (pr *PullRequest) LoadAttributes() error {
  86. return pr.loadAttributes(x)
  87. }
  88. func (pr *PullRequest) loadHeadRepo(e Engine) (err error) {
  89. if !pr.isHeadRepoLoaded && pr.HeadRepo == nil && pr.HeadRepoID > 0 {
  90. if pr.HeadRepoID == pr.BaseRepoID {
  91. if pr.BaseRepo != nil {
  92. pr.HeadRepo = pr.BaseRepo
  93. return nil
  94. } else if pr.Issue != nil && pr.Issue.Repo != nil {
  95. pr.HeadRepo = pr.Issue.Repo
  96. return nil
  97. }
  98. }
  99. pr.HeadRepo, err = getRepositoryByID(e, pr.HeadRepoID)
  100. if err != nil && !IsErrRepoNotExist(err) { // Head repo maybe deleted, but it should still work
  101. return fmt.Errorf("getRepositoryByID(head): %v", err)
  102. }
  103. pr.isHeadRepoLoaded = true
  104. }
  105. return nil
  106. }
  107. // LoadHeadRepo loads the head repository
  108. func (pr *PullRequest) LoadHeadRepo() error {
  109. return pr.loadHeadRepo(x)
  110. }
  111. // LoadBaseRepo loads the target repository
  112. func (pr *PullRequest) LoadBaseRepo() error {
  113. return pr.loadBaseRepo(x)
  114. }
  115. func (pr *PullRequest) loadBaseRepo(e Engine) (err error) {
  116. if pr.BaseRepo != nil {
  117. return nil
  118. }
  119. if pr.HeadRepoID == pr.BaseRepoID && pr.HeadRepo != nil {
  120. pr.BaseRepo = pr.HeadRepo
  121. return nil
  122. }
  123. if pr.Issue != nil && pr.Issue.Repo != nil {
  124. pr.BaseRepo = pr.Issue.Repo
  125. return nil
  126. }
  127. pr.BaseRepo, err = getRepositoryByID(e, pr.BaseRepoID)
  128. if err != nil {
  129. return fmt.Errorf("GetRepositoryByID(base): %v", err)
  130. }
  131. return nil
  132. }
  133. // LoadIssue loads issue information from database
  134. func (pr *PullRequest) LoadIssue() (err error) {
  135. return pr.loadIssue(x)
  136. }
  137. func (pr *PullRequest) loadIssue(e Engine) (err error) {
  138. if pr.Issue != nil {
  139. return nil
  140. }
  141. pr.Issue, err = getIssueByID(e, pr.IssueID)
  142. if err == nil {
  143. pr.Issue.PullRequest = pr
  144. }
  145. return err
  146. }
  147. // LoadProtectedBranch loads the protected branch of the base branch
  148. func (pr *PullRequest) LoadProtectedBranch() (err error) {
  149. return pr.loadProtectedBranch(x)
  150. }
  151. func (pr *PullRequest) loadProtectedBranch(e Engine) (err error) {
  152. if pr.ProtectedBranch == nil {
  153. if pr.BaseRepo == nil {
  154. if pr.BaseRepoID == 0 {
  155. return nil
  156. }
  157. pr.BaseRepo, err = getRepositoryByID(e, pr.BaseRepoID)
  158. if err != nil {
  159. return
  160. }
  161. }
  162. pr.ProtectedBranch, err = getProtectedBranchBy(e, pr.BaseRepo.ID, pr.BaseBranch)
  163. }
  164. return
  165. }
  166. // GetDefaultMergeMessage returns default message used when merging pull request
  167. func (pr *PullRequest) GetDefaultMergeMessage() string {
  168. if pr.HeadRepo == nil {
  169. var err error
  170. pr.HeadRepo, err = GetRepositoryByID(pr.HeadRepoID)
  171. if err != nil {
  172. log.Error("GetRepositoryById[%d]: %v", pr.HeadRepoID, err)
  173. return ""
  174. }
  175. }
  176. if err := pr.LoadIssue(); err != nil {
  177. log.Error("Cannot load issue %d for PR id %d: Error: %v", pr.IssueID, pr.ID, err)
  178. return ""
  179. }
  180. if pr.BaseRepoID == pr.HeadRepoID {
  181. return fmt.Sprintf("Merge pull request '%s' (#%d) from %s into %s", pr.Issue.Title, pr.Issue.Index, pr.HeadBranch, pr.BaseBranch)
  182. }
  183. return fmt.Sprintf("Merge pull request '%s' (#%d) from %s:%s into %s", pr.Issue.Title, pr.Issue.Index, pr.HeadRepo.FullName(), pr.HeadBranch, pr.BaseBranch)
  184. }
  185. // GetCommitMessages returns the commit messages between head and merge base (if there is one)
  186. func (pr *PullRequest) GetCommitMessages() string {
  187. if err := pr.LoadIssue(); err != nil {
  188. log.Error("Cannot load issue %d for PR id %d: Error: %v", pr.IssueID, pr.ID, err)
  189. return ""
  190. }
  191. if err := pr.Issue.LoadPoster(); err != nil {
  192. log.Error("Cannot load poster %d for pr id %d, index %d Error: %v", pr.Issue.PosterID, pr.ID, pr.Index, err)
  193. return ""
  194. }
  195. if pr.HeadRepo == nil {
  196. var err error
  197. pr.HeadRepo, err = GetRepositoryByID(pr.HeadRepoID)
  198. if err != nil {
  199. log.Error("GetRepositoryById[%d]: %v", pr.HeadRepoID, err)
  200. return ""
  201. }
  202. }
  203. gitRepo, err := git.OpenRepository(pr.HeadRepo.RepoPath())
  204. if err != nil {
  205. log.Error("Unable to open head repository: Error: %v", err)
  206. return ""
  207. }
  208. defer gitRepo.Close()
  209. headCommit, err := gitRepo.GetBranchCommit(pr.HeadBranch)
  210. if err != nil {
  211. log.Error("Unable to get head commit: %s Error: %v", pr.HeadBranch, err)
  212. return ""
  213. }
  214. mergeBase, err := gitRepo.GetCommit(pr.MergeBase)
  215. if err != nil {
  216. log.Error("Unable to get merge base commit: %s Error: %v", pr.MergeBase, err)
  217. return ""
  218. }
  219. limit := setting.Repository.PullRequest.DefaultMergeMessageCommitsLimit
  220. list, err := gitRepo.CommitsBetweenLimit(headCommit, mergeBase, limit, 0)
  221. if err != nil {
  222. log.Error("Unable to get commits between: %s %s Error: %v", pr.HeadBranch, pr.MergeBase, err)
  223. return ""
  224. }
  225. maxSize := setting.Repository.PullRequest.DefaultMergeMessageSize
  226. posterSig := pr.Issue.Poster.NewGitSig().String()
  227. authorsMap := map[string]bool{}
  228. authors := make([]string, 0, list.Len())
  229. stringBuilder := strings.Builder{}
  230. element := list.Front()
  231. for element != nil {
  232. commit := element.Value.(*git.Commit)
  233. if maxSize < 0 || stringBuilder.Len() < maxSize {
  234. toWrite := []byte(commit.CommitMessage)
  235. if len(toWrite) > maxSize-stringBuilder.Len() && maxSize > -1 {
  236. toWrite = append(toWrite[:maxSize-stringBuilder.Len()], "..."...)
  237. }
  238. if _, err := stringBuilder.Write(toWrite); err != nil {
  239. log.Error("Unable to write commit message Error: %v", err)
  240. return ""
  241. }
  242. if _, err := stringBuilder.WriteRune('\n'); err != nil {
  243. log.Error("Unable to write commit message Error: %v", err)
  244. return ""
  245. }
  246. }
  247. authorString := commit.Author.String()
  248. if !authorsMap[authorString] && authorString != posterSig {
  249. authors = append(authors, authorString)
  250. authorsMap[authorString] = true
  251. }
  252. element = element.Next()
  253. }
  254. // Consider collecting the remaining authors
  255. if limit >= 0 && setting.Repository.PullRequest.DefaultMergeMessageAllAuthors {
  256. skip := limit
  257. limit = 30
  258. for {
  259. list, err := gitRepo.CommitsBetweenLimit(headCommit, mergeBase, limit, skip)
  260. if err != nil {
  261. log.Error("Unable to get commits between: %s %s Error: %v", pr.HeadBranch, pr.MergeBase, err)
  262. return ""
  263. }
  264. if list.Len() == 0 {
  265. break
  266. }
  267. element := list.Front()
  268. for element != nil {
  269. commit := element.Value.(*git.Commit)
  270. authorString := commit.Author.String()
  271. if !authorsMap[authorString] && authorString != posterSig {
  272. authors = append(authors, authorString)
  273. authorsMap[authorString] = true
  274. }
  275. element = element.Next()
  276. }
  277. }
  278. }
  279. if len(authors) > 0 {
  280. if _, err := stringBuilder.WriteRune('\n'); err != nil {
  281. log.Error("Unable to write to string builder Error: %v", err)
  282. return ""
  283. }
  284. }
  285. for _, author := range authors {
  286. if _, err := stringBuilder.Write([]byte("Co-authored-by: ")); err != nil {
  287. log.Error("Unable to write to string builder Error: %v", err)
  288. return ""
  289. }
  290. if _, err := stringBuilder.Write([]byte(author)); err != nil {
  291. log.Error("Unable to write to string builder Error: %v", err)
  292. return ""
  293. }
  294. if _, err := stringBuilder.WriteRune('\n'); err != nil {
  295. log.Error("Unable to write to string builder Error: %v", err)
  296. return ""
  297. }
  298. }
  299. return stringBuilder.String()
  300. }
  301. // ReviewCount represents a count of Reviews
  302. type ReviewCount struct {
  303. IssueID int64
  304. Type ReviewType
  305. Count int64
  306. }
  307. // GetApprovalCounts returns the approval counts by type
  308. // FIXME: Only returns official counts due to double counting of non-official counts
  309. func (pr *PullRequest) GetApprovalCounts() ([]*ReviewCount, error) {
  310. return pr.getApprovalCounts(x)
  311. }
  312. func (pr *PullRequest) getApprovalCounts(e Engine) ([]*ReviewCount, error) {
  313. rCounts := make([]*ReviewCount, 0, 6)
  314. sess := e.Where("issue_id = ?", pr.IssueID)
  315. return rCounts, sess.Select("issue_id, type, count(id) as `count`").Where("official = ?", true).GroupBy("issue_id, type").Table("review").Find(&rCounts)
  316. }
  317. // GetApprovers returns the approvers of the pull request
  318. func (pr *PullRequest) GetApprovers() string {
  319. stringBuilder := strings.Builder{}
  320. if err := pr.getReviewedByLines(&stringBuilder); err != nil {
  321. log.Error("Unable to getReviewedByLines: Error: %v", err)
  322. return ""
  323. }
  324. return stringBuilder.String()
  325. }
  326. func (pr *PullRequest) getReviewedByLines(writer io.Writer) error {
  327. maxReviewers := setting.Repository.PullRequest.DefaultMergeMessageMaxApprovers
  328. if maxReviewers == 0 {
  329. return nil
  330. }
  331. sess := x.NewSession()
  332. defer sess.Close()
  333. if err := sess.Begin(); err != nil {
  334. return err
  335. }
  336. // Note: This doesn't page as we only expect a very limited number of reviews
  337. reviews, err := findReviews(sess, FindReviewOptions{
  338. Type: ReviewTypeApprove,
  339. IssueID: pr.IssueID,
  340. OfficialOnly: setting.Repository.PullRequest.DefaultMergeMessageOfficialApproversOnly,
  341. })
  342. if err != nil {
  343. log.Error("Unable to FindReviews for PR ID %d: %v", pr.ID, err)
  344. return err
  345. }
  346. reviewersWritten := 0
  347. for _, review := range reviews {
  348. if maxReviewers > 0 && reviewersWritten > maxReviewers {
  349. break
  350. }
  351. if err := review.loadReviewer(sess); err != nil && !IsErrUserNotExist(err) {
  352. log.Error("Unable to LoadReviewer[%d] for PR ID %d : %v", review.ReviewerID, pr.ID, err)
  353. return err
  354. } else if review.Reviewer == nil {
  355. continue
  356. }
  357. if _, err := writer.Write([]byte("Reviewed-by: ")); err != nil {
  358. return err
  359. }
  360. if _, err := writer.Write([]byte(review.Reviewer.NewGitSig().String())); err != nil {
  361. return err
  362. }
  363. if _, err := writer.Write([]byte{'\n'}); err != nil {
  364. return err
  365. }
  366. reviewersWritten++
  367. }
  368. return sess.Commit()
  369. }
  370. // GetDefaultSquashMessage returns default message used when squash and merging pull request
  371. func (pr *PullRequest) GetDefaultSquashMessage() string {
  372. if err := pr.LoadIssue(); err != nil {
  373. log.Error("LoadIssue: %v", err)
  374. return ""
  375. }
  376. if err := pr.LoadBaseRepo(); err != nil {
  377. log.Error("LoadBaseRepo: %v", err)
  378. return ""
  379. }
  380. if pr.BaseRepo.UnitEnabled(UnitTypeExternalTracker) {
  381. return fmt.Sprintf("%s (!%d)", pr.Issue.Title, pr.Issue.Index)
  382. }
  383. return fmt.Sprintf("%s (#%d)", pr.Issue.Title, pr.Issue.Index)
  384. }
  385. // GetGitRefName returns git ref for hidden pull request branch
  386. func (pr *PullRequest) GetGitRefName() string {
  387. return fmt.Sprintf("refs/pull/%d/head", pr.Index)
  388. }
  389. // IsChecking returns true if this pull request is still checking conflict.
  390. func (pr *PullRequest) IsChecking() bool {
  391. return pr.Status == PullRequestStatusChecking
  392. }
  393. // CanAutoMerge returns true if this pull request can be merged automatically.
  394. func (pr *PullRequest) CanAutoMerge() bool {
  395. return pr.Status == PullRequestStatusMergeable
  396. }
  397. // GetLastCommitStatus returns the last commit status for this pull request.
  398. func (pr *PullRequest) GetLastCommitStatus() (status *CommitStatus, err error) {
  399. if err = pr.LoadHeadRepo(); err != nil {
  400. return nil, err
  401. }
  402. if pr.HeadRepo == nil {
  403. return nil, ErrPullRequestHeadRepoMissing{pr.ID, pr.HeadRepoID}
  404. }
  405. headGitRepo, err := git.OpenRepository(pr.HeadRepo.RepoPath())
  406. if err != nil {
  407. return nil, err
  408. }
  409. defer headGitRepo.Close()
  410. lastCommitID, err := headGitRepo.GetBranchCommitID(pr.HeadBranch)
  411. if err != nil {
  412. return nil, err
  413. }
  414. err = pr.LoadBaseRepo()
  415. if err != nil {
  416. return nil, err
  417. }
  418. statusList, err := GetLatestCommitStatus(pr.BaseRepo, lastCommitID, 0)
  419. if err != nil {
  420. return nil, err
  421. }
  422. return CalcCommitStatus(statusList), nil
  423. }
  424. // MergeStyle represents the approach to merge commits into base branch.
  425. type MergeStyle string
  426. const (
  427. // MergeStyleMerge create merge commit
  428. MergeStyleMerge MergeStyle = "merge"
  429. // MergeStyleRebase rebase before merging
  430. MergeStyleRebase MergeStyle = "rebase"
  431. // MergeStyleRebaseMerge rebase before merging with merge commit (--no-ff)
  432. MergeStyleRebaseMerge MergeStyle = "rebase-merge"
  433. // MergeStyleSquash squash commits into single commit before merging
  434. MergeStyleSquash MergeStyle = "squash"
  435. )
  436. // SetMerged sets a pull request to merged and closes the corresponding issue
  437. func (pr *PullRequest) SetMerged() (bool, error) {
  438. if pr.HasMerged {
  439. return false, fmt.Errorf("PullRequest[%d] already merged", pr.Index)
  440. }
  441. if pr.MergedCommitID == "" || pr.MergedUnix == 0 || pr.Merger == nil {
  442. return false, fmt.Errorf("Unable to merge PullRequest[%d], some required fields are empty", pr.Index)
  443. }
  444. pr.HasMerged = true
  445. sess := x.NewSession()
  446. defer sess.Close()
  447. if err := sess.Begin(); err != nil {
  448. return false, err
  449. }
  450. if _, err := sess.Exec("UPDATE `issue` SET `repo_id` = `repo_id` WHERE `id` = ?", pr.IssueID); err != nil {
  451. return false, err
  452. }
  453. if _, err := sess.Exec("UPDATE `pull_request` SET `issue_id` = `issue_id` WHERE `id` = ?", pr.ID); err != nil {
  454. return false, err
  455. }
  456. pr.Issue = nil
  457. if err := pr.loadIssue(sess); err != nil {
  458. return false, err
  459. }
  460. if tmpPr, err := getPullRequestByID(sess, pr.ID); err != nil {
  461. return false, err
  462. } else if tmpPr.HasMerged {
  463. if pr.Issue.IsClosed {
  464. return false, nil
  465. }
  466. return false, fmt.Errorf("PullRequest[%d] already merged but it's associated issue [%d] is not closed", pr.Index, pr.IssueID)
  467. } else if pr.Issue.IsClosed {
  468. return false, fmt.Errorf("PullRequest[%d] already closed", pr.Index)
  469. }
  470. if err := pr.Issue.loadRepo(sess); err != nil {
  471. return false, err
  472. }
  473. if err := pr.Issue.Repo.getOwner(sess); err != nil {
  474. return false, err
  475. }
  476. if _, err := pr.Issue.changeStatus(sess, pr.Merger, true); err != nil {
  477. return false, fmt.Errorf("Issue.changeStatus: %v", err)
  478. }
  479. if _, err := sess.Where("id = ?", pr.ID).Cols("has_merged, status, merged_commit_id, merger_id, merged_unix").Update(pr); err != nil {
  480. return false, fmt.Errorf("Failed to update pr[%d]: %v", pr.ID, err)
  481. }
  482. if err := sess.Commit(); err != nil {
  483. return false, fmt.Errorf("Commit: %v", err)
  484. }
  485. return true, nil
  486. }
  487. // NewPullRequest creates new pull request with labels for repository.
  488. func NewPullRequest(repo *Repository, pull *Issue, labelIDs []int64, uuids []string, pr *PullRequest) (err error) {
  489. // Retry several times in case INSERT fails due to duplicate key for (repo_id, index); see #7887
  490. i := 0
  491. for {
  492. if err = newPullRequestAttempt(repo, pull, labelIDs, uuids, pr); err == nil {
  493. return nil
  494. }
  495. if !IsErrNewIssueInsert(err) {
  496. return err
  497. }
  498. if i++; i == issueMaxDupIndexAttempts {
  499. break
  500. }
  501. log.Error("NewPullRequest: error attempting to insert the new issue; will retry. Original error: %v", err)
  502. }
  503. return fmt.Errorf("NewPullRequest: too many errors attempting to insert the new issue. Last error was: %v", err)
  504. }
  505. func newPullRequestAttempt(repo *Repository, pull *Issue, labelIDs []int64, uuids []string, pr *PullRequest) (err error) {
  506. sess := x.NewSession()
  507. defer sess.Close()
  508. if err = sess.Begin(); err != nil {
  509. return err
  510. }
  511. if err = newIssue(sess, pull.Poster, NewIssueOptions{
  512. Repo: repo,
  513. Issue: pull,
  514. LabelIDs: labelIDs,
  515. Attachments: uuids,
  516. IsPull: true,
  517. }); err != nil {
  518. if IsErrUserDoesNotHaveAccessToRepo(err) || IsErrNewIssueInsert(err) {
  519. return err
  520. }
  521. return fmt.Errorf("newIssue: %v", err)
  522. }
  523. pr.Index = pull.Index
  524. pr.BaseRepo = repo
  525. pr.IssueID = pull.ID
  526. if _, err = sess.Insert(pr); err != nil {
  527. return fmt.Errorf("insert pull repo: %v", err)
  528. }
  529. if err = sess.Commit(); err != nil {
  530. return fmt.Errorf("Commit: %v", err)
  531. }
  532. return nil
  533. }
  534. // GetUnmergedPullRequest returns a pull request that is open and has not been merged
  535. // by given head/base and repo/branch.
  536. func GetUnmergedPullRequest(headRepoID, baseRepoID int64, headBranch, baseBranch string) (*PullRequest, error) {
  537. pr := new(PullRequest)
  538. has, err := x.
  539. Where("head_repo_id=? AND head_branch=? AND base_repo_id=? AND base_branch=? AND has_merged=? AND issue.is_closed=?",
  540. headRepoID, headBranch, baseRepoID, baseBranch, false, false).
  541. Join("INNER", "issue", "issue.id=pull_request.issue_id").
  542. Get(pr)
  543. if err != nil {
  544. return nil, err
  545. } else if !has {
  546. return nil, ErrPullRequestNotExist{0, 0, headRepoID, baseRepoID, headBranch, baseBranch}
  547. }
  548. return pr, nil
  549. }
  550. // GetLatestPullRequestByHeadInfo returns the latest pull request (regardless of its status)
  551. // by given head information (repo and branch).
  552. func GetLatestPullRequestByHeadInfo(repoID int64, branch string) (*PullRequest, error) {
  553. pr := new(PullRequest)
  554. has, err := x.
  555. Where("head_repo_id = ? AND head_branch = ?", repoID, branch).
  556. OrderBy("id DESC").
  557. Get(pr)
  558. if !has {
  559. return nil, err
  560. }
  561. return pr, err
  562. }
  563. // GetPullRequestByIndex returns a pull request by the given index
  564. func GetPullRequestByIndex(repoID int64, index int64) (*PullRequest, error) {
  565. pr := &PullRequest{
  566. BaseRepoID: repoID,
  567. Index: index,
  568. }
  569. has, err := x.Get(pr)
  570. if err != nil {
  571. return nil, err
  572. } else if !has {
  573. return nil, ErrPullRequestNotExist{0, 0, 0, repoID, "", ""}
  574. }
  575. if err = pr.LoadAttributes(); err != nil {
  576. return nil, err
  577. }
  578. if err = pr.LoadIssue(); err != nil {
  579. return nil, err
  580. }
  581. return pr, nil
  582. }
  583. func getPullRequestByID(e Engine, id int64) (*PullRequest, error) {
  584. pr := new(PullRequest)
  585. has, err := e.ID(id).Get(pr)
  586. if err != nil {
  587. return nil, err
  588. } else if !has {
  589. return nil, ErrPullRequestNotExist{id, 0, 0, 0, "", ""}
  590. }
  591. return pr, pr.loadAttributes(e)
  592. }
  593. // GetPullRequestByID returns a pull request by given ID.
  594. func GetPullRequestByID(id int64) (*PullRequest, error) {
  595. return getPullRequestByID(x, id)
  596. }
  597. // GetPullRequestByIssueIDWithNoAttributes returns pull request with no attributes loaded by given issue ID.
  598. func GetPullRequestByIssueIDWithNoAttributes(issueID int64) (*PullRequest, error) {
  599. var pr PullRequest
  600. has, err := x.Where("issue_id = ?", issueID).Get(&pr)
  601. if err != nil {
  602. return nil, err
  603. }
  604. if !has {
  605. return nil, ErrPullRequestNotExist{0, issueID, 0, 0, "", ""}
  606. }
  607. return &pr, nil
  608. }
  609. func getPullRequestByIssueID(e Engine, issueID int64) (*PullRequest, error) {
  610. pr := &PullRequest{
  611. IssueID: issueID,
  612. }
  613. has, err := e.Get(pr)
  614. if err != nil {
  615. return nil, err
  616. } else if !has {
  617. return nil, ErrPullRequestNotExist{0, issueID, 0, 0, "", ""}
  618. }
  619. return pr, pr.loadAttributes(e)
  620. }
  621. // GetPullRequestByIssueID returns pull request by given issue ID.
  622. func GetPullRequestByIssueID(issueID int64) (*PullRequest, error) {
  623. return getPullRequestByIssueID(x, issueID)
  624. }
  625. // Update updates all fields of pull request.
  626. func (pr *PullRequest) Update() error {
  627. _, err := x.ID(pr.ID).AllCols().Update(pr)
  628. return err
  629. }
  630. // UpdateCols updates specific fields of pull request.
  631. func (pr *PullRequest) UpdateCols(cols ...string) error {
  632. _, err := x.ID(pr.ID).Cols(cols...).Update(pr)
  633. return err
  634. }
  635. // UpdateColsIfNotMerged updates specific fields of a pull request if it has not been merged
  636. func (pr *PullRequest) UpdateColsIfNotMerged(cols ...string) error {
  637. _, err := x.Where("id = ? AND has_merged = ?", pr.ID, false).Cols(cols...).Update(pr)
  638. return err
  639. }
  640. // IsWorkInProgress determine if the Pull Request is a Work In Progress by its title
  641. func (pr *PullRequest) IsWorkInProgress() bool {
  642. if err := pr.LoadIssue(); err != nil {
  643. log.Error("LoadIssue: %v", err)
  644. return false
  645. }
  646. for _, prefix := range setting.Repository.PullRequest.WorkInProgressPrefixes {
  647. if strings.HasPrefix(strings.ToUpper(pr.Issue.Title), prefix) {
  648. return true
  649. }
  650. }
  651. return false
  652. }
  653. // IsFilesConflicted determines if the Pull Request has changes conflicting with the target branch.
  654. func (pr *PullRequest) IsFilesConflicted() bool {
  655. return len(pr.ConflictedFiles) > 0
  656. }
  657. // GetWorkInProgressPrefix returns the prefix used to mark the pull request as a work in progress.
  658. // It returns an empty string when none were found
  659. func (pr *PullRequest) GetWorkInProgressPrefix() string {
  660. if err := pr.LoadIssue(); err != nil {
  661. log.Error("LoadIssue: %v", err)
  662. return ""
  663. }
  664. for _, prefix := range setting.Repository.PullRequest.WorkInProgressPrefixes {
  665. if strings.HasPrefix(strings.ToUpper(pr.Issue.Title), prefix) {
  666. return pr.Issue.Title[0:len(prefix)]
  667. }
  668. }
  669. return ""
  670. }
  671. // IsHeadEqualWithBranch returns if the commits of branchName are available in pull request head
  672. func (pr *PullRequest) IsHeadEqualWithBranch(branchName string) (bool, error) {
  673. var err error
  674. if err = pr.LoadBaseRepo(); err != nil {
  675. return false, err
  676. }
  677. baseGitRepo, err := git.OpenRepository(pr.BaseRepo.RepoPath())
  678. if err != nil {
  679. return false, err
  680. }
  681. baseCommit, err := baseGitRepo.GetBranchCommit(branchName)
  682. if err != nil {
  683. return false, err
  684. }
  685. if err = pr.LoadHeadRepo(); err != nil {
  686. return false, err
  687. }
  688. headGitRepo, err := git.OpenRepository(pr.HeadRepo.RepoPath())
  689. if err != nil {
  690. return false, err
  691. }
  692. headCommit, err := headGitRepo.GetBranchCommit(pr.HeadBranch)
  693. if err != nil {
  694. return false, err
  695. }
  696. return baseCommit.HasPreviousCommit(headCommit.ID)
  697. }
  698. // IsSameRepo returns true if base repo and head repo is the same
  699. func (pr *PullRequest) IsSameRepo() bool {
  700. return pr.BaseRepoID == pr.HeadRepoID
  701. }