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_label.go 23KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830
  1. // Copyright 2016 The Gogs Authors. All rights reserved.
  2. // Copyright 2020 The Gitea Authors.
  3. // Use of this source code is governed by a MIT-style
  4. // license that can be found in the LICENSE file.
  5. package models
  6. import (
  7. "context"
  8. "fmt"
  9. "html/template"
  10. "math"
  11. "regexp"
  12. "strconv"
  13. "strings"
  14. "code.gitea.io/gitea/models/db"
  15. "code.gitea.io/gitea/modules/timeutil"
  16. "xorm.io/builder"
  17. )
  18. // LabelColorPattern is a regexp witch can validate LabelColor
  19. var LabelColorPattern = regexp.MustCompile("^#[0-9a-fA-F]{6}$")
  20. // Label represents a label of repository for issues.
  21. type Label struct {
  22. ID int64 `xorm:"pk autoincr"`
  23. RepoID int64 `xorm:"INDEX"`
  24. OrgID int64 `xorm:"INDEX"`
  25. Name string
  26. Description string
  27. Color string `xorm:"VARCHAR(7)"`
  28. NumIssues int
  29. NumClosedIssues int
  30. CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
  31. UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
  32. NumOpenIssues int `xorm:"-"`
  33. NumOpenRepoIssues int64 `xorm:"-"`
  34. IsChecked bool `xorm:"-"`
  35. QueryString string `xorm:"-"`
  36. IsSelected bool `xorm:"-"`
  37. IsExcluded bool `xorm:"-"`
  38. }
  39. func init() {
  40. db.RegisterModel(new(Label))
  41. db.RegisterModel(new(IssueLabel))
  42. }
  43. // GetLabelTemplateFile loads the label template file by given name,
  44. // then parses and returns a list of name-color pairs and optionally description.
  45. func GetLabelTemplateFile(name string) ([][3]string, error) {
  46. data, err := GetRepoInitFile("label", name)
  47. if err != nil {
  48. return nil, ErrIssueLabelTemplateLoad{name, fmt.Errorf("GetRepoInitFile: %v", err)}
  49. }
  50. lines := strings.Split(string(data), "\n")
  51. list := make([][3]string, 0, len(lines))
  52. for i := 0; i < len(lines); i++ {
  53. line := strings.TrimSpace(lines[i])
  54. if len(line) == 0 {
  55. continue
  56. }
  57. parts := strings.SplitN(line, ";", 2)
  58. fields := strings.SplitN(parts[0], " ", 2)
  59. if len(fields) != 2 {
  60. return nil, ErrIssueLabelTemplateLoad{name, fmt.Errorf("line is malformed: %s", line)}
  61. }
  62. color := strings.Trim(fields[0], " ")
  63. if len(color) == 6 {
  64. color = "#" + color
  65. }
  66. if !LabelColorPattern.MatchString(color) {
  67. return nil, ErrIssueLabelTemplateLoad{name, fmt.Errorf("bad HTML color code in line: %s", line)}
  68. }
  69. var description string
  70. if len(parts) > 1 {
  71. description = strings.TrimSpace(parts[1])
  72. }
  73. fields[1] = strings.TrimSpace(fields[1])
  74. list = append(list, [3]string{fields[1], color, description})
  75. }
  76. return list, nil
  77. }
  78. // CalOpenIssues sets the number of open issues of a label based on the already stored number of closed issues.
  79. func (label *Label) CalOpenIssues() {
  80. label.NumOpenIssues = label.NumIssues - label.NumClosedIssues
  81. }
  82. // CalOpenOrgIssues calculates the open issues of a label for a specific repo
  83. func (label *Label) CalOpenOrgIssues(repoID, labelID int64) {
  84. repoIDs := []int64{repoID}
  85. labelIDs := []int64{labelID}
  86. counts, _ := CountIssuesByRepo(&IssuesOptions{
  87. RepoIDs: repoIDs,
  88. LabelIDs: labelIDs,
  89. })
  90. for _, count := range counts {
  91. label.NumOpenRepoIssues += count
  92. }
  93. }
  94. // LoadSelectedLabelsAfterClick calculates the set of selected labels when a label is clicked
  95. func (label *Label) LoadSelectedLabelsAfterClick(currentSelectedLabels []int64) {
  96. var labelQuerySlice []string
  97. labelSelected := false
  98. labelID := strconv.FormatInt(label.ID, 10)
  99. for _, s := range currentSelectedLabels {
  100. if s == label.ID {
  101. labelSelected = true
  102. } else if -s == label.ID {
  103. labelSelected = true
  104. label.IsExcluded = true
  105. } else if s != 0 {
  106. labelQuerySlice = append(labelQuerySlice, strconv.FormatInt(s, 10))
  107. }
  108. }
  109. if !labelSelected {
  110. labelQuerySlice = append(labelQuerySlice, labelID)
  111. }
  112. label.IsSelected = labelSelected
  113. label.QueryString = strings.Join(labelQuerySlice, ",")
  114. }
  115. // BelongsToOrg returns true if label is an organization label
  116. func (label *Label) BelongsToOrg() bool {
  117. return label.OrgID > 0
  118. }
  119. // BelongsToRepo returns true if label is a repository label
  120. func (label *Label) BelongsToRepo() bool {
  121. return label.RepoID > 0
  122. }
  123. // SrgbToLinear converts a component of an sRGB color to its linear intensity
  124. // See: https://en.wikipedia.org/wiki/SRGB#The_reverse_transformation_(sRGB_to_CIE_XYZ)
  125. func SrgbToLinear(color uint8) float64 {
  126. flt := float64(color) / 255
  127. if flt <= 0.04045 {
  128. return flt / 12.92
  129. }
  130. return math.Pow((flt+0.055)/1.055, 2.4)
  131. }
  132. // Luminance returns the luminance of an sRGB color
  133. func Luminance(color uint32) float64 {
  134. r := SrgbToLinear(uint8(0xFF & (color >> 16)))
  135. g := SrgbToLinear(uint8(0xFF & (color >> 8)))
  136. b := SrgbToLinear(uint8(0xFF & color))
  137. // luminance ratios for sRGB
  138. return 0.2126*r + 0.7152*g + 0.0722*b
  139. }
  140. // LuminanceThreshold is the luminance at which white and black appear to have the same contrast
  141. // i.e. x such that 1.05 / (x + 0.05) = (x + 0.05) / 0.05
  142. // i.e. math.Sqrt(1.05*0.05) - 0.05
  143. const LuminanceThreshold float64 = 0.179
  144. // ForegroundColor calculates the text color for labels based
  145. // on their background color.
  146. func (label *Label) ForegroundColor() template.CSS {
  147. if strings.HasPrefix(label.Color, "#") {
  148. if color, err := strconv.ParseUint(label.Color[1:], 16, 64); err == nil {
  149. // NOTE: see web_src/js/components/ContextPopup.vue for similar implementation
  150. luminance := Luminance(uint32(color))
  151. // prefer white or black based upon contrast
  152. if luminance < LuminanceThreshold {
  153. return template.CSS("#fff")
  154. }
  155. return template.CSS("#000")
  156. }
  157. }
  158. // default to black
  159. return template.CSS("#000")
  160. }
  161. // .____ ___. .__
  162. // | | _____ \_ |__ ____ | |
  163. // | | \__ \ | __ \_/ __ \| |
  164. // | |___ / __ \| \_\ \ ___/| |__
  165. // >_______ (____ /___ /\___ >____/
  166. func loadLabels(labelTemplate string) ([]string, error) {
  167. list, err := GetLabelTemplateFile(labelTemplate)
  168. if err != nil {
  169. return nil, err
  170. }
  171. labels := make([]string, len(list))
  172. for i := 0; i < len(list); i++ {
  173. labels[i] = list[i][0]
  174. }
  175. return labels, nil
  176. }
  177. // LoadLabelsFormatted loads the labels' list of a template file as a string separated by comma
  178. func LoadLabelsFormatted(labelTemplate string) (string, error) {
  179. labels, err := loadLabels(labelTemplate)
  180. return strings.Join(labels, ", "), err
  181. }
  182. func initializeLabels(e db.Engine, id int64, labelTemplate string, isOrg bool) error {
  183. list, err := GetLabelTemplateFile(labelTemplate)
  184. if err != nil {
  185. return err
  186. }
  187. labels := make([]*Label, len(list))
  188. for i := 0; i < len(list); i++ {
  189. labels[i] = &Label{
  190. Name: list[i][0],
  191. Description: list[i][2],
  192. Color: list[i][1],
  193. }
  194. if isOrg {
  195. labels[i].OrgID = id
  196. } else {
  197. labels[i].RepoID = id
  198. }
  199. }
  200. for _, label := range labels {
  201. if err = newLabel(e, label); err != nil {
  202. return err
  203. }
  204. }
  205. return nil
  206. }
  207. // InitializeLabels adds a label set to a repository using a template
  208. func InitializeLabels(ctx context.Context, repoID int64, labelTemplate string, isOrg bool) error {
  209. return initializeLabels(db.GetEngine(ctx), repoID, labelTemplate, isOrg)
  210. }
  211. func newLabel(e db.Engine, label *Label) error {
  212. _, err := e.Insert(label)
  213. return err
  214. }
  215. // NewLabel creates a new label
  216. func NewLabel(label *Label) error {
  217. if !LabelColorPattern.MatchString(label.Color) {
  218. return fmt.Errorf("bad color code: %s", label.Color)
  219. }
  220. return newLabel(db.GetEngine(db.DefaultContext), label)
  221. }
  222. // NewLabels creates new labels
  223. func NewLabels(labels ...*Label) error {
  224. ctx, committer, err := db.TxContext()
  225. if err != nil {
  226. return err
  227. }
  228. defer committer.Close()
  229. for _, label := range labels {
  230. if !LabelColorPattern.MatchString(label.Color) {
  231. return fmt.Errorf("bad color code: %s", label.Color)
  232. }
  233. if err := newLabel(db.GetEngine(ctx), label); err != nil {
  234. return err
  235. }
  236. }
  237. return committer.Commit()
  238. }
  239. // UpdateLabel updates label information.
  240. func UpdateLabel(l *Label) error {
  241. if !LabelColorPattern.MatchString(l.Color) {
  242. return fmt.Errorf("bad color code: %s", l.Color)
  243. }
  244. return updateLabelCols(db.GetEngine(db.DefaultContext), l, "name", "description", "color")
  245. }
  246. // DeleteLabel delete a label
  247. func DeleteLabel(id, labelID int64) error {
  248. label, err := GetLabelByID(labelID)
  249. if err != nil {
  250. if IsErrLabelNotExist(err) {
  251. return nil
  252. }
  253. return err
  254. }
  255. ctx, committer, err := db.TxContext()
  256. if err != nil {
  257. return err
  258. }
  259. defer committer.Close()
  260. sess := db.GetEngine(ctx)
  261. if label.BelongsToOrg() && label.OrgID != id {
  262. return nil
  263. }
  264. if label.BelongsToRepo() && label.RepoID != id {
  265. return nil
  266. }
  267. if _, err = sess.ID(labelID).Delete(new(Label)); err != nil {
  268. return err
  269. } else if _, err = sess.
  270. Where("label_id = ?", labelID).
  271. Delete(new(IssueLabel)); err != nil {
  272. return err
  273. }
  274. // delete comments about now deleted label_id
  275. if _, err = sess.Where("label_id = ?", labelID).Cols("label_id").Delete(&Comment{}); err != nil {
  276. return err
  277. }
  278. return committer.Commit()
  279. }
  280. // getLabelByID returns a label by label id
  281. func getLabelByID(e db.Engine, labelID int64) (*Label, error) {
  282. if labelID <= 0 {
  283. return nil, ErrLabelNotExist{labelID}
  284. }
  285. l := &Label{}
  286. has, err := e.ID(labelID).Get(l)
  287. if err != nil {
  288. return nil, err
  289. } else if !has {
  290. return nil, ErrLabelNotExist{l.ID}
  291. }
  292. return l, nil
  293. }
  294. // GetLabelByID returns a label by given ID.
  295. func GetLabelByID(id int64) (*Label, error) {
  296. return getLabelByID(db.GetEngine(db.DefaultContext), id)
  297. }
  298. // GetLabelsByIDs returns a list of labels by IDs
  299. func GetLabelsByIDs(labelIDs []int64) ([]*Label, error) {
  300. labels := make([]*Label, 0, len(labelIDs))
  301. return labels, db.GetEngine(db.DefaultContext).Table("label").
  302. In("id", labelIDs).
  303. Asc("name").
  304. Cols("id", "repo_id", "org_id").
  305. Find(&labels)
  306. }
  307. // __________ .__ __
  308. // \______ \ ____ ______ ____ _____|__|/ |_ ___________ ___.__.
  309. // | _// __ \\____ \ / _ \/ ___/ \ __\/ _ \_ __ < | |
  310. // | | \ ___/| |_> > <_> )___ \| || | ( <_> ) | \/\___ |
  311. // |____|_ /\___ > __/ \____/____ >__||__| \____/|__| / ____|
  312. // \/ \/|__| \/ \/
  313. // getLabelInRepoByName returns a label by Name in given repository.
  314. func getLabelInRepoByName(e db.Engine, repoID int64, labelName string) (*Label, error) {
  315. if len(labelName) == 0 || repoID <= 0 {
  316. return nil, ErrRepoLabelNotExist{0, repoID}
  317. }
  318. l := &Label{
  319. Name: labelName,
  320. RepoID: repoID,
  321. }
  322. has, err := e.Get(l)
  323. if err != nil {
  324. return nil, err
  325. } else if !has {
  326. return nil, ErrRepoLabelNotExist{0, l.RepoID}
  327. }
  328. return l, nil
  329. }
  330. // getLabelInRepoByID returns a label by ID in given repository.
  331. func getLabelInRepoByID(e db.Engine, repoID, labelID int64) (*Label, error) {
  332. if labelID <= 0 || repoID <= 0 {
  333. return nil, ErrRepoLabelNotExist{labelID, repoID}
  334. }
  335. l := &Label{
  336. ID: labelID,
  337. RepoID: repoID,
  338. }
  339. has, err := e.Get(l)
  340. if err != nil {
  341. return nil, err
  342. } else if !has {
  343. return nil, ErrRepoLabelNotExist{l.ID, l.RepoID}
  344. }
  345. return l, nil
  346. }
  347. // GetLabelInRepoByName returns a label by name in given repository.
  348. func GetLabelInRepoByName(repoID int64, labelName string) (*Label, error) {
  349. return getLabelInRepoByName(db.GetEngine(db.DefaultContext), repoID, labelName)
  350. }
  351. // GetLabelIDsInRepoByNames returns a list of labelIDs by names in a given
  352. // repository.
  353. // it silently ignores label names that do not belong to the repository.
  354. func GetLabelIDsInRepoByNames(repoID int64, labelNames []string) ([]int64, error) {
  355. labelIDs := make([]int64, 0, len(labelNames))
  356. return labelIDs, db.GetEngine(db.DefaultContext).Table("label").
  357. Where("repo_id = ?", repoID).
  358. In("name", labelNames).
  359. Asc("name").
  360. Cols("id").
  361. Find(&labelIDs)
  362. }
  363. // BuildLabelNamesIssueIDsCondition returns a builder where get issue ids match label names
  364. func BuildLabelNamesIssueIDsCondition(labelNames []string) *builder.Builder {
  365. return builder.Select("issue_label.issue_id").
  366. From("issue_label").
  367. InnerJoin("label", "label.id = issue_label.label_id").
  368. Where(
  369. builder.In("label.name", labelNames),
  370. ).
  371. GroupBy("issue_label.issue_id")
  372. }
  373. // GetLabelInRepoByID returns a label by ID in given repository.
  374. func GetLabelInRepoByID(repoID, labelID int64) (*Label, error) {
  375. return getLabelInRepoByID(db.GetEngine(db.DefaultContext), repoID, labelID)
  376. }
  377. // GetLabelsInRepoByIDs returns a list of labels by IDs in given repository,
  378. // it silently ignores label IDs that do not belong to the repository.
  379. func GetLabelsInRepoByIDs(repoID int64, labelIDs []int64) ([]*Label, error) {
  380. labels := make([]*Label, 0, len(labelIDs))
  381. return labels, db.GetEngine(db.DefaultContext).
  382. Where("repo_id = ?", repoID).
  383. In("id", labelIDs).
  384. Asc("name").
  385. Find(&labels)
  386. }
  387. func getLabelsByRepoID(e db.Engine, repoID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  388. if repoID <= 0 {
  389. return nil, ErrRepoLabelNotExist{0, repoID}
  390. }
  391. labels := make([]*Label, 0, 10)
  392. sess := e.Where("repo_id = ?", repoID)
  393. switch sortType {
  394. case "reversealphabetically":
  395. sess.Desc("name")
  396. case "leastissues":
  397. sess.Asc("num_issues")
  398. case "mostissues":
  399. sess.Desc("num_issues")
  400. default:
  401. sess.Asc("name")
  402. }
  403. if listOptions.Page != 0 {
  404. sess = db.SetSessionPagination(sess, &listOptions)
  405. }
  406. return labels, sess.Find(&labels)
  407. }
  408. // GetLabelsByRepoID returns all labels that belong to given repository by ID.
  409. func GetLabelsByRepoID(repoID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  410. return getLabelsByRepoID(db.GetEngine(db.DefaultContext), repoID, sortType, listOptions)
  411. }
  412. // CountLabelsByRepoID count number of all labels that belong to given repository by ID.
  413. func CountLabelsByRepoID(repoID int64) (int64, error) {
  414. return db.GetEngine(db.DefaultContext).Where("repo_id = ?", repoID).Count(&Label{})
  415. }
  416. // ________
  417. // \_____ \_______ ____
  418. // / | \_ __ \/ ___\
  419. // / | \ | \/ /_/ >
  420. // \_______ /__| \___ /
  421. // \/ /_____/
  422. // getLabelInOrgByName returns a label by Name in given organization
  423. func getLabelInOrgByName(e db.Engine, orgID int64, labelName string) (*Label, error) {
  424. if len(labelName) == 0 || orgID <= 0 {
  425. return nil, ErrOrgLabelNotExist{0, orgID}
  426. }
  427. l := &Label{
  428. Name: labelName,
  429. OrgID: orgID,
  430. }
  431. has, err := e.Get(l)
  432. if err != nil {
  433. return nil, err
  434. } else if !has {
  435. return nil, ErrOrgLabelNotExist{0, l.OrgID}
  436. }
  437. return l, nil
  438. }
  439. // getLabelInOrgByID returns a label by ID in given organization.
  440. func getLabelInOrgByID(e db.Engine, orgID, labelID int64) (*Label, error) {
  441. if labelID <= 0 || orgID <= 0 {
  442. return nil, ErrOrgLabelNotExist{labelID, orgID}
  443. }
  444. l := &Label{
  445. ID: labelID,
  446. OrgID: orgID,
  447. }
  448. has, err := e.Get(l)
  449. if err != nil {
  450. return nil, err
  451. } else if !has {
  452. return nil, ErrOrgLabelNotExist{l.ID, l.OrgID}
  453. }
  454. return l, nil
  455. }
  456. // GetLabelInOrgByName returns a label by name in given organization.
  457. func GetLabelInOrgByName(orgID int64, labelName string) (*Label, error) {
  458. return getLabelInOrgByName(db.GetEngine(db.DefaultContext), orgID, labelName)
  459. }
  460. // GetLabelIDsInOrgByNames returns a list of labelIDs by names in a given
  461. // organization.
  462. func GetLabelIDsInOrgByNames(orgID int64, labelNames []string) ([]int64, error) {
  463. if orgID <= 0 {
  464. return nil, ErrOrgLabelNotExist{0, orgID}
  465. }
  466. labelIDs := make([]int64, 0, len(labelNames))
  467. return labelIDs, db.GetEngine(db.DefaultContext).Table("label").
  468. Where("org_id = ?", orgID).
  469. In("name", labelNames).
  470. Asc("name").
  471. Cols("id").
  472. Find(&labelIDs)
  473. }
  474. // GetLabelInOrgByID returns a label by ID in given organization.
  475. func GetLabelInOrgByID(orgID, labelID int64) (*Label, error) {
  476. return getLabelInOrgByID(db.GetEngine(db.DefaultContext), orgID, labelID)
  477. }
  478. // GetLabelsInOrgByIDs returns a list of labels by IDs in given organization,
  479. // it silently ignores label IDs that do not belong to the organization.
  480. func GetLabelsInOrgByIDs(orgID int64, labelIDs []int64) ([]*Label, error) {
  481. labels := make([]*Label, 0, len(labelIDs))
  482. return labels, db.GetEngine(db.DefaultContext).
  483. Where("org_id = ?", orgID).
  484. In("id", labelIDs).
  485. Asc("name").
  486. Find(&labels)
  487. }
  488. func getLabelsByOrgID(e db.Engine, orgID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  489. if orgID <= 0 {
  490. return nil, ErrOrgLabelNotExist{0, orgID}
  491. }
  492. labels := make([]*Label, 0, 10)
  493. sess := e.Where("org_id = ?", orgID)
  494. switch sortType {
  495. case "reversealphabetically":
  496. sess.Desc("name")
  497. case "leastissues":
  498. sess.Asc("num_issues")
  499. case "mostissues":
  500. sess.Desc("num_issues")
  501. default:
  502. sess.Asc("name")
  503. }
  504. if listOptions.Page != 0 {
  505. sess = db.SetSessionPagination(sess, &listOptions)
  506. }
  507. return labels, sess.Find(&labels)
  508. }
  509. // GetLabelsByOrgID returns all labels that belong to given organization by ID.
  510. func GetLabelsByOrgID(orgID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  511. return getLabelsByOrgID(db.GetEngine(db.DefaultContext), orgID, sortType, listOptions)
  512. }
  513. // CountLabelsByOrgID count all labels that belong to given organization by ID.
  514. func CountLabelsByOrgID(orgID int64) (int64, error) {
  515. return db.GetEngine(db.DefaultContext).Where("org_id = ?", orgID).Count(&Label{})
  516. }
  517. // .___
  518. // | | ______ ________ __ ____
  519. // | |/ ___// ___/ | \_/ __ \
  520. // | |\___ \ \___ \| | /\ ___/
  521. // |___/____ >____ >____/ \___ |
  522. // \/ \/ \/
  523. func getLabelsByIssueID(e db.Engine, issueID int64) ([]*Label, error) {
  524. var labels []*Label
  525. return labels, e.Where("issue_label.issue_id = ?", issueID).
  526. Join("LEFT", "issue_label", "issue_label.label_id = label.id").
  527. Asc("label.name").
  528. Find(&labels)
  529. }
  530. // GetLabelsByIssueID returns all labels that belong to given issue by ID.
  531. func GetLabelsByIssueID(issueID int64) ([]*Label, error) {
  532. return getLabelsByIssueID(db.GetEngine(db.DefaultContext), issueID)
  533. }
  534. func updateLabelCols(e db.Engine, l *Label, cols ...string) error {
  535. _, err := e.ID(l.ID).
  536. SetExpr("num_issues",
  537. builder.Select("count(*)").From("issue_label").
  538. Where(builder.Eq{"label_id": l.ID}),
  539. ).
  540. SetExpr("num_closed_issues",
  541. builder.Select("count(*)").From("issue_label").
  542. InnerJoin("issue", "issue_label.issue_id = issue.id").
  543. Where(builder.Eq{
  544. "issue_label.label_id": l.ID,
  545. "issue.is_closed": true,
  546. }),
  547. ).
  548. Cols(cols...).Update(l)
  549. return err
  550. }
  551. // .___ .____ ___. .__
  552. // | | ______ ________ __ ____ | | _____ \_ |__ ____ | |
  553. // | |/ ___// ___/ | \_/ __ \| | \__ \ | __ \_/ __ \| |
  554. // | |\___ \ \___ \| | /\ ___/| |___ / __ \| \_\ \ ___/| |__
  555. // |___/____ >____ >____/ \___ >_______ (____ /___ /\___ >____/
  556. // \/ \/ \/ \/ \/ \/ \/
  557. // IssueLabel represents an issue-label relation.
  558. type IssueLabel struct {
  559. ID int64 `xorm:"pk autoincr"`
  560. IssueID int64 `xorm:"UNIQUE(s)"`
  561. LabelID int64 `xorm:"UNIQUE(s)"`
  562. }
  563. func hasIssueLabel(e db.Engine, issueID, labelID int64) bool {
  564. has, _ := e.Where("issue_id = ? AND label_id = ?", issueID, labelID).Get(new(IssueLabel))
  565. return has
  566. }
  567. // HasIssueLabel returns true if issue has been labeled.
  568. func HasIssueLabel(issueID, labelID int64) bool {
  569. return hasIssueLabel(db.GetEngine(db.DefaultContext), issueID, labelID)
  570. }
  571. // newIssueLabel this function creates a new label it does not check if the label is valid for the issue
  572. // YOU MUST CHECK THIS BEFORE THIS FUNCTION
  573. func newIssueLabel(ctx context.Context, issue *Issue, label *Label, doer *User) (err error) {
  574. e := db.GetEngine(ctx)
  575. if _, err = e.Insert(&IssueLabel{
  576. IssueID: issue.ID,
  577. LabelID: label.ID,
  578. }); err != nil {
  579. return err
  580. }
  581. if err = issue.loadRepo(e); err != nil {
  582. return
  583. }
  584. opts := &CreateCommentOptions{
  585. Type: CommentTypeLabel,
  586. Doer: doer,
  587. Repo: issue.Repo,
  588. Issue: issue,
  589. Label: label,
  590. Content: "1",
  591. }
  592. if _, err = createComment(ctx, opts); err != nil {
  593. return err
  594. }
  595. return updateLabelCols(e, label, "num_issues", "num_closed_issue")
  596. }
  597. // NewIssueLabel creates a new issue-label relation.
  598. func NewIssueLabel(issue *Issue, label *Label, doer *User) (err error) {
  599. if HasIssueLabel(issue.ID, label.ID) {
  600. return nil
  601. }
  602. ctx, committer, err := db.TxContext()
  603. if err != nil {
  604. return err
  605. }
  606. defer committer.Close()
  607. sess := db.GetEngine(ctx)
  608. if err = issue.loadRepo(sess); err != nil {
  609. return err
  610. }
  611. // Do NOT add invalid labels
  612. if issue.RepoID != label.RepoID && issue.Repo.OwnerID != label.OrgID {
  613. return nil
  614. }
  615. if err = newIssueLabel(ctx, issue, label, doer); err != nil {
  616. return err
  617. }
  618. issue.Labels = nil
  619. if err = issue.loadLabels(sess); err != nil {
  620. return err
  621. }
  622. return committer.Commit()
  623. }
  624. // newIssueLabels add labels to an issue. It will check if the labels are valid for the issue
  625. func newIssueLabels(ctx context.Context, issue *Issue, labels []*Label, doer *User) (err error) {
  626. e := db.GetEngine(ctx)
  627. if err = issue.loadRepo(e); err != nil {
  628. return err
  629. }
  630. for _, label := range labels {
  631. // Don't add already present labels and invalid labels
  632. if hasIssueLabel(e, issue.ID, label.ID) ||
  633. (label.RepoID != issue.RepoID && label.OrgID != issue.Repo.OwnerID) {
  634. continue
  635. }
  636. if err = newIssueLabel(ctx, issue, label, doer); err != nil {
  637. return fmt.Errorf("newIssueLabel: %v", err)
  638. }
  639. }
  640. return nil
  641. }
  642. // NewIssueLabels creates a list of issue-label relations.
  643. func NewIssueLabels(issue *Issue, labels []*Label, doer *User) (err error) {
  644. ctx, committer, err := db.TxContext()
  645. if err != nil {
  646. return err
  647. }
  648. defer committer.Close()
  649. if err = newIssueLabels(ctx, issue, labels, doer); err != nil {
  650. return err
  651. }
  652. issue.Labels = nil
  653. if err = issue.loadLabels(db.GetEngine(ctx)); err != nil {
  654. return err
  655. }
  656. return committer.Commit()
  657. }
  658. func deleteIssueLabel(ctx context.Context, issue *Issue, label *Label, doer *User) (err error) {
  659. e := db.GetEngine(ctx)
  660. if count, err := e.Delete(&IssueLabel{
  661. IssueID: issue.ID,
  662. LabelID: label.ID,
  663. }); err != nil {
  664. return err
  665. } else if count == 0 {
  666. return nil
  667. }
  668. if err = issue.loadRepo(e); err != nil {
  669. return
  670. }
  671. opts := &CreateCommentOptions{
  672. Type: CommentTypeLabel,
  673. Doer: doer,
  674. Repo: issue.Repo,
  675. Issue: issue,
  676. Label: label,
  677. }
  678. if _, err = createComment(ctx, opts); err != nil {
  679. return err
  680. }
  681. return updateLabelCols(e, label, "num_issues", "num_closed_issue")
  682. }
  683. // DeleteIssueLabel deletes issue-label relation.
  684. func DeleteIssueLabel(issue *Issue, label *Label, doer *User) (err error) {
  685. ctx, committer, err := db.TxContext()
  686. if err != nil {
  687. return err
  688. }
  689. defer committer.Close()
  690. if err = deleteIssueLabel(ctx, issue, label, doer); err != nil {
  691. return err
  692. }
  693. issue.Labels = nil
  694. if err = issue.loadLabels(db.GetEngine(ctx)); err != nil {
  695. return err
  696. }
  697. return committer.Commit()
  698. }
  699. func deleteLabelsByRepoID(sess db.Engine, repoID int64) error {
  700. deleteCond := builder.Select("id").From("label").Where(builder.Eq{"label.repo_id": repoID})
  701. if _, err := sess.In("label_id", deleteCond).
  702. Delete(&IssueLabel{}); err != nil {
  703. return err
  704. }
  705. _, err := sess.Delete(&Label{RepoID: repoID})
  706. return err
  707. }