const { Client } = require('ssh2');
const fs = require('fs');

const conn = new Client();

conn.on('ready', () => {
  console.log('=== 上传server.js ===');
  
  conn.sftp((err, sftp) => {
    if (err) { console.log('SFTP ERR:', err); conn.end(); return; }
    
    const serverContent = fs.readFileSync('server/server.js', 'utf-8');
    const writeStream = sftp.createWriteStream('/www/huishou/server/server.js');
    writeStream.write(serverContent);
    writeStream.end();
    
    writeStream.on('close', () => {
      console.log('server.js上传成功');
      sftp.end();
      
      console.log('\n=== 重启后端容器 ===');
      conn.exec('cd /www/huishou && docker-compose restart backend', (err2, stream2) => {
        if (err2) { console.log('ERR2:', err2); conn.end(); return; }
        stream2.on('data', (d) => console.log(d.toString()));
        stream2.on('close', () => {
          setTimeout(() => {
            console.log('\n=== 测试用户API ===');
            conn.exec('curl -s -X POST http://127.0.0.1:4000/api/users -H "Content-Type: application/json" -d \'{"name":"测试用户","phone":"13800138000","role":"collector"}\'', (err3, stream3) => {
              if (err3) { console.log('ERR3:', err3); conn.end(); return; }
              let data = '';
              stream3.on('data', (d) => data += d.toString());
              stream3.on('close', () => {
                console.log('创建用户结果:', data);
                
                const result = JSON.parse(data);
                if (result.success) {
                  console.log('\n✅ 用户管理API测试成功！');
                  
                  console.log('\n=== 删除测试用户 ===');
                  conn.exec(`curl -s -X DELETE http://127.0.0.1:4000/api/users/${result.data.id}`, (err4, stream4) => {
                    if (err4) { console.log('ERR4:', err4); conn.end(); return; }
                    let delData = '';
                    stream4.on('data', (d) => delData += d.toString());
                    stream4.on('close', () => {
                      console.log('删除结果:', delData);
                      conn.end();
                    });
                  });
                } else {
                  console.log('\n⚠️ 用户创建失败:', result.message);
                  conn.end();
                }
              });
            });
          }, 5000);
        });
      });
    });
  });
});

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