forgejo/services/f3/driver/review.go

180 lines
4.8 KiB
Go
Raw Normal View History

// 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.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 = &review{}
type review struct {
common
forgejoReview *issues_model.Review
}
func (o *review) SetNative(review any) {
o.forgejoReview = review.(*issues_model.Review)
}
func (o *review) GetNativeID() string {
return fmt.Sprintf("%d", o.forgejoReview.ID)
}
func (o *review) NewFormat() f3.Interface {
node := o.GetNode()
return node.GetTree().(f3_tree.TreeInterface).NewFormat(node.GetKind())
}
func (o *review) ToFormat() f3.Interface {
if o.forgejoReview == nil {
return o.NewFormat()
}
review := &f3.Review{
Common: f3.NewCommon(o.GetNativeID()),
ReviewerID: f3_tree.NewUserReference(o.forgejoReview.ReviewerID),
Official: o.forgejoReview.Official,
CommitID: o.forgejoReview.CommitID,
Content: o.forgejoReview.Content,
CreatedAt: o.forgejoReview.CreatedUnix.AsTime(),
}
switch o.forgejoReview.Type {
case issues_model.ReviewTypeApprove:
review.State = f3.ReviewStateApproved
case issues_model.ReviewTypeReject:
review.State = f3.ReviewStateChangesRequested
case issues_model.ReviewTypeComment:
review.State = f3.ReviewStateCommented
case issues_model.ReviewTypePending:
review.State = f3.ReviewStatePending
case issues_model.ReviewTypeRequest:
review.State = f3.ReviewStateRequestReview
default:
review.State = f3.ReviewStateUnknown
}
if o.forgejoReview.Reviewer != nil {
review.ReviewerID = f3_tree.NewUserReference(o.forgejoReview.Reviewer.ID)
}
return review
}
func (o *review) FromFormat(content f3.Interface) {
review := content.(*f3.Review)
o.forgejoReview = &issues_model.Review{
ID: f3_util.ParseInt(review.GetID()),
ReviewerID: review.ReviewerID.GetIDAsInt(),
Reviewer: &user_model.User{
ID: review.ReviewerID.GetIDAsInt(),
},
Official: review.Official,
CommitID: review.CommitID,
Content: review.Content,
CreatedUnix: timeutil.TimeStamp(review.CreatedAt.Unix()),
}
switch review.State {
case f3.ReviewStateApproved:
o.forgejoReview.Type = issues_model.ReviewTypeApprove
case f3.ReviewStateChangesRequested:
o.forgejoReview.Type = issues_model.ReviewTypeReject
case f3.ReviewStateCommented:
o.forgejoReview.Type = issues_model.ReviewTypeComment
case f3.ReviewStatePending:
o.forgejoReview.Type = issues_model.ReviewTypePending
case f3.ReviewStateRequestReview:
o.forgejoReview.Type = issues_model.ReviewTypeRequest
default:
o.forgejoReview.Type = issues_model.ReviewTypeUnknown
}
}
func (o *review) Get(ctx context.Context) bool {
node := o.GetNode()
o.Trace("%s", node.GetID())
id := f3_util.ParseInt(string(node.GetID()))
review, err := issues_model.GetReviewByID(ctx, id)
if issues_model.IsErrReviewNotExist(err) {
return false
}
if err != nil {
panic(fmt.Errorf("review %v %w", id, err))
}
if err := review.LoadReviewer(ctx); err != nil {
panic(fmt.Errorf("LoadReviewer %v %w", *review, err))
}
o.forgejoReview = review
return true
}
func (o *review) Patch(ctx context.Context) {
o.Trace("%d", o.forgejoReview.ID)
if _, err := db.GetEngine(ctx).ID(o.forgejoReview.ID).Cols("content").Update(o.forgejoReview); err != nil {
panic(fmt.Errorf("UpdateReviewCols: %v %v", o.forgejoReview, err))
}
}
func (o *review) Put(ctx context.Context) generic.NodeID {
node := o.GetNode()
o.Trace("%s", node.GetID())
project := f3_tree.GetProjectID(o.GetNode())
pullRequest := f3_tree.GetPullRequestID(o.GetNode())
issue, err := issues_model.GetIssueByIndex(ctx, project, pullRequest)
if err != nil {
panic(fmt.Errorf("GetIssueByIndex %v", err))
}
o.forgejoReview.IssueID = issue.ID
sess := db.GetEngine(ctx)
if _, err := sess.NoAutoTime().Insert(o.forgejoReview); err != nil {
panic(err)
}
o.Trace("review created %d", o.forgejoReview.ID)
return generic.NodeID(fmt.Sprintf("%d", o.forgejoReview.ID))
}
func (o *review) Delete(ctx context.Context) {
node := o.GetNode()
o.Trace("%s", node.GetID())
project := f3_tree.GetProjectID(o.GetNode())
pullRequest := f3_tree.GetPullRequestID(o.GetNode())
issue, err := issues_model.GetIssueByIndex(ctx, project, pullRequest)
if err != nil {
panic(fmt.Errorf("GetIssueByIndex %v", err))
}
o.forgejoReview.IssueID = issue.ID
if err := issues_model.DeleteReview(ctx, o.forgejoReview); err != nil {
panic(err)
}
}
func newReview() generic.NodeDriverInterface {
return &review{}
}