import { Client } from 'ssh2';

const conn = new Client();

conn.on('ready', () => {
  console.log('=== 修复现有订单的collectorId字段 ===\n');
  
  console.log('1. 获取所有订单:');
  conn.exec('curl -s http://localhost:4000/api/orders', (err, stream) => {
    if (err) { console.log('ERR:', err); conn.end(); return; }
    let output = '';
    stream.on('data', (d) => output += d.toString());
    stream.on('close', () => {
      try {
        const result = JSON.parse(output);
        const orders = result.data || [];
        console.log(`总订单数: ${orders.length}`);
        
        const ordersWithCollector = orders.filter(o => o.collector);
        console.log(`有collector字段的订单数: ${ordersWithCollector.length}`);
        
        ordersWithCollector.forEach(order => {
          console.log(`  - 订单${order.id}: collector=${order.collector}, collectorId=${order.collectorId || '空'}`);
        });
        
        console.log('\n2. 获取用户列表:');
        conn.exec('curl -s http://localhost:4000/api/users', (err, stream2) => {
          if (err) { console.log('ERR:', err); conn.end(); return; }
          let output2 = '';
          stream2.on('data', (d) => output2 += d.toString());
          stream2.on('close', () => {
            try {
              const userResult = JSON.parse(output2);
              const users = userResult.data || [];
              
              const updatePromises = [];
              ordersWithCollector.forEach(order => {
                if (!order.collectorId) {
                  const user = users.find(u => u.name === order.collector);
                  if (user) {
                    console.log(`  更新订单${order.id}: collector=${order.collector} -> collectorId=${user.id}`);
                    updatePromises.push(new Promise((resolve) => {
                      conn.exec(`curl -s -X PUT "http://localhost:4000/api/orders/${order.id}" -H "Content-Type: application/json" -d '{"collectorId":"${user.id}"}'`, (err, stream3) => {
                        if (err) { console.log('ERR:', err); resolve(); return; }
                        let output3 = '';
                        stream3.on('data', (d) => output3 += d.toString());
                        stream3.on('close', () => {
                          console.log(`    订单${order.id}更新结果: ${output3}`);
                          resolve();
                        });
                      });
                    }));
                  } else {
                    console.log(`  订单${order.id}: collector=${order.collector} 未找到匹配用户`);
                  }
                }
              });
              
              Promise.all(updatePromises).then(() => {
                console.log('\n3. 验证修复结果:');
                conn.exec(`curl -s "http://localhost:4000/api/orders?collector=3"`, (err, stream4) => {
                  if (err) { console.log('ERR:', err); conn.end(); return; }
                  let output4 = '';
                  stream4.on('data', (d) => output4 += d.toString());
                  stream4.on('close', () => {
                    try {
                      const result4 = JSON.parse(output4);
                      console.log(`宁赜的订单数: ${result4.data?.length || 0}`);
                    } catch (e) {
                      console.log('输出:', output4);
                    }
                    conn.end();
                  });
                });
              });
            } catch (e) {
              console.log('解析错误:', e.message);
              conn.end();
            }
          });
        });
      } catch (e) {
        console.log('解析错误:', e.message);
        conn.end();
      }
    });
  });
});

conn.on('error', (err) => {
  console.log('连接错误:', err.message);
});

conn.connect({
  host: 'hs.yifenbao.cn',
  port: 22,
  username: 'root',
  password: 'Torch1112'
});