server.js

const express = require('express');
const multer = require('multer');
const { VM } = require('vm2');
const fs = require('fs');
const path = require('path');

const app = express();
app.use(express.static(path.join(__dirname, 'public')));
app.use(express.json());
const upload = multer({ dest: 'uploads/' });

// 确保上传目录存在
if (!fs.existsSync('uploads')) {
  fs.mkdirSync('uploads');
}

// 处理文件上传的API
app.post('/execute', upload.single('file'), (req, res) => {
  if (!req.file) {
    return res.status(400).json({ error: '请上传JS文件' });
  }

  try {
    // 读取上传的文件内容
    const code = fs.readFileSync(req.file.path, 'utf8');

    // 创建VM实例
    const vm = new VM({
      timeout: 5000, // 5秒超时
      sandbox: {}
    });

    // 执行代码并捕获输出
    const result = vm.run(code);

    // 删除临时文件
    fs.unlinkSync(req.file.path);

    res.json({ result });
  } catch (error) {
    // 确保清理临时文件
    if (req.file && fs.existsSync(req.file.path)) {
      fs.unlinkSync(req.file.path);
    }
    res.status(500).json({ error: error.message });
  }
});

// 处理代码字符串的API
app.post('/execute-code', (req, res) => {
  const { code } = req.body;
  
  if (!code) {
    return res.status(400).json({ error: '请提供JavaScript代码' });
  }

  try {
    // 创建VM实例
    const vm = new VM({
      timeout: 5000, // 5秒超时
      sandbox: {}
    });

    // 执行代码并捕获输出
    const result = vm.run(code);
    res.json({ result });
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
  console.log(`服务器运行在 http://localhost:${PORT}`);
});