huishou/upload-all.js

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;
    }
    
    const localPaths = [
      { local: 'dist/build/h5', remote: '/www/huishou/dist/build/h5' },
      { local: 'dist/build/mp-weixin', remote: '/www/huishou/dist/build/mp-weixin' },
      { local: 'server/server.js', remote: '/www/huishou/server/server.js' },
      { local: 'admin', remote: '/www/huishou/admin' }
    ];
    
    let uploadedCount = 0;
    
    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 new Promise((resolve) => {
            const readStream = fs.createReadStream(localItem);
            const writeStream = sftp.createWriteStream(remoteItem);
            
            writeStream.on('close', () => {
              console.log(`上传: ${item}`);
              resolve();
            });
            
            readStream.pipe(writeStream);
          });
        }
      }
    }
    
    async function processNext() {
      if (uploadedCount >= localPaths.length) {
        console.log('\n所有文件上传完成');
        
        conn.exec('cd /www/huishou && docker-compose up -d --build', (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('容器构建并启动完成');
            
            setTimeout(() => {
              conn.exec('docker ps | grep huishou', (err, stream) => {
                if (err) {
                  console.error('执行命令错误:', err.message);
                  conn.end();
                  return;
                }
                
                let output2 = '';
                stream.stdout.on('data', (data) => {
                  output2 += data.toString();
                });
                
                stream.on('close', () => {
                  console.log('容器状态:');
                  console.log(output2);
                  
                  conn.exec('curl -s http://localhost:4000/api/prices', (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('后端API测试:');
                      console.log(output3);
                      conn.end();
                    });
                  });
                });
              });
            }, 10000);
          });
        });
        
        return;
      }
      
      const current = localPaths[uploadedCount];
      console.log(`\n开始上传 ${current.local} -> ${current.remote}`);
      
      await uploadDir(current.local, current.remote);
      uploadedCount++;
      processNext();
    }
    
    processNext();
  });
});

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

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

conn.connect(config);