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

const conn = new Client();

conn.on('ready', () => {
  console.log('=== 部署后端代码 ===');
  
  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: 重启后端容器');
        
        conn.exec('docker restart huishou-backend', (err, stream) => {
          if (err) {
            console.log('重启容器失败:', err);
            conn.end();
            return;
          }
          
          let output = '';
          stream.on('data', (d) => output += d.toString());
          stream.on('close', (code) => {
            console.log('容器重启完成, 退出码:', code);
            console.log('输出:', output);
            
            setTimeout(() => {
              console.log('Step 3: 检查容器状态');
              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);
          });
        });
      });
      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'
});