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数据库连接成功');
    insertProjectData();
  }
});

// 插入测试项目数据
function insertProjectData() {
  console.log('开始插入测试项目数据...');
  
  // 插入测试项目
  const projects = [
    ['项目A', 'PROJ001', 1, 1, 100000, '2024-01-01', '2024-12-31', '这是第一个测试项目', '进行中', '北京市'],
    ['项目B', 'PROJ002', 2, 1, 200000, '2023-01-01', '2023-12-31', '这是第二个测试项目', '已完成', '上海市'],
    ['项目C', 'PROJ003', 3, 2, 150000, '2024-06-01', '2025-06-30', '这是第三个测试项目', '未开始', '广州市'],
    ['项目D', 'PROJ004', 4, 2, 300000, '2024-03-01', '2024-12-31', '这是第四个测试项目', '进行中', '深圳市'],
    ['项目E', 'PROJ005', 5, 3, 80000, '2023-06-01', '2023-12-31', '这是第五个测试项目', '已完成', '杭州市']
  ];

  projects.forEach(project => {
    db.run(
      'INSERT INTO projects (name, code, customer_id, manager_id, contract_amount, start_date, end_date, description, status, location) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)',
      project,
      (err) => {
        if (err) {
          console.error('插入项目数据失败:', err.message);
        } else {
          console.log('插入项目数据成功:', project[0]);
        }
      }
    );
  });
  
  console.log('测试项目数据插入完成');
}