151 lines
4.1 KiB
Go
151 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 == "closed",
|
|
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 := f3_util.ParseInt(string(node.GetID()))
|
|
|
|
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.NodeID(fmt.Sprintf("%d", 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{}
|
|
}
|