HackMD/lib/models/index.js
2016-12-12 11:12:59 +08:00

42 lines
1 KiB
JavaScript

"use strict";
// external modules
var fs = require("fs");
var path = require("path");
var Sequelize = require("sequelize");
// core
var config = require('../config.js');
var logger = require("../logger.js");
var dbconfig = config.db;
dbconfig.logging = config.debug ? logger.info : false;
var sequelize = null;
if (dbconfig.hasOwnProperty('username') || dbconfig.hasOwnProperty('password'))
sequelize = new Sequelize(dbconfig.database, dbconfig.username, dbconfig.password, dbconfig);
else
sequelize = new Sequelize(dbconfig.database, dbconfig);
var db = {};
fs
.readdirSync(__dirname)
.filter(function (file) {
return (file.indexOf(".") !== 0) && (file !== "index.js");
})
.forEach(function (file) {
var model = sequelize.import(path.join(__dirname, file));
db[model.name] = model;
});
Object.keys(db).forEach(function (modelName) {
if ("associate" in db[modelName]) {
db[modelName].associate(db);
}
});
db.sequelize = sequelize;
db.Sequelize = Sequelize;
module.exports = db;