const db = require('./db-sqlite');

async function resetData() {
  try {
    console.log('开始重置数据...');
    
    // 1. 清空项目管理里的数据
    console.log('清空项目相关数据...');
    
    // 删除质保金数据
    await db.query('DELETE FROM warranty_deposits');
    console.log('已删除质保金数据');
    
    // 删除项目财务信息数据
    await db.query('DELETE FROM project_finances');
    console.log('已删除项目财务信息数据');
    
    // 删除施工节点数据
    await db.query('DELETE FROM project_milestones');
    console.log('已删除施工节点数据');
    
    // 删除项目材料数据
    await db.query('DELETE FROM project_materials');
    console.log('已删除项目材料数据');
    
    // 删除分包合同数据
    await db.query('DELETE FROM subcontracts');
    console.log('已删除分包合同数据');
    
    // 删除项目合同数据
    await db.query('DELETE FROM project_contracts');
    console.log('已删除项目合同数据');
    
    // 删除项目数据
    await db.query('DELETE FROM projects');
    console.log('已删除项目数据');
    
    // 2. 修改预算项目的状态
    console.log('修改预算项目状态...');
    
    // 把所有预算项目状态改为商谈中
    await db.query('UPDATE budget_projects SET status = ?', ['negotiating']);
    console.log('已将所有预算项目状态改为商谈中');
    
    // 3. 检查预算项目列表
    const budgetProjectsResult = await db.query('SELECT * FROM budget_projects');
    console.log('\n预算项目列表:');
    budgetProjectsResult.rows.forEach(project => {
      console.log(`ID: ${project.id}, 名称: ${project.name}, 状态: ${project.status}`);
    });
    
    console.log('\n✅ 数据重置完成!');
    process.exit(0);
    
  } catch (error) {
    console.error('重置数据失败:', error);
    process.exit(1);
  }
}

// 执行重置
resetData();