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 21KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810
  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. "bytes"
  7. "errors"
  8. "strings"
  9. "time"
  10. "github.com/go-xorm/xorm"
  11. "github.com/gogits/gogs/modules/base"
  12. )
  13. var (
  14. ErrIssueNotExist = errors.New("Issue does not exist")
  15. ErrLabelNotExist = errors.New("Label does not exist")
  16. ErrMilestoneNotExist = errors.New("Milestone does not exist")
  17. )
  18. // Issue represents an issue or pull request of repository.
  19. type Issue struct {
  20. Id int64
  21. RepoId int64 `xorm:"INDEX"`
  22. Index int64 // Index in one repository.
  23. Name string
  24. Repo *Repository `xorm:"-"`
  25. PosterId int64
  26. Poster *User `xorm:"-"`
  27. LabelIds string `xorm:"TEXT"`
  28. Labels []*Label `xorm:"-"`
  29. MilestoneId int64
  30. AssigneeId int64
  31. Assignee *User `xorm:"-"`
  32. IsRead bool `xorm:"-"`
  33. IsPull bool // Indicates whether is a pull request or not.
  34. IsClosed bool
  35. Content string `xorm:"TEXT"`
  36. RenderedContent string `xorm:"-"`
  37. Priority int
  38. NumComments int
  39. Deadline time.Time
  40. Created time.Time `xorm:"CREATED"`
  41. Updated time.Time `xorm:"UPDATED"`
  42. }
  43. func (i *Issue) GetPoster() (err error) {
  44. i.Poster, err = GetUserById(i.PosterId)
  45. if err == ErrUserNotExist {
  46. i.Poster = &User{Name: "FakeUser"}
  47. return nil
  48. }
  49. return err
  50. }
  51. func (i *Issue) GetLabels() error {
  52. if len(i.LabelIds) < 3 {
  53. return nil
  54. }
  55. strIds := strings.Split(strings.TrimSuffix(i.LabelIds[1:], "|"), "|$")
  56. i.Labels = make([]*Label, 0, len(strIds))
  57. for _, strId := range strIds {
  58. id, _ := base.StrTo(strId).Int64()
  59. if id > 0 {
  60. l, err := GetLabelById(id)
  61. if err != nil {
  62. if err == ErrLabelNotExist {
  63. continue
  64. }
  65. return err
  66. }
  67. i.Labels = append(i.Labels, l)
  68. }
  69. }
  70. return nil
  71. }
  72. func (i *Issue) GetAssignee() (err error) {
  73. if i.AssigneeId == 0 {
  74. return nil
  75. }
  76. i.Assignee, err = GetUserById(i.AssigneeId)
  77. if err == ErrUserNotExist {
  78. return nil
  79. }
  80. return err
  81. }
  82. // CreateIssue creates new issue for repository.
  83. func NewIssue(issue *Issue) (err error) {
  84. sess := orm.NewSession()
  85. defer sess.Close()
  86. if err = sess.Begin(); err != nil {
  87. return err
  88. }
  89. if _, err = sess.Insert(issue); err != nil {
  90. sess.Rollback()
  91. return err
  92. }
  93. rawSql := "UPDATE `repository` SET num_issues = num_issues + 1 WHERE id = ?"
  94. if _, err = sess.Exec(rawSql, issue.RepoId); err != nil {
  95. sess.Rollback()
  96. return err
  97. }
  98. return sess.Commit()
  99. }
  100. // GetIssueByIndex returns issue by given index in repository.
  101. func GetIssueByIndex(rid, index int64) (*Issue, error) {
  102. issue := &Issue{RepoId: rid, Index: index}
  103. has, err := orm.Get(issue)
  104. if err != nil {
  105. return nil, err
  106. } else if !has {
  107. return nil, ErrIssueNotExist
  108. }
  109. return issue, nil
  110. }
  111. // GetIssueById returns an issue by ID.
  112. func GetIssueById(id int64) (*Issue, error) {
  113. issue := &Issue{Id: id}
  114. has, err := orm.Get(issue)
  115. if err != nil {
  116. return nil, err
  117. } else if !has {
  118. return nil, ErrIssueNotExist
  119. }
  120. return issue, nil
  121. }
  122. // GetIssues returns a list of issues by given conditions.
  123. func GetIssues(uid, rid, pid, mid int64, page int, isClosed bool, labelIds, sortType string) ([]Issue, error) {
  124. sess := orm.Limit(20, (page-1)*20)
  125. if rid > 0 {
  126. sess.Where("repo_id=?", rid).And("is_closed=?", isClosed)
  127. } else {
  128. sess.Where("is_closed=?", isClosed)
  129. }
  130. if uid > 0 {
  131. sess.And("assignee_id=?", uid)
  132. } else if pid > 0 {
  133. sess.And("poster_id=?", pid)
  134. }
  135. if mid > 0 {
  136. sess.And("milestone_id=?", mid)
  137. }
  138. if len(labelIds) > 0 {
  139. for _, label := range strings.Split(labelIds, ",") {
  140. sess.And("label_ids like '%$" + label + "|%'")
  141. }
  142. }
  143. switch sortType {
  144. case "oldest":
  145. sess.Asc("created")
  146. case "recentupdate":
  147. sess.Desc("updated")
  148. case "leastupdate":
  149. sess.Asc("updated")
  150. case "mostcomment":
  151. sess.Desc("num_comments")
  152. case "leastcomment":
  153. sess.Asc("num_comments")
  154. case "priority":
  155. sess.Desc("priority")
  156. default:
  157. sess.Desc("created")
  158. }
  159. var issues []Issue
  160. err := sess.Find(&issues)
  161. return issues, err
  162. }
  163. // GetIssuesByLabel returns a list of issues by given label and repository.
  164. func GetIssuesByLabel(repoId int64, label string) ([]*Issue, error) {
  165. issues := make([]*Issue, 0, 10)
  166. err := orm.Where("repo_id=?", repoId).And("label_ids like '%$" + label + "|%'").Find(&issues)
  167. return issues, err
  168. }
  169. // GetIssueCountByPoster returns number of issues of repository by poster.
  170. func GetIssueCountByPoster(uid, rid int64, isClosed bool) int64 {
  171. count, _ := orm.Where("repo_id=?", rid).And("poster_id=?", uid).And("is_closed=?", isClosed).Count(new(Issue))
  172. return count
  173. }
  174. // .___ ____ ___
  175. // | | ______ ________ __ ____ | | \______ ___________
  176. // | |/ ___// ___/ | \_/ __ \| | / ___// __ \_ __ \
  177. // | |\___ \ \___ \| | /\ ___/| | /\___ \\ ___/| | \/
  178. // |___/____ >____ >____/ \___ >______//____ >\___ >__|
  179. // \/ \/ \/ \/ \/
  180. // IssueUser represents an issue-user relation.
  181. type IssueUser struct {
  182. Id int64
  183. Uid int64 // User ID.
  184. IssueId int64
  185. RepoId int64
  186. MilestoneId int64
  187. IsRead bool
  188. IsAssigned bool
  189. IsMentioned bool
  190. IsPoster bool
  191. IsClosed bool
  192. }
  193. // NewIssueUserPairs adds new issue-user pairs for new issue of repository.
  194. func NewIssueUserPairs(rid, iid, oid, pid, aid int64, repoName string) (err error) {
  195. iu := &IssueUser{IssueId: iid, RepoId: rid}
  196. us, err := GetCollaborators(repoName)
  197. if err != nil {
  198. return err
  199. }
  200. isNeedAddPoster := true
  201. for _, u := range us {
  202. iu.Uid = u.Id
  203. iu.IsPoster = iu.Uid == pid
  204. if isNeedAddPoster && iu.IsPoster {
  205. isNeedAddPoster = false
  206. }
  207. iu.IsAssigned = iu.Uid == aid
  208. if _, err = orm.Insert(iu); err != nil {
  209. return err
  210. }
  211. }
  212. if isNeedAddPoster {
  213. iu.Uid = pid
  214. iu.IsPoster = true
  215. iu.IsAssigned = iu.Uid == aid
  216. if _, err = orm.Insert(iu); err != nil {
  217. return err
  218. }
  219. }
  220. return nil
  221. }
  222. // PairsContains returns true when pairs list contains given issue.
  223. func PairsContains(ius []*IssueUser, issueId int64) int {
  224. for i := range ius {
  225. if ius[i].IssueId == issueId {
  226. return i
  227. }
  228. }
  229. return -1
  230. }
  231. // GetIssueUserPairs returns issue-user pairs by given repository and user.
  232. func GetIssueUserPairs(rid, uid int64, isClosed bool) ([]*IssueUser, error) {
  233. ius := make([]*IssueUser, 0, 10)
  234. err := orm.Where("is_closed=?", isClosed).Find(&ius, &IssueUser{RepoId: rid, Uid: uid})
  235. return ius, err
  236. }
  237. // GetIssueUserPairsByRepoIds returns issue-user pairs by given repository IDs.
  238. func GetIssueUserPairsByRepoIds(rids []int64, isClosed bool, page int) ([]*IssueUser, error) {
  239. buf := bytes.NewBufferString("")
  240. for _, rid := range rids {
  241. buf.WriteString("repo_id=")
  242. buf.WriteString(base.ToStr(rid))
  243. buf.WriteString(" OR ")
  244. }
  245. cond := strings.TrimSuffix(buf.String(), " OR ")
  246. ius := make([]*IssueUser, 0, 10)
  247. sess := orm.Limit(20, (page-1)*20).Where("is_closed=?", isClosed)
  248. if len(cond) > 0 {
  249. sess.And(cond)
  250. }
  251. err := sess.Find(&ius)
  252. return ius, err
  253. }
  254. // GetIssueUserPairsByMode returns issue-user pairs by given repository and user.
  255. func GetIssueUserPairsByMode(uid, rid int64, isClosed bool, page, filterMode int) ([]*IssueUser, error) {
  256. ius := make([]*IssueUser, 0, 10)
  257. sess := orm.Limit(20, (page-1)*20).Where("uid=?", uid).And("is_closed=?", isClosed)
  258. if rid > 0 {
  259. sess.And("repo_id=?", rid)
  260. }
  261. switch filterMode {
  262. case FM_ASSIGN:
  263. sess.And("is_assigned=?", true)
  264. case FM_CREATE:
  265. sess.And("is_poster=?", true)
  266. default:
  267. return ius, nil
  268. }
  269. err := sess.Find(&ius)
  270. return ius, err
  271. }
  272. // IssueStats represents issue statistic information.
  273. type IssueStats struct {
  274. OpenCount, ClosedCount int64
  275. AllCount int64
  276. AssignCount int64
  277. CreateCount int64
  278. MentionCount int64
  279. }
  280. // Filter modes.
  281. const (
  282. FM_ASSIGN = iota + 1
  283. FM_CREATE
  284. FM_MENTION
  285. )
  286. // GetIssueStats returns issue statistic information by given conditions.
  287. func GetIssueStats(rid, uid int64, isShowClosed bool, filterMode int) *IssueStats {
  288. stats := &IssueStats{}
  289. issue := new(Issue)
  290. tmpSess := &xorm.Session{}
  291. sess := orm.Where("repo_id=?", rid)
  292. *tmpSess = *sess
  293. stats.OpenCount, _ = tmpSess.And("is_closed=?", false).Count(issue)
  294. *tmpSess = *sess
  295. stats.ClosedCount, _ = tmpSess.And("is_closed=?", true).Count(issue)
  296. if isShowClosed {
  297. stats.AllCount = stats.ClosedCount
  298. } else {
  299. stats.AllCount = stats.OpenCount
  300. }
  301. if filterMode != FM_MENTION {
  302. sess = orm.Where("repo_id=?", rid)
  303. switch filterMode {
  304. case FM_ASSIGN:
  305. sess.And("assignee_id=?", uid)
  306. case FM_CREATE:
  307. sess.And("poster_id=?", uid)
  308. default:
  309. goto nofilter
  310. }
  311. *tmpSess = *sess
  312. stats.OpenCount, _ = tmpSess.And("is_closed=?", false).Count(issue)
  313. *tmpSess = *sess
  314. stats.ClosedCount, _ = tmpSess.And("is_closed=?", true).Count(issue)
  315. } else {
  316. sess := orm.Where("repo_id=?", rid).And("uid=?", uid).And("is_mentioned=?", true)
  317. *tmpSess = *sess
  318. stats.OpenCount, _ = tmpSess.And("is_closed=?", false).Count(new(IssueUser))
  319. *tmpSess = *sess
  320. stats.ClosedCount, _ = tmpSess.And("is_closed=?", true).Count(new(IssueUser))
  321. }
  322. nofilter:
  323. stats.AssignCount, _ = orm.Where("repo_id=?", rid).And("is_closed=?", isShowClosed).And("assignee_id=?", uid).Count(issue)
  324. stats.CreateCount, _ = orm.Where("repo_id=?", rid).And("is_closed=?", isShowClosed).And("poster_id=?", uid).Count(issue)
  325. stats.MentionCount, _ = orm.Where("repo_id=?", rid).And("uid=?", uid).And("is_closed=?", isShowClosed).And("is_mentioned=?", true).Count(new(IssueUser))
  326. return stats
  327. }
  328. // GetUserIssueStats returns issue statistic information for dashboard by given conditions.
  329. func GetUserIssueStats(uid int64, filterMode int) *IssueStats {
  330. stats := &IssueStats{}
  331. issue := new(Issue)
  332. stats.AssignCount, _ = orm.Where("assignee_id=?", uid).And("is_closed=?", false).Count(issue)
  333. stats.CreateCount, _ = orm.Where("poster_id=?", uid).And("is_closed=?", false).Count(issue)
  334. return stats
  335. }
  336. // UpdateIssue updates information of issue.
  337. func UpdateIssue(issue *Issue) error {
  338. _, err := orm.Id(issue.Id).AllCols().Update(issue)
  339. return err
  340. }
  341. // UpdateIssueUserByStatus updates issue-user pairs by issue status.
  342. func UpdateIssueUserPairsByStatus(iid int64, isClosed bool) error {
  343. rawSql := "UPDATE `issue_user` SET is_closed = ? WHERE issue_id = ?"
  344. _, err := orm.Exec(rawSql, isClosed, iid)
  345. return err
  346. }
  347. // UpdateIssueUserPairByAssignee updates issue-user pair for assigning.
  348. func UpdateIssueUserPairByAssignee(aid, iid int64) error {
  349. rawSql := "UPDATE `issue_user` SET is_assigned = ? WHERE issue_id = ?"
  350. if _, err := orm.Exec(rawSql, false, iid); err != nil {
  351. return err
  352. }
  353. // Assignee ID equals to 0 means clear assignee.
  354. if aid == 0 {
  355. return nil
  356. }
  357. rawSql = "UPDATE `issue_user` SET is_assigned = true WHERE uid = ? AND issue_id = ?"
  358. _, err := orm.Exec(rawSql, aid, iid)
  359. return err
  360. }
  361. // UpdateIssueUserPairByRead updates issue-user pair for reading.
  362. func UpdateIssueUserPairByRead(uid, iid int64) error {
  363. rawSql := "UPDATE `issue_user` SET is_read = ? WHERE uid = ? AND issue_id = ?"
  364. _, err := orm.Exec(rawSql, true, uid, iid)
  365. return err
  366. }
  367. // UpdateIssueUserPairsByMentions updates issue-user pairs by mentioning.
  368. func UpdateIssueUserPairsByMentions(uids []int64, iid int64) error {
  369. for _, uid := range uids {
  370. iu := &IssueUser{Uid: uid, IssueId: iid}
  371. has, err := orm.Get(iu)
  372. if err != nil {
  373. return err
  374. }
  375. iu.IsMentioned = true
  376. if has {
  377. _, err = orm.Id(iu.Id).AllCols().Update(iu)
  378. } else {
  379. _, err = orm.Insert(iu)
  380. }
  381. if err != nil {
  382. return err
  383. }
  384. }
  385. return nil
  386. }
  387. // .____ ___. .__
  388. // | | _____ \_ |__ ____ | |
  389. // | | \__ \ | __ \_/ __ \| |
  390. // | |___ / __ \| \_\ \ ___/| |__
  391. // |_______ (____ /___ /\___ >____/
  392. // \/ \/ \/ \/
  393. // Label represents a label of repository for issues.
  394. type Label struct {
  395. Id int64
  396. RepoId int64 `xorm:"INDEX"`
  397. Name string
  398. Color string `xorm:"VARCHAR(7)"`
  399. NumIssues int
  400. NumClosedIssues int
  401. NumOpenIssues int `xorm:"-"`
  402. IsChecked bool `xorm:"-"`
  403. }
  404. // CalOpenIssues calculates the open issues of label.
  405. func (m *Label) CalOpenIssues() {
  406. m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
  407. }
  408. // NewLabel creates new label of repository.
  409. func NewLabel(l *Label) error {
  410. _, err := orm.Insert(l)
  411. return err
  412. }
  413. // GetLabelById returns a label by given ID.
  414. func GetLabelById(id int64) (*Label, error) {
  415. l := &Label{Id: id}
  416. has, err := orm.Get(l)
  417. if err != nil {
  418. return nil, err
  419. } else if !has {
  420. return nil, ErrLabelNotExist
  421. }
  422. return l, nil
  423. }
  424. // GetLabels returns a list of labels of given repository ID.
  425. func GetLabels(repoId int64) ([]*Label, error) {
  426. labels := make([]*Label, 0, 10)
  427. err := orm.Where("repo_id=?", repoId).Find(&labels)
  428. return labels, err
  429. }
  430. // UpdateLabel updates label information.
  431. func UpdateLabel(l *Label) error {
  432. _, err := orm.Id(l.Id).Update(l)
  433. return err
  434. }
  435. // DeleteLabel delete a label of given repository.
  436. func DeleteLabel(repoId int64, strId string) error {
  437. id, _ := base.StrTo(strId).Int64()
  438. l, err := GetLabelById(id)
  439. if err != nil {
  440. if err == ErrLabelNotExist {
  441. return nil
  442. }
  443. return err
  444. }
  445. issues, err := GetIssuesByLabel(repoId, strId)
  446. if err != nil {
  447. return err
  448. }
  449. sess := orm.NewSession()
  450. defer sess.Close()
  451. if err = sess.Begin(); err != nil {
  452. return err
  453. }
  454. for _, issue := range issues {
  455. issue.LabelIds = strings.Replace(issue.LabelIds, "$"+strId+"|", "", -1)
  456. if _, err = sess.Id(issue.Id).AllCols().Update(issue); err != nil {
  457. sess.Rollback()
  458. return err
  459. }
  460. }
  461. if _, err = sess.Delete(l); err != nil {
  462. sess.Rollback()
  463. return err
  464. }
  465. return sess.Commit()
  466. }
  467. // _____ .__.__ __
  468. // / \ |__| | ____ _______/ |_ ____ ____ ____
  469. // / \ / \| | | _/ __ \ / ___/\ __\/ _ \ / \_/ __ \
  470. // / Y \ | |_\ ___/ \___ \ | | ( <_> ) | \ ___/
  471. // \____|__ /__|____/\___ >____ > |__| \____/|___| /\___ >
  472. // \/ \/ \/ \/ \/
  473. // Milestone represents a milestone of repository.
  474. type Milestone struct {
  475. Id int64
  476. RepoId int64 `xorm:"INDEX"`
  477. Index int64
  478. Name string
  479. Content string
  480. RenderedContent string `xorm:"-"`
  481. IsClosed bool
  482. NumIssues int
  483. NumClosedIssues int
  484. NumOpenIssues int `xorm:"-"`
  485. Completeness int // Percentage(1-100).
  486. Deadline time.Time
  487. DeadlineString string `xorm:"-"`
  488. ClosedDate time.Time
  489. }
  490. // CalOpenIssues calculates the open issues of milestone.
  491. func (m *Milestone) CalOpenIssues() {
  492. m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
  493. }
  494. // NewMilestone creates new milestone of repository.
  495. func NewMilestone(m *Milestone) (err error) {
  496. sess := orm.NewSession()
  497. defer sess.Close()
  498. if err = sess.Begin(); err != nil {
  499. return err
  500. }
  501. if _, err = sess.Insert(m); err != nil {
  502. sess.Rollback()
  503. return err
  504. }
  505. rawSql := "UPDATE `repository` SET num_milestones = num_milestones + 1 WHERE id = ?"
  506. if _, err = sess.Exec(rawSql, m.RepoId); err != nil {
  507. sess.Rollback()
  508. return err
  509. }
  510. return sess.Commit()
  511. }
  512. // GetMilestoneById returns the milestone by given ID.
  513. func GetMilestoneById(id int64) (*Milestone, error) {
  514. m := &Milestone{Id: id}
  515. has, err := orm.Get(m)
  516. if err != nil {
  517. return nil, err
  518. } else if !has {
  519. return nil, ErrMilestoneNotExist
  520. }
  521. return m, nil
  522. }
  523. // GetMilestoneByIndex returns the milestone of given repository and index.
  524. func GetMilestoneByIndex(repoId, idx int64) (*Milestone, error) {
  525. m := &Milestone{RepoId: repoId, Index: idx}
  526. has, err := orm.Get(m)
  527. if err != nil {
  528. return nil, err
  529. } else if !has {
  530. return nil, ErrMilestoneNotExist
  531. }
  532. return m, nil
  533. }
  534. // GetMilestones returns a list of milestones of given repository and status.
  535. func GetMilestones(repoId int64, isClosed bool) ([]*Milestone, error) {
  536. miles := make([]*Milestone, 0, 10)
  537. err := orm.Where("repo_id=?", repoId).And("is_closed=?", isClosed).Find(&miles)
  538. return miles, err
  539. }
  540. // UpdateMilestone updates information of given milestone.
  541. func UpdateMilestone(m *Milestone) error {
  542. _, err := orm.Id(m.Id).Update(m)
  543. return err
  544. }
  545. // ChangeMilestoneStatus changes the milestone open/closed status.
  546. func ChangeMilestoneStatus(m *Milestone, isClosed bool) (err error) {
  547. repo, err := GetRepositoryById(m.RepoId)
  548. if err != nil {
  549. return err
  550. }
  551. sess := orm.NewSession()
  552. defer sess.Close()
  553. if err = sess.Begin(); err != nil {
  554. return err
  555. }
  556. m.IsClosed = isClosed
  557. if _, err = sess.Id(m.Id).AllCols().Update(m); err != nil {
  558. sess.Rollback()
  559. return err
  560. }
  561. if isClosed {
  562. repo.NumClosedMilestones++
  563. } else {
  564. repo.NumClosedMilestones--
  565. }
  566. if _, err = sess.Id(repo.Id).Update(repo); err != nil {
  567. sess.Rollback()
  568. return err
  569. }
  570. return sess.Commit()
  571. }
  572. // ChangeMilestoneAssign changes assignment of milestone for issue.
  573. func ChangeMilestoneAssign(oldMid, mid int64, issue *Issue) (err error) {
  574. sess := orm.NewSession()
  575. defer sess.Close()
  576. if err = sess.Begin(); err != nil {
  577. return err
  578. }
  579. if oldMid > 0 {
  580. m, err := GetMilestoneById(oldMid)
  581. if err != nil {
  582. return err
  583. }
  584. m.NumIssues--
  585. if issue.IsClosed {
  586. m.NumClosedIssues--
  587. }
  588. if m.NumIssues > 0 {
  589. m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
  590. } else {
  591. m.Completeness = 0
  592. }
  593. if _, err = sess.Id(m.Id).Update(m); err != nil {
  594. sess.Rollback()
  595. return err
  596. }
  597. rawSql := "UPDATE `issue_user` SET milestone_id = 0 WHERE issue_id = ?"
  598. if _, err = sess.Exec(rawSql, issue.Id); err != nil {
  599. sess.Rollback()
  600. return err
  601. }
  602. }
  603. if mid > 0 {
  604. m, err := GetMilestoneById(mid)
  605. if err != nil {
  606. return err
  607. }
  608. m.NumIssues++
  609. if issue.IsClosed {
  610. m.NumClosedIssues++
  611. }
  612. m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
  613. if _, err = sess.Id(m.Id).Update(m); err != nil {
  614. sess.Rollback()
  615. return err
  616. }
  617. rawSql := "UPDATE `issue_user` SET milestone_id = ? WHERE issue_id = ?"
  618. if _, err = sess.Exec(rawSql, m.Id, issue.Id); err != nil {
  619. sess.Rollback()
  620. return err
  621. }
  622. }
  623. return sess.Commit()
  624. }
  625. // DeleteMilestone deletes a milestone.
  626. func DeleteMilestone(m *Milestone) (err error) {
  627. sess := orm.NewSession()
  628. defer sess.Close()
  629. if err = sess.Begin(); err != nil {
  630. return err
  631. }
  632. if _, err = sess.Delete(m); err != nil {
  633. sess.Rollback()
  634. return err
  635. }
  636. rawSql := "UPDATE `repository` SET num_milestones = num_milestones - 1 WHERE id = ?"
  637. if _, err = sess.Exec(rawSql, m.RepoId); err != nil {
  638. sess.Rollback()
  639. return err
  640. }
  641. rawSql = "UPDATE `issue` SET milestone_id = 0 WHERE milestone_id = ?"
  642. if _, err = sess.Exec(rawSql, m.Id); err != nil {
  643. sess.Rollback()
  644. return err
  645. }
  646. rawSql = "UPDATE `issue_user` SET milestone_id = 0 WHERE milestone_id = ?"
  647. if _, err = sess.Exec(rawSql, m.Id); err != nil {
  648. sess.Rollback()
  649. return err
  650. }
  651. return sess.Commit()
  652. }
  653. // _________ __
  654. // \_ ___ \ ____ _____ _____ ____ _____/ |_
  655. // / \ \/ / _ \ / \ / \_/ __ \ / \ __\
  656. // \ \___( <_> ) Y Y \ Y Y \ ___/| | \ |
  657. // \______ /\____/|__|_| /__|_| /\___ >___| /__|
  658. // \/ \/ \/ \/ \/
  659. // Issue types.
  660. const (
  661. IT_PLAIN = iota // Pure comment.
  662. IT_REOPEN // Issue reopen status change prompt.
  663. IT_CLOSE // Issue close status change prompt.
  664. )
  665. // Comment represents a comment in commit and issue page.
  666. type Comment struct {
  667. Id int64
  668. Type int
  669. PosterId int64
  670. Poster *User `xorm:"-"`
  671. IssueId int64
  672. CommitId int64
  673. Line int64
  674. Content string
  675. Created time.Time `xorm:"CREATED"`
  676. }
  677. // CreateComment creates comment of issue or commit.
  678. func CreateComment(userId, repoId, issueId, commitId, line int64, cmtType int, content string) error {
  679. sess := orm.NewSession()
  680. defer sess.Close()
  681. if err := sess.Begin(); err != nil {
  682. return err
  683. }
  684. if _, err := sess.Insert(&Comment{PosterId: userId, Type: cmtType, IssueId: issueId,
  685. CommitId: commitId, Line: line, Content: content}); err != nil {
  686. sess.Rollback()
  687. return err
  688. }
  689. // Check comment type.
  690. switch cmtType {
  691. case IT_PLAIN:
  692. rawSql := "UPDATE `issue` SET num_comments = num_comments + 1 WHERE id = ?"
  693. if _, err := sess.Exec(rawSql, issueId); err != nil {
  694. sess.Rollback()
  695. return err
  696. }
  697. case IT_REOPEN:
  698. rawSql := "UPDATE `repository` SET num_closed_issues = num_closed_issues - 1 WHERE id = ?"
  699. if _, err := sess.Exec(rawSql, repoId); err != nil {
  700. sess.Rollback()
  701. return err
  702. }
  703. case IT_CLOSE:
  704. rawSql := "UPDATE `repository` SET num_closed_issues = num_closed_issues + 1 WHERE id = ?"
  705. if _, err := sess.Exec(rawSql, repoId); err != nil {
  706. sess.Rollback()
  707. return err
  708. }
  709. }
  710. return sess.Commit()
  711. }
  712. // GetIssueComments returns list of comment by given issue id.
  713. func GetIssueComments(issueId int64) ([]Comment, error) {
  714. comments := make([]Comment, 0, 10)
  715. err := orm.Asc("created").Find(&comments, &Comment{IssueId: issueId})
  716. return comments, err
  717. }