mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-11-03 16:01:11 +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
		
			
				
	
	
		
			133 lines
		
	
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			133 lines
		
	
	
	
		
			3.4 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"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	issues_model "code.gitea.io/gitea/models/issues"
 | 
						|
	user_model "code.gitea.io/gitea/models/user"
 | 
						|
 | 
						|
	"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 = &reaction{}
 | 
						|
 | 
						|
type reaction struct {
 | 
						|
	common
 | 
						|
 | 
						|
	forgejoReaction *issues_model.Reaction
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) SetNative(reaction any) {
 | 
						|
	o.forgejoReaction = reaction.(*issues_model.Reaction)
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) GetNativeID() string {
 | 
						|
	return fmt.Sprintf("%d", o.forgejoReaction.ID)
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) NewFormat() f3.Interface {
 | 
						|
	node := o.GetNode()
 | 
						|
	return node.GetTree().(f3_tree.TreeInterface).NewFormat(node.GetKind())
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) ToFormat() f3.Interface {
 | 
						|
	if o.forgejoReaction == nil {
 | 
						|
		return o.NewFormat()
 | 
						|
	}
 | 
						|
	return &f3.Reaction{
 | 
						|
		Common:  f3.NewCommon(fmt.Sprintf("%d", o.forgejoReaction.ID)),
 | 
						|
		UserID:  f3_tree.NewUserReference(o.forgejoReaction.User.ID),
 | 
						|
		Content: o.forgejoReaction.Type,
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) FromFormat(content f3.Interface) {
 | 
						|
	reaction := content.(*f3.Reaction)
 | 
						|
 | 
						|
	o.forgejoReaction = &issues_model.Reaction{
 | 
						|
		ID:     f3_util.ParseInt(reaction.GetID()),
 | 
						|
		UserID: reaction.UserID.GetIDAsInt(),
 | 
						|
		User: &user_model.User{
 | 
						|
			ID: reaction.UserID.GetIDAsInt(),
 | 
						|
		},
 | 
						|
		Type: reaction.Content,
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) Get(ctx context.Context) bool {
 | 
						|
	node := o.GetNode()
 | 
						|
	o.Trace("%s", node.GetID())
 | 
						|
 | 
						|
	id := node.GetID().Int64()
 | 
						|
 | 
						|
	if has, err := db.GetEngine(ctx).Where("ID = ?", id).Get(o.forgejoReaction); err != nil {
 | 
						|
		panic(fmt.Errorf("reaction %v %w", id, err))
 | 
						|
	} else if !has {
 | 
						|
		return false
 | 
						|
	}
 | 
						|
	if _, err := o.forgejoReaction.LoadUser(ctx); err != nil {
 | 
						|
		panic(fmt.Errorf("LoadUser %v %w", *o.forgejoReaction, err))
 | 
						|
	}
 | 
						|
	return true
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) Patch(ctx context.Context) {
 | 
						|
	o.Trace("%d", o.forgejoReaction.ID)
 | 
						|
	if _, err := db.GetEngine(ctx).ID(o.forgejoReaction.ID).Cols("type").Update(o.forgejoReaction); err != nil {
 | 
						|
		panic(fmt.Errorf("UpdateReactionCols: %v %v", o.forgejoReaction, err))
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) Put(ctx context.Context) generic.NodeID {
 | 
						|
	o.Error("%v", o.forgejoReaction.User)
 | 
						|
 | 
						|
	sess := db.GetEngine(ctx)
 | 
						|
 | 
						|
	reactionable := f3_tree.GetReactionable(o.GetNode())
 | 
						|
	reactionableID := f3_tree.GetReactionableID(o.GetNode())
 | 
						|
 | 
						|
	switch reactionable.GetKind() {
 | 
						|
	case f3_tree.KindIssue, f3_tree.KindPullRequest:
 | 
						|
		project := f3_tree.GetProjectID(o.GetNode())
 | 
						|
		issue, err := issues_model.GetIssueByIndex(ctx, project, reactionableID)
 | 
						|
		if err != nil {
 | 
						|
			panic(fmt.Errorf("GetIssueByIndex %v %w", reactionableID, err))
 | 
						|
		}
 | 
						|
		o.forgejoReaction.IssueID = issue.ID
 | 
						|
	case f3_tree.KindComment:
 | 
						|
		o.forgejoReaction.CommentID = reactionableID
 | 
						|
	default:
 | 
						|
		panic(fmt.Errorf("unexpected type %v", reactionable.GetKind()))
 | 
						|
	}
 | 
						|
 | 
						|
	o.Error("%v", o.forgejoReaction)
 | 
						|
 | 
						|
	if _, err := sess.Insert(o.forgejoReaction); err != nil {
 | 
						|
		panic(err)
 | 
						|
	}
 | 
						|
	o.Trace("reaction created %d", o.forgejoReaction.ID)
 | 
						|
	return generic.NewNodeID(o.forgejoReaction.ID)
 | 
						|
}
 | 
						|
 | 
						|
func (o *reaction) Delete(ctx context.Context) {
 | 
						|
	node := o.GetNode()
 | 
						|
	o.Trace("%s", node.GetID())
 | 
						|
 | 
						|
	sess := db.GetEngine(ctx)
 | 
						|
	if _, err := sess.Delete(o.forgejoReaction); err != nil {
 | 
						|
		panic(err)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func newReaction() generic.NodeDriverInterface {
 | 
						|
	return &reaction{}
 | 
						|
}
 |