user registration, bal, change_bal, transfer

db.ts

@@ -1,12 +1,101 @@
 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, income;
+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,
+    },
+  });
+}
+
+//
+