-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdbObjects.js
38 lines (30 loc) · 1.07 KB
/
dbObjects.js
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
const Sequelize = require("sequelize");
const sequelize = new Sequelize({
host: "localhost",
dialect: "sqlite",
logging: false,
storage: "./.data/database.sqlite"
});
const Users = require("./tables/Users")(sequelize, Sequelize.DataTypes);
const CurrencyShop = require("./tables/CurrencyShop")(sequelize, Sequelize.DataTypes);
const UserItems = require("./tables/UserItems")(sequelize, Sequelize.DataTypes);
UserItems.belongsTo(CurrencyShop, { foreignKey: "item_id", as: "item" });
/* eslint-disable-next-line func-names */
Users.prototype.addItem = async function(item) {
const userItem = await UserItems.findOne({
where: { user_id: this.user_id, item_id: item.id },
});
if (userItem) {
userItem.amount += 1;
return userItem.save();
}
return UserItems.create({ user_id: this.user_id, item_id: item.id, amount: 1 });
};
/* eslint-disable-next-line func-names */
Users.prototype.getItems = function() {
return UserItems.findAll({
where: { user_id: this.user_id },
include: ["item"],
});
};
module.exports = { Users, CurrencyShop, UserItems };