yunhaifinance/backend/db-sqlite.js.disabled

const sqlite3 = require('sqlite3').verbose();
const path = require('path');

// 创建SQLite数据库连接
const dbPath = path.join(__dirname, 'company_finance.db');
const db = new sqlite3.Database(dbPath, (err) => {
  if (err) {
    console.error('数据库连接失败:', err.message);
  } else {
    console.log('SQLite数据库连接成功');
    initializeDatabase();
  }
});

// 初始化数据库
function initializeDatabase() {
  const tables = [
    // 用户表
    `CREATE TABLE IF NOT EXISTS users (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      username TEXT UNIQUE NOT NULL,
      password TEXT NOT NULL,
      name TEXT,
      email TEXT,
      phone TEXT,
      role TEXT NOT NULL DEFAULT 'user',
      avatar TEXT,
      passport TEXT,
      driverLicense TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 项目表
    `CREATE TABLE IF NOT EXISTS projects (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      code TEXT UNIQUE NOT NULL,
      customer_id INTEGER,
      manager_id INTEGER,
      contract_amount REAL DEFAULT 0.0,
      start_date DATE,
      end_date DATE,
      description TEXT,
      status TEXT DEFAULT 'active',
      location TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (manager_id) REFERENCES users(id)
    )`,
    // 客户表
    `CREATE TABLE IF NOT EXISTS customers (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      contact TEXT,
      position TEXT,
      phone TEXT,
      email TEXT,
      address TEXT,
      remark TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 供应商表
    `CREATE TABLE IF NOT EXISTS suppliers (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      contact TEXT,
      position TEXT,
      phone TEXT,
      email TEXT,
      address TEXT,
      supply_category TEXT,
      country TEXT,
      remark TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 分包商表
    `CREATE TABLE IF NOT EXISTS subcontractors (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      contact TEXT,
      position TEXT,
      phone TEXT,
      email TEXT,
      address TEXT,
      scope TEXT,
      features TEXT,
      country TEXT,
      remark TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 商品表
    `CREATE TABLE IF NOT EXISTS products (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      category_id INTEGER,
      unit TEXT,
      price REAL,
      description TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 分类表
    `CREATE TABLE IF NOT EXISTS categories (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      parent_id INTEGER,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 预算项目表
    `CREATE TABLE IF NOT EXISTS budget_projects (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      name TEXT NOT NULL,
      customer_id INTEGER,
      manager_id INTEGER,
      location TEXT,
      survey_date DATE,
      intermediary TEXT,
      intermediary_fee_type TEXT,
      intermediary_fee_value REAL,
      customer_requirements TEXT,
      project_overview TEXT,
      attachments TEXT,
      survey_photos TEXT,
      status TEXT DEFAULT 'negotiating',
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (customer_id) REFERENCES customers(id),
      FOREIGN KEY (manager_id) REFERENCES users(id)
    )`,
    // 报价表
    `CREATE TABLE IF NOT EXISTS budget_quotations (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      version INTEGER,
      quotation_date DATE,
      amount REAL,
      currency TEXT DEFAULT 'CNY',
      status TEXT DEFAULT 'draft',
      file_url TEXT,
      remark TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES budget_projects(id)
    )`,
    // 项目合同表
    `CREATE TABLE IF NOT EXISTS project_contracts (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      contract_code TEXT UNIQUE NOT NULL,
      contract_amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      settlement_method TEXT,
      contract_period INTEGER,
      start_date DATE,
      end_date DATE,
      warranty_deposit_percentage REAL DEFAULT 5,
      warranty_period INTEGER DEFAULT 12,
      contract_file TEXT,
      other_info TEXT,
      tax_included INTEGER DEFAULT 0,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 分包合同表
    `CREATE TABLE IF NOT EXISTS subcontracts (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      subcontractor_id INTEGER,
      subcontractor_name TEXT,
      contract_amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      settlement_type TEXT DEFAULT 'lump_sum',
      other_terms TEXT,
      payment_description TEXT,
      start_date DATE,
      end_date DATE,
      paid_amount REAL DEFAULT 0,
      status TEXT DEFAULT 'active',
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id),
      FOREIGN KEY (subcontractor_id) REFERENCES subcontractors(id)
    )`,
    // 项目材料表
    `CREATE TABLE IF NOT EXISTS project_materials (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      product_id INTEGER,
      product_name TEXT,
      unit TEXT,
      budget_quantity REAL,
      purchase_quantity REAL,
      used_quantity REAL,
      average_price REAL,
      total_amount REAL,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id),
      FOREIGN KEY (product_id) REFERENCES products(id)
    )`,
    // 施工节点表
    `CREATE TABLE IF NOT EXISTS project_milestones (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      milestone_name TEXT,
      condition TEXT,
      percentage REAL,
      amount REAL,
      expected_date DATE,
      actual_date DATE,
      completion_progress INTEGER DEFAULT 0,
      status TEXT DEFAULT 'pending',
      voucher TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 项目财务信息表
    `CREATE TABLE IF NOT EXISTS project_finances (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      payment_type TEXT,
      amount REAL DEFAULT 0,
      currency TEXT DEFAULT 'CNY',
      payment_date DATE,
      status TEXT DEFAULT 'pending',
      description TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 质保金表
    `CREATE TABLE IF NOT EXISTS warranty_deposits (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      amount REAL NOT NULL,
      percentage REAL DEFAULT 5,
      currency TEXT DEFAULT 'CNY',
      warranty_period INTEGER DEFAULT 12,
      start_date DATE,
      end_date DATE,
      status TEXT DEFAULT 'active',
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 施工日志表
    `CREATE TABLE IF NOT EXISTS construction_logs (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      log_date DATE,
      weather TEXT,
      work_content TEXT,
      photos TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 联系人表
    `CREATE TABLE IF NOT EXISTS contacts (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      entity_id INTEGER NOT NULL,
      entity_type TEXT NOT NULL,
      name TEXT NOT NULL,
      position TEXT,
      phone TEXT,
      is_primary INTEGER DEFAULT 0,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 分包商收款信息表
    `CREATE TABLE IF NOT EXISTS subcontractor_payment_infos (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      subcontractor_id INTEGER NOT NULL,
      account_name TEXT NOT NULL,
      bank_account TEXT NOT NULL,
      bank_name TEXT NOT NULL,
      qr_code TEXT,
      is_primary INTEGER DEFAULT 0,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 汇率表
    `CREATE TABLE IF NOT EXISTS exchange_rates (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      pair_key TEXT NOT NULL,
      rate REAL NOT NULL,
      effective_date DATE NOT NULL,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 付款节点表
    `CREATE TABLE IF NOT EXISTS payment_nodes (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      project_id INTEGER,
      node_name TEXT NOT NULL,
      due_date DATE,
      amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      status TEXT DEFAULT 'pending',
      description TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 付款记录表
    `CREATE TABLE IF NOT EXISTS payment_records (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      node_id INTEGER,
      payment_date DATE,
      amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      payment_method TEXT,
      status TEXT DEFAULT 'completed',
      description TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (node_id) REFERENCES payment_nodes(id)
    )`,
    // 预支款表
    `CREATE TABLE IF NOT EXISTS advances (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      user_id INTEGER,
      project_id INTEGER,
      amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      amount_cny REAL DEFAULT 0,
      total_reimbursed REAL DEFAULT 0,
      reason TEXT NOT NULL,
      advance_date DATE,
      advance_code TEXT UNIQUE NOT NULL,
      status TEXT DEFAULT 'pending',
      applicant TEXT,
      attachments TEXT,
      approval_remark TEXT,
      execute_date DATE,
      execute_method TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (user_id) REFERENCES users(id),
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 报销表
    `CREATE TABLE IF NOT EXISTS reimbursements (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      user_id INTEGER,
      project_id INTEGER,
      amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      amount_cny REAL DEFAULT 0,
      reason TEXT NOT NULL,
      reimbursement_date DATE,
      reimbursement_code TEXT UNIQUE NOT NULL,
      status TEXT DEFAULT 'pending',
      applicant TEXT,
      expense_type TEXT NOT NULL,
      detail_items TEXT,
      attachments TEXT,
      approval_remark TEXT,
      execute_date DATE,
      execute_method TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (user_id) REFERENCES users(id),
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 付款申请表
    `CREATE TABLE IF NOT EXISTS payment_requests (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      request_code TEXT UNIQUE NOT NULL,
      applicant TEXT NOT NULL,
      payee TEXT NOT NULL,
      bank_account TEXT NOT NULL,
      bank_name TEXT NOT NULL,
      amount REAL NOT NULL,
      amount_cny REAL DEFAULT 0,
      currency TEXT DEFAULT 'CNY',
      payment_date DATE NOT NULL,
      reason TEXT NOT NULL,
      detail_items TEXT,
      attachments TEXT,
      status TEXT DEFAULT 'pending',
      approval_remark TEXT,
      execute_date DATE,
      execute_method TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 核销申请表
    `CREATE TABLE IF NOT EXISTS verifications (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      verification_code TEXT UNIQUE NOT NULL,
      applicant TEXT NOT NULL,
      advance_code TEXT NOT NULL,
      advance_amount REAL NOT NULL,
      amount REAL NOT NULL,
      amount_cny REAL DEFAULT 0,
      currency TEXT DEFAULT 'CNY',
      verification_date DATE NOT NULL,
      reason TEXT NOT NULL,
      expense_type TEXT NOT NULL,
      project_id INTEGER,
      settlement INTEGER DEFAULT 0,
      settlement_amount REAL DEFAULT 0,
      detail_items TEXT,
      attachments TEXT,
      status TEXT DEFAULT 'pending',
      approval_remark TEXT,
      execute_date DATE,
      execute_method TEXT,
      user_id INTEGER,
      advance_id INTEGER,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (project_id) REFERENCES projects(id)
    )`,
    // 库存管理表
    `CREATE TABLE IF NOT EXISTS inventory (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      product_id INTEGER,
      product_name TEXT NOT NULL,
      quantity REAL NOT NULL,
      unit TEXT NOT NULL,
      price REAL,
      total_value REAL,
      location TEXT,
      status TEXT DEFAULT 'in_stock',
      last_updated DATE,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (product_id) REFERENCES products(id)
    )`,
    // 采购订单表
    `CREATE TABLE IF NOT EXISTS purchase_orders (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      code TEXT NOT NULL,
      purchase_request_id INTEGER,
      supplier_id INTEGER,
      supplier_name TEXT,
      total_amount REAL DEFAULT 0,
      currency TEXT DEFAULT 'CNY',
      order_date TEXT,
      delivery_date TEXT,
      status TEXT DEFAULT 'pending',
      created_by TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 采购订单明细表
    `CREATE TABLE IF NOT EXISTS purchase_order_items (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      purchase_order_id INTEGER NOT NULL,
      product_id INTEGER,
      product_name TEXT NOT NULL,
      specification TEXT,
      quantity REAL NOT NULL,
      unit TEXT NOT NULL,
      unit_price REAL NOT NULL,
      total_price REAL NOT NULL,
      remark TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 付款计划表
    `CREATE TABLE IF NOT EXISTS payment_plans (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      purchase_order_id INTEGER NOT NULL,
      code TEXT NOT NULL,
      payment_date TEXT,
      amount REAL NOT NULL,
      currency TEXT DEFAULT 'CNY',
      payment_type TEXT DEFAULT 'partial',
      status TEXT DEFAULT 'pending',
      description TEXT,
      created_by TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`,
    // 库存记录表
    `CREATE TABLE IF NOT EXISTS inventory_records (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      record_type TEXT NOT NULL,
      purchase_request_id INTEGER,
      project_id INTEGER,
      product_id INTEGER,
      quantity REAL NOT NULL,
      unit_price REAL,
      total_amount REAL,
      record_date TEXT,
      operator TEXT,
      remark TEXT,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )`
  ];

  let index = 0;
  
  function createNextTable() {
    if (index >= tables.length) {
      console.log('所有表创建成功');
      insertTestData();
      return;
    }
    
    const sql = tables[index];
    db.run(sql, (err) => {
      if (err) {
        console.error(`创建表 ${index + 1} 失败:`, err.message);
      }
      index++;
      createNextTable();
    });
  }

  createNextTable();
}

// 插入测试数据
function insertTestData() {
  // 检查是否已有用户数据
  db.get('SELECT COUNT(*) as count FROM users', (err, row) => {
    if (err) {
      console.error('查询用户数据失败:', err.message);
      return;
    }

    if (row.count === 0) {
      const users = [
        ['admin', 'X123c321@', '系统管理员', 'admin'],
        ['finance', 'X123c321@', '财务专员', 'finance'],
        ['manager', 'X123c321@', '项目经理', 'manager'],
        ['employee', 'X123c321@', '普通员工', 'employee']
      ];

      users.forEach(user => {
        db.run(
          'INSERT INTO users (username, password, name, role) VALUES (?, ?, ?, ?)',
          user,
          (err) => {
            if (err) {
              console.error('插入用户数据失败:', err.message);
            }
          }
        );
      });
    }
  });

  db.get('SELECT COUNT(*) as count FROM categories', (err, row) => {
    if (err) {
      console.error('查询分类数据失败:', err.message);
      return;
    }

    if (row.count === 0) {
      const categories = [
        ['电线电缆', null],
        ['高压绝缘线', 1],
        ['低压电缆', 1],
        ['钢绞线', 1],
        ['绝缘子', null],
        ['陶瓷绝缘子', 5],
        ['复合绝缘子', 5],
        ['金具', null],
        ['线夹', 8],
        ['间隔棒', 8]
      ];

      categories.forEach(category => {
        db.run(
          'INSERT INTO categories (name, parent_id) VALUES (?, ?)',
          category,
          (err) => {
            if (err) {
              console.error('插入分类数据失败:', err.message);
            }
          }
        );
      });
    }
  });

  db.get('SELECT COUNT(*) as count FROM products', (err, row) => {
    if (err) {
      console.error('查询商品数据失败:', err.message);
      return;
    }

    if (row.count === 0) {
      const products = [
        ['JKLYJ-35-22kV', 2, '米', 15.5, '高压绝缘线'],
        ['JKLYJ-50-22kV', 2, '米', 18.8, '高压绝缘线'],
        ['VV-3x25+1x16', 3, '米', 22.5, '低压电缆'],
        ['GJ-35', 4, '米', 8.2, '钢绞线'],
        ['XP-70', 6, '个', 25.0, '陶瓷绝缘子'],
        ['FXBW-10/70', 7, '个', 85.0, '复合绝缘子'],
        ['NLL-1', 9, '个', 12.5, '线夹'],
        ['JGX-35', 9, '个', 18.0, '线夹'],
        ['FJB-2', 10, '个', 22.0, '间隔棒']
      ];

      products.forEach(product => {
        db.run(
          'INSERT INTO products (name, category_id, unit, price, description) VALUES (?, ?, ?, ?, ?)',
          product,
          (err) => {
            if (err) {
              console.error('插入商品数据失败:', err.message);
            }
          }
        );
      });
    }
  });

  db.get('SELECT COUNT(*) as count FROM exchange_rates', (err, row) => {
    if (err) {
      console.error('查询汇率数据失败:', err.message);
      return;
    }

    if (row.count === 0) {
      const rates = [
        ['CNY_LAK', 2900, '2024-01-01'],
        ['CNY_USD', 0.143, '2024-01-01'],
        ['CNY_THB', 4.8, '2024-01-01'],
        ['USD_LAK', 20300, '2024-01-01'],
        ['THB_LAK', 604, '2024-01-01']
      ];

      rates.forEach(rate => {
        db.run(
          'INSERT INTO exchange_rates (pair_key, rate, effective_date) VALUES (?, ?, ?)',
          rate,
          (err) => {
            if (err) {
              console.error('插入汇率数据失败:', err.message);
            }
          }
        );
      });
    }
  });

  db.get('SELECT COUNT(*) as count FROM inventory', (err, row) => {
    if (err) {
      console.error('查询库存数据失败:', err.message);
      return;
    }

    if (row.count === 0) {
      const inventoryItems = [
        [1, 'JKLYJ-35-22kV', 1000, '米', 15.5, 15500, '仓库A', 'in_stock', '2024-01-01'],
        [2, 'JKLYJ-50-22kV', 800, '米', 18.8, 15040, '仓库A', 'in_stock', '2024-01-01'],
        [3, 'VV-3x25+1x16', 500, '米', 22.5, 11250, '仓库B', 'in_stock', '2024-01-01'],
        [4, 'GJ-35', 1200, '米', 8.2, 9840, '仓库B', 'in_stock', '2024-01-01'],
        [5, 'XP-70', 200, '个', 25.0, 5000, '仓库C', 'in_stock', '2024-01-01']
      ];

      inventoryItems.forEach(item => {
        db.run(
          'INSERT INTO inventory (product_id, product_name, quantity, unit, price, total_value, location, status, last_updated) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)',
          item,
          (err) => {
            if (err) {
              console.error('插入库存数据失败:', err.message);
            }
          }
        );
      });
    }
  });
}

// 为sqlite3.Database添加query方法，使其与PostgreSQL的接口兼容
db.query = function(text, params) {
  return new Promise((resolve, reject) => {
    if (text.trim().startsWith('SELECT')) {
      db.all(text, params, (err, rows) => {
        if (err) {
          reject(err);
        } else {
          resolve({ rows });
        }
      });
    } else {
      db.run(text, params, function(err) {
        if (err) {
          reject(err);
        } else {
          resolve({ rows: [], lastID: this.lastID, changes: this.changes });
        }
      });
    }
  });
};

// 导出数据库连接
module.exports = db;