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

const conn = new Client();

conn.on('ready', () => {
  console.log('=== 创建H5构建目录 ===');
  conn.exec('mkdir -p /www/huishou/dist/build/h5/static/icons', (err) => {
    if (err) { console.log('ERR:', err); conn.end(); return; }
    
    console.log('\n=== 上传H5构建文件 ===');
    const h5Dir = 'dist/build/h5';
    
    fs.readdir(h5Dir, { withFileTypes: true }, (err, entries) => {
      if (err) { console.log('READDIR ERR:', err); conn.end(); return; }
      
      let uploaded = 0;
      const total = entries.length;
      
      entries.forEach(entry => {
        const srcPath = path.join(h5Dir, entry.name);
        const destPath = `/www/huishou/${h5Dir}/${entry.name}`;
        
        if (entry.isFile()) {
          const content = fs.readFileSync(srcPath);
          conn.sftp((err, sftp) => {
            if (err) { console.log('SFTP ERR:', err); return; }
            const writeStream = sftp.createWriteStream(destPath);
            writeStream.write(content);
            writeStream.end();
            writeStream.on('close', () => {
              sftp.end();
              uploaded++;
              console.log(`已上传: ${entry.name}`);
              if (uploaded === total) finishUpload();
            });
          });
        } else {
          conn.exec(`mkdir -p ${destPath}`, (err) => {
            if (err) { console.log('MKDIR ERR:', err); return; }
            uploaded++;
            if (uploaded === total) finishUpload();
          });
        }
      });
    });
  });
  
  const finishUpload = () => {
    console.log('\n=== 上传图标文件 ===');
    const icons = ['paper.png', 'metal.png', 'plastic.png', 'fabric.png', 'glass.png'];
    let iconsUploaded = 0;
    
    icons.forEach(icon => {
      const srcPath = path.join('src', 'static', 'icons', icon);
      const destPath = `/www/huishou/dist/build/h5/static/icons/${icon}`;
      
      fs.readFile(srcPath, (err, content) => {
        if (err) { console.log('READ ERR:', err); return; }
        conn.sftp((err, sftp) => {
          if (err) { console.log('SFTP ERR:', err); return; }
          const writeStream = sftp.createWriteStream(destPath);
          writeStream.write(content);
          writeStream.end();
          writeStream.on('close', () => {
            sftp.end();
            iconsUploaded++;
            console.log(`已上传图标: ${icon}`);
            if (iconsUploaded === icons.length) {
              console.log('\n✅ 前端部署完成！');
              conn.end();
            }
          });
        });
      });
    });
  };
});

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