You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

issue.go 21 kB

11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
11 years ago
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 := 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. 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 := orm.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, _ := orm.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 // User ID.
  189. IssueId int64
  190. RepoId int64
  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 = orm.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 = orm.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 := orm.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 := orm.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 := orm.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 := orm.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 = orm.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 := orm.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, _ = orm.Where("repo_id=?", rid).And("is_closed=?", isShowClosed).And("assignee_id=?", uid).Count(issue)
  329. stats.CreateCount, _ = orm.Where("repo_id=?", rid).And("is_closed=?", isShowClosed).And("poster_id=?", uid).Count(issue)
  330. stats.MentionCount, _ = orm.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, _ = orm.Where("assignee_id=?", uid).And("is_closed=?", false).Count(issue)
  338. stats.CreateCount, _ = orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.Get(iu)
  377. if err != nil {
  378. return err
  379. }
  380. iu.IsMentioned = true
  381. if has {
  382. _, err = orm.Id(iu.Id).AllCols().Update(iu)
  383. } else {
  384. _, err = orm.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 := orm.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 := orm.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 := orm.Where("repo_id=?", repoId).Find(&labels)
  436. return labels, err
  437. }
  438. // UpdateLabel updates label information.
  439. func UpdateLabel(l *Label) error {
  440. _, err := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.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 := orm.Asc("created").Find(&comments, &Comment{IssueId: issueId})
  724. return comments, err
  725. }