From d1fb95883e331f16f15d8d6698cc2db6613732c9 Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Fri, 15 Nov 2024 23:06:02 +0000
Subject: allow path to db as arg

---
 scripts/gen-invite.js | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

(limited to 'scripts/gen-invite.js')

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");
-}
-- 
cgit v1.2.3