102 lines
2.0 KiB
TypeScript
102 lines
2.0 KiB
TypeScript
import { MongoClient } from "mongodb";
|
|
|
|
import { did_update } from "./util";
|
|
|
|
//figure out the options and whatnot later
|
|
const client = new MongoClient(process.env.MONGO_CONNECTION_STRING);
|
|
|
|
let store, users, roleincome;
|
|
|
|
client.connect().then(() => {
|
|
console.log("Connected to the database");
|
|
const db = client.db("db");
|
|
store = db.collection("store");
|
|
users = db.collection("users");
|
|
roleincome = db.collection("roleincome");
|
|
//
|
|
});
|
|
|
|
//db structure types
|
|
|
|
export interface StoreItem {
|
|
//
|
|
};
|
|
|
|
export interface User {
|
|
user: `${number}`; //discord user id
|
|
balance: number;
|
|
items: Record<string, number>;
|
|
//
|
|
};
|
|
|
|
export interface RoleIncome {
|
|
//
|
|
};
|
|
|
|
//default
|
|
|
|
const DEFAULT_USER: Omit<User, "user"> = {
|
|
balance: 0,
|
|
items: {},
|
|
//
|
|
};
|
|
|
|
//users collection db functions
|
|
|
|
export async function add_new_user(user: string) {
|
|
return await users.insertOne({ user, ...DEFAULT_USER });
|
|
}
|
|
|
|
export async function get_user(user: string): Promise<User | undefined> {
|
|
return await users.findOne({ user });
|
|
}
|
|
|
|
export async function add_balance(user: string, amount: number) {
|
|
return await users.updateOne({ user }, {
|
|
$inc: {
|
|
balance: amount,
|
|
},
|
|
});
|
|
}
|
|
|
|
//if false, not enough balance
|
|
export async function sub_balance(user: string, amount: number, negative_allowed: boolean = false): Promise<boolean> {
|
|
let query: any = {
|
|
user,
|
|
};
|
|
if (!negative_allowed) {
|
|
query.balance = {
|
|
$gte: amount,
|
|
};
|
|
}
|
|
return did_update(await users.updateOne(query, {
|
|
$inc: {
|
|
balance: -amount,
|
|
},
|
|
}));
|
|
}
|
|
|
|
export async function add_item_to_user(user: string, item: string, amount: number) {
|
|
return await users.updateOne({ user }, {
|
|
$inc: {
|
|
[`items.${item}`]: amount,
|
|
},
|
|
});
|
|
}
|
|
|
|
export async function sub_item_to_user(user: string, item: string, amount: number): Promise<boolean> {
|
|
return await users.updateOne({
|
|
user,
|
|
[`items.${item}`]: {
|
|
$gte: amount,
|
|
},
|
|
}, {
|
|
$inc: {
|
|
[`items.${item}`]: -amount,
|
|
},
|
|
});
|
|
}
|
|
|
|
//
|
|
|