123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337 |
- // Copyright 2017 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package models
-
- import (
- "code.gitea.io/gitea/modules/setting"
- "code.gitea.io/gitea/modules/util"
- api "code.gitea.io/sdk/gitea"
-
- "github.com/go-xorm/xorm"
- )
-
- // Milestone represents a milestone of repository.
- type Milestone struct {
- ID int64 `xorm:"pk autoincr"`
- RepoID int64 `xorm:"INDEX"`
- Name string
- Content string `xorm:"TEXT"`
- RenderedContent string `xorm:"-"`
- IsClosed bool
- NumIssues int
- NumClosedIssues int
- NumOpenIssues int `xorm:"-"`
- Completeness int // Percentage(1-100).
- IsOverDue bool `xorm:"-"`
-
- DeadlineString string `xorm:"-"`
- DeadlineUnix util.TimeStamp
- ClosedDateUnix util.TimeStamp
- }
-
- // BeforeInsert is invoked from XORM before inserting an object of this type.
- func (m *Milestone) BeforeInsert() {
- m.DeadlineUnix = util.TimeStampNow()
- }
-
- // BeforeUpdate is invoked from XORM before updating this object.
- func (m *Milestone) BeforeUpdate() {
- if m.NumIssues > 0 {
- m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
- } else {
- m.Completeness = 0
- }
- }
-
- // AfterLoad is invoked from XORM after setting the value of a field of
- // this object.
- func (m *Milestone) AfterLoad() {
- m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
- if m.DeadlineUnix.Year() == 9999 {
- return
- }
-
- m.DeadlineString = m.DeadlineUnix.Format("2006-01-02")
- if util.TimeStampNow() >= m.DeadlineUnix {
- m.IsOverDue = true
- }
- }
-
- // State returns string representation of milestone status.
- func (m *Milestone) State() api.StateType {
- if m.IsClosed {
- return api.StateClosed
- }
- return api.StateOpen
- }
-
- // APIFormat returns this Milestone in API format.
- func (m *Milestone) APIFormat() *api.Milestone {
- apiMilestone := &api.Milestone{
- ID: m.ID,
- State: m.State(),
- Title: m.Name,
- Description: m.Content,
- OpenIssues: m.NumOpenIssues,
- ClosedIssues: m.NumClosedIssues,
- }
- if m.IsClosed {
- apiMilestone.Closed = m.ClosedDateUnix.AsTimePtr()
- }
- if m.DeadlineUnix.Year() < 9999 {
- apiMilestone.Deadline = m.DeadlineUnix.AsTimePtr()
- }
- return apiMilestone
- }
-
- // NewMilestone creates new milestone of repository.
- func NewMilestone(m *Milestone) (err error) {
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
-
- if _, err = sess.Insert(m); err != nil {
- return err
- }
-
- if _, err = sess.Exec("UPDATE `repository` SET num_milestones = num_milestones + 1 WHERE id = ?", m.RepoID); err != nil {
- return err
- }
- return sess.Commit()
- }
-
- func getMilestoneByRepoID(e Engine, repoID, id int64) (*Milestone, error) {
- m := &Milestone{
- ID: id,
- RepoID: repoID,
- }
- has, err := e.Get(m)
- if err != nil {
- return nil, err
- } else if !has {
- return nil, ErrMilestoneNotExist{id, repoID}
- }
- return m, nil
- }
-
- // GetMilestoneByRepoID returns the milestone in a repository.
- func GetMilestoneByRepoID(repoID, id int64) (*Milestone, error) {
- return getMilestoneByRepoID(x, repoID, id)
- }
-
- // GetMilestonesByRepoID returns all milestones of a repository.
- func GetMilestonesByRepoID(repoID int64) ([]*Milestone, error) {
- miles := make([]*Milestone, 0, 10)
- return miles, x.Where("repo_id = ?", repoID).Find(&miles)
- }
-
- // GetMilestones returns a list of milestones of given repository and status.
- func GetMilestones(repoID int64, page int, isClosed bool, sortType string) ([]*Milestone, error) {
- miles := make([]*Milestone, 0, setting.UI.IssuePagingNum)
- sess := x.Where("repo_id = ? AND is_closed = ?", repoID, isClosed)
- if page > 0 {
- sess = sess.Limit(setting.UI.IssuePagingNum, (page-1)*setting.UI.IssuePagingNum)
- }
-
- switch sortType {
- case "furthestduedate":
- sess.Desc("deadline_unix")
- case "leastcomplete":
- sess.Asc("completeness")
- case "mostcomplete":
- sess.Desc("completeness")
- case "leastissues":
- sess.Asc("num_issues")
- case "mostissues":
- sess.Desc("num_issues")
- default:
- sess.Asc("deadline_unix")
- }
-
- return miles, sess.Find(&miles)
- }
-
- func updateMilestone(e Engine, m *Milestone) error {
- _, err := e.ID(m.ID).AllCols().Update(m)
- return err
- }
-
- // UpdateMilestone updates information of given milestone.
- func UpdateMilestone(m *Milestone) error {
- return updateMilestone(x, m)
- }
-
- func countRepoMilestones(e Engine, repoID int64) int64 {
- count, _ := e.
- Where("repo_id=?", repoID).
- Count(new(Milestone))
- return count
- }
-
- func countRepoClosedMilestones(e Engine, repoID int64) int64 {
- closed, _ := e.
- Where("repo_id=? AND is_closed=?", repoID, true).
- Count(new(Milestone))
- return closed
- }
-
- // CountRepoClosedMilestones returns number of closed milestones in given repository.
- func CountRepoClosedMilestones(repoID int64) int64 {
- return countRepoClosedMilestones(x, repoID)
- }
-
- // MilestoneStats returns number of open and closed milestones of given repository.
- func MilestoneStats(repoID int64) (open int64, closed int64) {
- open, _ = x.
- Where("repo_id=? AND is_closed=?", repoID, false).
- Count(new(Milestone))
- return open, CountRepoClosedMilestones(repoID)
- }
-
- // ChangeMilestoneStatus changes the milestone open/closed status.
- func ChangeMilestoneStatus(m *Milestone, isClosed bool) (err error) {
- repo, err := GetRepositoryByID(m.RepoID)
- if err != nil {
- return err
- }
-
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
-
- m.IsClosed = isClosed
- if err = updateMilestone(sess, m); err != nil {
- return err
- }
-
- repo.NumMilestones = int(countRepoMilestones(sess, repo.ID))
- repo.NumClosedMilestones = int(countRepoClosedMilestones(sess, repo.ID))
- if _, err = sess.ID(repo.ID).Cols("num_milestones, num_closed_milestones").Update(repo); err != nil {
- return err
- }
- return sess.Commit()
- }
-
- func changeMilestoneIssueStats(e *xorm.Session, issue *Issue) error {
- if issue.MilestoneID == 0 {
- return nil
- }
-
- m, err := getMilestoneByRepoID(e, issue.RepoID, issue.MilestoneID)
- if err != nil {
- return err
- }
-
- if issue.IsClosed {
- m.NumOpenIssues--
- m.NumClosedIssues++
- } else {
- m.NumOpenIssues++
- m.NumClosedIssues--
- }
-
- return updateMilestone(e, m)
- }
-
- func changeMilestoneAssign(e *xorm.Session, doer *User, issue *Issue, oldMilestoneID int64) error {
- if oldMilestoneID > 0 {
- m, err := getMilestoneByRepoID(e, issue.RepoID, oldMilestoneID)
- if err != nil {
- return err
- }
-
- m.NumIssues--
- if issue.IsClosed {
- m.NumClosedIssues--
- }
-
- if err = updateMilestone(e, m); err != nil {
- return err
- }
- }
-
- if issue.MilestoneID > 0 {
- m, err := getMilestoneByRepoID(e, issue.RepoID, issue.MilestoneID)
- if err != nil {
- return err
- }
-
- m.NumIssues++
- if issue.IsClosed {
- m.NumClosedIssues++
- }
-
- if err = updateMilestone(e, m); err != nil {
- return err
- }
- }
-
- if err := issue.loadRepo(e); err != nil {
- return err
- }
-
- if oldMilestoneID > 0 || issue.MilestoneID > 0 {
- if _, err := createMilestoneComment(e, doer, issue.Repo, issue, oldMilestoneID, issue.MilestoneID); err != nil {
- return err
- }
- }
-
- return updateIssue(e, issue)
- }
-
- // ChangeMilestoneAssign changes assignment of milestone for issue.
- func ChangeMilestoneAssign(issue *Issue, doer *User, oldMilestoneID int64) (err error) {
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
-
- if err = changeMilestoneAssign(sess, doer, issue, oldMilestoneID); err != nil {
- return err
- }
- return sess.Commit()
- }
-
- // DeleteMilestoneByRepoID deletes a milestone from a repository.
- func DeleteMilestoneByRepoID(repoID, id int64) error {
- m, err := GetMilestoneByRepoID(repoID, id)
- if err != nil {
- if IsErrMilestoneNotExist(err) {
- return nil
- }
- return err
- }
-
- repo, err := GetRepositoryByID(m.RepoID)
- if err != nil {
- return err
- }
-
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
-
- if _, err = sess.ID(m.ID).Delete(new(Milestone)); err != nil {
- return err
- }
-
- repo.NumMilestones = int(countRepoMilestones(sess, repo.ID))
- repo.NumClosedMilestones = int(countRepoClosedMilestones(sess, repo.ID))
- if _, err = sess.ID(repo.ID).Cols("num_milestones, num_closed_milestones").Update(repo); err != nil {
- return err
- }
-
- if _, err = sess.Exec("UPDATE `issue` SET milestone_id = 0 WHERE milestone_id = ?", m.ID); err != nil {
- return err
- }
- return sess.Commit()
- }
|