api/graph2/resolvers/story.go

package resolvers

import (
	"context"
	"errors"
	"time"

	"git.aiterp.net/rpdata/api/graph2/graphcore"
	"git.aiterp.net/rpdata/api/internal/auth"
	"git.aiterp.net/rpdata/api/models"
	"git.aiterp.net/rpdata/api/models/changekeys"
	"git.aiterp.net/rpdata/api/models/changes"
	"git.aiterp.net/rpdata/api/models/chapters"
	"git.aiterp.net/rpdata/api/models/stories"
)

func (r *queryResolver) Story(ctx context.Context, id string) (*models.Story, error) {
	story, err := stories.FindID(id)
	if err != nil {
		return nil, err
	}

	return &story, nil
}

func (r *queryResolver) Stories(ctx context.Context, filter *stories.Filter) ([]*models.Story, error) {
	if filter != nil {
		if filter.Unlisted != nil && *filter.Unlisted == true {
			token := auth.TokenFromContext(ctx)
			if !token.Authenticated() {
				return nil, errors.New("You are not permitted to view unlisted stories")
			}

			if !token.Permitted("story.unlisted") {
				filter.Author = &token.UserID
			}
		}
	}

	stories, err := stories.List(filter)
	if err != nil {
		return nil, err
	}

	stories2 := make([]*models.Story, len(stories))
	for i := range stories {
		stories2[i] = &stories[i]
	}

	return stories2, nil
}

// Mutations

func (r *mutationResolver) AddStory(ctx context.Context, input graphcore.StoryAddInput) (*models.Story, error) {
	token := auth.TokenFromContext(ctx)
	if token == nil || !token.Permitted("member", "story.add") {
		return nil, errors.New("Permission denied")
	}

	author := token.UserID
	if input.Author != nil && *input.Author != author {
		if !token.Permitted("story.add") {
			return nil, errors.New("You are not permitted to add a story in another author's name")
		}

		author = *input.Author
	}

	fictionalDate := time.Time{}
	if input.FictionalDate != nil {
		fictionalDate = *input.FictionalDate
	}

	listed := input.Listed != nil && *input.Listed
	open := input.Open != nil && *input.Open

	tags := make([]models.Tag, len(input.Tags))
	for i := range input.Tags {
		tags[i] = *input.Tags[i]
	}

	story, err := stories.Add(input.Name, author, input.Category, listed, open, tags, time.Now(), fictionalDate)
	if err != nil {
		return nil, errors.New("Failed to add story: " + err.Error())
	}

	go changes.Submit("Story", "add", token.UserID, story.Listed, changekeys.Listed(story), story)

	return &story, nil
}

func (r *mutationResolver) AddStoryTag(ctx context.Context, input graphcore.StoryTagAddInput) (*models.Story, error) {
	token := auth.TokenFromContext(ctx)

	story, err := stories.FindID(input.ID)
	if err != nil {
		return nil, errors.New("Story not found")
	}

	if story.Open {
		if !token.Permitted("member") {
			return nil, errors.New("You are not permitted to edit this story")
		}
	} else {
		if !token.PermittedUser(story.Author, "member", "story.edit") {
			return nil, errors.New("You are not permitted to edit this story")
		}
	}

	story, err = stories.AddTag(story, *input.Tag)
	if err != nil {
		return nil, errors.New("Failed to add story: " + err.Error())
	}

	go changes.Submit("Story", "tag", token.UserID, story.Listed, changekeys.Listed(story), story, input.Tag)

	return &story, nil
}

func (r *mutationResolver) RemoveStoryTag(ctx context.Context, input graphcore.StoryTagRemoveInput) (*models.Story, error) {
	token := auth.TokenFromContext(ctx)

	story, err := stories.FindID(input.ID)
	if err != nil {
		return nil, errors.New("Story not found")
	}

	if story.Open {
		if !token.Permitted("member") {
			return nil, errors.New("You are not permitted to edit this story")
		}
	} else {
		if !token.PermittedUser(story.Author, "member", "story.edit") {
			return nil, errors.New("You are not permitted to edit this story")
		}
	}

	story, err = stories.RemoveTag(story, *input.Tag)
	if err != nil {
		return nil, errors.New("Failed to add story: " + err.Error())
	}

	go changes.Submit("Story", "untag", token.UserID, story.Listed, changekeys.Listed(story), story, input.Tag)

	return &story, nil
}

func (r *mutationResolver) EditStory(ctx context.Context, input graphcore.StoryEditInput) (*models.Story, error) {
	token := auth.TokenFromContext(ctx)

	story, err := stories.FindID(input.ID)
	if err != nil {
		return nil, errors.New("Story not found")
	}

	if !token.PermittedUser(story.Author, "member", "story.edit") {
		return nil, errors.New("You are not permitted to remove this story")
	}

	if input.ClearFictionalDate != nil && *input.ClearFictionalDate {
		input.FictionalDate = &time.Time{}
	}

	story, err = stories.Edit(story, input.Name, input.Category, input.Listed, input.Open, input.FictionalDate)
	if err != nil {
		return nil, errors.New("Failed to add story: " + err.Error())
	}

	go changes.Submit("Story", "edit", token.UserID, story.Listed, changekeys.Listed(story), story)

	return &story, nil
}

func (r *mutationResolver) RemoveStory(ctx context.Context, input graphcore.StoryRemoveInput) (*models.Story, error) {
	token := auth.TokenFromContext(ctx)

	story, err := stories.FindID(input.ID)
	if err != nil {
		return nil, errors.New("Story not found")
	}

	if !token.PermittedUser(story.Author, "member", "story.remove") {
		return nil, errors.New("You are not permitted to remove this story")
	}

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

	err = chapters.RemoveStory(story)
	if err != nil {
		return nil, errors.New("Failed to remove chapters, but story is removed: " + err.Error())
	}

	go changes.Submit("Story", "remove", token.UserID, story.Listed, changekeys.Listed(story), story)

	return &story, nil
}