some UI work, editPOST handler

storage/mongo/storage.go

@@ -0,0 +1,206 @@
+package mongo
+
+import (
+	"github.com/dalu/wiki/storage"
+	"gopkg.in/mgo.v2"
+	"gopkg.in/mgo.v2/bson"
+)
+
+const (
+	databaseDefault          = "wiki"
+	termCollectionDefault    = "term"
+	revisionCollectionDefault = "revision"
+)
+
+type mongoStorage struct {
+	ms                *mgo.Session
+	database          string
+	termCollection    string
+	revisionCollection string
+}
+
+type Config struct {
+	Database          string
+	TermCollection    string
+	RevisionCollection string
+}
+
+func New(ms *mgo.Session, config Config) mongoStorage {
+	if config.Database == "" {
+		config.Database = databaseDefault
+	}
+	if config.TermCollection == "" {
+		config.TermCollection = termCollectionDefault
+	}
+	if config.RevisionCollection == "" {
+		config.RevisionCollection = revisionCollectionDefault
+	}
+
+	return mongoStorage{
+		ms:                ms.Clone(),
+		database:          config.Database,
+		termCollection:    config.TermCollection,
+		revisionCollection: config.RevisionCollection,
+	}
+}
+
+// --- Term ---
+
+func (s mongoStorage) CreateTerm(term *storage.Term) error {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	return c.Insert(term)
+}
+
+func (s mongoStorage) UpdateTerm(term *storage.Term) error {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	return c.UpdateId(term.ID, term)
+}
+
+func (s mongoStorage) RemoveTerm(term *storage.Term) error {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	return c.RemoveId(term.ID)
+}
+
+func (s mongoStorage) GetTermByName(name string) (*storage.Term, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	term := new(storage.Term)
+	if e := c.Find(bson.M{"name": name}).One(term); e != nil {
+		return nil, e
+	}
+	return term, nil
+}
+
+func (s mongoStorage) GetTermBySlug(slug string) (*storage.Term, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	term := new(storage.Term)
+	if e := c.Find(bson.M{"slug": slug}).One(term); e != nil {
+		return nil, e
+	}
+	return term, nil
+}
+
+func (s mongoStorage) GetTermByID(id string) (*storage.Term, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	term := new(storage.Term)
+	if e := c.FindId(bson.ObjectIdHex(id)).One(term); e != nil {
+		return nil, e
+	}
+	return term, nil
+}
+
+func (s mongoStorage) GetTerms() ([]*storage.Term, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.termCollection)
+	terms := []*storage.Term{}
+	if e := c.Find(nil).All(&terms); e != nil {
+		return terms, e
+	}
+	return terms, nil
+}
+
+// --- Revision ---
+
+func (s mongoStorage) CreateRevision(Revision *storage.Revision) error {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	return c.Insert(Revision)
+}
+
+func (s mongoStorage) RemoveRevision(Revision *storage.Revision) error {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	return c.RemoveId(Revision.ID)
+}
+
+func (s mongoStorage) RemoveRevisionsByTermID(termID string) error {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	_, e := c.RemoveAll(bson.M{"term_id": termID})
+	if e != nil {
+		return e
+	}
+	return nil
+}
+
+func (s mongoStorage) GetRevisionByTermName(termName string) (*storage.Revision, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	term, e := s.GetTermByName(termName)
+	if e != nil {
+		return nil, e
+	}
+	Revision := new(storage.Revision)
+	if e := c.Find(bson.M{"term_id": term.ID.Hex()}).One(Revision); e != nil {
+		return nil, e
+	}
+	return Revision, nil
+}
+
+func (s mongoStorage) GetRevisionByTermSlug(termSlug string) (*storage.Revision, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	term, e := s.GetTermByName(termSlug)
+	if e != nil {
+		return nil, e
+	}
+	Revision := new(storage.Revision)
+	if e := c.Find(bson.M{"term_id": term.ID.Hex()}).One(Revision); e != nil {
+		return nil, e
+	}
+	return Revision, nil
+}
+
+func (s mongoStorage) GetRevisionByID(id string) (*storage.Revision, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	Revision := new(storage.Revision)
+	if e := c.FindId(bson.ObjectIdHex(id)).One(Revision); e != nil {
+		return nil, e
+	}
+	return Revision, nil
+}
+
+func (s mongoStorage) GetRevisionByTermID(termID string) (*storage.Revision, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	term, e := s.GetTermByID(termID)
+	if e != nil {
+		return nil, e
+	}
+	Revision := new(storage.Revision)
+	if e := c.Find(bson.M{"term_id": term.ID.Hex()}).One(Revision); e != nil {
+		return nil, e
+	}
+	return Revision, nil
+}
+
+func (s mongoStorage) GetRevisions() ([]*storage.Revision, error) {
+	ms := s.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(s.database).C(s.revisionCollection)
+	terms := []*storage.Revision{}
+	if e := c.Find(nil).All(&terms); e != nil {
+		return terms, e
+	}
+	return terms, nil
+}