-
Notifications
You must be signed in to change notification settings - Fork 106
/
index.hbs
96 lines (79 loc) · 2.42 KB
/
index.hbs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
{{#if isMongoDB}}
const mongoose = require('mongoose');
const requireAll = require('require-all');
const models = {};
mongoose.connect(process.env.DATABASE_URL, { useNewUrlParser: true, useUnifiedTopology: true });
function filterFile(filename) {
if (filename.endsWith('.js') && filename !== 'index.js') {
return filename;
}
return false;
}
const modelsList = requireAll({
dirname: __dirname,
filter: (filename) => filterFile(filename),
});
Object.keys(modelsList).forEach((filename) => {
const model = modelsList[filename];
models[model.modelName] = model;
});
module.exports = models;
{{else}}
const fs = require('fs');
const path = require('path');
const Sequelize = require('sequelize');
if (!process.env.DATABASE_URL) {
console.error('Cannot connect to the database. Please declare the DATABASE_URL environment variable with the correct database connection string.');
process.exit();
}
const databaseOptions = {
logging: process.env.NODE_ENV === 'development' ? console.log : false,
pool: { maxConnections: 10, minConnections: 1 },
dialectOptions: {},
};
{{#if isMySQL}}
databaseOptions.dialectOptions.typeCast = (field, useDefaultTypeCasting) => {
if ((field.type === "BIT") && (field.length === 1)) {
const bytes = field.buffer();
return bytes ? bytes[0] === 1 : bytes;
}
return useDefaultTypeCasting();
};
{{/if}}
if (process.env.DATABASE_SSL && JSON.parse(process.env.DATABASE_SSL.toLowerCase())) {
{{#if isMySQL}}
databaseOptions.dialectOptions.ssl = { rejectUnauthorized: true };
{{else if isMSSQL}}
databaseOptions.dialectOptions.options = { encrypt: true };
{{else}}
if (process.env.DATABASE_REJECT_UNAUTHORIZED === false) {
databaseOptions.dialectOptions.ssl = { rejectUnauthorized: false };
} else {
databaseOptions.dialectOptions.ssl = true;
}
{{/if}}
}
const sequelize = new Sequelize(process.env.DATABASE_URL, databaseOptions);
const db = {};
fs
.readdirSync(__dirname)
.filter((file) => {
return (file.indexOf('.') !== 0) && (file !== 'index.js');
})
.forEach((file) => {
try {
const model = sequelize.import(path.join(__dirname, file));
db[model.name] = model;
} catch (error) {
console.error('Model creation error: ' + error);
}
});
Object.keys(db).forEach((modelName) => {
if ('associate' in db[modelName]) {
db[modelName].associate(db);
}
});
db.sequelize = sequelize;
db.Sequelize = Sequelize;
module.exports = db;
{{/if}}