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

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;
    }
    
    async function uploadFile(localPath, remotePath) {
      return new Promise((resolve) => {
        const readStream = fs.createReadStream(localPath);
        const writeStream = sftp.createWriteStream(remotePath);
        
        writeStream.on('close', () => {
          console.log(`上传: ${path.basename(localPath)}`);
          resolve();
        });
        
        readStream.pipe(writeStream);
      });
    }
    
    async function uploadDir(localDir, remoteDir) {
      const items = fs.readdirSync(localDir);
      
      for (const item of items) {
        const localItem = path.join(localDir, item);
        const remoteItem = remoteDir + '/' + item;
        const stat = fs.statSync(localItem);
        
        if (stat.isDirectory()) {
          await new Promise((resolve) => {
            sftp.mkdir(remoteItem, { recursive: true }, () => resolve());
          });
          await uploadDir(localItem, remoteItem);
        } else {
          await uploadFile(localItem, remoteItem);
        }
      }
    }
    
    async function run() {
      await uploadFile('./Dockerfile', '/www/huishou/Dockerfile');
      await uploadFile('./.dockerignore', '/www/huishou/.dockerignore');
      await uploadFile('./server/server.js', '/www/huishou/server/server.js');
      
      console.log('\n开始上传node_modules...');
      await uploadDir('./server/node_modules', '/www/huishou/server/node_modules');
      
      console.log('\n所有文件上传完成');
      
      conn.exec('cd /www/huishou && docker-compose down && docker rmi huishou-backend --force 2>&1', (err, stream) => {
        if (err) {
          console.error('执行命令错误:', err.message);
          conn.end();
          return;
        }
        
        let output = '';
        stream.stdout.on('data', (data) => {
          output += data.toString();
          process.stdout.write(data);
        });
        
        stream.stderr.on('data', (data) => {
          output += data.toString();
          process.stdout.write(data);
        });
        
        stream.on('close', () => {
          console.log('\n清理完成');
          
          conn.exec('cd /www/huishou && docker-compose up -d --build 2>&1', (err, stream) => {
            if (err) {
              console.error('执行命令错误:', err.message);
              conn.end();
              return;
            }
            
            let output2 = '';
            stream.stdout.on('data', (data) => {
              output2 += data.toString();
              process.stdout.write(data);
            });
            
            stream.stderr.on('data', (data) => {
              output2 += data.toString();
              process.stdout.write(data);
            });
            
            stream.on('close', () => {
              console.log('\n构建完成');
              
              setTimeout(() => {
                conn.exec('docker ps | grep huishou', (err, stream) => {
                  if (err) {
                    console.error('执行命令错误:', err.message);
                    conn.end();
                    return;
                  }
                  
                  let output3 = '';
                  stream.stdout.on('data', (data) => {
                    output3 += data.toString();
                  });
                  
                  stream.on('close', () => {
                    console.log('\n容器状态:');
                    console.log(output3);
                    
                    conn.exec('curl -s http://localhost:4000/api/prices 2>&1', (err, stream) => {
                      if (err) {
                        console.error('执行命令错误:', err.message);
                        conn.end();
                        return;
                      }
                      
                      let output4 = '';
                      stream.stdout.on('data', (data) => {
                        output4 += data.toString();
                      });
                      
                      stream.stderr.on('data', (data) => {
                        output4 += data.toString();
                      });
                      
                      stream.on('close', () => {
                        console.log('\n后端API测试:');
                        console.log(output4);
                        conn.end();
                      });
                    });
                  });
                });
              }, 15000);
            });
          });
        });
      });
    }
    
    run();
  });
});

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

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

conn.connect(config);