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
|
const config = require("./config");
const Sequelize = require("sequelize");
const SequelizeAuto = require("sequelize-auto");
const conf = config.db[process.env.NODE_ENV || "development"];
const auto = new SequelizeAuto(
conf.database,
conf.username,
conf.password,
config.auto
);
const tables = {
User: "users",
Payment: "payments"
};
const db = new Sequelize(conf.database, conf.username, conf.password, {
host: conf.host,
dialect: conf.dialect,
logging: false,
dialectOptions: { multipleStatements: true }
});
const p = new Promise((resolve, reject) => {
return auto.run(() => {
Object.keys(tables).forEach(k => {
let t = tables[k];
let fields = {};
Object.keys(auto.tables[t]).forEach(f => {
let isKey = f === "id";
let rawtype = auto.tables[t][f].type.toLowerCase();
let type = Sequelize.STRING;
if (rawtype.match(/^int/)) type = Sequelize.INTEGER;
if (rawtype.match(/^date/)) type = Sequelize.DATE;
fields[f] = {
type: type,
field: f,
primaryKey: isKey,
autoIncrement: isKey
};
});
db[k] = db.define(k, fields, { tableName: t });
let typefields = {};
let options = {};
if (t === "users") {
options = {
before: (findOptions, args, context) => {
findOptions.where = { id: context.user.id };
return findOptions;
}
};
}
if (t === "payments") {
db["Payment"].belongsTo(db["User"], {
as: "user",
foreignKey: "user_id"
});
options = {
before: (findOptions, args, context) => {
findOptions.where = { user_id: context.user.id };
return findOptions;
}
};
}
});
db["User"].hasMany(db["Payment"], {
as: "payments",
foreignKey: "user_id"
});
resolve(db);
});
});
module.exports = p;
|
