Initial Commit

lib/db.ts

@@ -0,0 +1,87 @@
+import Database from 'better-sqlite3';
+import path from 'path';
+
+const db = new Database(path.join(process.cwd(), 'autotrader.sqlite'));
+
+db.pragma('foreign_keys = ON');
+
+db.exec(`
+    CREATE TABLE IF NOT EXISTS firms (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT NOT NULL UNIQUE,
+        username TEXT NOT NULL,
+        password TEXT NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS account_configs (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        firm_id INTEGER NOT NULL REFERENCES firms(id) ON DELETE CASCADE,
+        prefix TEXT NOT NULL,
+        profit_target REAL NOT NULL DEFAULT 3000,
+        consistency REAL NOT NULL DEFAULT 0.5,
+        min_day_pnl REAL NOT NULL DEFAULT -999,
+        min_trading_days INTEGER NOT NULL DEFAULT 5
+    );
+`);
+
+// Seed default data if empty
+const firmCount = (db.prepare('SELECT COUNT(*) as count FROM firms').get() as { count: number }).count;
+if (firmCount === 0) {
+    const insertFirm = db.prepare('INSERT INTO firms (name, username, password) VALUES (?, ?, ?)');
+    const insertAccount = db.prepare(
+        'INSERT INTO account_configs (firm_id, prefix, profit_target, consistency, min_day_pnl, min_trading_days) VALUES (?, ?, ?, ?, ?, ?)'
+    );
+
+    const alpha = insertFirm.run('Alpha', 'brandonsenoli72786', '-Z2kPm7nBg');
+    insertAccount.run(alpha.lastInsertRowid, 'AFSTDEV',  9000, 0.51, -999, 5);
+    insertAccount.run(alpha.lastInsertRowid, 'AFSTDQA',  4500, 0.40, -999, 7);
+    insertAccount.run(alpha.lastInsertRowid, 'AFZEROEV', 3000, 0.50, -999, 5);
+    insertAccount.run(alpha.lastInsertRowid, 'AFZEROQA', 3000, 0.50, -999, 5);
+    insertAccount.run(alpha.lastInsertRowid, 'AF',       3000, 0.50, -999, 5);
+
+    const tpt = insertFirm.run('TakeProfitTrader', 'BRANDONLI1', 'W4592F5512U2817tv=');
+    insertAccount.run(tpt.lastInsertRowid, 'TAKEPROFIT', 9000, 0.50, -999, 5);
+
+    console.log('[db] Seeded default firms.');
+}
+
+export interface AccountConfigRow {
+    id: number;
+    firm_id: number;
+    prefix: string;
+    profit_target: number;
+    consistency: number;
+    min_day_pnl: number;
+    min_trading_days: number;
+}
+
+export interface FirmRow {
+    id: number;
+    name: string;
+    username: string;
+    password: string;
+}
+
+export interface FirmWithAccounts extends FirmRow {
+    accounts: AccountConfigRow[];
+}
+
+export function getFirms(): FirmWithAccounts[] {
+    const firms = db.prepare('SELECT * FROM firms ORDER BY id').all() as FirmRow[];
+    const getAccounts = db.prepare('SELECT * FROM account_configs WHERE firm_id = ? ORDER BY id');
+    return firms.map((firm) => ({
+        ...firm,
+        accounts: getAccounts.all(firm.id) as AccountConfigRow[],
+    }));
+}
+
+export function createFirm(name: string, username: string, password: string): FirmRow {
+    const stmt = db.prepare('INSERT INTO firms (name, username, password) VALUES (?, ?, ?)');
+    const result = stmt.run(name, username, password);
+    return db.prepare('SELECT * FROM firms WHERE id = ?').get(result.lastInsertRowid) as FirmRow;
+}
+
+export function deleteFirm(id: number): boolean {
+    const result = db.prepare('DELETE FROM firms WHERE id = ?').run(id);
+    return result.changes > 0;
+}