const Database = require('better-sqlite3');
const path = require('path');
const fs = require('fs');

const dataDir = '/app/data';
if (!fs.existsSync(dataDir)) {
  fs.mkdirSync(dataDir, { recursive: true });
}

const dbPath = path.join(dataDir, 'xiaocai.db');
const db = new Database(dbPath);
db.pragma('journal_mode = WAL');

// 创建基础表（如果不存在）
db.exec(`
  CREATE TABLE IF NOT EXISTS users (
    username TEXT PRIMARY KEY,
    password TEXT NOT NULL,
    is_admin INTEGER DEFAULT 0
  );

  CREATE TABLE IF NOT EXISTS rooms (
    id TEXT PRIMARY KEY,
    name TEXT NOT NULL,
    created_by TEXT NOT NULL,
    white_list TEXT DEFAULT '',
    created_at TEXT DEFAULT (datetime('now','localtime'))
  );

  CREATE TABLE IF NOT EXISTS messages (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    room_id TEXT NOT NULL,
    user TEXT NOT NULL,
    text TEXT,
    attachments TEXT,
    timestamp TEXT DEFAULT (datetime('now','localtime')),
    FOREIGN KEY(room_id) REFERENCES rooms(id)
  );

  CREATE TABLE IF NOT EXISTS purchases (
    id TEXT PRIMARY KEY,
    room_id TEXT NOT NULL,
    item TEXT,
    quantity INTEGER DEFAULT 1,
    unit_price REAL DEFAULT 0,
    amount REAL DEFAULT 0,
    freight REAL DEFAULT 0,
    payment_method TEXT DEFAULT '未指定',
    invoice_type TEXT DEFAULT '无票',
    status TEXT DEFAULT '待处理',
    applicant TEXT,
    remarks TEXT DEFAULT '',
    created_at TEXT DEFAULT (datetime('now','localtime')),
    updated_at TEXT DEFAULT (datetime('now','localtime')),
    FOREIGN KEY(room_id) REFERENCES rooms(id)
  );

  CREATE TABLE IF NOT EXISTS purchase_history (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    purchase_id TEXT NOT NULL,
    action TEXT NOT NULL,
    user TEXT NOT NULL,
    timestamp TEXT DEFAULT (datetime('now','localtime')),
    attachments TEXT,
    FOREIGN KEY(purchase_id) REFERENCES purchases(id)
  );

  CREATE TABLE IF NOT EXISTS purchase_attachments (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    purchase_id TEXT NOT NULL,
    file_path TEXT NOT NULL,
    uploaded_by TEXT NOT NULL,
    timestamp TEXT DEFAULT (datetime('now','localtime')),
    FOREIGN KEY(purchase_id) REFERENCES purchases(id)
  );
`);

// 为已有数据库添加可能缺失的列（安全迁移，忽略错误）
const migrations = [
  `ALTER TABLE purchases ADD COLUMN quantity INTEGER DEFAULT 1`,
  `ALTER TABLE purchases ADD COLUMN unit_price REAL DEFAULT 0`,
  `ALTER TABLE purchases ADD COLUMN freight REAL DEFAULT 0`,
  `ALTER TABLE purchases ADD COLUMN invoice_type TEXT DEFAULT '无票'`,
  `ALTER TABLE purchases ADD COLUMN payment_method TEXT DEFAULT '未指定'`,
  `ALTER TABLE purchases ADD COLUMN remarks TEXT DEFAULT ''`,
  `ALTER TABLE rooms ADD COLUMN white_list TEXT DEFAULT ''`,
];

for (const sql of migrations) {
  try {
    db.exec(sql);
  } catch (e) {
    // 列已存在或其他错误，忽略
  }
}

module.exports = db;