45 lines
1.2 KiB
JavaScript
45 lines
1.2 KiB
JavaScript
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;
|