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

const conn = new Client();

conn.on('ready', () => {
  console.log('=== 修改后端代码 ===\n');
  
  conn.exec('cat /www/huishou/server/server.js | head -35', (err, stream) => {
    if (err) {
      console.log('读取失败:', err.message);
      conn.end();
      return;
    }
    
    let content = '';
    stream.on('data', (data) => {
      content += data.toString();
    });
    
    stream.on('close', () => {
      console.log('当前代码:', content);
      
      const newRootHandler = `app.get('/', (req, res) => {
  const h5Index = path.join(__dirname, '../dist/build/h5/index.html')
  res.sendFile(h5Index)
})`;
      
      conn.exec('sed -i \'20,31c\\' + newRootHandler.replace(/\n/g, '\\n') + '\\' /www/huishou/server/server.js', (err, stream2) => {
        if (err) {
          console.log('修改失败:', err.message);
          
          conn.exec('cat > /tmp/server-fix.js << "ENDOFFILE"\n' + newRootHandler + '\nENDOFFILE', (err, stream3) => {
            if (err) {
              console.log('创建临时文件失败:', err.message);
            } else {
              console.log('临时文件创建成功');
            }
            
            stream3.on('close', () => {
              conn.end();
            });
          });
          return;
        }
        
        stream2.on('close', () => {
          console.log('代码已修改');
          
          conn.exec('cat /www/huishou/server/server.js | head -35', (err, stream3) => {
            if (err) {
              console.log('验证失败:', err.message);
            } else {
              stream3.on('data', (data3) => {
                console.log('\n修改后:', data3.toString());
              });
            }
            
            stream3.on('close', () => {
              conn.exec('pkill -f "node server/server.js" && sleep 2 && cd /www/huishou && node server/server.js &', (err, stream4) => {
                if (err) {
                  console.log('重启失败:', err.message);
                } else {
                  console.log('服务器已重启');
                }
                
                stream4.on('close', () => {
                  setTimeout(() => {
                    conn.exec('curl -s -I https://hs.yifenbao.cn/', (err, stream5) => {
                      if (err) {
                        console.log('测试失败:', err.message);
                      } else {
                        stream5.on('data', (data5) => {
                          console.log('\n根路径响应头:', data5.toString());
                        });
                      }
                      
                      stream5.on('close', () => {
                        conn.end();
                      });
                    });
                  }, 3000);
                });
              });
            });
          });
        });
      });
    });
  });
});

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

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