graph2: Added Chapter mutations

models/chapters/add.go

@@ -0,0 +1,38 @@
+package chapters
+
+import (
+	"time"
+
+	"git.aiterp.net/rpdata/api/models"
+	"github.com/globalsign/mgo/bson"
+)
+
+// Add adds a new chapter.
+func Add(story models.Story, title, author, source string, createdDate time.Time, finctionalDate *time.Time) (models.Chapter, error) {
+	chapter := models.Chapter{
+		ID:          makeChapterID(),
+		StoryID:     story.ID,
+		Title:       title,
+		Author:      author,
+		Source:      source,
+		CreatedDate: createdDate,
+		EditedDate:  createdDate,
+	}
+
+	if finctionalDate != nil {
+		chapter.FictionalDate = *finctionalDate
+	}
+
+	err := collection.Insert(chapter)
+	if err != nil {
+		return models.Chapter{}, err
+	}
+
+	if createdDate.After(story.UpdatedDate) {
+		if err := storyCollection.UpdateId(story.ID, bson.M{"$set": bson.M{"updatedDate": createdDate}}); err == nil {
+			story.UpdatedDate = createdDate
+		}
+	}
+
+	return chapter, nil
+}

models/chapters/db.go

@@ -1,12 +1,17 @@
 package chapters
 
 import (
+	"crypto/rand"
+	"encoding/binary"
+	"strconv"
+
 	"git.aiterp.net/rpdata/api/internal/store"
 	"git.aiterp.net/rpdata/api/models"
 	"github.com/globalsign/mgo"
 )
 
 var collection *mgo.Collection
+var storyCollection *mgo.Collection
 
 func find(query interface{}) (models.Chapter, error) {
 	chapter := models.Chapter{}
@@ -25,9 +30,29 @@ func list(query interface{}) ([]models.Chapter, error) {
 	return chapters, nil
 }
 
+func makeChapterID() string {
+	result := "SC"
+	offset := 0
+	data := make([]byte, 32)
+
+	rand.Read(data)
+	for len(result) < 24 {
+		result += strconv.FormatUint(binary.LittleEndian.Uint64(data[offset:]), 36)
+		offset += 8
+
+		if offset >= 32 {
+			rand.Read(data)
+			offset = 0
+		}
+	}
+
+	return result[:24]
+}
+
 func init() {
 	store.HandleInit(func(db *mgo.Database) {
 		collection = db.C("story.chapters")
+		storyCollection = db.C("story.stories")
 
 		collection.EnsureIndexKey("storyId")
 		collection.EnsureIndexKey("author")

models/chapters/edit.go

@@ -0,0 +1,38 @@
+package chapters
+
+import (
+	"time"
+
+	"git.aiterp.net/rpdata/api/models"
+	"github.com/globalsign/mgo/bson"
+)
+
+// Edit edits a chapter, and updates EditedDate. While many Edit functions cheat if there's nothing to
+// change, this functill will due to EditedDate.
+func Edit(chapter models.Chapter, title, source *string, fictionalDate *time.Time) (models.Chapter, error) {
+	now := time.Now()
+	changes := bson.M{"editedDate": now}
+
+	edited := chapter
+	edited.EditedDate = now
+
+	if title != nil && *title != chapter.Title {
+		changes["title"] = *title
+		edited.Title = *title
+	}
+	if source != nil && *source != chapter.Source {
+		changes["source"] = *source
+		edited.Source = *source
+	}
+	if fictionalDate != nil && !fictionalDate.Equal(chapter.FictionalDate) {
+		changes["fictionalDate"] = *fictionalDate
+		edited.FictionalDate = *fictionalDate
+	}
+
+	err := collection.UpdateId(chapter.ID, bson.M{"$set": changes})
+	if err != nil {
+		return chapter, err
+	}
+
+	return edited, nil
+}

models/chapters/remove.go

@@ -0,0 +1,12 @@
+package chapters
+
+import "git.aiterp.net/rpdata/api/models"
+
+// Remove removes a chapter.
+func Remove(chapter models.Chapter) (models.Chapter, error) {
+	if err := collection.RemoveId(chapter.ID); err != nil {
+		return models.Chapter{}, err
+	}
+
+	return chapter, nil
+}