database

type BaseDatabase interface {
	Connect(connection *Connection) error
	Disconnect()
}

type Connection struct {
	Host         string `json:"host"`
	DatabaseName string `json:"databaseName"`
}

type DatabaseUploadResponse struct {
	BytesWritten int           `json:"bytesWritten"`
	FileID       bson.ObjectId `json:"fileID"`
	FileName     string        `json:"fileName"`
	Height       int           `json:"height"`
	Width        int           `json:"width"`
}

type DatabaseUploader interface {
	Upload(reader io.Reader, name, path string) (*DatabaseUploadResponse, error)
}

type DocumentDatabase interface {
	BaseDatabase

	GetCollection(name string) *mgo.Collection
	GetDB() *mgo.Database
	GridFS(prefix string) *mgo.GridFS
}

type MockDocumentDatabase struct {
	ConnectFunc    func(connection *Connection) error
	DisconnectFunc func()

	GetCollectionFunc func(name string) *mgo.Collection
	GetDBFunc         func() *mgo.Database
	GridFSFunc        func(prefix string) *mgo.GridFS
}

type MockMongoCollection struct {
	CountFunc       func() (int, error)
	EnsureIndexFunc func(index mgo.Index) error
	FindFunc        func(query interface{}) *mgo.Query
	FindIdFunc      func(id interface{}) *mgo.Query
	InsertFunc      func(docs ...interface{}) error
	RemoveFunc      func(selector interface{}) error
	RemoveAllFunc   func(selector interface{}) (*mgo.ChangeInfo, error)
	RemoveIdFunc    func(id interface{}) error
	UpdateFunc      func(selector, update interface{}) error
	UpdateAllFunc   func(selector, update interface{}) (*mgo.ChangeInfo, error)
	UpdateIdFunc    func(id, update interface{}) error
	UpsertFunc      func(selector, update interface{}) (*mgo.ChangeInfo, error)
	UpsertIdFunc    func(id, update interface{}) (*mgo.ChangeInfo, error)
}

type MockMongoDatabase struct {
	CFunc      func(name string) MongoCollection
	GridFSFunc func(prefix string) MongoGridFS
}

type MockMongoQuery struct {
	AllFunc      func(result interface{}) error
	CountFunc    func() (int, error)
	DistinctFunc func(key string, result interface{}) error
	LimitFunc    func(n int) MongoQuery
	OneFunc      func(result interface{}) error
	SelectFunc   func(selector interface{}) MongoQuery
	SkipFunc     func(n int) MongoQuery
	SortFunc     func(fields ...string) MongoQuery
}

type MockMongoSession struct {
	DBFunc func(name string) MongoDatabase
}

type MongoCollection interface {
	Count() (int, error)
	EnsureIndex(index mgo.Index) error
	Find(query interface{}) *MongoQueryWrapper
	FindId(id interface{}) *mgo.Query
	Insert(docs ...interface{}) error
	Remove(selector interface{}) error
	RemoveAll(selector interface{}) (*mgo.ChangeInfo, error)
	RemoveId(id interface{}) error
	Update(selector, update interface{}) error
	UpdateAll(selector, update interface{}) (*mgo.ChangeInfo, error)
	UpdateId(id, update interface{}) error
	Upsert(selector, update interface{}) (*mgo.ChangeInfo, error)
	UpsertId(id, update interface{}) (*mgo.ChangeInfo, error)
}

type MongoDatabase interface {
	C(name string) MongoCollection
	GridFS(prefix string) *mgo.GridFS
}

type MongoDatabaseService struct {
	DB      *mgo.Database
	Session *mgo.Session
}

type MongoGridFS interface {
	Create(name string) (MongoGridFile, error)
	Find(query interface{}) MongoQuery
	Open(name string) (MongoGridFile, error)
	OpenId(id interface{}) (MongoGridFile, error)
	OpenNext(MongoIter, MongoGridFile) bool
	Remove(name string) error
	RemoveId(id interface{}) error
}

type MongoGridFile interface {
	Abort()
	Close() error
	ContentType() string
	GetMeta(result interface{}) error
	Id() interface{}
	MD5() (md5 string)
	Name() string
	Read(b []byte) (int, error)
	Seek(offset int64, whence int) (int64, error)
	SetChunkSize(bytes int)
	SetContentType(ctype string)
	SetId(id interface{})
	SetMeta(metadata interface{})
	SetName(name string)
	Size() (bytes int64)
	UploadDate() time.Time
	Write(data []byte) (int, error)
}

type MongoIter interface {
	All(result interface{}) error
	Close() error
	Err() error
	For(result interface{}, f func() error) error
	Next(result interface{}) bool
	Timeout() bool
}

type MongoQuery interface {
	All(result interface{}) error
	Count() (int, error)
	Distinct(key string, result interface{}) error
	Limit(n int) MongoQuery
	One(result interface{}) error
	Select(selector interface{}) MongoQuery
	Skip(n int) MongoQuery
	Sort(fields ...string) MongoQuery
}

type MongoQueryWrapper struct {
	*mgo.Query
}

type MongoSession interface {
	Close()
	DB(name string) MongoDatabase
}

type MongoUploader struct {
	DB DocumentDatabase
}