api/graph2/queries/chapter.go

package queries

import (
	"context"
	"errors"
	"time"

	"git.aiterp.net/rpdata/api/internal/auth"
	"git.aiterp.net/rpdata/api/models/stories"

	"git.aiterp.net/rpdata/api/graph2/input"
	"git.aiterp.net/rpdata/api/models"
	"git.aiterp.net/rpdata/api/models/chapters"
)

func (r *resolver) Chapter(ctx context.Context, id string) (models.Chapter, error) {
	return chapters.FindID(id)
}

func (r *mutationResolver) AddChapter(ctx context.Context, input input.ChapterAddInput) (models.Chapter, error) {
	story, err := stories.FindID(input.StoryID)
	if err != nil {
		return models.Chapter{}, errors.New("Story not found")
	}

	token := auth.TokenFromContext(ctx)
	if !token.Authenticated() {
		return models.Chapter{}, errors.New("Unauthorized")
	}

	author := token.UserID
	if input.Author != nil && *input.Author != author {
		if !token.Permitted("story.add") {
			return models.Chapter{}, errors.New("False pretender")
		}

		author = *input.Author
	}

	if !story.Open && story.Author != author {
		return models.Chapter{}, errors.New("Story is not open")
	}

	return chapters.Add(story, input.Title, author, input.Source, time.Now(), input.FictionalDate)
}

func (r *mutationResolver) EditChapter(ctx context.Context, input input.ChapterEditInput) (models.Chapter, error) {
	chapter, err := chapters.FindID(input.ID)
	if err != nil {
		return models.Chapter{}, errors.New("Chapter not found")
	}

	token := auth.TokenFromContext(ctx)
	if !token.Authenticated() || !token.PermittedUser(chapter.Author, "member", "story.edit") {
		return models.Chapter{}, errors.New("Unauthorized")
	}

	return chapters.Edit(chapter, input.Title, input.Source, input.FictionalDate)
}

func (r *mutationResolver) RemoveChapter(ctx context.Context, input input.ChapterRemoveInput) (models.Chapter, error) {
	chapter, err := chapters.FindID(input.ID)
	if err != nil {
		return models.Chapter{}, errors.New("Chapter not found")
	}

	token := auth.TokenFromContext(ctx)
	if !token.Authenticated() || !token.PermittedUser(chapter.Author, "member", "story.remove") {
		return models.Chapter{}, errors.New("Unauthorized")
	}

	return chapters.Remove(chapter)
}