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.exec('mkdir -p /www/huishou/dist/build/h5/assets', (err, stream) => {
    stream.on('close', () => {
      console.log('目录创建完成');
      
      // 读取index.html
      const indexHtml = fs.readFileSync('./dist/build/h5/index.html', 'utf8');
      
      // 使用cat命令写入文件
      const escapedContent = indexHtml.replace(/'/g, "'\\''");
      conn.exec(`cat > /www/huishou/dist/build/h5/index.html << 'EOF'\n${indexHtml}\nEOF`, (err, stream) => {
        stream.on('close', () => {
          console.log('index.html已上传');
          
          // 上传首页JS文件
          const jsFiles = fs.readdirSync('./dist/build/h5/assets', { withFileTypes: true })
            .filter(f => f.isFile() && f.name.includes('pages-index-index'));
          
          jsFiles.forEach(jsFile => {
            const jsContent = fs.readFileSync(`./dist/build/h5/assets/${jsFile.name}`, 'utf8');
            const base64Content = Buffer.from(jsContent).toString('base64');
            
            console.log(`上传 ${jsFile.name} (${base64Content.length} bytes)`);
            
            conn.exec(`echo '${base64Content}' | base64 -d > /www/huishou/dist/build/h5/assets/${jsFile.name}`, (err, stream) => {
              if (err) {
                console.error(`上传失败: ${err.message}`);
              } else {
                stream.on('close', () => {
                  console.log(`✓ ${jsFile.name} 上传成功`);
                  
                  // 验证文件大小
                  conn.exec(`ls -la /www/huishou/dist/build/h5/assets/${jsFile.name}`, (err, stream) => {
                    let output = '';
                    stream.stdout.on('data', data => output += data);
                    stream.on('close', () => {
                      console.log('文件详情:', output);
                      
                      // 重启容器
                      conn.exec('docker restart huishou', (err, stream) => {
                        stream.on('close', () => {
                          console.log('容器已重启');
                          conn.end();
                        });
                      });
                    });
                  });
                });
              }
            });
          });
        });
      });
    });
  });
});

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

conn.connect(config);