issue.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821
  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 := x.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 := x.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 := x.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 := x.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. type IssueStatus int
  164. const (
  165. IS_OPEN = iota + 1
  166. IS_CLOSE
  167. )
  168. // GetIssuesByLabel returns a list of issues by given label and repository.
  169. func GetIssuesByLabel(repoId int64, label string) ([]*Issue, error) {
  170. issues := make([]*Issue, 0, 10)
  171. err := x.Where("repo_id=?", repoId).And("label_ids like '%$" + label + "|%'").Find(&issues)
  172. return issues, err
  173. }
  174. // GetIssueCountByPoster returns number of issues of repository by poster.
  175. func GetIssueCountByPoster(uid, rid int64, isClosed bool) int64 {
  176. count, _ := x.Where("repo_id=?", rid).And("poster_id=?", uid).And("is_closed=?", isClosed).Count(new(Issue))
  177. return count
  178. }
  179. // .___ ____ ___
  180. // | | ______ ________ __ ____ | | \______ ___________
  181. // | |/ ___// ___/ | \_/ __ \| | / ___// __ \_ __ \
  182. // | |\___ \ \___ \| | /\ ___/| | /\___ \\ ___/| | \/
  183. // |___/____ >____ >____/ \___ >______//____ >\___ >__|
  184. // \/ \/ \/ \/ \/
  185. // IssueUser represents an issue-user relation.
  186. type IssueUser struct {
  187. Id int64
  188. Uid int64 `xorm:"INDEX"` // User ID.
  189. IssueId int64
  190. RepoId int64 `xorm:"INDEX"`
  191. MilestoneId int64
  192. IsRead bool
  193. IsAssigned bool
  194. IsMentioned bool
  195. IsPoster bool
  196. IsClosed bool
  197. }
  198. // NewIssueUserPairs adds new issue-user pairs for new issue of repository.
  199. func NewIssueUserPairs(rid, iid, oid, pid, aid int64, repoName string) (err error) {
  200. iu := &IssueUser{IssueId: iid, RepoId: rid}
  201. us, err := GetCollaborators(repoName)
  202. if err != nil {
  203. return err
  204. }
  205. isNeedAddPoster := true
  206. for _, u := range us {
  207. iu.Uid = u.Id
  208. iu.IsPoster = iu.Uid == pid
  209. if isNeedAddPoster && iu.IsPoster {
  210. isNeedAddPoster = false
  211. }
  212. iu.IsAssigned = iu.Uid == aid
  213. if _, err = x.Insert(iu); err != nil {
  214. return err
  215. }
  216. }
  217. if isNeedAddPoster {
  218. iu.Uid = pid
  219. iu.IsPoster = true
  220. iu.IsAssigned = iu.Uid == aid
  221. if _, err = x.Insert(iu); err != nil {
  222. return err
  223. }
  224. }
  225. return nil
  226. }
  227. // PairsContains returns true when pairs list contains given issue.
  228. func PairsContains(ius []*IssueUser, issueId int64) int {
  229. for i := range ius {
  230. if ius[i].IssueId == issueId {
  231. return i
  232. }
  233. }
  234. return -1
  235. }
  236. // GetIssueUserPairs returns issue-user pairs by given repository and user.
  237. func GetIssueUserPairs(rid, uid int64, isClosed bool) ([]*IssueUser, error) {
  238. ius := make([]*IssueUser, 0, 10)
  239. err := x.Where("is_closed=?", isClosed).Find(&ius, &IssueUser{RepoId: rid, Uid: uid})
  240. return ius, err
  241. }
  242. // GetIssueUserPairsByRepoIds returns issue-user pairs by given repository IDs.
  243. func GetIssueUserPairsByRepoIds(rids []int64, isClosed bool, page int) ([]*IssueUser, error) {
  244. buf := bytes.NewBufferString("")
  245. for _, rid := range rids {
  246. buf.WriteString("repo_id=")
  247. buf.WriteString(base.ToStr(rid))
  248. buf.WriteString(" OR ")
  249. }
  250. cond := strings.TrimSuffix(buf.String(), " OR ")
  251. ius := make([]*IssueUser, 0, 10)
  252. sess := x.Limit(20, (page-1)*20).Where("is_closed=?", isClosed)
  253. if len(cond) > 0 {
  254. sess.And(cond)
  255. }
  256. err := sess.Find(&ius)
  257. return ius, err
  258. }
  259. // GetIssueUserPairsByMode returns issue-user pairs by given repository and user.
  260. func GetIssueUserPairsByMode(uid, rid int64, isClosed bool, page, filterMode int) ([]*IssueUser, error) {
  261. ius := make([]*IssueUser, 0, 10)
  262. sess := x.Limit(20, (page-1)*20).Where("uid=?", uid).And("is_closed=?", isClosed)
  263. if rid > 0 {
  264. sess.And("repo_id=?", rid)
  265. }
  266. switch filterMode {
  267. case FM_ASSIGN:
  268. sess.And("is_assigned=?", true)
  269. case FM_CREATE:
  270. sess.And("is_poster=?", true)
  271. default:
  272. return ius, nil
  273. }
  274. err := sess.Find(&ius)
  275. return ius, err
  276. }
  277. // IssueStats represents issue statistic information.
  278. type IssueStats struct {
  279. OpenCount, ClosedCount int64
  280. AllCount int64
  281. AssignCount int64
  282. CreateCount int64
  283. MentionCount int64
  284. }
  285. // Filter modes.
  286. const (
  287. FM_ASSIGN = iota + 1
  288. FM_CREATE
  289. FM_MENTION
  290. )
  291. // GetIssueStats returns issue statistic information by given conditions.
  292. func GetIssueStats(rid, uid int64, isShowClosed bool, filterMode int) *IssueStats {
  293. stats := &IssueStats{}
  294. issue := new(Issue)
  295. tmpSess := &xorm.Session{}
  296. sess := x.Where("repo_id=?", rid)
  297. *tmpSess = *sess
  298. stats.OpenCount, _ = tmpSess.And("is_closed=?", false).Count(issue)
  299. *tmpSess = *sess
  300. stats.ClosedCount, _ = tmpSess.And("is_closed=?", true).Count(issue)
  301. if isShowClosed {
  302. stats.AllCount = stats.ClosedCount
  303. } else {
  304. stats.AllCount = stats.OpenCount
  305. }
  306. if filterMode != FM_MENTION {
  307. sess = x.Where("repo_id=?", rid)
  308. switch filterMode {
  309. case FM_ASSIGN:
  310. sess.And("assignee_id=?", uid)
  311. case FM_CREATE:
  312. sess.And("poster_id=?", uid)
  313. default:
  314. goto nofilter
  315. }
  316. *tmpSess = *sess
  317. stats.OpenCount, _ = tmpSess.And("is_closed=?", false).Count(issue)
  318. *tmpSess = *sess
  319. stats.ClosedCount, _ = tmpSess.And("is_closed=?", true).Count(issue)
  320. } else {
  321. sess := x.Where("repo_id=?", rid).And("uid=?", uid).And("is_mentioned=?", true)
  322. *tmpSess = *sess
  323. stats.OpenCount, _ = tmpSess.And("is_closed=?", false).Count(new(IssueUser))
  324. *tmpSess = *sess
  325. stats.ClosedCount, _ = tmpSess.And("is_closed=?", true).Count(new(IssueUser))
  326. }
  327. nofilter:
  328. stats.AssignCount, _ = x.Where("repo_id=?", rid).And("is_closed=?", isShowClosed).And("assignee_id=?", uid).Count(issue)
  329. stats.CreateCount, _ = x.Where("repo_id=?", rid).And("is_closed=?", isShowClosed).And("poster_id=?", uid).Count(issue)
  330. stats.MentionCount, _ = x.Where("repo_id=?", rid).And("uid=?", uid).And("is_closed=?", isShowClosed).And("is_mentioned=?", true).Count(new(IssueUser))
  331. return stats
  332. }
  333. // GetUserIssueStats returns issue statistic information for dashboard by given conditions.
  334. func GetUserIssueStats(uid int64, filterMode int) *IssueStats {
  335. stats := &IssueStats{}
  336. issue := new(Issue)
  337. stats.AssignCount, _ = x.Where("assignee_id=?", uid).And("is_closed=?", false).Count(issue)
  338. stats.CreateCount, _ = x.Where("poster_id=?", uid).And("is_closed=?", false).Count(issue)
  339. return stats
  340. }
  341. // UpdateIssue updates information of issue.
  342. func UpdateIssue(issue *Issue) error {
  343. _, err := x.Id(issue.Id).AllCols().Update(issue)
  344. return err
  345. }
  346. // UpdateIssueUserByStatus updates issue-user pairs by issue status.
  347. func UpdateIssueUserPairsByStatus(iid int64, isClosed bool) error {
  348. rawSql := "UPDATE `issue_user` SET is_closed = ? WHERE issue_id = ?"
  349. _, err := x.Exec(rawSql, isClosed, iid)
  350. return err
  351. }
  352. // UpdateIssueUserPairByAssignee updates issue-user pair for assigning.
  353. func UpdateIssueUserPairByAssignee(aid, iid int64) error {
  354. rawSql := "UPDATE `issue_user` SET is_assigned = ? WHERE issue_id = ?"
  355. if _, err := x.Exec(rawSql, false, iid); err != nil {
  356. return err
  357. }
  358. // Assignee ID equals to 0 means clear assignee.
  359. if aid == 0 {
  360. return nil
  361. }
  362. rawSql = "UPDATE `issue_user` SET is_assigned = true WHERE uid = ? AND issue_id = ?"
  363. _, err := x.Exec(rawSql, aid, iid)
  364. return err
  365. }
  366. // UpdateIssueUserPairByRead updates issue-user pair for reading.
  367. func UpdateIssueUserPairByRead(uid, iid int64) error {
  368. rawSql := "UPDATE `issue_user` SET is_read = ? WHERE uid = ? AND issue_id = ?"
  369. _, err := x.Exec(rawSql, true, uid, iid)
  370. return err
  371. }
  372. // UpdateIssueUserPairsByMentions updates issue-user pairs by mentioning.
  373. func UpdateIssueUserPairsByMentions(uids []int64, iid int64) error {
  374. for _, uid := range uids {
  375. iu := &IssueUser{Uid: uid, IssueId: iid}
  376. has, err := x.Get(iu)
  377. if err != nil {
  378. return err
  379. }
  380. iu.IsMentioned = true
  381. if has {
  382. _, err = x.Id(iu.Id).AllCols().Update(iu)
  383. } else {
  384. _, err = x.Insert(iu)
  385. }
  386. if err != nil {
  387. return err
  388. }
  389. }
  390. return nil
  391. }
  392. // .____ ___. .__
  393. // | | _____ \_ |__ ____ | |
  394. // | | \__ \ | __ \_/ __ \| |
  395. // | |___ / __ \| \_\ \ ___/| |__
  396. // |_______ (____ /___ /\___ >____/
  397. // \/ \/ \/ \/
  398. // Label represents a label of repository for issues.
  399. type Label struct {
  400. Id int64
  401. RepoId int64 `xorm:"INDEX"`
  402. Name string
  403. Color string `xorm:"VARCHAR(7)"`
  404. NumIssues int
  405. NumClosedIssues int
  406. NumOpenIssues int `xorm:"-"`
  407. IsChecked bool `xorm:"-"`
  408. }
  409. // CalOpenIssues calculates the open issues of label.
  410. func (m *Label) CalOpenIssues() {
  411. m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
  412. }
  413. // NewLabel creates new label of repository.
  414. func NewLabel(l *Label) error {
  415. _, err := x.Insert(l)
  416. return err
  417. }
  418. // GetLabelById returns a label by given ID.
  419. func GetLabelById(id int64) (*Label, error) {
  420. if id <= 0 {
  421. return nil, ErrLabelNotExist
  422. }
  423. l := &Label{Id: id}
  424. has, err := x.Get(l)
  425. if err != nil {
  426. return nil, err
  427. } else if !has {
  428. return nil, ErrLabelNotExist
  429. }
  430. return l, nil
  431. }
  432. // GetLabels returns a list of labels of given repository ID.
  433. func GetLabels(repoId int64) ([]*Label, error) {
  434. labels := make([]*Label, 0, 10)
  435. err := x.Where("repo_id=?", repoId).Find(&labels)
  436. return labels, err
  437. }
  438. // UpdateLabel updates label information.
  439. func UpdateLabel(l *Label) error {
  440. _, err := x.Id(l.Id).Update(l)
  441. return err
  442. }
  443. // DeleteLabel delete a label of given repository.
  444. func DeleteLabel(repoId int64, strId string) error {
  445. id, _ := base.StrTo(strId).Int64()
  446. l, err := GetLabelById(id)
  447. if err != nil {
  448. if err == ErrLabelNotExist {
  449. return nil
  450. }
  451. return err
  452. }
  453. issues, err := GetIssuesByLabel(repoId, strId)
  454. if err != nil {
  455. return err
  456. }
  457. sess := x.NewSession()
  458. defer sess.Close()
  459. if err = sess.Begin(); err != nil {
  460. return err
  461. }
  462. for _, issue := range issues {
  463. issue.LabelIds = strings.Replace(issue.LabelIds, "$"+strId+"|", "", -1)
  464. if _, err = sess.Id(issue.Id).AllCols().Update(issue); err != nil {
  465. sess.Rollback()
  466. return err
  467. }
  468. }
  469. if _, err = sess.Delete(l); err != nil {
  470. sess.Rollback()
  471. return err
  472. }
  473. return sess.Commit()
  474. }
  475. // _____ .__.__ __
  476. // / \ |__| | ____ _______/ |_ ____ ____ ____
  477. // / \ / \| | | _/ __ \ / ___/\ __\/ _ \ / \_/ __ \
  478. // / Y \ | |_\ ___/ \___ \ | | ( <_> ) | \ ___/
  479. // \____|__ /__|____/\___ >____ > |__| \____/|___| /\___ >
  480. // \/ \/ \/ \/ \/
  481. // Milestone represents a milestone of repository.
  482. type Milestone struct {
  483. Id int64
  484. RepoId int64 `xorm:"INDEX"`
  485. Index int64
  486. Name string
  487. Content string
  488. RenderedContent string `xorm:"-"`
  489. IsClosed bool
  490. NumIssues int
  491. NumClosedIssues int
  492. NumOpenIssues int `xorm:"-"`
  493. Completeness int // Percentage(1-100).
  494. Deadline time.Time
  495. DeadlineString string `xorm:"-"`
  496. ClosedDate time.Time
  497. }
  498. // CalOpenIssues calculates the open issues of milestone.
  499. func (m *Milestone) CalOpenIssues() {
  500. m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
  501. }
  502. // NewMilestone creates new milestone of repository.
  503. func NewMilestone(m *Milestone) (err error) {
  504. sess := x.NewSession()
  505. defer sess.Close()
  506. if err = sess.Begin(); err != nil {
  507. return err
  508. }
  509. if _, err = sess.Insert(m); err != nil {
  510. sess.Rollback()
  511. return err
  512. }
  513. rawSql := "UPDATE `repository` SET num_milestones = num_milestones + 1 WHERE id = ?"
  514. if _, err = sess.Exec(rawSql, m.RepoId); err != nil {
  515. sess.Rollback()
  516. return err
  517. }
  518. return sess.Commit()
  519. }
  520. // GetMilestoneById returns the milestone by given ID.
  521. func GetMilestoneById(id int64) (*Milestone, error) {
  522. m := &Milestone{Id: id}
  523. has, err := x.Get(m)
  524. if err != nil {
  525. return nil, err
  526. } else if !has {
  527. return nil, ErrMilestoneNotExist
  528. }
  529. return m, nil
  530. }
  531. // GetMilestoneByIndex returns the milestone of given repository and index.
  532. func GetMilestoneByIndex(repoId, idx int64) (*Milestone, error) {
  533. m := &Milestone{RepoId: repoId, Index: idx}
  534. has, err := x.Get(m)
  535. if err != nil {
  536. return nil, err
  537. } else if !has {
  538. return nil, ErrMilestoneNotExist
  539. }
  540. return m, nil
  541. }
  542. // GetMilestones returns a list of milestones of given repository and status.
  543. func GetMilestones(repoId int64, isClosed bool) ([]*Milestone, error) {
  544. miles := make([]*Milestone, 0, 10)
  545. err := x.Where("repo_id=?", repoId).And("is_closed=?", isClosed).Find(&miles)
  546. return miles, err
  547. }
  548. // UpdateMilestone updates information of given milestone.
  549. func UpdateMilestone(m *Milestone) error {
  550. _, err := x.Id(m.Id).Update(m)
  551. return err
  552. }
  553. // ChangeMilestoneStatus changes the milestone open/closed status.
  554. func ChangeMilestoneStatus(m *Milestone, isClosed bool) (err error) {
  555. repo, err := GetRepositoryById(m.RepoId)
  556. if err != nil {
  557. return err
  558. }
  559. sess := x.NewSession()
  560. defer sess.Close()
  561. if err = sess.Begin(); err != nil {
  562. return err
  563. }
  564. m.IsClosed = isClosed
  565. if _, err = sess.Id(m.Id).AllCols().Update(m); err != nil {
  566. sess.Rollback()
  567. return err
  568. }
  569. if isClosed {
  570. repo.NumClosedMilestones++
  571. } else {
  572. repo.NumClosedMilestones--
  573. }
  574. if _, err = sess.Id(repo.Id).Update(repo); err != nil {
  575. sess.Rollback()
  576. return err
  577. }
  578. return sess.Commit()
  579. }
  580. // ChangeMilestoneAssign changes assignment of milestone for issue.
  581. func ChangeMilestoneAssign(oldMid, mid int64, issue *Issue) (err error) {
  582. sess := x.NewSession()
  583. defer sess.Close()
  584. if err = sess.Begin(); err != nil {
  585. return err
  586. }
  587. if oldMid > 0 {
  588. m, err := GetMilestoneById(oldMid)
  589. if err != nil {
  590. return err
  591. }
  592. m.NumIssues--
  593. if issue.IsClosed {
  594. m.NumClosedIssues--
  595. }
  596. if m.NumIssues > 0 {
  597. m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
  598. } else {
  599. m.Completeness = 0
  600. }
  601. if _, err = sess.Id(m.Id).Update(m); err != nil {
  602. sess.Rollback()
  603. return err
  604. }
  605. rawSql := "UPDATE `issue_user` SET milestone_id = 0 WHERE issue_id = ?"
  606. if _, err = sess.Exec(rawSql, issue.Id); err != nil {
  607. sess.Rollback()
  608. return err
  609. }
  610. }
  611. if mid > 0 {
  612. m, err := GetMilestoneById(mid)
  613. if err != nil {
  614. return err
  615. }
  616. m.NumIssues++
  617. if issue.IsClosed {
  618. m.NumClosedIssues++
  619. }
  620. m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
  621. if _, err = sess.Id(m.Id).Update(m); err != nil {
  622. sess.Rollback()
  623. return err
  624. }
  625. rawSql := "UPDATE `issue_user` SET milestone_id = ? WHERE issue_id = ?"
  626. if _, err = sess.Exec(rawSql, m.Id, issue.Id); err != nil {
  627. sess.Rollback()
  628. return err
  629. }
  630. }
  631. return sess.Commit()
  632. }
  633. // DeleteMilestone deletes a milestone.
  634. func DeleteMilestone(m *Milestone) (err error) {
  635. sess := x.NewSession()
  636. defer sess.Close()
  637. if err = sess.Begin(); err != nil {
  638. return err
  639. }
  640. if _, err = sess.Delete(m); err != nil {
  641. sess.Rollback()
  642. return err
  643. }
  644. rawSql := "UPDATE `repository` SET num_milestones = num_milestones - 1 WHERE id = ?"
  645. if _, err = sess.Exec(rawSql, m.RepoId); err != nil {
  646. sess.Rollback()
  647. return err
  648. }
  649. rawSql = "UPDATE `issue` SET milestone_id = 0 WHERE milestone_id = ?"
  650. if _, err = sess.Exec(rawSql, m.Id); err != nil {
  651. sess.Rollback()
  652. return err
  653. }
  654. rawSql = "UPDATE `issue_user` SET milestone_id = 0 WHERE milestone_id = ?"
  655. if _, err = sess.Exec(rawSql, m.Id); err != nil {
  656. sess.Rollback()
  657. return err
  658. }
  659. return sess.Commit()
  660. }
  661. // _________ __
  662. // \_ ___ \ ____ _____ _____ ____ _____/ |_
  663. // / \ \/ / _ \ / \ / \_/ __ \ / \ __\
  664. // \ \___( <_> ) Y Y \ Y Y \ ___/| | \ |
  665. // \______ /\____/|__|_| /__|_| /\___ >___| /__|
  666. // \/ \/ \/ \/ \/
  667. // Issue types.
  668. const (
  669. IT_PLAIN = iota // Pure comment.
  670. IT_REOPEN // Issue reopen status change prompt.
  671. IT_CLOSE // Issue close status change prompt.
  672. )
  673. // Comment represents a comment in commit and issue page.
  674. type Comment struct {
  675. Id int64
  676. Type int
  677. PosterId int64
  678. Poster *User `xorm:"-"`
  679. IssueId int64
  680. CommitId int64
  681. Line int64
  682. Content string `xorm:"TEXT"`
  683. Created time.Time `xorm:"CREATED"`
  684. }
  685. // CreateComment creates comment of issue or commit.
  686. func CreateComment(userId, repoId, issueId, commitId, line int64, cmtType int, content string) error {
  687. sess := x.NewSession()
  688. defer sess.Close()
  689. if err := sess.Begin(); err != nil {
  690. return err
  691. }
  692. if _, err := sess.Insert(&Comment{PosterId: userId, Type: cmtType, IssueId: issueId,
  693. CommitId: commitId, Line: line, Content: content}); err != nil {
  694. sess.Rollback()
  695. return err
  696. }
  697. // Check comment type.
  698. switch cmtType {
  699. case IT_PLAIN:
  700. rawSql := "UPDATE `issue` SET num_comments = num_comments + 1 WHERE id = ?"
  701. if _, err := sess.Exec(rawSql, issueId); err != nil {
  702. sess.Rollback()
  703. return err
  704. }
  705. case IT_REOPEN:
  706. rawSql := "UPDATE `repository` SET num_closed_issues = num_closed_issues - 1 WHERE id = ?"
  707. if _, err := sess.Exec(rawSql, repoId); err != nil {
  708. sess.Rollback()
  709. return err
  710. }
  711. case IT_CLOSE:
  712. rawSql := "UPDATE `repository` SET num_closed_issues = num_closed_issues + 1 WHERE id = ?"
  713. if _, err := sess.Exec(rawSql, repoId); err != nil {
  714. sess.Rollback()
  715. return err
  716. }
  717. }
  718. return sess.Commit()
  719. }
  720. // GetIssueComments returns list of comment by given issue id.
  721. func GetIssueComments(issueId int64) ([]Comment, error) {
  722. comments := make([]Comment, 0, 10)
  723. err := x.Asc("created").Find(&comments, &Comment{IssueId: issueId})
  724. return comments, err
  725. }