smartdata/ts/smartdata.classes.db.ts

import * as plugins from './smartdata.plugins'
import { Objectmap } from 'lik'

import { DbCollection } from './smartdata.classes.dbcollection'

export type TConnectionStatus = 'disconnected' | 'connected' | 'failed'

export class Db {
    dbUrl: string
    db: plugins.mongodb.Db
    status: TConnectionStatus
    collections = new Objectmap<DbCollection<any>>()

    constructor(dbUrl: string) {
        this.dbUrl = dbUrl
    }

    // basic connection stuff ----------------------------------------------

    /**
     * connects to the database that was specified during instance creation
     */
    connect(): plugins.q.Promise<any> {
        let done = plugins.q.defer()
        plugins.mongodb.MongoClient.connect(this.dbUrl, (err, db) => {
            if (err) { console.log(err) }
            plugins.assert.equal(null, err)
            this.db = db
            plugins.beautylog.success(`connected to database at ${this.dbUrl}`)
            done.resolve(this.db)
        })
        return done.promise
    }

    /**
     * closes the connection to the databse
     */
    close(): plugins.q.Promise<any> {
        let done = plugins.q.defer()
        this.db.close()
        plugins.beautylog.ok(`disconnected to database at ${this.dbUrl}`)
        done.resolve()
        return done.promise
    }

    // advanced communication with the database --------------------------------

    /**
     * gets a collection by name: string
     */
    getCollectionByName<T>(nameArg: string): plugins.q.Promise<DbCollection<T>> {
        let done = plugins.q.defer<DbCollection<any>>()
        let resultCollection = this.collections.find((dbCollectionArg) => {
            return dbCollectionArg.name === nameArg
        })
        if (resultCollection !== null) {
            done.resolve(resultCollection)
        }
        return done.promise
    };

    addCollection(dbCollectionArg: DbCollection<any>) {
        this.collections.add(dbCollectionArg)
    }
}