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

const config = {
  host: 'hs.yifenbao.cn',
  port: 22,
  username: 'root',
  password: 'Torch1112'
};

const conn = new Client();

conn.on('ready', () => {
  console.log('SSH连接成功');
  
  conn.sftp((err, sftp) => {
    if (err) {
      console.error('SFTP错误:', err.message);
      conn.end();
      return;
    }
    
    const localPath = './server/server.js';
    const remotePath = '/root/huishou/server/server.js';
    
    const readStream = fs.createReadStream(localPath);
    const writeStream = sftp.createWriteStream(remotePath);
    
    writeStream.on('close', () => {
      console.log('server.js上传完成');
      
      conn.exec('cd /root/huishou && docker-compose down', (err, stream) => {
        if (err) {
          console.error('执行命令错误:', err.message);
          conn.end();
          return;
        }
        
        stream.on('close', () => {
          console.log('容器停止成功');
          
          conn.exec('cd /root/huishou && docker-compose up -d', (err, stream) => {
            if (err) {
              console.error('执行命令错误:', err.message);
              conn.end();
              return;
            }
            
            stream.on('close', () => {
              console.log('容器启动成功');
              
              setTimeout(() => {
                conn.exec('docker ps | grep huishou', (err, stream) => {
                  if (err) {
                    console.error('执行命令错误:', err.message);
                    conn.end();
                    return;
                  }
                  
                  let output = '';
                  stream.stdout.on('data', (data) => {
                    output += data.toString();
                  });
                  
                  stream.on('close', () => {
                    console.log('容器状态:');
                    console.log(output);
                    conn.end();
                  });
                });
              }, 5000);
            });
          });
        });
      });
    });
    
    writeStream.on('error', (err) => {
      console.error('上传错误:', err.message);
      conn.end();
    });
    
    readStream.pipe(writeStream);
  });
});

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

conn.on('end', () => {
  console.log('SSH连接已关闭');
});

conn.connect(config);