package chapters

import (
	"git.aiterp.net/rpdata/api/models"
	"github.com/globalsign/mgo/bson"
)

// Remove removes a chapter.
func Remove(chapter models.Chapter) (*models.Chapter, error) {
	if err := collection.RemoveId(chapter.ID); err != nil {
		return nil, err
	}

	return chapter, nil
}

// RemoveStory removes all chapters belonging to a story
func RemoveStory(story models.Story) error {
	if _, err := collection.RemoveAll(bson.M{"storyId": story.ID}); err != nil {
		return err
	}

	return nil
}