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

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824
  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. sess := db.NewSession(db.DefaultContext)
  256. defer sess.Close()
  257. if err = sess.Begin(); err != nil {
  258. return err
  259. }
  260. if label.BelongsToOrg() && label.OrgID != id {
  261. return nil
  262. }
  263. if label.BelongsToRepo() && label.RepoID != id {
  264. return nil
  265. }
  266. if _, err = sess.ID(labelID).Delete(new(Label)); err != nil {
  267. return err
  268. } else if _, err = sess.
  269. Where("label_id = ?", labelID).
  270. Delete(new(IssueLabel)); err != nil {
  271. return err
  272. }
  273. // delete comments about now deleted label_id
  274. if _, err = sess.Where("label_id = ?", labelID).Cols("label_id").Delete(&Comment{}); err != nil {
  275. return err
  276. }
  277. return sess.Commit()
  278. }
  279. // getLabelByID returns a label by label id
  280. func getLabelByID(e db.Engine, labelID int64) (*Label, error) {
  281. if labelID <= 0 {
  282. return nil, ErrLabelNotExist{labelID}
  283. }
  284. l := &Label{}
  285. has, err := e.ID(labelID).Get(l)
  286. if err != nil {
  287. return nil, err
  288. } else if !has {
  289. return nil, ErrLabelNotExist{l.ID}
  290. }
  291. return l, nil
  292. }
  293. // GetLabelByID returns a label by given ID.
  294. func GetLabelByID(id int64) (*Label, error) {
  295. return getLabelByID(db.GetEngine(db.DefaultContext), id)
  296. }
  297. // GetLabelsByIDs returns a list of labels by IDs
  298. func GetLabelsByIDs(labelIDs []int64) ([]*Label, error) {
  299. labels := make([]*Label, 0, len(labelIDs))
  300. return labels, db.GetEngine(db.DefaultContext).Table("label").
  301. In("id", labelIDs).
  302. Asc("name").
  303. Cols("id", "repo_id", "org_id").
  304. Find(&labels)
  305. }
  306. // __________ .__ __
  307. // \______ \ ____ ______ ____ _____|__|/ |_ ___________ ___.__.
  308. // | _// __ \\____ \ / _ \/ ___/ \ __\/ _ \_ __ < | |
  309. // | | \ ___/| |_> > <_> )___ \| || | ( <_> ) | \/\___ |
  310. // |____|_ /\___ > __/ \____/____ >__||__| \____/|__| / ____|
  311. // \/ \/|__| \/ \/
  312. // getLabelInRepoByName returns a label by Name in given repository.
  313. func getLabelInRepoByName(e db.Engine, repoID int64, labelName string) (*Label, error) {
  314. if len(labelName) == 0 || repoID <= 0 {
  315. return nil, ErrRepoLabelNotExist{0, repoID}
  316. }
  317. l := &Label{
  318. Name: labelName,
  319. RepoID: repoID,
  320. }
  321. has, err := e.Get(l)
  322. if err != nil {
  323. return nil, err
  324. } else if !has {
  325. return nil, ErrRepoLabelNotExist{0, l.RepoID}
  326. }
  327. return l, nil
  328. }
  329. // getLabelInRepoByID returns a label by ID in given repository.
  330. func getLabelInRepoByID(e db.Engine, repoID, labelID int64) (*Label, error) {
  331. if labelID <= 0 || repoID <= 0 {
  332. return nil, ErrRepoLabelNotExist{labelID, repoID}
  333. }
  334. l := &Label{
  335. ID: labelID,
  336. RepoID: repoID,
  337. }
  338. has, err := e.Get(l)
  339. if err != nil {
  340. return nil, err
  341. } else if !has {
  342. return nil, ErrRepoLabelNotExist{l.ID, l.RepoID}
  343. }
  344. return l, nil
  345. }
  346. // GetLabelInRepoByName returns a label by name in given repository.
  347. func GetLabelInRepoByName(repoID int64, labelName string) (*Label, error) {
  348. return getLabelInRepoByName(db.GetEngine(db.DefaultContext), repoID, labelName)
  349. }
  350. // GetLabelIDsInRepoByNames returns a list of labelIDs by names in a given
  351. // repository.
  352. // it silently ignores label names that do not belong to the repository.
  353. func GetLabelIDsInRepoByNames(repoID int64, labelNames []string) ([]int64, error) {
  354. labelIDs := make([]int64, 0, len(labelNames))
  355. return labelIDs, db.GetEngine(db.DefaultContext).Table("label").
  356. Where("repo_id = ?", repoID).
  357. In("name", labelNames).
  358. Asc("name").
  359. Cols("id").
  360. Find(&labelIDs)
  361. }
  362. // BuildLabelNamesIssueIDsCondition returns a builder where get issue ids match label names
  363. func BuildLabelNamesIssueIDsCondition(labelNames []string) *builder.Builder {
  364. return builder.Select("issue_label.issue_id").
  365. From("issue_label").
  366. InnerJoin("label", "label.id = issue_label.label_id").
  367. Where(
  368. builder.In("label.name", labelNames),
  369. ).
  370. GroupBy("issue_label.issue_id")
  371. }
  372. // GetLabelInRepoByID returns a label by ID in given repository.
  373. func GetLabelInRepoByID(repoID, labelID int64) (*Label, error) {
  374. return getLabelInRepoByID(db.GetEngine(db.DefaultContext), repoID, labelID)
  375. }
  376. // GetLabelsInRepoByIDs returns a list of labels by IDs in given repository,
  377. // it silently ignores label IDs that do not belong to the repository.
  378. func GetLabelsInRepoByIDs(repoID int64, labelIDs []int64) ([]*Label, error) {
  379. labels := make([]*Label, 0, len(labelIDs))
  380. return labels, db.GetEngine(db.DefaultContext).
  381. Where("repo_id = ?", repoID).
  382. In("id", labelIDs).
  383. Asc("name").
  384. Find(&labels)
  385. }
  386. func getLabelsByRepoID(e db.Engine, repoID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  387. if repoID <= 0 {
  388. return nil, ErrRepoLabelNotExist{0, repoID}
  389. }
  390. labels := make([]*Label, 0, 10)
  391. sess := e.Where("repo_id = ?", repoID)
  392. switch sortType {
  393. case "reversealphabetically":
  394. sess.Desc("name")
  395. case "leastissues":
  396. sess.Asc("num_issues")
  397. case "mostissues":
  398. sess.Desc("num_issues")
  399. default:
  400. sess.Asc("name")
  401. }
  402. if listOptions.Page != 0 {
  403. sess = db.SetSessionPagination(sess, &listOptions)
  404. }
  405. return labels, sess.Find(&labels)
  406. }
  407. // GetLabelsByRepoID returns all labels that belong to given repository by ID.
  408. func GetLabelsByRepoID(repoID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  409. return getLabelsByRepoID(db.GetEngine(db.DefaultContext), repoID, sortType, listOptions)
  410. }
  411. // CountLabelsByRepoID count number of all labels that belong to given repository by ID.
  412. func CountLabelsByRepoID(repoID int64) (int64, error) {
  413. return db.GetEngine(db.DefaultContext).Where("repo_id = ?", repoID).Count(&Label{})
  414. }
  415. // ________
  416. // \_____ \_______ ____
  417. // / | \_ __ \/ ___\
  418. // / | \ | \/ /_/ >
  419. // \_______ /__| \___ /
  420. // \/ /_____/
  421. // getLabelInOrgByName returns a label by Name in given organization
  422. func getLabelInOrgByName(e db.Engine, orgID int64, labelName string) (*Label, error) {
  423. if len(labelName) == 0 || orgID <= 0 {
  424. return nil, ErrOrgLabelNotExist{0, orgID}
  425. }
  426. l := &Label{
  427. Name: labelName,
  428. OrgID: orgID,
  429. }
  430. has, err := e.Get(l)
  431. if err != nil {
  432. return nil, err
  433. } else if !has {
  434. return nil, ErrOrgLabelNotExist{0, l.OrgID}
  435. }
  436. return l, nil
  437. }
  438. // getLabelInOrgByID returns a label by ID in given organization.
  439. func getLabelInOrgByID(e db.Engine, orgID, labelID int64) (*Label, error) {
  440. if labelID <= 0 || orgID <= 0 {
  441. return nil, ErrOrgLabelNotExist{labelID, orgID}
  442. }
  443. l := &Label{
  444. ID: labelID,
  445. OrgID: orgID,
  446. }
  447. has, err := e.Get(l)
  448. if err != nil {
  449. return nil, err
  450. } else if !has {
  451. return nil, ErrOrgLabelNotExist{l.ID, l.OrgID}
  452. }
  453. return l, nil
  454. }
  455. // GetLabelInOrgByName returns a label by name in given organization.
  456. func GetLabelInOrgByName(orgID int64, labelName string) (*Label, error) {
  457. return getLabelInOrgByName(db.GetEngine(db.DefaultContext), orgID, labelName)
  458. }
  459. // GetLabelIDsInOrgByNames returns a list of labelIDs by names in a given
  460. // organization.
  461. func GetLabelIDsInOrgByNames(orgID int64, labelNames []string) ([]int64, error) {
  462. if orgID <= 0 {
  463. return nil, ErrOrgLabelNotExist{0, orgID}
  464. }
  465. labelIDs := make([]int64, 0, len(labelNames))
  466. return labelIDs, db.GetEngine(db.DefaultContext).Table("label").
  467. Where("org_id = ?", orgID).
  468. In("name", labelNames).
  469. Asc("name").
  470. Cols("id").
  471. Find(&labelIDs)
  472. }
  473. // GetLabelInOrgByID returns a label by ID in given organization.
  474. func GetLabelInOrgByID(orgID, labelID int64) (*Label, error) {
  475. return getLabelInOrgByID(db.GetEngine(db.DefaultContext), orgID, labelID)
  476. }
  477. // GetLabelsInOrgByIDs returns a list of labels by IDs in given organization,
  478. // it silently ignores label IDs that do not belong to the organization.
  479. func GetLabelsInOrgByIDs(orgID int64, labelIDs []int64) ([]*Label, error) {
  480. labels := make([]*Label, 0, len(labelIDs))
  481. return labels, db.GetEngine(db.DefaultContext).
  482. Where("org_id = ?", orgID).
  483. In("id", labelIDs).
  484. Asc("name").
  485. Find(&labels)
  486. }
  487. func getLabelsByOrgID(e db.Engine, orgID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  488. if orgID <= 0 {
  489. return nil, ErrOrgLabelNotExist{0, orgID}
  490. }
  491. labels := make([]*Label, 0, 10)
  492. sess := e.Where("org_id = ?", orgID)
  493. switch sortType {
  494. case "reversealphabetically":
  495. sess.Desc("name")
  496. case "leastissues":
  497. sess.Asc("num_issues")
  498. case "mostissues":
  499. sess.Desc("num_issues")
  500. default:
  501. sess.Asc("name")
  502. }
  503. if listOptions.Page != 0 {
  504. sess = db.SetSessionPagination(sess, &listOptions)
  505. }
  506. return labels, sess.Find(&labels)
  507. }
  508. // GetLabelsByOrgID returns all labels that belong to given organization by ID.
  509. func GetLabelsByOrgID(orgID int64, sortType string, listOptions db.ListOptions) ([]*Label, error) {
  510. return getLabelsByOrgID(db.GetEngine(db.DefaultContext), orgID, sortType, listOptions)
  511. }
  512. // CountLabelsByOrgID count all labels that belong to given organization by ID.
  513. func CountLabelsByOrgID(orgID int64) (int64, error) {
  514. return db.GetEngine(db.DefaultContext).Where("org_id = ?", orgID).Count(&Label{})
  515. }
  516. // .___
  517. // | | ______ ________ __ ____
  518. // | |/ ___// ___/ | \_/ __ \
  519. // | |\___ \ \___ \| | /\ ___/
  520. // |___/____ >____ >____/ \___ |
  521. // \/ \/ \/
  522. func getLabelsByIssueID(e db.Engine, issueID int64) ([]*Label, error) {
  523. var labels []*Label
  524. return labels, e.Where("issue_label.issue_id = ?", issueID).
  525. Join("LEFT", "issue_label", "issue_label.label_id = label.id").
  526. Asc("label.name").
  527. Find(&labels)
  528. }
  529. // GetLabelsByIssueID returns all labels that belong to given issue by ID.
  530. func GetLabelsByIssueID(issueID int64) ([]*Label, error) {
  531. return getLabelsByIssueID(db.GetEngine(db.DefaultContext), issueID)
  532. }
  533. func updateLabelCols(e db.Engine, l *Label, cols ...string) error {
  534. _, err := e.ID(l.ID).
  535. SetExpr("num_issues",
  536. builder.Select("count(*)").From("issue_label").
  537. Where(builder.Eq{"label_id": l.ID}),
  538. ).
  539. SetExpr("num_closed_issues",
  540. builder.Select("count(*)").From("issue_label").
  541. InnerJoin("issue", "issue_label.issue_id = issue.id").
  542. Where(builder.Eq{
  543. "issue_label.label_id": l.ID,
  544. "issue.is_closed": true,
  545. }),
  546. ).
  547. Cols(cols...).Update(l)
  548. return err
  549. }
  550. // .___ .____ ___. .__
  551. // | | ______ ________ __ ____ | | _____ \_ |__ ____ | |
  552. // | |/ ___// ___/ | \_/ __ \| | \__ \ | __ \_/ __ \| |
  553. // | |\___ \ \___ \| | /\ ___/| |___ / __ \| \_\ \ ___/| |__
  554. // |___/____ >____ >____/ \___ >_______ (____ /___ /\___ >____/
  555. // \/ \/ \/ \/ \/ \/ \/
  556. // IssueLabel represents an issue-label relation.
  557. type IssueLabel struct {
  558. ID int64 `xorm:"pk autoincr"`
  559. IssueID int64 `xorm:"UNIQUE(s)"`
  560. LabelID int64 `xorm:"UNIQUE(s)"`
  561. }
  562. func hasIssueLabel(e db.Engine, issueID, labelID int64) bool {
  563. has, _ := e.Where("issue_id = ? AND label_id = ?", issueID, labelID).Get(new(IssueLabel))
  564. return has
  565. }
  566. // HasIssueLabel returns true if issue has been labeled.
  567. func HasIssueLabel(issueID, labelID int64) bool {
  568. return hasIssueLabel(db.GetEngine(db.DefaultContext), issueID, labelID)
  569. }
  570. // newIssueLabel this function creates a new label it does not check if the label is valid for the issue
  571. // YOU MUST CHECK THIS BEFORE THIS FUNCTION
  572. func newIssueLabel(e db.Engine, issue *Issue, label *Label, doer *User) (err error) {
  573. if _, err = e.Insert(&IssueLabel{
  574. IssueID: issue.ID,
  575. LabelID: label.ID,
  576. }); err != nil {
  577. return err
  578. }
  579. if err = issue.loadRepo(e); err != nil {
  580. return
  581. }
  582. opts := &CreateCommentOptions{
  583. Type: CommentTypeLabel,
  584. Doer: doer,
  585. Repo: issue.Repo,
  586. Issue: issue,
  587. Label: label,
  588. Content: "1",
  589. }
  590. if _, err = createComment(e, opts); err != nil {
  591. return err
  592. }
  593. return updateLabelCols(e, label, "num_issues", "num_closed_issue")
  594. }
  595. // NewIssueLabel creates a new issue-label relation.
  596. func NewIssueLabel(issue *Issue, label *Label, doer *User) (err error) {
  597. if HasIssueLabel(issue.ID, label.ID) {
  598. return nil
  599. }
  600. sess := db.NewSession(db.DefaultContext)
  601. defer sess.Close()
  602. if err = sess.Begin(); err != nil {
  603. return err
  604. }
  605. if err = issue.loadRepo(sess); err != nil {
  606. return err
  607. }
  608. // Do NOT add invalid labels
  609. if issue.RepoID != label.RepoID && issue.Repo.OwnerID != label.OrgID {
  610. return nil
  611. }
  612. if err = newIssueLabel(sess, issue, label, doer); err != nil {
  613. return err
  614. }
  615. issue.Labels = nil
  616. if err = issue.loadLabels(sess); err != nil {
  617. return err
  618. }
  619. return sess.Commit()
  620. }
  621. // newIssueLabels add labels to an issue. It will check if the labels are valid for the issue
  622. func newIssueLabels(e db.Engine, issue *Issue, labels []*Label, doer *User) (err error) {
  623. if err = issue.loadRepo(e); err != nil {
  624. return err
  625. }
  626. for _, label := range labels {
  627. // Don't add already present labels and invalid labels
  628. if hasIssueLabel(e, issue.ID, label.ID) ||
  629. (label.RepoID != issue.RepoID && label.OrgID != issue.Repo.OwnerID) {
  630. continue
  631. }
  632. if err = newIssueLabel(e, issue, label, doer); err != nil {
  633. return fmt.Errorf("newIssueLabel: %v", err)
  634. }
  635. }
  636. return nil
  637. }
  638. // NewIssueLabels creates a list of issue-label relations.
  639. func NewIssueLabels(issue *Issue, labels []*Label, doer *User) (err error) {
  640. ctx, committer, err := db.TxContext()
  641. if err != nil {
  642. return err
  643. }
  644. defer committer.Close()
  645. if err = newIssueLabels(db.GetEngine(ctx), issue, labels, doer); err != nil {
  646. return err
  647. }
  648. issue.Labels = nil
  649. if err = issue.loadLabels(db.GetEngine(ctx)); err != nil {
  650. return err
  651. }
  652. return committer.Commit()
  653. }
  654. func deleteIssueLabel(e db.Engine, issue *Issue, label *Label, doer *User) (err error) {
  655. if count, err := e.Delete(&IssueLabel{
  656. IssueID: issue.ID,
  657. LabelID: label.ID,
  658. }); err != nil {
  659. return err
  660. } else if count == 0 {
  661. return nil
  662. }
  663. if err = issue.loadRepo(e); err != nil {
  664. return
  665. }
  666. opts := &CreateCommentOptions{
  667. Type: CommentTypeLabel,
  668. Doer: doer,
  669. Repo: issue.Repo,
  670. Issue: issue,
  671. Label: label,
  672. }
  673. if _, err = createComment(e, opts); err != nil {
  674. return err
  675. }
  676. return updateLabelCols(e, label, "num_issues", "num_closed_issue")
  677. }
  678. // DeleteIssueLabel deletes issue-label relation.
  679. func DeleteIssueLabel(issue *Issue, label *Label, doer *User) (err error) {
  680. sess := db.NewSession(db.DefaultContext)
  681. defer sess.Close()
  682. if err = sess.Begin(); err != nil {
  683. return err
  684. }
  685. if err = deleteIssueLabel(sess, issue, label, doer); err != nil {
  686. return err
  687. }
  688. issue.Labels = nil
  689. if err = issue.loadLabels(sess); err != nil {
  690. return err
  691. }
  692. return sess.Commit()
  693. }
  694. func deleteLabelsByRepoID(sess db.Engine, repoID int64) error {
  695. deleteCond := builder.Select("id").From("label").Where(builder.Eq{"label.repo_id": repoID})
  696. if _, err := sess.In("label_id", deleteCond).
  697. Delete(&IssueLabel{}); err != nil {
  698. return err
  699. }
  700. _, err := sess.Delete(&Label{RepoID: repoID})
  701. return err
  702. }