estimation-flow/src/lib/db.ts

import Database from 'better-sqlite3';

// Use a file-based database in development
const db = new Database('local.db');

// --- SCHEMA CREATION ---
db.exec(`
  CREATE TABLE IF NOT EXISTS users (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    email TEXT UNIQUE NOT NULL,
    password TEXT NOT NULL,
    name TEXT NOT NULL
  )
`);

db.exec(`
  CREATE TABLE IF NOT EXISTS settings (
    key TEXT PRIMARY KEY,
    value TEXT
  )
`);

db.exec(`
  CREATE TABLE IF NOT EXISTS email_templates (
    id INTEGER PRIMARY KEY,
    subject TEXT,
    body TEXT
  )
`);

db.exec(`
  CREATE TABLE IF NOT EXISTS leads (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    name TEXT NOT NULL,
    email TEXT NOT NULL,
    phone TEXT,
    createdAt DATETIME DEFAULT CURRENT_TIMESTAMP
  )
`);

db.exec(`
  CREATE TABLE IF NOT EXISTS flows (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    name TEXT NOT NULL,
    description TEXT,
    path TEXT NOT NULL UNIQUE,
    createdAt DATETIME DEFAULT CURRENT_TIMESTAMP,
    updatedAt DATETIME DEFAULT CURRENT_TIMESTAMP
  )
`);

// --- SEEDING LOGIC ---
console.log('Running database checks and seeding if necessary...');

// Seed default user
const userStmt = db.prepare('SELECT id FROM users WHERE id = ?');
const defaultUser = userStmt.get(1);
if (!defaultUser) {
  const insertUser = db.prepare(
    "INSERT INTO users (id, email, password, name) VALUES (?, ?, ?, ?)"
  );
  insertUser.run(1, 'admin@example.com', 'password', 'Admin User');
  console.log('Default user created.');
}

// Seed default flow
const flowStmt = db.prepare("SELECT id FROM flows WHERE path = ?");
const defaultFlow = flowStmt.get('/');
if (!defaultFlow) {
  const insertFlow = db.prepare(
    "INSERT INTO flows (name, description, path) VALUES (?, ?, ?)"
  );
  insertFlow.run('Cost Estimator', 'The main cost estimation tool for clients.', '/');
  console.log('Default flow created.');
}

console.log('Database setup complete.');

export default db;