/
stock.js
58 lines (37 loc) · 1.04 KB
/
stock.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
const { EventEmitter } = require('events')
const { Client } = require('discord.js')
var Stock = require('./models/Stock')
var User = require('./models/User')
class StockMarket extends EventEmitter {
constructor(client) {
super()
this.client = client;
this.users = {
}
this.client.users.cache.forEach((user) => {
this.users[user.id] = {
stocks: []
}
})
}
/**
* @param {String} user_id
* @param {String} stockId
* @param {Number} numberOfStocks
*/
async buyStock(user_id, guild_id, stockId, numberOfStocks) {
numberOfStocks = parseInt(numberOfStocks);
if (isNaN(numberOfStocks)) return;
var stock = await Stock.findOne({ id: stockId });
var user = await User.findOne({ id: user_id });
var price = stock.price * numberOfStocks
if (price > user.coins.guilds[guild_id]) return false;
user.coins.guilds[guild_id]-= price;
user.stocks[guild_id][stockId]+= numberOfStocks;
user.save()
return true;
}
sellStock(user_id, guild_id, stockId, numberOfStocks) {
}
}
module.exports = StockMarket