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.

issue.go 47KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843
  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. "io"
  9. "mime/multipart"
  10. "os"
  11. "path"
  12. "strings"
  13. "time"
  14. api "code.gitea.io/sdk/gitea"
  15. "github.com/Unknwon/com"
  16. "github.com/go-xorm/xorm"
  17. gouuid "github.com/satori/go.uuid"
  18. "code.gitea.io/gitea/modules/base"
  19. "code.gitea.io/gitea/modules/log"
  20. "code.gitea.io/gitea/modules/setting"
  21. )
  22. var (
  23. errMissingIssueNumber = errors.New("No issue number specified")
  24. )
  25. // Issue represents an issue or pull request of repository.
  26. type Issue struct {
  27. ID int64 `xorm:"pk autoincr"`
  28. RepoID int64 `xorm:"INDEX UNIQUE(repo_index)"`
  29. Repo *Repository `xorm:"-"`
  30. Index int64 `xorm:"UNIQUE(repo_index)"` // Index in one repository.
  31. PosterID int64
  32. Poster *User `xorm:"-"`
  33. Title string `xorm:"name"`
  34. Content string `xorm:"TEXT"`
  35. RenderedContent string `xorm:"-"`
  36. Labels []*Label `xorm:"-"`
  37. MilestoneID int64
  38. Milestone *Milestone `xorm:"-"`
  39. Priority int
  40. AssigneeID int64
  41. Assignee *User `xorm:"-"`
  42. IsClosed bool
  43. IsRead bool `xorm:"-"`
  44. IsPull bool // Indicates whether is a pull request or not.
  45. PullRequest *PullRequest `xorm:"-"`
  46. NumComments int
  47. Deadline time.Time `xorm:"-"`
  48. DeadlineUnix int64
  49. Created time.Time `xorm:"-"`
  50. CreatedUnix int64
  51. Updated time.Time `xorm:"-"`
  52. UpdatedUnix int64
  53. Attachments []*Attachment `xorm:"-"`
  54. Comments []*Comment `xorm:"-"`
  55. }
  56. // BeforeInsert ...
  57. func (issue *Issue) BeforeInsert() {
  58. issue.CreatedUnix = time.Now().Unix()
  59. issue.UpdatedUnix = issue.CreatedUnix
  60. }
  61. // BeforeUpdate ...
  62. func (issue *Issue) BeforeUpdate() {
  63. issue.UpdatedUnix = time.Now().Unix()
  64. issue.DeadlineUnix = issue.Deadline.Unix()
  65. }
  66. // AfterSet ...
  67. func (issue *Issue) AfterSet(colName string, _ xorm.Cell) {
  68. switch colName {
  69. case "deadline_unix":
  70. issue.Deadline = time.Unix(issue.DeadlineUnix, 0).Local()
  71. case "created_unix":
  72. issue.Created = time.Unix(issue.CreatedUnix, 0).Local()
  73. case "updated_unix":
  74. issue.Updated = time.Unix(issue.UpdatedUnix, 0).Local()
  75. }
  76. }
  77. // loadAttributes ...
  78. func (issue *Issue) loadAttributes(e Engine) (err error) {
  79. if issue.Repo == nil {
  80. issue.Repo, err = getRepositoryByID(e, issue.RepoID)
  81. if err != nil {
  82. return fmt.Errorf("getRepositoryByID [%d]: %v", issue.RepoID, err)
  83. }
  84. }
  85. if issue.Poster == nil {
  86. issue.Poster, err = getUserByID(e, issue.PosterID)
  87. if err != nil {
  88. issue.PosterID = -1
  89. issue.Poster = NewGhostUser()
  90. if !IsErrUserNotExist(err) {
  91. return fmt.Errorf("getUserByID.(poster) [%d]: %v", issue.PosterID, err)
  92. }
  93. err = nil
  94. return
  95. }
  96. }
  97. if issue.Labels == nil {
  98. issue.Labels, err = getLabelsByIssueID(e, issue.ID)
  99. if err != nil {
  100. return fmt.Errorf("getLabelsByIssueID [%d]: %v", issue.ID, err)
  101. }
  102. }
  103. if issue.Milestone == nil && issue.MilestoneID > 0 {
  104. issue.Milestone, err = getMilestoneByRepoID(e, issue.RepoID, issue.MilestoneID)
  105. if err != nil {
  106. return fmt.Errorf("getMilestoneByRepoID [repo_id: %d, milestone_id: %d]: %v", issue.RepoID, issue.MilestoneID, err)
  107. }
  108. }
  109. if issue.Assignee == nil && issue.AssigneeID > 0 {
  110. issue.Assignee, err = getUserByID(e, issue.AssigneeID)
  111. if err != nil {
  112. return fmt.Errorf("getUserByID.(assignee) [%d]: %v", issue.AssigneeID, err)
  113. }
  114. }
  115. if issue.IsPull && issue.PullRequest == nil {
  116. // It is possible pull request is not yet created.
  117. issue.PullRequest, err = getPullRequestByIssueID(e, issue.ID)
  118. if err != nil && !IsErrPullRequestNotExist(err) {
  119. return fmt.Errorf("getPullRequestByIssueID [%d]: %v", issue.ID, err)
  120. }
  121. }
  122. if issue.Attachments == nil {
  123. issue.Attachments, err = getAttachmentsByIssueID(e, issue.ID)
  124. if err != nil {
  125. return fmt.Errorf("getAttachmentsByIssueID [%d]: %v", issue.ID, err)
  126. }
  127. }
  128. if issue.Comments == nil {
  129. issue.Comments, err = getCommentsByIssueID(e, issue.ID)
  130. if err != nil {
  131. return fmt.Errorf("getCommentsByIssueID [%d]: %v", issue.ID, err)
  132. }
  133. }
  134. return nil
  135. }
  136. // LoadAttributes ...
  137. func (issue *Issue) LoadAttributes() error {
  138. return issue.loadAttributes(x)
  139. }
  140. // HTMLURL ...
  141. func (issue *Issue) HTMLURL() string {
  142. var path string
  143. if issue.IsPull {
  144. path = "pulls"
  145. } else {
  146. path = "issues"
  147. }
  148. return fmt.Sprintf("%s/%s/%d", issue.Repo.HTMLURL(), path, issue.Index)
  149. }
  150. // State returns string representation of issue status.
  151. func (issue *Issue) State() api.StateType {
  152. if issue.IsClosed {
  153. return api.STATE_CLOSED
  154. }
  155. return api.STATE_OPEN
  156. }
  157. // APIFormat assumes some fields assigned with values:
  158. // Required - Poster, Labels,
  159. // Optional - Milestone, Assignee, PullRequest
  160. func (issue *Issue) APIFormat() *api.Issue {
  161. apiLabels := make([]*api.Label, len(issue.Labels))
  162. for i := range issue.Labels {
  163. apiLabels[i] = issue.Labels[i].APIFormat()
  164. }
  165. apiIssue := &api.Issue{
  166. ID: issue.ID,
  167. Index: issue.Index,
  168. Poster: issue.Poster.APIFormat(),
  169. Title: issue.Title,
  170. Body: issue.Content,
  171. Labels: apiLabels,
  172. State: issue.State(),
  173. Comments: issue.NumComments,
  174. Created: issue.Created,
  175. Updated: issue.Updated,
  176. }
  177. if issue.Milestone != nil {
  178. apiIssue.Milestone = issue.Milestone.APIFormat()
  179. }
  180. if issue.Assignee != nil {
  181. apiIssue.Assignee = issue.Assignee.APIFormat()
  182. }
  183. if issue.IsPull {
  184. apiIssue.PullRequest = &api.PullRequestMeta{
  185. HasMerged: issue.PullRequest.HasMerged,
  186. }
  187. if issue.PullRequest.HasMerged {
  188. apiIssue.PullRequest.Merged = &issue.PullRequest.Merged
  189. }
  190. }
  191. return apiIssue
  192. }
  193. // HashTag returns unique hash tag for issue.
  194. func (issue *Issue) HashTag() string {
  195. return "issue-" + com.ToStr(issue.ID)
  196. }
  197. // IsPoster returns true if given user by ID is the poster.
  198. func (issue *Issue) IsPoster(uid int64) bool {
  199. return issue.PosterID == uid
  200. }
  201. func (issue *Issue) hasLabel(e Engine, labelID int64) bool {
  202. return hasIssueLabel(e, issue.ID, labelID)
  203. }
  204. // HasLabel returns true if issue has been labeled by given ID.
  205. func (issue *Issue) HasLabel(labelID int64) bool {
  206. return issue.hasLabel(x, labelID)
  207. }
  208. // sendLabelUpdatedWebhook ...
  209. func (issue *Issue) sendLabelUpdatedWebhook(doer *User) {
  210. var err error
  211. if issue.IsPull {
  212. err = issue.PullRequest.LoadIssue()
  213. if err != nil {
  214. log.Error(4, "LoadIssue: %v", err)
  215. return
  216. }
  217. err = PrepareWebhooks(issue.Repo, HookEventPullRequest, &api.PullRequestPayload{
  218. Action: api.HookIssueLabelUpdated,
  219. Index: issue.Index,
  220. PullRequest: issue.PullRequest.APIFormat(),
  221. Repository: issue.Repo.APIFormat(nil),
  222. Sender: doer.APIFormat(),
  223. })
  224. }
  225. if err != nil {
  226. log.Error(4, "PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
  227. } else {
  228. go HookQueue.Add(issue.RepoID)
  229. }
  230. }
  231. // addLabel ...
  232. func (issue *Issue) addLabel(e *xorm.Session, label *Label) error {
  233. return newIssueLabel(e, issue, label)
  234. }
  235. // AddLabel adds a new label to the issue.
  236. func (issue *Issue) AddLabel(doer *User, label *Label) error {
  237. if err := NewIssueLabel(issue, label); err != nil {
  238. return err
  239. }
  240. issue.sendLabelUpdatedWebhook(doer)
  241. return nil
  242. }
  243. // addLabels ...
  244. func (issue *Issue) addLabels(e *xorm.Session, labels []*Label) error {
  245. return newIssueLabels(e, issue, labels)
  246. }
  247. // AddLabels adds a list of new labels to the issue.
  248. func (issue *Issue) AddLabels(doer *User, labels []*Label) error {
  249. if err := NewIssueLabels(issue, labels); err != nil {
  250. return err
  251. }
  252. issue.sendLabelUpdatedWebhook(doer)
  253. return nil
  254. }
  255. // getLabels ...
  256. func (issue *Issue) getLabels(e Engine) (err error) {
  257. if len(issue.Labels) > 0 {
  258. return nil
  259. }
  260. issue.Labels, err = getLabelsByIssueID(e, issue.ID)
  261. if err != nil {
  262. return fmt.Errorf("getLabelsByIssueID: %v", err)
  263. }
  264. return nil
  265. }
  266. // removeLabel ...
  267. func (issue *Issue) removeLabel(e *xorm.Session, label *Label) error {
  268. return deleteIssueLabel(e, issue, label)
  269. }
  270. // RemoveLabel removes a label from issue by given ID.
  271. func (issue *Issue) RemoveLabel(doer *User, label *Label) error {
  272. if err := DeleteIssueLabel(issue, label); err != nil {
  273. return err
  274. }
  275. issue.sendLabelUpdatedWebhook(doer)
  276. return nil
  277. }
  278. // clearLabels ...
  279. func (issue *Issue) clearLabels(e *xorm.Session) (err error) {
  280. if err = issue.getLabels(e); err != nil {
  281. return fmt.Errorf("getLabels: %v", err)
  282. }
  283. for i := range issue.Labels {
  284. if err = issue.removeLabel(e, issue.Labels[i]); err != nil {
  285. return fmt.Errorf("removeLabel: %v", err)
  286. }
  287. }
  288. return nil
  289. }
  290. // ClearLabels ...
  291. func (issue *Issue) ClearLabels(doer *User) (err error) {
  292. sess := x.NewSession()
  293. defer sessionRelease(sess)
  294. if err = sess.Begin(); err != nil {
  295. return err
  296. }
  297. if err = issue.clearLabels(sess); err != nil {
  298. return err
  299. }
  300. if err = sess.Commit(); err != nil {
  301. return fmt.Errorf("Commit: %v", err)
  302. }
  303. if issue.IsPull {
  304. err = issue.PullRequest.LoadIssue()
  305. if err != nil {
  306. log.Error(4, "LoadIssue: %v", err)
  307. return
  308. }
  309. err = PrepareWebhooks(issue.Repo, HookEventPullRequest, &api.PullRequestPayload{
  310. Action: api.HookIssueLabelCleared,
  311. Index: issue.Index,
  312. PullRequest: issue.PullRequest.APIFormat(),
  313. Repository: issue.Repo.APIFormat(nil),
  314. Sender: doer.APIFormat(),
  315. })
  316. }
  317. if err != nil {
  318. log.Error(4, "PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
  319. } else {
  320. go HookQueue.Add(issue.RepoID)
  321. }
  322. return nil
  323. }
  324. // ReplaceLabels removes all current labels and add new labels to the issue.
  325. func (issue *Issue) ReplaceLabels(labels []*Label) (err error) {
  326. sess := x.NewSession()
  327. defer sessionRelease(sess)
  328. if err = sess.Begin(); err != nil {
  329. return err
  330. }
  331. if err = issue.clearLabels(sess); err != nil {
  332. return fmt.Errorf("clearLabels: %v", err)
  333. } else if err = issue.addLabels(sess, labels); err != nil {
  334. return fmt.Errorf("addLabels: %v", err)
  335. }
  336. return sess.Commit()
  337. }
  338. // GetAssignee ...
  339. func (issue *Issue) GetAssignee() (err error) {
  340. if issue.AssigneeID == 0 || issue.Assignee != nil {
  341. return nil
  342. }
  343. issue.Assignee, err = GetUserByID(issue.AssigneeID)
  344. if IsErrUserNotExist(err) {
  345. return nil
  346. }
  347. return err
  348. }
  349. // ReadBy sets issue to be read by given user.
  350. func (issue *Issue) ReadBy(uid int64) error {
  351. return UpdateIssueUserByRead(uid, issue.ID)
  352. }
  353. func updateIssueCols(e Engine, issue *Issue, cols ...string) error {
  354. _, err := e.Id(issue.ID).Cols(cols...).Update(issue)
  355. return err
  356. }
  357. // UpdateIssueCols only updates values of specific columns for given issue.
  358. func UpdateIssueCols(issue *Issue, cols ...string) error {
  359. return updateIssueCols(x, issue, cols...)
  360. }
  361. // changeStatus ...
  362. func (issue *Issue) changeStatus(e *xorm.Session, doer *User, repo *Repository, isClosed bool) (err error) {
  363. // Nothing should be performed if current status is same as target status
  364. if issue.IsClosed == isClosed {
  365. return nil
  366. }
  367. issue.IsClosed = isClosed
  368. if err = updateIssueCols(e, issue, "is_closed"); err != nil {
  369. return err
  370. } else if err = updateIssueUsersByStatus(e, issue.ID, isClosed); err != nil {
  371. return err
  372. }
  373. // Update issue count of labels
  374. if err = issue.getLabels(e); err != nil {
  375. return err
  376. }
  377. for idx := range issue.Labels {
  378. if issue.IsClosed {
  379. issue.Labels[idx].NumClosedIssues++
  380. } else {
  381. issue.Labels[idx].NumClosedIssues--
  382. }
  383. if err = updateLabel(e, issue.Labels[idx]); err != nil {
  384. return err
  385. }
  386. }
  387. // Update issue count of milestone
  388. if err = changeMilestoneIssueStats(e, issue); err != nil {
  389. return err
  390. }
  391. // New action comment
  392. if _, err = createStatusComment(e, doer, repo, issue); err != nil {
  393. return err
  394. }
  395. return nil
  396. }
  397. // ChangeStatus changes issue status to open or closed.
  398. func (issue *Issue) ChangeStatus(doer *User, repo *Repository, isClosed bool) (err error) {
  399. sess := x.NewSession()
  400. defer sessionRelease(sess)
  401. if err = sess.Begin(); err != nil {
  402. return err
  403. }
  404. if err = issue.changeStatus(sess, doer, repo, isClosed); err != nil {
  405. return err
  406. }
  407. if err = sess.Commit(); err != nil {
  408. return fmt.Errorf("Commit: %v", err)
  409. }
  410. if issue.IsPull {
  411. // Merge pull request calls issue.changeStatus so we need to handle separately.
  412. issue.PullRequest.Issue = issue
  413. apiPullRequest := &api.PullRequestPayload{
  414. Index: issue.Index,
  415. PullRequest: issue.PullRequest.APIFormat(),
  416. Repository: repo.APIFormat(nil),
  417. Sender: doer.APIFormat(),
  418. }
  419. if isClosed {
  420. apiPullRequest.Action = api.HookIssueClosed
  421. } else {
  422. apiPullRequest.Action = api.HookIssueReopened
  423. }
  424. err = PrepareWebhooks(repo, HookEventPullRequest, apiPullRequest)
  425. }
  426. if err != nil {
  427. log.Error(4, "PrepareWebhooks [is_pull: %v, is_closed: %v]: %v", issue.IsPull, isClosed, err)
  428. } else {
  429. go HookQueue.Add(repo.ID)
  430. }
  431. return nil
  432. }
  433. // ChangeTitle ...
  434. func (issue *Issue) ChangeTitle(doer *User, title string) (err error) {
  435. oldTitle := issue.Title
  436. issue.Title = title
  437. if err = UpdateIssueCols(issue, "name"); err != nil {
  438. return fmt.Errorf("UpdateIssueCols: %v", err)
  439. }
  440. if issue.IsPull {
  441. issue.PullRequest.Issue = issue
  442. err = PrepareWebhooks(issue.Repo, HookEventPullRequest, &api.PullRequestPayload{
  443. Action: api.HookIssueEdited,
  444. Index: issue.Index,
  445. Changes: &api.ChangesPayload{
  446. Title: &api.ChangesFromPayload{
  447. From: oldTitle,
  448. },
  449. },
  450. PullRequest: issue.PullRequest.APIFormat(),
  451. Repository: issue.Repo.APIFormat(nil),
  452. Sender: doer.APIFormat(),
  453. })
  454. }
  455. if err != nil {
  456. log.Error(4, "PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
  457. } else {
  458. go HookQueue.Add(issue.RepoID)
  459. }
  460. return nil
  461. }
  462. // ChangeContent ...
  463. func (issue *Issue) ChangeContent(doer *User, content string) (err error) {
  464. oldContent := issue.Content
  465. issue.Content = content
  466. if err = UpdateIssueCols(issue, "content"); err != nil {
  467. return fmt.Errorf("UpdateIssueCols: %v", err)
  468. }
  469. if issue.IsPull {
  470. issue.PullRequest.Issue = issue
  471. err = PrepareWebhooks(issue.Repo, HookEventPullRequest, &api.PullRequestPayload{
  472. Action: api.HookIssueEdited,
  473. Index: issue.Index,
  474. Changes: &api.ChangesPayload{
  475. Body: &api.ChangesFromPayload{
  476. From: oldContent,
  477. },
  478. },
  479. PullRequest: issue.PullRequest.APIFormat(),
  480. Repository: issue.Repo.APIFormat(nil),
  481. Sender: doer.APIFormat(),
  482. })
  483. }
  484. if err != nil {
  485. log.Error(4, "PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
  486. } else {
  487. go HookQueue.Add(issue.RepoID)
  488. }
  489. return nil
  490. }
  491. // ChangeAssignee ...
  492. func (issue *Issue) ChangeAssignee(doer *User, assigneeID int64) (err error) {
  493. issue.AssigneeID = assigneeID
  494. if err = UpdateIssueUserByAssignee(issue); err != nil {
  495. return fmt.Errorf("UpdateIssueUserByAssignee: %v", err)
  496. }
  497. issue.Assignee, err = GetUserByID(issue.AssigneeID)
  498. if err != nil && !IsErrUserNotExist(err) {
  499. log.Error(4, "GetUserByID [assignee_id: %v]: %v", issue.AssigneeID, err)
  500. return nil
  501. }
  502. // Error not nil here means user does not exist, which is remove assignee.
  503. isRemoveAssignee := err != nil
  504. if issue.IsPull {
  505. issue.PullRequest.Issue = issue
  506. apiPullRequest := &api.PullRequestPayload{
  507. Index: issue.Index,
  508. PullRequest: issue.PullRequest.APIFormat(),
  509. Repository: issue.Repo.APIFormat(nil),
  510. Sender: doer.APIFormat(),
  511. }
  512. if isRemoveAssignee {
  513. apiPullRequest.Action = api.HookIssueUnassigned
  514. } else {
  515. apiPullRequest.Action = api.HookIssueAssigned
  516. }
  517. err = PrepareWebhooks(issue.Repo, HookEventPullRequest, apiPullRequest)
  518. }
  519. if err != nil {
  520. log.Error(4, "PrepareWebhooks [is_pull: %v, remove_assignee: %v]: %v", issue.IsPull, isRemoveAssignee, err)
  521. } else {
  522. go HookQueue.Add(issue.RepoID)
  523. }
  524. return nil
  525. }
  526. // NewIssueOptions ...
  527. type NewIssueOptions struct {
  528. Repo *Repository
  529. Issue *Issue
  530. LableIDs []int64
  531. Attachments []string // In UUID format.
  532. IsPull bool
  533. }
  534. func newIssue(e *xorm.Session, opts NewIssueOptions) (err error) {
  535. opts.Issue.Title = strings.TrimSpace(opts.Issue.Title)
  536. opts.Issue.Index = opts.Repo.NextIssueIndex()
  537. if opts.Issue.MilestoneID > 0 {
  538. milestone, err := getMilestoneByRepoID(e, opts.Issue.RepoID, opts.Issue.MilestoneID)
  539. if err != nil && !IsErrMilestoneNotExist(err) {
  540. return fmt.Errorf("getMilestoneByID: %v", err)
  541. }
  542. // Assume milestone is invalid and drop silently.
  543. opts.Issue.MilestoneID = 0
  544. if milestone != nil {
  545. opts.Issue.MilestoneID = milestone.ID
  546. opts.Issue.Milestone = milestone
  547. if err = changeMilestoneAssign(e, opts.Issue, -1); err != nil {
  548. return err
  549. }
  550. }
  551. }
  552. if opts.Issue.AssigneeID > 0 {
  553. assignee, err := getUserByID(e, opts.Issue.AssigneeID)
  554. if err != nil && !IsErrUserNotExist(err) {
  555. return fmt.Errorf("getUserByID: %v", err)
  556. }
  557. // Assume assignee is invalid and drop silently.
  558. opts.Issue.AssigneeID = 0
  559. if assignee != nil {
  560. valid, err := hasAccess(e, assignee, opts.Repo, AccessModeWrite)
  561. if err != nil {
  562. return fmt.Errorf("hasAccess [user_id: %d, repo_id: %d]: %v", assignee.ID, opts.Repo.ID, err)
  563. }
  564. if valid {
  565. opts.Issue.AssigneeID = assignee.ID
  566. opts.Issue.Assignee = assignee
  567. }
  568. }
  569. }
  570. // Milestone and assignee validation should happen before insert actual object.
  571. if _, err = e.Insert(opts.Issue); err != nil {
  572. return err
  573. }
  574. if opts.IsPull {
  575. _, err = e.Exec("UPDATE `repository` SET num_pulls = num_pulls + 1 WHERE id = ?", opts.Issue.RepoID)
  576. } else {
  577. _, err = e.Exec("UPDATE `repository` SET num_issues = num_issues + 1 WHERE id = ?", opts.Issue.RepoID)
  578. }
  579. if err != nil {
  580. return err
  581. }
  582. if len(opts.LableIDs) > 0 {
  583. // During the session, SQLite3 dirver cannot handle retrieve objects after update something.
  584. // So we have to get all needed labels first.
  585. labels := make([]*Label, 0, len(opts.LableIDs))
  586. if err = e.In("id", opts.LableIDs).Find(&labels); err != nil {
  587. return fmt.Errorf("find all labels [label_ids: %v]: %v", opts.LableIDs, err)
  588. }
  589. for _, label := range labels {
  590. // Silently drop invalid labels.
  591. if label.RepoID != opts.Repo.ID {
  592. continue
  593. }
  594. if err = opts.Issue.addLabel(e, label); err != nil {
  595. return fmt.Errorf("addLabel [id: %d]: %v", label.ID, err)
  596. }
  597. }
  598. }
  599. if err = newIssueUsers(e, opts.Repo, opts.Issue); err != nil {
  600. return err
  601. }
  602. if len(opts.Attachments) > 0 {
  603. attachments, err := getAttachmentsByUUIDs(e, opts.Attachments)
  604. if err != nil {
  605. return fmt.Errorf("getAttachmentsByUUIDs [uuids: %v]: %v", opts.Attachments, err)
  606. }
  607. for i := 0; i < len(attachments); i++ {
  608. attachments[i].IssueID = opts.Issue.ID
  609. if _, err = e.Id(attachments[i].ID).Update(attachments[i]); err != nil {
  610. return fmt.Errorf("update attachment [id: %d]: %v", attachments[i].ID, err)
  611. }
  612. }
  613. }
  614. return opts.Issue.loadAttributes(e)
  615. }
  616. // NewIssue creates new issue with labels for repository.
  617. func NewIssue(repo *Repository, issue *Issue, labelIDs []int64, uuids []string) (err error) {
  618. sess := x.NewSession()
  619. defer sessionRelease(sess)
  620. if err = sess.Begin(); err != nil {
  621. return err
  622. }
  623. if err = newIssue(sess, NewIssueOptions{
  624. Repo: repo,
  625. Issue: issue,
  626. LableIDs: labelIDs,
  627. Attachments: uuids,
  628. }); err != nil {
  629. return fmt.Errorf("newIssue: %v", err)
  630. }
  631. if err = sess.Commit(); err != nil {
  632. return fmt.Errorf("Commit: %v", err)
  633. }
  634. if err = NotifyWatchers(&Action{
  635. ActUserID: issue.Poster.ID,
  636. ActUserName: issue.Poster.Name,
  637. OpType: ActionCreateIssue,
  638. Content: fmt.Sprintf("%d|%s", issue.Index, issue.Title),
  639. RepoID: repo.ID,
  640. RepoUserName: repo.Owner.Name,
  641. RepoName: repo.Name,
  642. IsPrivate: repo.IsPrivate,
  643. }); err != nil {
  644. log.Error(4, "NotifyWatchers: %v", err)
  645. }
  646. if err = issue.MailParticipants(); err != nil {
  647. log.Error(4, "MailParticipants: %v", err)
  648. }
  649. return nil
  650. }
  651. // GetIssueByRef returns an Issue specified by a GFM reference.
  652. // See https://help.github.com/articles/writing-on-github#references for more information on the syntax.
  653. func GetIssueByRef(ref string) (*Issue, error) {
  654. n := strings.IndexByte(ref, byte('#'))
  655. if n == -1 {
  656. return nil, errMissingIssueNumber
  657. }
  658. index, err := com.StrTo(ref[n+1:]).Int64()
  659. if err != nil {
  660. return nil, err
  661. }
  662. repo, err := GetRepositoryByRef(ref[:n])
  663. if err != nil {
  664. return nil, err
  665. }
  666. issue, err := GetIssueByIndex(repo.ID, index)
  667. if err != nil {
  668. return nil, err
  669. }
  670. return issue, issue.LoadAttributes()
  671. }
  672. // GetRawIssueByIndex returns raw issue without loading attributes by index in a repository.
  673. func GetRawIssueByIndex(repoID, index int64) (*Issue, error) {
  674. issue := &Issue{
  675. RepoID: repoID,
  676. Index: index,
  677. }
  678. has, err := x.Get(issue)
  679. if err != nil {
  680. return nil, err
  681. } else if !has {
  682. return nil, ErrIssueNotExist{0, repoID, index}
  683. }
  684. return issue, nil
  685. }
  686. // GetIssueByIndex returns issue by index in a repository.
  687. func GetIssueByIndex(repoID, index int64) (*Issue, error) {
  688. issue, err := GetRawIssueByIndex(repoID, index)
  689. if err != nil {
  690. return nil, err
  691. }
  692. return issue, issue.LoadAttributes()
  693. }
  694. func getIssueByID(e Engine, id int64) (*Issue, error) {
  695. issue := new(Issue)
  696. has, err := e.Id(id).Get(issue)
  697. if err != nil {
  698. return nil, err
  699. } else if !has {
  700. return nil, ErrIssueNotExist{id, 0, 0}
  701. }
  702. return issue, issue.LoadAttributes()
  703. }
  704. // GetIssueByID returns an issue by given ID.
  705. func GetIssueByID(id int64) (*Issue, error) {
  706. return getIssueByID(x, id)
  707. }
  708. // IssuesOptions ...
  709. type IssuesOptions struct {
  710. UserID int64
  711. AssigneeID int64
  712. RepoID int64
  713. PosterID int64
  714. MilestoneID int64
  715. RepoIDs []int64
  716. Page int
  717. IsClosed bool
  718. IsMention bool
  719. IsPull bool
  720. Labels string
  721. SortType string
  722. }
  723. // Issues returns a list of issues by given conditions.
  724. func Issues(opts *IssuesOptions) ([]*Issue, error) {
  725. if opts.Page <= 0 {
  726. opts.Page = 1
  727. }
  728. sess := x.Limit(setting.UI.IssuePagingNum, (opts.Page-1)*setting.UI.IssuePagingNum)
  729. if opts.RepoID > 0 {
  730. sess.And("issue.repo_id=?", opts.RepoID)
  731. } else if len(opts.RepoIDs) > 0 {
  732. // In case repository IDs are provided but actually no repository has issue.
  733. sess.In("issue.repo_id", opts.RepoIDs)
  734. }
  735. sess.And("issue.is_closed=?", opts.IsClosed)
  736. if opts.AssigneeID > 0 {
  737. sess.And("issue.assignee_id=?", opts.AssigneeID)
  738. } else if opts.PosterID > 0 {
  739. sess.And("issue.poster_id=?", opts.PosterID)
  740. }
  741. if opts.MilestoneID > 0 {
  742. sess.And("issue.milestone_id=?", opts.MilestoneID)
  743. }
  744. sess.And("issue.is_pull=?", opts.IsPull)
  745. switch opts.SortType {
  746. case "oldest":
  747. sess.Asc("issue.created_unix")
  748. case "recentupdate":
  749. sess.Desc("issue.updated_unix")
  750. case "leastupdate":
  751. sess.Asc("issue.updated_unix")
  752. case "mostcomment":
  753. sess.Desc("issue.num_comments")
  754. case "leastcomment":
  755. sess.Asc("issue.num_comments")
  756. case "priority":
  757. sess.Desc("issue.priority")
  758. default:
  759. sess.Desc("issue.created_unix")
  760. }
  761. if len(opts.Labels) > 0 && opts.Labels != "0" {
  762. labelIDs := base.StringsToInt64s(strings.Split(opts.Labels, ","))
  763. if len(labelIDs) > 0 {
  764. sess.
  765. Join("INNER", "issue_label", "issue.id = issue_label.issue_id").
  766. In("issue_label.label_id", labelIDs)
  767. }
  768. }
  769. if opts.IsMention {
  770. sess.
  771. Join("INNER", "issue_user", "issue.id = issue_user.issue_id").
  772. And("issue_user.is_mentioned = ?", true)
  773. if opts.UserID > 0 {
  774. sess.And("issue_user.uid = ?", opts.UserID)
  775. }
  776. }
  777. issues := make([]*Issue, 0, setting.UI.IssuePagingNum)
  778. if err := sess.Find(&issues); err != nil {
  779. return nil, fmt.Errorf("Find: %v", err)
  780. }
  781. // FIXME: use IssueList to improve performance.
  782. for i := range issues {
  783. if err := issues[i].LoadAttributes(); err != nil {
  784. return nil, fmt.Errorf("LoadAttributes [%d]: %v", issues[i].ID, err)
  785. }
  786. }
  787. return issues, nil
  788. }
  789. // .___ ____ ___
  790. // | | ______ ________ __ ____ | | \______ ___________
  791. // | |/ ___// ___/ | \_/ __ \| | / ___// __ \_ __ \
  792. // | |\___ \ \___ \| | /\ ___/| | /\___ \\ ___/| | \/
  793. // |___/____ >____ >____/ \___ >______//____ >\___ >__|
  794. // \/ \/ \/ \/ \/
  795. // IssueUser represents an issue-user relation.
  796. type IssueUser struct {
  797. ID int64 `xorm:"pk autoincr"`
  798. UID int64 `xorm:"INDEX"` // User ID.
  799. IssueID int64
  800. RepoID int64 `xorm:"INDEX"`
  801. MilestoneID int64
  802. IsRead bool
  803. IsAssigned bool
  804. IsMentioned bool
  805. IsPoster bool
  806. IsClosed bool
  807. }
  808. func newIssueUsers(e *xorm.Session, repo *Repository, issue *Issue) error {
  809. assignees, err := repo.getAssignees(e)
  810. if err != nil {
  811. return fmt.Errorf("getAssignees: %v", err)
  812. }
  813. // Poster can be anyone, append later if not one of assignees.
  814. isPosterAssignee := false
  815. // Leave a seat for poster itself to append later, but if poster is one of assignee
  816. // and just waste 1 unit is cheaper than re-allocate memory once.
  817. issueUsers := make([]*IssueUser, 0, len(assignees)+1)
  818. for _, assignee := range assignees {
  819. isPoster := assignee.ID == issue.PosterID
  820. issueUsers = append(issueUsers, &IssueUser{
  821. IssueID: issue.ID,
  822. RepoID: repo.ID,
  823. UID: assignee.ID,
  824. IsPoster: isPoster,
  825. IsAssigned: assignee.ID == issue.AssigneeID,
  826. })
  827. if !isPosterAssignee && isPoster {
  828. isPosterAssignee = true
  829. }
  830. }
  831. if !isPosterAssignee {
  832. issueUsers = append(issueUsers, &IssueUser{
  833. IssueID: issue.ID,
  834. RepoID: repo.ID,
  835. UID: issue.PosterID,
  836. IsPoster: true,
  837. })
  838. }
  839. if _, err = e.Insert(issueUsers); err != nil {
  840. return err
  841. }
  842. return nil
  843. }
  844. // NewIssueUsers adds new issue-user relations for new issue of repository.
  845. func NewIssueUsers(repo *Repository, issue *Issue) (err error) {
  846. sess := x.NewSession()
  847. defer sessionRelease(sess)
  848. if err = sess.Begin(); err != nil {
  849. return err
  850. }
  851. if err = newIssueUsers(sess, repo, issue); err != nil {
  852. return err
  853. }
  854. return sess.Commit()
  855. }
  856. // PairsContains returns true when pairs list contains given issue.
  857. func PairsContains(ius []*IssueUser, issueID, uid int64) int {
  858. for i := range ius {
  859. if ius[i].IssueID == issueID &&
  860. ius[i].UID == uid {
  861. return i
  862. }
  863. }
  864. return -1
  865. }
  866. // GetIssueUsers returns issue-user pairs by given repository and user.
  867. func GetIssueUsers(rid, uid int64, isClosed bool) ([]*IssueUser, error) {
  868. ius := make([]*IssueUser, 0, 10)
  869. err := x.Where("is_closed=?", isClosed).Find(&ius, &IssueUser{RepoID: rid, UID: uid})
  870. return ius, err
  871. }
  872. // GetIssueUserPairsByRepoIds returns issue-user pairs by given repository IDs.
  873. func GetIssueUserPairsByRepoIds(rids []int64, isClosed bool, page int) ([]*IssueUser, error) {
  874. if len(rids) == 0 {
  875. return []*IssueUser{}, nil
  876. }
  877. ius := make([]*IssueUser, 0, 10)
  878. sess := x.
  879. Limit(20, (page-1)*20).
  880. Where("is_closed=?", isClosed).
  881. In("repo_id", rids)
  882. err := sess.Find(&ius)
  883. return ius, err
  884. }
  885. // GetIssueUserPairsByMode returns issue-user pairs by given repository and user.
  886. func GetIssueUserPairsByMode(uid, rid int64, isClosed bool, page, filterMode int) ([]*IssueUser, error) {
  887. ius := make([]*IssueUser, 0, 10)
  888. sess := x.
  889. Limit(20, (page-1)*20).
  890. Where("uid=?", uid).
  891. And("is_closed=?", isClosed)
  892. if rid > 0 {
  893. sess.And("repo_id=?", rid)
  894. }
  895. switch filterMode {
  896. case FilterModeAssign:
  897. sess.And("is_assigned=?", true)
  898. case FilterModeCreate:
  899. sess.And("is_poster=?", true)
  900. default:
  901. return ius, nil
  902. }
  903. err := sess.Find(&ius)
  904. return ius, err
  905. }
  906. // UpdateIssueMentions extracts mentioned people from content and
  907. // updates issue-user relations for them.
  908. func UpdateIssueMentions(issueID int64, mentions []string) error {
  909. if len(mentions) == 0 {
  910. return nil
  911. }
  912. for i := range mentions {
  913. mentions[i] = strings.ToLower(mentions[i])
  914. }
  915. users := make([]*User, 0, len(mentions))
  916. if err := x.In("lower_name", mentions).Asc("lower_name").Find(&users); err != nil {
  917. return fmt.Errorf("find mentioned users: %v", err)
  918. }
  919. ids := make([]int64, 0, len(mentions))
  920. for _, user := range users {
  921. ids = append(ids, user.ID)
  922. if !user.IsOrganization() || user.NumMembers == 0 {
  923. continue
  924. }
  925. memberIDs := make([]int64, 0, user.NumMembers)
  926. orgUsers, err := GetOrgUsersByOrgID(user.ID)
  927. if err != nil {
  928. return fmt.Errorf("GetOrgUsersByOrgID [%d]: %v", user.ID, err)
  929. }
  930. for _, orgUser := range orgUsers {
  931. memberIDs = append(memberIDs, orgUser.ID)
  932. }
  933. ids = append(ids, memberIDs...)
  934. }
  935. if err := UpdateIssueUsersByMentions(issueID, ids); err != nil {
  936. return fmt.Errorf("UpdateIssueUsersByMentions: %v", err)
  937. }
  938. return nil
  939. }
  940. // IssueStats represents issue statistic information.
  941. type IssueStats struct {
  942. OpenCount, ClosedCount int64
  943. AllCount int64
  944. AssignCount int64
  945. CreateCount int64
  946. MentionCount int64
  947. }
  948. // Filter modes.
  949. const (
  950. FilterModeAll = iota
  951. FilterModeAssign
  952. FilterModeCreate
  953. FilterModeMention
  954. )
  955. func parseCountResult(results []map[string][]byte) int64 {
  956. if len(results) == 0 {
  957. return 0
  958. }
  959. for _, result := range results[0] {
  960. return com.StrTo(string(result)).MustInt64()
  961. }
  962. return 0
  963. }
  964. // IssueStatsOptions ...
  965. type IssueStatsOptions struct {
  966. RepoID int64
  967. UserID int64
  968. Labels string
  969. MilestoneID int64
  970. AssigneeID int64
  971. FilterMode int
  972. IsPull bool
  973. }
  974. // GetIssueStats returns issue statistic information by given conditions.
  975. func GetIssueStats(opts *IssueStatsOptions) *IssueStats {
  976. stats := &IssueStats{}
  977. countSession := func(opts *IssueStatsOptions) *xorm.Session {
  978. sess := x.
  979. Where("issue.repo_id = ?", opts.RepoID).
  980. And("is_pull = ?", opts.IsPull)
  981. if len(opts.Labels) > 0 && opts.Labels != "0" {
  982. labelIDs := base.StringsToInt64s(strings.Split(opts.Labels, ","))
  983. if len(labelIDs) > 0 {
  984. sess.
  985. Join("INNER", "issue_label", "issue.id = issue_id").
  986. In("label_id", labelIDs)
  987. }
  988. }
  989. if opts.MilestoneID > 0 {
  990. sess.And("issue.milestone_id = ?", opts.MilestoneID)
  991. }
  992. if opts.AssigneeID > 0 {
  993. sess.And("assignee_id = ?", opts.AssigneeID)
  994. }
  995. return sess
  996. }
  997. switch opts.FilterMode {
  998. case FilterModeAll, FilterModeAssign:
  999. stats.OpenCount, _ = countSession(opts).
  1000. And("is_closed = ?", false).
  1001. Count(&Issue{})
  1002. stats.ClosedCount, _ = countSession(opts).
  1003. And("is_closed = ?", true).
  1004. Count(&Issue{})
  1005. case FilterModeCreate:
  1006. stats.OpenCount, _ = countSession(opts).
  1007. And("poster_id = ?", opts.UserID).
  1008. And("is_closed = ?", false).
  1009. Count(&Issue{})
  1010. stats.ClosedCount, _ = countSession(opts).
  1011. And("poster_id = ?", opts.UserID).
  1012. And("is_closed = ?", true).
  1013. Count(&Issue{})
  1014. case FilterModeMention:
  1015. stats.OpenCount, _ = countSession(opts).
  1016. Join("INNER", "issue_user", "issue.id = issue_user.issue_id").
  1017. And("issue_user.uid = ?", opts.UserID).
  1018. And("issue_user.is_mentioned = ?", true).
  1019. And("issue.is_closed = ?", false).
  1020. Count(&Issue{})
  1021. stats.ClosedCount, _ = countSession(opts).
  1022. Join("INNER", "issue_user", "issue.id = issue_user.issue_id").
  1023. And("issue_user.uid = ?", opts.UserID).
  1024. And("issue_user.is_mentioned = ?", true).
  1025. And("issue.is_closed = ?", true).
  1026. Count(&Issue{})
  1027. }
  1028. return stats
  1029. }
  1030. // GetUserIssueStats returns issue statistic information for dashboard by given conditions.
  1031. func GetUserIssueStats(repoID, uid int64, repoIDs []int64, filterMode int, isPull bool) *IssueStats {
  1032. stats := &IssueStats{}
  1033. countSession := func(isClosed, isPull bool, repoID int64, repoIDs []int64) *xorm.Session {
  1034. sess := x.
  1035. Where("issue.is_closed = ?", isClosed).
  1036. And("issue.is_pull = ?", isPull)
  1037. if repoID > 0 {
  1038. sess.And("repo_id = ?", repoID)
  1039. } else if len(repoIDs) > 0 {
  1040. sess.In("repo_id", repoIDs)
  1041. }
  1042. return sess
  1043. }
  1044. stats.AssignCount, _ = countSession(false, isPull, repoID, repoIDs).
  1045. And("assignee_id = ?", uid).
  1046. Count(&Issue{})
  1047. stats.CreateCount, _ = countSession(false, isPull, repoID, repoIDs).
  1048. And("poster_id = ?", uid).
  1049. Count(&Issue{})
  1050. openCountSession := countSession(false, isPull, repoID, repoIDs)
  1051. closedCountSession := countSession(true, isPull, repoID, repoIDs)
  1052. switch filterMode {
  1053. case FilterModeAssign:
  1054. openCountSession.And("assignee_id = ?", uid)
  1055. closedCountSession.And("assignee_id = ?", uid)
  1056. case FilterModeCreate:
  1057. openCountSession.And("poster_id = ?", uid)
  1058. closedCountSession.And("poster_id = ?", uid)
  1059. }
  1060. stats.OpenCount, _ = openCountSession.Count(&Issue{})
  1061. stats.ClosedCount, _ = closedCountSession.Count(&Issue{})
  1062. return stats
  1063. }
  1064. // GetRepoIssueStats returns number of open and closed repository issues by given filter mode.
  1065. func GetRepoIssueStats(repoID, uid int64, filterMode int, isPull bool) (numOpen int64, numClosed int64) {
  1066. countSession := func(isClosed, isPull bool, repoID int64) *xorm.Session {
  1067. sess := x.
  1068. Where("issue.repo_id = ?", isClosed).
  1069. And("is_pull = ?", isPull).
  1070. And("repo_id = ?", repoID)
  1071. return sess
  1072. }
  1073. openCountSession := countSession(false, isPull, repoID)
  1074. closedCountSession := countSession(true, isPull, repoID)
  1075. switch filterMode {
  1076. case FilterModeAssign:
  1077. openCountSession.And("assignee_id = ?", uid)
  1078. closedCountSession.And("assignee_id = ?", uid)
  1079. case FilterModeCreate:
  1080. openCountSession.And("poster_id = ?", uid)
  1081. closedCountSession.And("poster_id = ?", uid)
  1082. }
  1083. openResult, _ := openCountSession.Count(&Issue{})
  1084. closedResult, _ := closedCountSession.Count(&Issue{})
  1085. return openResult, closedResult
  1086. }
  1087. func updateIssue(e Engine, issue *Issue) error {
  1088. _, err := e.Id(issue.ID).AllCols().Update(issue)
  1089. return err
  1090. }
  1091. // UpdateIssue updates all fields of given issue.
  1092. func UpdateIssue(issue *Issue) error {
  1093. return updateIssue(x, issue)
  1094. }
  1095. func updateIssueUsersByStatus(e Engine, issueID int64, isClosed bool) error {
  1096. _, err := e.Exec("UPDATE `issue_user` SET is_closed=? WHERE issue_id=?", isClosed, issueID)
  1097. return err
  1098. }
  1099. // UpdateIssueUsersByStatus updates issue-user relations by issue status.
  1100. func UpdateIssueUsersByStatus(issueID int64, isClosed bool) error {
  1101. return updateIssueUsersByStatus(x, issueID, isClosed)
  1102. }
  1103. func updateIssueUserByAssignee(e *xorm.Session, issue *Issue) (err error) {
  1104. if _, err = e.Exec("UPDATE `issue_user` SET is_assigned = ? WHERE issue_id = ?", false, issue.ID); err != nil {
  1105. return err
  1106. }
  1107. // Assignee ID equals to 0 means clear assignee.
  1108. if issue.AssigneeID > 0 {
  1109. if _, err = e.Exec("UPDATE `issue_user` SET is_assigned = ? WHERE uid = ? AND issue_id = ?", true, issue.AssigneeID, issue.ID); err != nil {
  1110. return err
  1111. }
  1112. }
  1113. return updateIssue(e, issue)
  1114. }
  1115. // UpdateIssueUserByAssignee updates issue-user relation for assignee.
  1116. func UpdateIssueUserByAssignee(issue *Issue) (err error) {
  1117. sess := x.NewSession()
  1118. defer sessionRelease(sess)
  1119. if err = sess.Begin(); err != nil {
  1120. return err
  1121. }
  1122. if err = updateIssueUserByAssignee(sess, issue); err != nil {
  1123. return err
  1124. }
  1125. return sess.Commit()
  1126. }
  1127. // UpdateIssueUserByRead updates issue-user relation for reading.
  1128. func UpdateIssueUserByRead(uid, issueID int64) error {
  1129. _, err := x.Exec("UPDATE `issue_user` SET is_read=? WHERE uid=? AND issue_id=?", true, uid, issueID)
  1130. return err
  1131. }
  1132. // UpdateIssueUsersByMentions updates issue-user pairs by mentioning.
  1133. func UpdateIssueUsersByMentions(issueID int64, uids []int64) error {
  1134. for _, uid := range uids {
  1135. iu := &IssueUser{
  1136. UID: uid,
  1137. IssueID: issueID,
  1138. }
  1139. has, err := x.Get(iu)
  1140. if err != nil {
  1141. return err
  1142. }
  1143. iu.IsMentioned = true
  1144. if has {
  1145. _, err = x.Id(iu.ID).AllCols().Update(iu)
  1146. } else {
  1147. _, err = x.Insert(iu)
  1148. }
  1149. if err != nil {
  1150. return err
  1151. }
  1152. }
  1153. return nil
  1154. }
  1155. // _____ .__.__ __
  1156. // / \ |__| | ____ _______/ |_ ____ ____ ____
  1157. // / \ / \| | | _/ __ \ / ___/\ __\/ _ \ / \_/ __ \
  1158. // / Y \ | |_\ ___/ \___ \ | | ( <_> ) | \ ___/
  1159. // \____|__ /__|____/\___ >____ > |__| \____/|___| /\___ >
  1160. // \/ \/ \/ \/ \/
  1161. // Milestone represents a milestone of repository.
  1162. type Milestone struct {
  1163. ID int64 `xorm:"pk autoincr"`
  1164. RepoID int64 `xorm:"INDEX"`
  1165. Name string
  1166. Content string `xorm:"TEXT"`
  1167. RenderedContent string `xorm:"-"`
  1168. IsClosed bool
  1169. NumIssues int
  1170. NumClosedIssues int
  1171. NumOpenIssues int `xorm:"-"`
  1172. Completeness int // Percentage(1-100).
  1173. IsOverDue bool `xorm:"-"`
  1174. DeadlineString string `xorm:"-"`
  1175. Deadline time.Time `xorm:"-"`
  1176. DeadlineUnix int64
  1177. ClosedDate time.Time `xorm:"-"`
  1178. ClosedDateUnix int64
  1179. }
  1180. // BeforeInsert ...
  1181. func (m *Milestone) BeforeInsert() {
  1182. m.DeadlineUnix = m.Deadline.Unix()
  1183. }
  1184. // BeforeUpdate ...
  1185. func (m *Milestone) BeforeUpdate() {
  1186. if m.NumIssues > 0 {
  1187. m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
  1188. } else {
  1189. m.Completeness = 0
  1190. }
  1191. m.DeadlineUnix = m.Deadline.Unix()
  1192. m.ClosedDateUnix = m.ClosedDate.Unix()
  1193. }
  1194. // AfterSet ...
  1195. func (m *Milestone) AfterSet(colName string, _ xorm.Cell) {
  1196. switch colName {
  1197. case "num_closed_issues":
  1198. m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
  1199. case "deadline_unix":
  1200. m.Deadline = time.Unix(m.DeadlineUnix, 0).Local()
  1201. if m.Deadline.Year() == 9999 {
  1202. return
  1203. }
  1204. m.DeadlineString = m.Deadline.Format("2006-01-02")
  1205. if time.Now().Local().After(m.Deadline) {
  1206. m.IsOverDue = true
  1207. }
  1208. case "closed_date_unix":
  1209. m.ClosedDate = time.Unix(m.ClosedDateUnix, 0).Local()
  1210. }
  1211. }
  1212. // State returns string representation of milestone status.
  1213. func (m *Milestone) State() api.StateType {
  1214. if m.IsClosed {
  1215. return api.STATE_CLOSED
  1216. }
  1217. return api.STATE_OPEN
  1218. }
  1219. // APIFormat ...
  1220. func (m *Milestone) APIFormat() *api.Milestone {
  1221. apiMilestone := &api.Milestone{
  1222. ID: m.ID,
  1223. State: m.State(),
  1224. Title: m.Name,
  1225. Description: m.Content,
  1226. OpenIssues: m.NumOpenIssues,
  1227. ClosedIssues: m.NumClosedIssues,
  1228. }
  1229. if m.IsClosed {
  1230. apiMilestone.Closed = &m.ClosedDate
  1231. }
  1232. if m.Deadline.Year() < 9999 {
  1233. apiMilestone.Deadline = &m.Deadline
  1234. }
  1235. return apiMilestone
  1236. }
  1237. // NewMilestone creates new milestone of repository.
  1238. func NewMilestone(m *Milestone) (err error) {
  1239. sess := x.NewSession()
  1240. defer sessionRelease(sess)
  1241. if err = sess.Begin(); err != nil {
  1242. return err
  1243. }
  1244. if _, err = sess.Insert(m); err != nil {
  1245. return err
  1246. }
  1247. if _, err = sess.Exec("UPDATE `repository` SET num_milestones = num_milestones + 1 WHERE id = ?", m.RepoID); err != nil {
  1248. return err
  1249. }
  1250. return sess.Commit()
  1251. }
  1252. func getMilestoneByRepoID(e Engine, repoID, id int64) (*Milestone, error) {
  1253. m := &Milestone{
  1254. ID: id,
  1255. RepoID: repoID,
  1256. }
  1257. has, err := e.Get(m)
  1258. if err != nil {
  1259. return nil, err
  1260. } else if !has {
  1261. return nil, ErrMilestoneNotExist{id, repoID}
  1262. }
  1263. return m, nil
  1264. }
  1265. // GetMilestoneByRepoID returns the milestone in a repository.
  1266. func GetMilestoneByRepoID(repoID, id int64) (*Milestone, error) {
  1267. return getMilestoneByRepoID(x, repoID, id)
  1268. }
  1269. // GetMilestonesByRepoID returns all milestones of a repository.
  1270. func GetMilestonesByRepoID(repoID int64) ([]*Milestone, error) {
  1271. miles := make([]*Milestone, 0, 10)
  1272. return miles, x.Where("repo_id = ?", repoID).Find(&miles)
  1273. }
  1274. // GetMilestones returns a list of milestones of given repository and status.
  1275. func GetMilestones(repoID int64, page int, isClosed bool) ([]*Milestone, error) {
  1276. miles := make([]*Milestone, 0, setting.UI.IssuePagingNum)
  1277. sess := x.Where("repo_id = ? AND is_closed = ?", repoID, isClosed)
  1278. if page > 0 {
  1279. sess = sess.Limit(setting.UI.IssuePagingNum, (page-1)*setting.UI.IssuePagingNum)
  1280. }
  1281. return miles, sess.Find(&miles)
  1282. }
  1283. func updateMilestone(e Engine, m *Milestone) error {
  1284. _, err := e.Id(m.ID).AllCols().Update(m)
  1285. return err
  1286. }
  1287. // UpdateMilestone updates information of given milestone.
  1288. func UpdateMilestone(m *Milestone) error {
  1289. return updateMilestone(x, m)
  1290. }
  1291. func countRepoMilestones(e Engine, repoID int64) int64 {
  1292. count, _ := e.
  1293. Where("repo_id=?", repoID).
  1294. Count(new(Milestone))
  1295. return count
  1296. }
  1297. // CountRepoMilestones returns number of milestones in given repository.
  1298. func CountRepoMilestones(repoID int64) int64 {
  1299. return countRepoMilestones(x, repoID)
  1300. }
  1301. func countRepoClosedMilestones(e Engine, repoID int64) int64 {
  1302. closed, _ := e.
  1303. Where("repo_id=? AND is_closed=?", repoID, true).
  1304. Count(new(Milestone))
  1305. return closed
  1306. }
  1307. // CountRepoClosedMilestones returns number of closed milestones in given repository.
  1308. func CountRepoClosedMilestones(repoID int64) int64 {
  1309. return countRepoClosedMilestones(x, repoID)
  1310. }
  1311. // MilestoneStats returns number of open and closed milestones of given repository.
  1312. func MilestoneStats(repoID int64) (open int64, closed int64) {
  1313. open, _ = x.
  1314. Where("repo_id=? AND is_closed=?", repoID, false).
  1315. Count(new(Milestone))
  1316. return open, CountRepoClosedMilestones(repoID)
  1317. }
  1318. // ChangeMilestoneStatus changes the milestone open/closed status.
  1319. func ChangeMilestoneStatus(m *Milestone, isClosed bool) (err error) {
  1320. repo, err := GetRepositoryByID(m.RepoID)
  1321. if err != nil {
  1322. return err
  1323. }
  1324. sess := x.NewSession()
  1325. defer sessionRelease(sess)
  1326. if err = sess.Begin(); err != nil {
  1327. return err
  1328. }
  1329. m.IsClosed = isClosed
  1330. if err = updateMilestone(sess, m); err != nil {
  1331. return err
  1332. }
  1333. repo.NumMilestones = int(countRepoMilestones(sess, repo.ID))
  1334. repo.NumClosedMilestones = int(countRepoClosedMilestones(sess, repo.ID))
  1335. if _, err = sess.Id(repo.ID).AllCols().Update(repo); err != nil {
  1336. return err
  1337. }
  1338. return sess.Commit()
  1339. }
  1340. func changeMilestoneIssueStats(e *xorm.Session, issue *Issue) error {
  1341. if issue.MilestoneID == 0 {
  1342. return nil
  1343. }
  1344. m, err := getMilestoneByRepoID(e, issue.RepoID, issue.MilestoneID)
  1345. if err != nil {
  1346. return err
  1347. }
  1348. if issue.IsClosed {
  1349. m.NumOpenIssues--
  1350. m.NumClosedIssues++
  1351. } else {
  1352. m.NumOpenIssues++
  1353. m.NumClosedIssues--
  1354. }
  1355. return updateMilestone(e, m)
  1356. }
  1357. // ChangeMilestoneIssueStats updates the open/closed issues counter and progress
  1358. // for the milestone associated with the given issue.
  1359. func ChangeMilestoneIssueStats(issue *Issue) (err error) {
  1360. sess := x.NewSession()
  1361. defer sessionRelease(sess)
  1362. if err = sess.Begin(); err != nil {
  1363. return err
  1364. }
  1365. if err = changeMilestoneIssueStats(sess, issue); err != nil {
  1366. return err
  1367. }
  1368. return sess.Commit()
  1369. }
  1370. func changeMilestoneAssign(e *xorm.Session, issue *Issue, oldMilestoneID int64) error {
  1371. if oldMilestoneID > 0 {
  1372. m, err := getMilestoneByRepoID(e, issue.RepoID, oldMilestoneID)
  1373. if err != nil {
  1374. return err
  1375. }
  1376. m.NumIssues--
  1377. if issue.IsClosed {
  1378. m.NumClosedIssues--
  1379. }
  1380. if err = updateMilestone(e, m); err != nil {
  1381. return err
  1382. } else if _, err = e.Exec("UPDATE `issue_user` SET milestone_id = 0 WHERE issue_id = ?", issue.ID); err != nil {
  1383. return err
  1384. }
  1385. }
  1386. if issue.MilestoneID > 0 {
  1387. m, err := getMilestoneByRepoID(e, issue.RepoID, issue.MilestoneID)
  1388. if err != nil {
  1389. return err
  1390. }
  1391. m.NumIssues++
  1392. if issue.IsClosed {
  1393. m.NumClosedIssues++
  1394. }
  1395. if err = updateMilestone(e, m); err != nil {
  1396. return err
  1397. } else if _, err = e.Exec("UPDATE `issue_user` SET milestone_id = ? WHERE issue_id = ?", m.ID, issue.ID); err != nil {
  1398. return err
  1399. }
  1400. }
  1401. return updateIssue(e, issue)
  1402. }
  1403. // ChangeMilestoneAssign changes assignment of milestone for issue.
  1404. func ChangeMilestoneAssign(issue *Issue, oldMilestoneID int64) (err error) {
  1405. sess := x.NewSession()
  1406. defer sess.Close()
  1407. if err = sess.Begin(); err != nil {
  1408. return err
  1409. }
  1410. if err = changeMilestoneAssign(sess, issue, oldMilestoneID); err != nil {
  1411. return err
  1412. }
  1413. return sess.Commit()
  1414. }
  1415. // DeleteMilestoneByRepoID deletes a milestone from a repository.
  1416. func DeleteMilestoneByRepoID(repoID, id int64) error {
  1417. m, err := GetMilestoneByRepoID(repoID, id)
  1418. if err != nil {
  1419. if IsErrMilestoneNotExist(err) {
  1420. return nil
  1421. }
  1422. return err
  1423. }
  1424. repo, err := GetRepositoryByID(m.RepoID)
  1425. if err != nil {
  1426. return err
  1427. }
  1428. sess := x.NewSession()
  1429. defer sessionRelease(sess)
  1430. if err = sess.Begin(); err != nil {
  1431. return err
  1432. }
  1433. if _, err = sess.Id(m.ID).Delete(new(Milestone)); err != nil {
  1434. return err
  1435. }
  1436. repo.NumMilestones = int(countRepoMilestones(sess, repo.ID))
  1437. repo.NumClosedMilestones = int(countRepoClosedMilestones(sess, repo.ID))
  1438. if _, err = sess.Id(repo.ID).AllCols().Update(repo); err != nil {
  1439. return err
  1440. }
  1441. if _, err = sess.Exec("UPDATE `issue` SET milestone_id = 0 WHERE milestone_id = ?", m.ID); err != nil {
  1442. return err
  1443. } else if _, err = sess.Exec("UPDATE `issue_user` SET milestone_id = 0 WHERE milestone_id = ?", m.ID); err != nil {
  1444. return err
  1445. }
  1446. return sess.Commit()
  1447. }
  1448. // Attachment represent a attachment of issue/comment/release.
  1449. type Attachment struct {
  1450. ID int64 `xorm:"pk autoincr"`
  1451. UUID string `xorm:"uuid UNIQUE"`
  1452. IssueID int64 `xorm:"INDEX"`
  1453. CommentID int64
  1454. ReleaseID int64 `xorm:"INDEX"`
  1455. Name string
  1456. Created time.Time `xorm:"-"`
  1457. CreatedUnix int64
  1458. }
  1459. // BeforeInsert ...
  1460. func (a *Attachment) BeforeInsert() {
  1461. a.CreatedUnix = time.Now().Unix()
  1462. }
  1463. // AfterSet ...
  1464. func (a *Attachment) AfterSet(colName string, _ xorm.Cell) {
  1465. switch colName {
  1466. case "created_unix":
  1467. a.Created = time.Unix(a.CreatedUnix, 0).Local()
  1468. }
  1469. }
  1470. // AttachmentLocalPath returns where attachment is stored in local file system based on given UUID.
  1471. func AttachmentLocalPath(uuid string) string {
  1472. return path.Join(setting.AttachmentPath, uuid[0:1], uuid[1:2], uuid)
  1473. }
  1474. // LocalPath returns where attachment is stored in local file system.
  1475. func (a *Attachment) LocalPath() string {
  1476. return AttachmentLocalPath(a.UUID)
  1477. }
  1478. // NewAttachment creates a new attachment object.
  1479. func NewAttachment(name string, buf []byte, file multipart.File) (_ *Attachment, err error) {
  1480. attach := &Attachment{
  1481. UUID: gouuid.NewV4().String(),
  1482. Name: name,
  1483. }
  1484. localPath := attach.LocalPath()
  1485. if err = os.MkdirAll(path.Dir(localPath), os.ModePerm); err != nil {
  1486. return nil, fmt.Errorf("MkdirAll: %v", err)
  1487. }
  1488. fw, err := os.Create(localPath)
  1489. if err != nil {
  1490. return nil, fmt.Errorf("Create: %v", err)
  1491. }
  1492. defer fw.Close()
  1493. if _, err = fw.Write(buf); err != nil {
  1494. return nil, fmt.Errorf("Write: %v", err)
  1495. } else if _, err = io.Copy(fw, file); err != nil {
  1496. return nil, fmt.Errorf("Copy: %v", err)
  1497. }
  1498. if _, err := x.Insert(attach); err != nil {
  1499. return nil, err
  1500. }
  1501. return attach, nil
  1502. }
  1503. func getAttachmentByUUID(e Engine, uuid string) (*Attachment, error) {
  1504. attach := &Attachment{UUID: uuid}
  1505. has, err := x.Get(attach)
  1506. if err != nil {
  1507. return nil, err
  1508. } else if !has {
  1509. return nil, ErrAttachmentNotExist{0, uuid}
  1510. }
  1511. return attach, nil
  1512. }
  1513. func getAttachmentsByUUIDs(e Engine, uuids []string) ([]*Attachment, error) {
  1514. if len(uuids) == 0 {
  1515. return []*Attachment{}, nil
  1516. }
  1517. // Silently drop invalid uuids.
  1518. attachments := make([]*Attachment, 0, len(uuids))
  1519. return attachments, e.In("uuid", uuids).Find(&attachments)
  1520. }
  1521. // GetAttachmentByUUID returns attachment by given UUID.
  1522. func GetAttachmentByUUID(uuid string) (*Attachment, error) {
  1523. return getAttachmentByUUID(x, uuid)
  1524. }
  1525. func getAttachmentsByIssueID(e Engine, issueID int64) ([]*Attachment, error) {
  1526. attachments := make([]*Attachment, 0, 10)
  1527. return attachments, e.Where("issue_id = ? AND comment_id = 0", issueID).Find(&attachments)
  1528. }
  1529. // GetAttachmentsByIssueID returns all attachments of an issue.
  1530. func GetAttachmentsByIssueID(issueID int64) ([]*Attachment, error) {
  1531. return getAttachmentsByIssueID(x, issueID)
  1532. }
  1533. // GetAttachmentsByCommentID returns all attachments if comment by given ID.
  1534. func GetAttachmentsByCommentID(commentID int64) ([]*Attachment, error) {
  1535. attachments := make([]*Attachment, 0, 10)
  1536. return attachments, x.Where("comment_id=?", commentID).Find(&attachments)
  1537. }
  1538. // DeleteAttachment deletes the given attachment and optionally the associated file.
  1539. func DeleteAttachment(a *Attachment, remove bool) error {
  1540. _, err := DeleteAttachments([]*Attachment{a}, remove)
  1541. return err
  1542. }
  1543. // DeleteAttachments deletes the given attachments and optionally the associated files.
  1544. func DeleteAttachments(attachments []*Attachment, remove bool) (int, error) {
  1545. for i, a := range attachments {
  1546. if remove {
  1547. if err := os.Remove(a.LocalPath()); err != nil {
  1548. return i, err
  1549. }
  1550. }
  1551. if _, err := x.Delete(a); err != nil {
  1552. return i, err
  1553. }
  1554. }
  1555. return len(attachments), nil
  1556. }
  1557. // DeleteAttachmentsByIssue deletes all attachments associated with the given issue.
  1558. func DeleteAttachmentsByIssue(issueID int64, remove bool) (int, error) {
  1559. attachments, err := GetAttachmentsByIssueID(issueID)
  1560. if err != nil {
  1561. return 0, err
  1562. }
  1563. return DeleteAttachments(attachments, remove)
  1564. }
  1565. // DeleteAttachmentsByComment deletes all attachments associated with the given comment.
  1566. func DeleteAttachmentsByComment(commentID int64, remove bool) (int, error) {
  1567. attachments, err := GetAttachmentsByCommentID(commentID)
  1568. if err != nil {
  1569. return 0, err
  1570. }
  1571. return DeleteAttachments(attachments, remove)
  1572. }