Saving token used and time joined

server/db/model/Tokens.js

@@ -1,10 +1,28 @@
 import OrderedObject from "./OrderedObject.js"
+const { z } = require("zod")
 
 export default class Tokens extends OrderedObject {
 
-    add(token) {
+    schema = z.object({
+        index: z.number(),
+        url: z.string(),
+        uuid: z.string().regex(
+            /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i,
+            "Invalid UUID"
+        ),
+        used: z.boolean(),
+    })
+
+    markUsed(uuid) {
+        let token = this.get(uuid)
+        token.used = true
+        super.update(`TOKEN-${uuid}`, token)
+    }
+
+    save(token) {
         let id = `TOKEN-${token.uuid}`
-        if(this.validate(id, token)) {
+        let result = this.schema.safeParse(token)
+        if(result.success) {
             try {
                 super.add(id, token)
             } catch(e) {
@@ -12,6 +30,7 @@ export default class Tokens extends OrderedObject {
                 throw e
             }
         } else {
+            console.error(result.error)
             throw new global.ServerError(400, "Invalid Member Data!");
         }
     }
@@ -19,44 +38,4 @@ export default class Tokens extends OrderedObject {
     get(uuid) {
         return super.get(`TOKEN-${uuid}`)
     }
-
-    validate(id, node) {
-        let idTraits = {
-            firstWord: "TOKEN"
-        }
-
-        let fields = [
-            "index",
-            "url",
-            "used"
-        ]
-
-        let checkID = () => {
-            let split = id.split("-")
-            return (
-                split[0] === idTraits.firstWord
-            )
-        }
-        let idres = checkID()
-        if(!idres) {
-            return false
-        }
-
-        let checkFields = () => {
-            for(let i = 0; i < fields.length; i++) {
-                if(!node[fields[i]]) {
-                    throw new Error(`Token ${node.email} is missing trait ${fields[i]}`)
-                    return false
-                } else {
-                    return true
-                }
-            }
-        }
-        let fieldres = checkFields()
-        if(!fieldres) {
-            return false
-        }
-
-        return true
-    }
 }