graph2: Added Chapter mutations

graph2/graph.go

@@ -22,6 +22,10 @@ func (r *rootResolver) Query() QueryResolver {
 	return &queries.Resolver
 }
 
+func (r *rootResolver) Mutation() MutationResolver {
+	return queries.MutationResolver
+}
+
 func (r *rootResolver) Log() LogResolver {
 	return &types.LogResolver
 }

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

graph2/schema/root.gql

@@ -1,5 +1,6 @@
 schema {
   query: Query
+  mutation: Mutation
 }
 
 type Query {
@@ -57,6 +58,17 @@ type Query {
   token: Token!
 }
 
+type Mutation {
+  # Add a chapter to a story
+  addChapter(input: ChapterAddInput!): Chapter!
+
+  # Edit a chapter
+  editChapter(input: ChapterEditInput!): Chapter!
+
+  # Remove a chapter
+  removeChapter(input: ChapterRemoveInput!): Chapter!
+}
+
 # A Date represents a RFC3339 encoded date with up to millisecond precision.
 scalar Date
 

graph2/schema/types/Chapter.gql

@@ -56,4 +56,10 @@ input ChapterEditInput {
 
   # Set the fictional date for a chapter
   fictionalDate: Date
+}
+
+# Input for removeChapter mutation
+input ChapterRemoveInput {
+  # The chapter to remove
+  id: String!
 }