add invite system

1 files changed, 37 insertions, 0 deletions

diff --git a/src/invite.js b/src/invite.js
new file mode 100644
index 0000000..0f6624c
--- /dev/null
+++ b/src/invite.js
@@ -0,0 +1,37 @@
+const { db } = require("./db");
+
+const validateInviteToken = async (req, res, next) => {
+        const token = req.query.token;
+
+        if (!token) {
+                return res.render("register", {
+                        message: "this instance requires an invite",
+                        isDisabled: true,
+                });
+        }
+
+        const invite = db
+                .query("SELECT * FROM invites WHERE token = $token AND usedAt IS null")
+                .get({ token });
+
+        if (!invite) {
+                return res.render("register", {
+                        message: "this invite token is invalid",
+                        isDisabled: true,
+                });
+        }
+
+        if (invite.usedAt) {
+                return res.render("register", {
+                        message: "this invite has been claimed",
+                        isDisabled: true,
+                });
+        }
+
+        req.invite = invite;
+        next();
+};
+
+module.exports = {
+        validateInviteToken,
+};