api/resolver/chapter.go

package resolver

import (
	"context"
	"time"

	"git.aiterp.net/rpdata/api/internal/session"
	"git.aiterp.net/rpdata/api/model/change"
	"git.aiterp.net/rpdata/api/model/story"
)

// ChapterResolver for the Chapter graphql type
type ChapterResolver struct{ C story.Chapter }

// ChapterArgs is args for chapter query
type ChapterArgs struct {
	ID string
}

// Chapter implements the chapter query
func (r *QueryResolver) Chapter(ctx context.Context, args *ChapterArgs) (*ChapterResolver, error) {
	chapter, err := story.FindChapterID(args.ID)
	if err != nil {
		return nil, err
	}

	return &ChapterResolver{C: chapter}, nil
}

// AddChapterArgs is args for the addChapter mutation
type AddChapterArgs struct {
	Input *AddChapterInput
}

// AddChapterInput is input for the addChapter mutation
type AddChapterInput struct {
	StoryID       string
	Title         string
	Author        *string
	Source        string
	FictionalDate *string
}

// AddChapter implements the addChapter mutation
func (r *MutationResolver) AddChapter(ctx context.Context, args *AddChapterArgs) (*ChapterResolver, error) {
	user := session.FromContext(ctx).User()
	if user == nil || !user.Permitted("member", "chapter.add") {
		return nil, ErrUnauthorized
	}

	story, err := story.FindID(args.Input.StoryID)
	if err != nil {
		return nil, err
	}

	author := user.ID
	if args.Input.Author != nil {
		author = *args.Input.Author

		if user.ID != author && !user.Permitted("chapter.add") {
			return nil, ErrPermissionDenied
		}
	}

	fictionalDate := time.Time{}
	if args.Input.FictionalDate != nil {
		fictionalDate, err = time.Parse(time.RFC3339Nano, *args.Input.FictionalDate)
		if err != nil {
			return nil, err
		}
	}

	chapter, err := story.AddChapter(args.Input.Title, author, args.Input.Source, time.Now(), fictionalDate)
	if err != nil {
		return nil, err
	}

	change.Submit("Chapter", "add", user.ID, chapter.ID, map[string]interface{}{
		"title":         args.Input.Title,
		"author":        author,
		"fictionalDate": fictionalDate,
	})

	return &ChapterResolver{C: chapter}, nil
}

// EditChapterArgs is args for the addChapter mutation
type EditChapterArgs struct {
	Input *EditChapterInput
}

// EditChapterInput is input for the addChapter mutation
type EditChapterInput struct {
	ID            string
	Title         *string
	Source        *string
	FictionalDate *string
}

// EditChapter implements the editChapter mutation
func (r *MutationResolver) EditChapter(ctx context.Context, args *EditChapterArgs) (*ChapterResolver, error) {
	user := session.FromContext(ctx).User()
	if user == nil || !user.Permitted("member", "chapter.edit") {
		return nil, ErrUnauthorized
	}

	chapter, err := story.FindChapterID(args.Input.ID)
	if err != nil {
		return nil, err
	}

	if chapter.Author != user.ID && !user.Permitted("chapter.edit") {
		return nil, ErrPermissionDenied
	}

	var fictionalDate *time.Time
	if args.Input.FictionalDate != nil {
		date, err := time.Parse(time.RFC3339Nano, *args.Input.FictionalDate)
		if err != nil {
			return nil, err
		}

		fictionalDate = &date
	}

	err = chapter.Edit(args.Input.Title, args.Input.Source, fictionalDate)
	if err != nil {
		return nil, err
	}

	change.Submit("Chapter", "edit", user.ID, chapter.ID, map[string]interface{}{
		"title":         args.Input.Title,
		"source":        args.Input.Source,
		"fictionalDate": fictionalDate,
	})

	return &ChapterResolver{C: chapter}, nil
}

// DeleteChapterArgs is args for the addChapter mutation
type DeleteChapterArgs struct {
	ID string
}

// RemoveChapter implements the removeChapter mutation
func (r *MutationResolver) RemoveChapter(ctx context.Context, args *DeleteChapterArgs) (*ChapterResolver, error) {
	user := session.FromContext(ctx).User()
	if user == nil || !user.Permitted("member", "chapter.edit") {
		return nil, ErrUnauthorized
	}

	chapter, err := story.FindChapterID(args.ID)
	if err != nil {
		return nil, err
	}

	err = chapter.Remove()
	if err != nil {
		return nil, err
	}

	change.Submit("Chapter", "remove", user.ID, chapter.ID, nil)

	return &ChapterResolver{C: chapter}, nil
}

// ID resolves Chapter.id
func (r *ChapterResolver) ID() string {
	return r.C.ID
}

// StoryID resolves Chapter.storyId
func (r *ChapterResolver) StoryID() string {
	return r.C.StoryID
}

// Title resolves Chapter.title
func (r *ChapterResolver) Title() string {
	return r.C.Title
}

// Author resolves Chapter.author
func (r *ChapterResolver) Author() string {
	return r.C.Author
}

// Source resolves Chapter.source
func (r *ChapterResolver) Source() string {
	return r.C.Source
}

// CreatedDate resolves Chapter.createdDate
func (r *ChapterResolver) CreatedDate() string {
	return r.C.CreatedDate.Format(time.RFC3339Nano)
}

// FictionalDate resolves Chapter.fictionalDate
func (r *ChapterResolver) FictionalDate() string {
	return r.C.FictionalDate.Format(time.RFC3339Nano)
}

// EditedDate resolves Chapter.editedDate
func (r *ChapterResolver) EditedDate() string {
	return r.C.EditedDate.Format(time.RFC3339Nano)
}