1 files changed, 10 insertions, 10 deletions

diff --git a/scripts/gen-invite.js b/scripts/gen-invite.js
index 7e40ffa..6b1e93d 100644
--- a/scripts/gen-invite.js
+++ b/scripts/gen-invite.js
@@ -1,9 +1,18 @@
 import { Database } from "bun:sqlite";
 
-const db = new Database("readit.db", {
+const command = process.argv[2];
+
+const dbPath = process.argv[3] ? process.argv[3] : "readit.db";
+const db = new Database(dbPath, {
         strict: true,
 });
 
+if (command === "create") {
+        createInvite();
+} else {
+        console.log("requires an arg");
+}
+
 db.run(`
         CREATE TABLE IF NOT EXISTS invites (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -23,12 +32,3 @@ function createInvite() {
         db.run("INSERT INTO invites (token) VALUES ($token)", { token });
         console.log(`Invite token created: ${token}`);
 }
-
-const command = process.argv[2];
-const arg = process.argv[3];
-
-if (command === "create") {
-        createInvite();
-} else {
-        console.log("requires an arg");
-}