api/graph2/types/chapter.go

package types

import (
	"context"
	"errors"
	"time"

	"git.aiterp.net/rpdata/api/models"
	"git.aiterp.net/rpdata/api/models/comments"
)

type chapterResolver struct{}

func (r *chapterResolver) FictionalDate(ctx context.Context, chapter *models.Chapter) (*time.Time, error) {
	if chapter.FictionalDate.IsZero() {
		return nil, nil
	}

	return &chapter.FictionalDate, nil
}

func (r *chapterResolver) Comments(ctx context.Context, chapter *models.Chapter, limit *int) ([]*models.Comment, error) {
	limitValue := 0
	if limit != nil {
		if *limit < 0 {
			return nil, errors.New("Limit cannot be negative")
		}

		limitValue = *limit
	}

	if !chapter.CommentMode.IsEnabled() {
		return nil, nil
	}

	comments, err := comments.ListChapterID(chapter.ID, limitValue)
	if err != nil {
		return nil, err
	}

	comments2 := make([]*models.Comment, len(comments))
	for i := range comments {
		comments2[i] = &comments[i]
	}

	return comments2, nil
}

// ChapterResolver is a resolver
var ChapterResolver chapterResolver