b3eaf2249d
Add new methods GetIDString() & SetIDString() & ToFormatInterface()
Change the prototype of the fixture function
(cherry picked from commit d7b263ff8b)
162 lines
4.3 KiB
Go
162 lines
4.3 KiB
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.ID == other.Comment.ID
|
|
}
|
|
|
|
func (o *Comment) ToFormatInterface() format.Interface {
|
|
return o.ToFormat()
|
|
}
|
|
|
|
func (o *Comment) ToFormat() *format.Comment {
|
|
return &format.Comment{
|
|
IssueIndex: o.Comment.IssueID,
|
|
Common: format.NewCommon(o.Comment.ID),
|
|
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(),
|
|
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 {
|
|
g *Forgejo
|
|
}
|
|
|
|
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 {
|
|
comments, err := issues_model.FindComments(ctx, &issues_model.FindCommentsOptions{
|
|
ListOptions: db.ListOptions{Page: page, PageSize: o.g.perPage},
|
|
RepoID: project.GetID(),
|
|
IssueID: commentable.GetID(),
|
|
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 *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))
|
|
}
|
|
c, err := issue_service.CreateIssueComment(ctx, o.g.GetDoer(), &project.Repository, issue, comment.Content, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return o.Get(ctx, user, project, commentable, CommentConverter(c))
|
|
}
|
|
|
|
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
|
|
}
|