api/model/story/chapter.go

package story

import (
	"crypto/rand"
	"encoding/binary"
	"strconv"
	"time"

	"git.aiterp.net/rpdata/api/internal/store"
	"github.com/globalsign/mgo"
	"github.com/globalsign/mgo/bson"
)

var chapterCollection *mgo.Collection

// A Chapter is a part of a story.
type Chapter struct {
	ID            string    `bson:"_id"`
	StoryID       string    `bson:"storyId"`
	Title         string    `bson:"title"`
	Author        string    `bson:"author"`
	Source        string    `bson:"source"`
	CreatedDate   time.Time `bson:"createdDate"`
	FictionalDate time.Time `bson:"fictionalDate,omitempty"`
	EditedDate    time.Time `bson:"editedDate"`
}

// 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 (chapter *Chapter) Edit(title, source *string, fictionalDate *time.Time) error {
	now := time.Now()
	changes := bson.M{"editedDate": now}
	changed := *chapter
	changed.EditedDate = now

	if title != nil && *title != chapter.Title {
		changes["title"] = *title
		changed.Title = *title
	}
	if source != nil && *source != chapter.Source {
		changes["source"] = *source
		changed.Source = *source
	}
	if fictionalDate != nil && !fictionalDate.Equal(chapter.FictionalDate) {
		changes["fictionalDate"] = *fictionalDate
		changed.FictionalDate = *fictionalDate
	}

	err := chapterCollection.UpdateId(chapter.ID, bson.M{"$set": changes})
	if err != nil {
		return err
	}

	*chapter = changed

	return nil
}

// Remove removes a chapter.
func (chapter *Chapter) Remove() error {
	return chapterCollection.RemoveId(chapter.ID)
}

// FindChapterID finds a chapter by its own ID
func FindChapterID(id string) (Chapter, error) {
	chapter := Chapter{}
	err := chapterCollection.FindId(id).One(&chapter)

	return chapter, err
}

// ListChapterStoryID lists all chapters for the story ID
func ListChapterStoryID(storyID string) ([]Chapter, error) {
	chapters := make([]Chapter, 0, 8)
	err := chapterCollection.Find(bson.M{"storyId": storyID}).Sort("createdDate").All(&chapters)
	if err != nil {
		return nil, err
	}

	return chapters, nil
}

// makeChapterID makes a random chapter ID that's 24 characters long
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) {
		chapterCollection = db.C("story.chapters")

		chapterCollection.EnsureIndexKey("storyId")
		chapterCollection.EnsureIndexKey("author")
		chapterCollection.EnsureIndexKey("createdDate")
	})
}