mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-11-04 00:11:04 +00:00 
			
		
		
		
	* support changing label colors * support changing issue state * use helpers to keep type conversions DRY * drop the x/exp license because it is no longer used The tests are performed by the gof3 compliance suite
		
			
				
	
	
		
			150 lines
		
	
	
	
		
			4.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			150 lines
		
	
	
	
		
			4.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright Earl Warren <contact@earl-warren.org>
 | 
						|
// Copyright Loïc Dachary <loic@dachary.org>
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package driver
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"fmt"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	issues_model "code.gitea.io/gitea/models/issues"
 | 
						|
	"code.gitea.io/gitea/modules/setting"
 | 
						|
	"code.gitea.io/gitea/modules/timeutil"
 | 
						|
 | 
						|
	"code.forgejo.org/f3/gof3/v3/f3"
 | 
						|
	f3_tree "code.forgejo.org/f3/gof3/v3/tree/f3"
 | 
						|
	"code.forgejo.org/f3/gof3/v3/tree/generic"
 | 
						|
	f3_util "code.forgejo.org/f3/gof3/v3/util"
 | 
						|
)
 | 
						|
 | 
						|
var _ f3_tree.ForgeDriverInterface = &milestone{}
 | 
						|
 | 
						|
type milestone struct {
 | 
						|
	common
 | 
						|
 | 
						|
	forgejoMilestone *issues_model.Milestone
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) SetNative(milestone any) {
 | 
						|
	o.forgejoMilestone = milestone.(*issues_model.Milestone)
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) GetNativeID() string {
 | 
						|
	return fmt.Sprintf("%d", o.forgejoMilestone.ID)
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) NewFormat() f3.Interface {
 | 
						|
	node := o.GetNode()
 | 
						|
	return node.GetTree().(f3_tree.TreeInterface).NewFormat(node.GetKind())
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) ToFormat() f3.Interface {
 | 
						|
	if o.forgejoMilestone == nil {
 | 
						|
		return o.NewFormat()
 | 
						|
	}
 | 
						|
	return &f3.Milestone{
 | 
						|
		Common:      f3.NewCommon(fmt.Sprintf("%d", o.forgejoMilestone.ID)),
 | 
						|
		Title:       o.forgejoMilestone.Name,
 | 
						|
		Description: o.forgejoMilestone.Content,
 | 
						|
		Created:     o.forgejoMilestone.CreatedUnix.AsTime(),
 | 
						|
		Updated:     o.forgejoMilestone.UpdatedUnix.AsTimePtr(),
 | 
						|
		Deadline:    o.forgejoMilestone.DeadlineUnix.AsTimePtr(),
 | 
						|
		State:       string(o.forgejoMilestone.State()),
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) FromFormat(content f3.Interface) {
 | 
						|
	milestone := content.(*f3.Milestone)
 | 
						|
 | 
						|
	var deadline timeutil.TimeStamp
 | 
						|
	if milestone.Deadline != nil {
 | 
						|
		deadline = timeutil.TimeStamp(milestone.Deadline.Unix())
 | 
						|
	}
 | 
						|
	if deadline == 0 {
 | 
						|
		deadline = timeutil.TimeStamp(time.Date(9999, 1, 1, 0, 0, 0, 0, setting.DefaultUILocation).Unix())
 | 
						|
	}
 | 
						|
 | 
						|
	var closed timeutil.TimeStamp
 | 
						|
	if milestone.Closed != nil {
 | 
						|
		closed = timeutil.TimeStamp(milestone.Closed.Unix())
 | 
						|
	}
 | 
						|
 | 
						|
	if milestone.Created.IsZero() {
 | 
						|
		if milestone.Updated != nil {
 | 
						|
			milestone.Created = *milestone.Updated
 | 
						|
		} else if milestone.Deadline != nil {
 | 
						|
			milestone.Created = *milestone.Deadline
 | 
						|
		} else {
 | 
						|
			milestone.Created = time.Now()
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if milestone.Updated == nil || milestone.Updated.IsZero() {
 | 
						|
		milestone.Updated = &milestone.Created
 | 
						|
	}
 | 
						|
 | 
						|
	o.forgejoMilestone = &issues_model.Milestone{
 | 
						|
		ID:             f3_util.ParseInt(milestone.GetID()),
 | 
						|
		Name:           milestone.Title,
 | 
						|
		Content:        milestone.Description,
 | 
						|
		IsClosed:       milestone.State == f3.MilestoneStateClosed,
 | 
						|
		CreatedUnix:    timeutil.TimeStamp(milestone.Created.Unix()),
 | 
						|
		UpdatedUnix:    timeutil.TimeStamp(milestone.Updated.Unix()),
 | 
						|
		ClosedDateUnix: closed,
 | 
						|
		DeadlineUnix:   deadline,
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) Get(ctx context.Context) bool {
 | 
						|
	node := o.GetNode()
 | 
						|
	o.Trace("%s", node.GetID())
 | 
						|
 | 
						|
	project := f3_tree.GetProjectID(o.GetNode())
 | 
						|
	id := node.GetID().Int64()
 | 
						|
 | 
						|
	milestone, err := issues_model.GetMilestoneByRepoID(ctx, project, id)
 | 
						|
	if issues_model.IsErrMilestoneNotExist(err) {
 | 
						|
		return false
 | 
						|
	}
 | 
						|
	if err != nil {
 | 
						|
		panic(fmt.Errorf("milestone %v %w", id, err))
 | 
						|
	}
 | 
						|
	o.forgejoMilestone = milestone
 | 
						|
	return true
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) Patch(ctx context.Context) {
 | 
						|
	o.Trace("%d", o.forgejoMilestone.ID)
 | 
						|
	if _, err := db.GetEngine(ctx).ID(o.forgejoMilestone.ID).Cols("name", "description").Update(o.forgejoMilestone); err != nil {
 | 
						|
		panic(fmt.Errorf("UpdateMilestoneCols: %v %v", o.forgejoMilestone, err))
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) Put(ctx context.Context) generic.NodeID {
 | 
						|
	node := o.GetNode()
 | 
						|
	o.Trace("%s", node.GetID())
 | 
						|
 | 
						|
	o.forgejoMilestone.RepoID = f3_tree.GetProjectID(o.GetNode())
 | 
						|
	if err := issues_model.NewMilestone(ctx, o.forgejoMilestone); err != nil {
 | 
						|
		panic(err)
 | 
						|
	}
 | 
						|
	o.Trace("milestone created %d", o.forgejoMilestone.ID)
 | 
						|
	return generic.NewNodeID(o.forgejoMilestone.ID)
 | 
						|
}
 | 
						|
 | 
						|
func (o *milestone) Delete(ctx context.Context) {
 | 
						|
	node := o.GetNode()
 | 
						|
	o.Trace("%s", node.GetID())
 | 
						|
 | 
						|
	project := f3_tree.GetProjectID(o.GetNode())
 | 
						|
 | 
						|
	if err := issues_model.DeleteMilestoneByRepoID(ctx, project, o.forgejoMilestone.ID); err != nil {
 | 
						|
		panic(err)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func newMilestone() generic.NodeDriverInterface {
 | 
						|
	return &milestone{}
 | 
						|
}
 |