Initial commit

lib/db.ts

@@ -0,0 +1,76 @@
+import { Pool } from "pg";
+
+const pool = new Pool({
+    connectionString: process.env.DATABASE_URL,
+    ssl:
+        process.env.NODE_ENV === "production"
+            ? { rejectUnauthorized: false }
+            : false,
+});
+
+export async function query(text: string, params?: any[]) {
+    const result = await pool.query(text, params);
+    return result;
+}
+
+export async function initDatabase() {
+    await query(`
+        CREATE TABLE IF NOT EXISTS tasks (
+            id SERIAL PRIMARY KEY,
+            text TEXT NOT NULL,
+            description TEXT,
+            checklist JSONB DEFAULT '[]',
+            category VARCHAR(20) NOT NULL CHECK(category IN ('urgent', 'deadline', 'admin', 'creative')),
+            status VARCHAR(20) NOT NULL DEFAULT 'todo' CHECK(status IN ('todo', 'doing', 'done', 'archived')),
+            added VARCHAR(50) NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+        )
+    `);
+
+    await query(`
+        CREATE TABLE IF NOT EXISTS sessions (
+            id SERIAL PRIMARY KEY,
+            mode VARCHAR(20) NOT NULL CHECK(mode IN ('normal', 'crisis')),
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            exported_at TIMESTAMP
+        )
+    `);
+
+    await query(`
+        CREATE TABLE IF NOT EXISTS session_answers (
+            id SERIAL PRIMARY KEY,
+            session_id INTEGER NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
+            question_index INTEGER NOT NULL,
+            category TEXT NOT NULL,
+            question TEXT NOT NULL,
+            answer TEXT NOT NULL
+        )
+    `);
+
+    await query(`
+        CREATE TABLE IF NOT EXISTS questions (
+            id SERIAL PRIMARY KEY,
+            category TEXT NOT NULL,
+            question TEXT NOT NULL,
+            mode VARCHAR(20) NOT NULL CHECK(mode IN ('normal', 'crisis')),
+            order_index INTEGER NOT NULL DEFAULT 0,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+        )
+    `);
+
+    await query(
+        `CREATE INDEX IF NOT EXISTS idx_tasks_category ON tasks(category)`
+    );
+    await query(`CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status)`);
+    await query(
+        `CREATE INDEX IF NOT EXISTS idx_sessions_mode ON sessions(mode)`
+    );
+    await query(
+        `CREATE INDEX IF NOT EXISTS idx_session_answers_session_id ON session_answers(session_id)`
+    );
+    await query(
+        `CREATE INDEX IF NOT EXISTS idx_questions_mode ON questions(mode)`
+    );
+}
+
+export default { query, initDatabase };