1 files changed, 179 insertions, 0 deletions
diff --git a/services/f3/driver/review.go b/services/f3/driver/review.go
new file mode 100644
index 0000000..a3c074b
--- /dev/null
+++ b/services/f3/driver/review.go
@@ -0,0 +1,179 @@
+// 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 := node.GetID().Int64()
+
+	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.NewNodeID(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{}
+}