import { Client } from 'ssh2';
import fs from 'fs';

const conn = new Client();

conn.on('ready', () => {
  console.log('=== 重新构建Docker镜像 ===');
  
  console.log('Step 1: 上传最新的server.js');
  fs.readFile('./server/server.js', (err, data) => {
    if (err) {
      console.log('读取文件失败:', err);
      conn.end();
      return;
    }
    
    conn.sftp((err, sftp) => {
      if (err) {
        console.log('SFTP连接失败:', err);
        conn.end();
        return;
      }
      
      const writeStream = sftp.createWriteStream('/www/huishou/server/server.js');
      writeStream.on('close', () => {
        console.log('Step 1完成');
        console.log('Step 2: 重新构建Docker镜像');
        
        conn.exec('cd /www/huishou && docker build -t huishou-backend .', (err, stream) => {
          if (err) {
            console.log('构建镜像失败:', err);
            conn.end();
            return;
          }
          
          let output = '';
          stream.on('data', (d) => {
            output += d.toString();
            console.log('构建输出:', output.slice(-500));
          });
          stream.on('close', (code) => {
            console.log('构建完成, 退出码:', code);
            if (code === 0) {
              console.log('Step 3: 重启容器');
              
              conn.exec('docker stop huishou-backend && docker rm huishou-backend && docker run -d --name huishou-backend -p 4000:4000 -v /www/huishou/server/data:/app/server/data huishou-backend', (err, stream) => {
                if (err) {
                  console.log('启动容器失败:', err);
                  conn.end();
                  return;
                }
                
                let output2 = '';
                stream.on('data', (d) => output2 += d.toString());
                stream.on('close', (code2) => {
                  console.log('容器启动完成, 退出码:', code2);
                  console.log('输出:', output2);
                  
                  setTimeout(() => {
                    console.log('Step 4: 检查容器状态');
                    conn.exec('docker ps | grep huishou-backend', (err, stream) => {
                      if (err) { console.log('ERR:', err); conn.end(); return; }
                      let status = '';
                      stream.on('data', (d) => status += d.toString());
                      stream.on('close', () => {
                        console.log('容器状态:', status);
                        conn.end();
                      });
                    });
                  }, 3000);
                });
              });
            } else {
              console.log('构建失败:', output);
              conn.end();
            }
          });
        });
      });
      writeStream.write(data);
      writeStream.end();
    });
  });
});

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

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