Add lib/db.js

lib/db.js

@@ -0,0 +1,44 @@
+import Database from 'better-sqlite3';
+import path from 'path';
+
+const dbPath = path.resolve(process.cwd(), 'devbin.db');
+const db = new Database(dbPath);
+
+db.exec(`
+  CREATE TABLE IF NOT EXISTS pastes (
+    id TEXT PRIMARY KEY,
+    content TEXT NOT NULL,
+    language TEXT NOT NULL,
+    filename TEXT NOT NULL,
+    created_at TEXT NOT NULL,
+    expires_at TEXT,
+    expiry_type TEXT DEFAULT 'time',
+    view_count INTEGER DEFAULT 0,
+    allow_discussions INTEGER DEFAULT 0,
+    delete_token TEXT
+  );
+
+  CREATE TABLE IF NOT EXISTS comments (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    paste_id TEXT NOT NULL,
+    author TEXT NOT NULL,
+    content TEXT NOT NULL,
+    created_at TEXT NOT NULL,
+    FOREIGN KEY (paste_id) REFERENCES pastes (id) ON DELETE CASCADE
+  );
+`);
+
+// Migration for delete_token
+try {
+  const tableInfo = db.prepare("PRAGMA table_info(pastes)").all();
+  if (!tableInfo.some(col => col.name === 'delete_token')) {
+    db.exec("ALTER TABLE pastes ADD COLUMN delete_token TEXT;");
+  }
+  if (!tableInfo.some(col => col.name === 'allow_discussions')) {
+    db.exec("ALTER TABLE pastes ADD COLUMN allow_discussions INTEGER DEFAULT 0;");
+  }
+} catch (err) {
+  console.error("Migration failed:", err);
+}
+
+export default db;