forgejo/services/f3/driver/comment.go

// 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"
	issue_service "code.gitea.io/gitea/services/issue"

	"lab.forgefriends.org/friendlyforgeformat/gof3/forges/common"
	"lab.forgefriends.org/friendlyforgeformat/gof3/format"
	"lab.forgefriends.org/friendlyforgeformat/gof3/util"
)

type Comment struct {
	issues_model.Comment
}

func CommentConverter(f *issues_model.Comment) *Comment {
	return &Comment{
		Comment: *f,
	}
}

func (o Comment) GetID() int64 {
	return o.Comment.ID
}

func (o Comment) GetIDString() string {
	return fmt.Sprintf("%d", o.GetID())
}

func (o *Comment) SetID(id int64) {
	o.Comment.ID = id
}

func (o *Comment) SetIDString(id string) {
	o.SetID(util.ParseInt(id))
}

func (o *Comment) IsNil() bool {
	return o.ID == 0
}

func (o *Comment) Equals(other *Comment) bool {
	return o.Comment.Content == other.Comment.Content
}

func (o *Comment) ToFormatInterface() format.Interface {
	return o.ToFormat()
}

func (o *Comment) ToFormat() *format.Comment {
	return &format.Comment{
		Common:     format.NewCommon(o.Comment.ID),
		IssueIndex: o.Comment.IssueID,
		PosterID:   format.NewUserReference(o.Comment.Poster.ID),
		Content:    o.Comment.Content,
		Created:    o.Comment.CreatedUnix.AsTime(),
		Updated:    o.Comment.UpdatedUnix.AsTime(),
	}
}

func (o *Comment) FromFormat(comment *format.Comment) {
	*o = Comment{
		Comment: issues_model.Comment{
			ID:      comment.GetID(),
			IssueID: comment.IssueIndex,
			Issue: &issues_model.Issue{
				ID: comment.IssueIndex,
			},
			PosterID: comment.PosterID.GetID(),
			Poster: &user_model.User{
				ID: comment.PosterID.GetID(),
			},
			Content:     comment.Content,
			CreatedUnix: timeutil.TimeStamp(comment.Created.Unix()),
			UpdatedUnix: timeutil.TimeStamp(comment.Updated.Unix()),
		},
	}
}

type CommentProvider struct {
	BaseProvider
}

func (o *CommentProvider) ToFormat(ctx context.Context, comment *Comment) *format.Comment {
	return comment.ToFormat()
}

func (o *CommentProvider) FromFormat(ctx context.Context, f *format.Comment) *Comment {
	var comment Comment
	comment.FromFormat(f)
	return &comment
}

func (o *CommentProvider) GetObjects(ctx context.Context, user *User, project *Project, commentable common.ContainerObjectInterface, page int) []*Comment {
	var issue *issues_model.Issue
	switch c := commentable.(type) {
	case *PullRequest:
		issue = c.PullRequest.Issue
	case *Issue:
		issue = &c.Issue
	default:
		panic(fmt.Errorf("unexpected type %T", commentable))
	}
	comments, err := issues_model.FindComments(ctx, &issues_model.FindCommentsOptions{
		ListOptions: db.ListOptions{Page: page, PageSize: o.g.perPage},
		RepoID:      project.GetID(),
		IssueID:     issue.ID,
		Type:        issues_model.CommentTypeComment,
	})
	if err != nil {
		panic(fmt.Errorf("error while listing comment: %v", err))
	}

	return util.ConvertMap[*issues_model.Comment, *Comment](comments, CommentConverter)
}

func (o *CommentProvider) ProcessObject(ctx context.Context, user *User, project *Project, commentable common.ContainerObjectInterface, comment *Comment) {
	if err := comment.LoadIssue(ctx); err != nil {
		panic(err)
	}
	if err := comment.LoadPoster(ctx); err != nil {
		panic(err)
	}
}

func (o *CommentProvider) Get(ctx context.Context, user *User, project *Project, commentable common.ContainerObjectInterface, comment *Comment) *Comment {
	id := comment.GetID()
	c, err := issues_model.GetCommentByID(ctx, id)
	if issues_model.IsErrCommentNotExist(err) {
		return &Comment{}
	}
	if err != nil {
		panic(err)
	}

	co := CommentConverter(c)
	o.ProcessObject(ctx, user, project, commentable, co)
	return co
}

func (o *CommentProvider) Put(ctx context.Context, user *User, project *Project, commentable common.ContainerObjectInterface, comment, existing *Comment) *Comment {
	var issue *issues_model.Issue
	switch c := commentable.(type) {
	case *PullRequest:
		issue = c.PullRequest.Issue
	case *Issue:
		issue = &c.Issue
	default:
		panic(fmt.Errorf("unexpected type %T", commentable))
	}

	var result *Comment

	if existing == nil || existing.IsNil() {
		c, err := issue_service.CreateIssueComment(ctx, o.g.GetDoer(), &project.Repository, issue, comment.Content, nil)
		if err != nil {
			panic(err)
		}
		result = CommentConverter(c)
	} else {
		var u issues_model.Comment
		u.ID = existing.GetID()
		cols := make([]string, 0, 10)

		if comment.Content != existing.Content {
			u.Content = comment.Content
			cols = append(cols, "content")
		}

		if len(cols) > 0 {
			if _, err := db.GetEngine(ctx).ID(existing.ID).Cols(cols...).Update(u); err != nil {
				panic(err)
			}
		}

		result = existing
	}

	return o.Get(ctx, user, project, commentable, result)
}

func (o *CommentProvider) Delete(ctx context.Context, user *User, project *Project, commentable common.ContainerObjectInterface, comment *Comment) *Comment {
	c := o.Get(ctx, user, project, commentable, comment)
	if !c.IsNil() {
		err := issues_model.DeleteComment(ctx, &c.Comment)
		if err != nil {
			panic(err)
		}
	}
	return c
}