graph2: Added Chapter mutations

graph2/queries/chapter.go

@@ -2,7 +2,13 @@ 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"
 )
@@ -10,3 +16,58 @@ import (
 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)
+}

graph2/queries/resolver.go

@@ -1,6 +1,10 @@
 package queries
 
 type resolver struct{}
+type mutationResolver struct{}
 
 // Resolver has all the queries
 var Resolver resolver
+
+// MutationResolver brings the mutagens.
+var MutationResolver *mutationResolver