import Data from './core/Data'
import { MongoClient, ObjectId, Db } from 'mongodb'

var db: Db

export async function connect(dbName: string, mongoUri: string) {
    if (db) {
        return db
    }
    const connection = await MongoClient.connect(mongoUri, { useUnifiedTopology: true })
    db = init(connection.db(dbName))
    return db
}

export function collection(name) {
    return db.collection(name)
}

export function id(i) {
    if (typeof i === 'string') return new ObjectId(i)
    else if (i instanceof ObjectId) return i
    else if (i instanceof Object) return '_id' in i ? new ObjectId(i._id) : null
    return null
}

function init(db: Db) {
    console.log('INIT DB')

    Data.setStore('Page', db.collection('pages'))
    Data.setStore('Article', db.collection('articles'))
    Data.setStore('Album', db.collection('albums'))
    Data.setStore('Image', db.collection('images'))
    Data.setStore('Teacher', db.collection('collective'))
    Data.setStore('Document', db.collection('documents'))
    Data.setStore('Settings', db.collection('settings'))

    return db
}