123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402 |
- // Copyright 2017 The Gogs 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 db
- import (
- "fmt"
- "time"
- log "unknwon.dev/clog/v2"
- "xorm.io/xorm"
- api "github.com/gogs/go-gogs-client"
- "gogs.io/gogs/internal/conf"
- )
- // Milestone represents a milestone of repository.
- type Milestone struct {
- ID int64
- RepoID int64 `xorm:"INDEX"`
- Name string
- Content string `xorm:"TEXT"`
- RenderedContent string `xorm:"-" json:"-"`
- IsClosed bool
- NumIssues int
- NumClosedIssues int
- NumOpenIssues int `xorm:"-" json:"-"`
- Completeness int // Percentage(1-100).
- IsOverDue bool `xorm:"-" json:"-"`
- DeadlineString string `xorm:"-" json:"-"`
- Deadline time.Time `xorm:"-" json:"-"`
- DeadlineUnix int64
- ClosedDate time.Time `xorm:"-" json:"-"`
- ClosedDateUnix int64
- }
- func (m *Milestone) BeforeInsert() {
- m.DeadlineUnix = m.Deadline.Unix()
- }
- func (m *Milestone) BeforeUpdate() {
- if m.NumIssues > 0 {
- m.Completeness = m.NumClosedIssues * 100 / m.NumIssues
- } else {
- m.Completeness = 0
- }
- m.DeadlineUnix = m.Deadline.Unix()
- m.ClosedDateUnix = m.ClosedDate.Unix()
- }
- func (m *Milestone) AfterSet(colName string, _ xorm.Cell) {
- switch colName {
- case "num_closed_issues":
- m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
- case "deadline_unix":
- m.Deadline = time.Unix(m.DeadlineUnix, 0).Local()
- if m.Deadline.Year() == 9999 {
- return
- }
- m.DeadlineString = m.Deadline.Format("2006-01-02")
- if time.Now().Local().After(m.Deadline) {
- m.IsOverDue = true
- }
- case "closed_date_unix":
- m.ClosedDate = time.Unix(m.ClosedDateUnix, 0).Local()
- }
- }
- // State returns string representation of milestone status.
- func (m *Milestone) State() api.StateType {
- if m.IsClosed {
- return api.STATE_CLOSED
- }
- return api.STATE_OPEN
- }
- func (m *Milestone) ChangeStatus(isClosed bool) error {
- return ChangeMilestoneStatus(m, isClosed)
- }
- 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.ClosedDate
- }
- if m.Deadline.Year() < 9999 {
- apiMilestone.Deadline = &m.Deadline
- }
- return apiMilestone
- }
- func (m *Milestone) CountIssues(isClosed, includePulls bool) int64 {
- sess := x.Where("milestone_id = ?", m.ID).And("is_closed = ?", isClosed)
- if !includePulls {
- sess.And("is_pull = ?", false)
- }
- count, _ := sess.Count(new(Issue))
- return count
- }
- // 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
- }
- // GetWebhookByRepoID 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) ([]*Milestone, error) {
- miles := make([]*Milestone, 0, conf.UI.IssuePagingNum)
- sess := x.Where("repo_id = ? AND is_closed = ?", repoID, isClosed)
- if page > 0 {
- sess = sess.Limit(conf.UI.IssuePagingNum, (page-1)*conf.UI.IssuePagingNum)
- }
- 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
- }
- // CountRepoMilestones returns number of milestones in given repository.
- func CountRepoMilestones(repoID int64) int64 {
- return countRepoMilestones(x, repoID)
- }
- 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.
- // If milestone passes with changed values, those values will be
- // updated to database as well.
- 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).AllCols().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)
- }
- // ChangeMilestoneIssueStats updates the open/closed issues counter and progress
- // for the milestone associated with the given issue.
- func ChangeMilestoneIssueStats(issue *Issue) (err error) {
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
- if err = changeMilestoneIssueStats(sess, issue); err != nil {
- return err
- }
- return sess.Commit()
- }
- func changeMilestoneAssign(e *xorm.Session, 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
- } else if _, err = e.Exec("UPDATE `issue_user` SET milestone_id = 0 WHERE issue_id = ?", issue.ID); err != nil {
- return err
- }
- issue.Milestone = nil
- }
- 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
- } else if _, err = e.Exec("UPDATE `issue_user` SET milestone_id = ? WHERE issue_id = ?", m.ID, issue.ID); err != nil {
- return err
- }
- issue.Milestone = m
- }
- return updateIssue(e, issue)
- }
- // ChangeMilestoneAssign changes assignment of milestone for issue.
- func ChangeMilestoneAssign(doer *User, issue *Issue, oldMilestoneID int64) (err error) {
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
- if err = changeMilestoneAssign(sess, issue, oldMilestoneID); err != nil {
- return err
- }
- if err = sess.Commit(); err != nil {
- return fmt.Errorf("Commit: %v", err)
- }
- var hookAction api.HookIssueAction
- if issue.MilestoneID > 0 {
- hookAction = api.HOOK_ISSUE_MILESTONED
- } else {
- hookAction = api.HOOK_ISSUE_DEMILESTONED
- }
- if issue.IsPull {
- err = issue.PullRequest.LoadIssue()
- if err != nil {
- log.Error("LoadIssue: %v", err)
- return
- }
- err = PrepareWebhooks(issue.Repo, HOOK_EVENT_PULL_REQUEST, &api.PullRequestPayload{
- Action: hookAction,
- Index: issue.Index,
- PullRequest: issue.PullRequest.APIFormat(),
- Repository: issue.Repo.APIFormat(nil),
- Sender: doer.APIFormat(),
- })
- } else {
- err = PrepareWebhooks(issue.Repo, HOOK_EVENT_ISSUES, &api.IssuesPayload{
- Action: hookAction,
- Index: issue.Index,
- Issue: issue.APIFormat(),
- Repository: issue.Repo.APIFormat(nil),
- Sender: doer.APIFormat(),
- })
- }
- if err != nil {
- log.Error("PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)
- }
- return nil
- }
- // DeleteMilestoneOfRepoByID deletes a milestone from a repository.
- func DeleteMilestoneOfRepoByID(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).AllCols().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
- } else if _, err = sess.Exec("UPDATE `issue_user` SET milestone_id = 0 WHERE milestone_id = ?", m.ID); err != nil {
- return err
- }
- return sess.Commit()
- }
|