Updated users model

Author: the-code-genin

models/users.ts

@@ -19,7 +19,7 @@ export function userToJSON(user: User): UserJSON {
 export async function getUsersCountWithEmail(email: string) {
     const total = (await global.db<User>("users")
         .where("email", email)
-        .count("*", {as: "total"})
+        .count("*", { as: "total" })
     )[0].total;
     return Number(total);
 }
@@ -43,4 +43,25 @@ export async function saveUser(user: User) {
     return getUserByID(insertId[0]) as User;
 }
 
+export async function checkUserForAuthToken(userId: number, token: string) {
+    const total = (await global.db("user_auth_tokens")
+        .where("user_id", userId)
+        .where("token", token)
+        .count("*", { as: "total" })
+    )[0].total;
+    return Number(total) != 0;
+}
+
+export async function addUserAuthToken(userId: number, token: string) {
+    if (await checkUserForAuthToken(userId, token)) {
+        throw new Error("User already has this auth token.");
+    }
+
+    const insertId = await global.db("user_auth_tokens").insert({
+        user_id: userId,
+        token
+    });
+    return insertId.length == 1;
+}
+
 export default () => global.db<User>("users");