60 lines
1.5 KiB
JavaScript
60 lines
1.5 KiB
JavaScript
const MongoClient = require('mongodb').MongoClient;
|
|
const mongodbUri = require('mongodb-uri');
|
|
|
|
let _db;
|
|
|
|
function initDb(dbUrl, callback){ // eslint-disable-line
|
|
if(_db){
|
|
console.warn('Trying to init DB again!');
|
|
return callback(null, _db);
|
|
}
|
|
MongoClient.connect(dbUrl, { useNewUrlParser: true, useUnifiedTopology: true }, connected);
|
|
function connected(err, client){
|
|
if(err){
|
|
return callback(err);
|
|
}
|
|
|
|
// Set the DB url
|
|
dbUrl = getDbUri(dbUrl);
|
|
|
|
// select DB
|
|
const dbUriObj = mongodbUri.parse(dbUrl);
|
|
|
|
// Set the DB depending on ENV
|
|
const db = client.db(dbUriObj.database);
|
|
|
|
// setup the collections
|
|
db.users = db.collection('users');
|
|
db.products = db.collection('products');
|
|
db.orders = db.collection('orders');
|
|
db.pages = db.collection('pages');
|
|
db.menu = db.collection('menu');
|
|
db.customers = db.collection('customers');
|
|
db.cart = db.collection('cart');
|
|
db.sessions = db.collection('sessions');
|
|
db.discounts = db.collection('discounts');
|
|
|
|
_db = db;
|
|
return callback(null, _db);
|
|
}
|
|
};
|
|
|
|
function getDbUri(dbUrl){
|
|
const dbUriObj = mongodbUri.parse(dbUrl);
|
|
// if in testing, set the testing DB
|
|
if(process.env.NODE_ENV === 'test'){
|
|
dbUriObj.database = 'expresscart-test';
|
|
}
|
|
return mongodbUri.format(dbUriObj);
|
|
}
|
|
|
|
function getDb(){
|
|
return _db;
|
|
}
|
|
|
|
module.exports = {
|
|
getDb,
|
|
initDb,
|
|
getDbUri
|
|
};
|