server.js

require("dotenv").config();

const path = require("path");
const express = require("express");
const cors = require("cors");

const app = express();
app.use(cors());
app.use(express.json());
app.use(express.static(path.join(__dirname, "public")));

const SYSTEM_PROMPT = `你是一个"世界规则推演引擎"。你的职责是根据当前世界状态和可能的神谕（玩家指令），推演出下一个时间步的世界状态。

规则：
1. 你会收到一个 JSON 对象，表示当前世界状态。
2. 如果玩家的神谕（user_prompt）为空，你需要推演自然时间流逝带来的变化：资源自然增减、人口自然增长或衰减、环境自然恢复或恶化等。
3. 如果玩家的神谕（user_prompt）不为空，你需要优先结算神谕带来的影响，然后再叠加自然演变。
4. 你可以修改现有属性的值，也可以根据推演逻辑大胆创造新的属性。
5. 所有数值变化应当合理，符合因果逻辑。

强制约束：
- 你必须且只能输出一个合法的 JSON 对象，表示推演后的新世界状态。
- 输出的 JSON 必须是嵌套格式：第一层 Key 为大类名称（如 "自然生态"、"人类社会"、"科技文明" 等），第二层为具体属性和数值。
- 示例格式：{"自然生态":{"树木":128,"动物":65},"人类社会":{"人口":25,"食物":80}}
- 你可以自由创造新的大类和属性。
- 不要输出任何解释、注释或多余文字，只输出纯 JSON。`;

function extractJSON(text) {
  const fenced = /```json([\s\S]*?)```/.exec(text);
  if (fenced) return JSON.parse(fenced[1].trim());

  const braces = /\{[\s\S]*\}/.exec(text);
  if (braces) return JSON.parse(braces[0]);

  return JSON.parse(text);
}

function normalizeApiUrl(rawUrl) {
  const url = rawUrl.trim().replace(/\/+$/, "");
  try {
    new URL(url);
  } catch {
    throw new Error("Invalid API URL format");
  }
  if (url.endsWith("/v1/chat/completions")) return url;
  if (url.endsWith("/v1")) return url + "/chat/completions";
  return url + "/v1/chat/completions";
}

app.post("/evolve", async (req, res) => {
  const body = req.body;
  const apiUrl = (body.apiUrl && body.apiUrl.trim()) || process.env.DEFAULT_API_URL || "";
  const apiKey = (body.apiKey && body.apiKey.trim()) || process.env.DEFAULT_API_KEY || "";
  const model  = (body.model  && body.model.trim())  || process.env.DEFAULT_MODEL   || "";
  const { current_state, user_prompt, temperature } = body;

  if (!apiUrl || !apiKey || !model || !current_state) {
    return res.status(400).json({ error: "Missing required fields: apiUrl, apiKey, model, current_state" });
  }

  let endpoint;
  try {
    endpoint = normalizeApiUrl(apiUrl);
  } catch {
    return res.status(400).json({ error: "Invalid API URL: please check the URL format" });
  }

  const userMessage =
    (user_prompt ? `神谕：${user_prompt}\n` : "") +
    `当前世界状态：\n${JSON.stringify(current_state, null, 2)}`;

  const payload = {
    model,
    messages: [
      { role: "system", content: SYSTEM_PROMPT },
      { role: "user", content: userMessage },
    ],
    response_format: { type: "json_object" },
  };

  if (temperature !== undefined && temperature !== null) {
    payload.temperature = Number(temperature);
  }

  try {
    const response = await fetch(endpoint, {
      method: "POST",
      headers: {
        "Content-Type": "application/json",
        Authorization: `Bearer ${apiKey}`,
      },
      body: JSON.stringify(payload),
    });

    if (!response.ok) {
      const text = await response.text();
      return res.status(response.status).json({ error: `LLM API error: ${text}` });
    }

    const data = await response.json();
    const content = data.choices?.[0]?.message?.content;

    if (!content) {
      return res.status(502).json({ error: "No content in LLM response" });
    }

    let newState;
    try {
      newState = extractJSON(content);
    } catch {
      return res.status(502).json({ error: "LLM returned invalid JSON", raw: content });
    }
    return res.json(newState);
  } catch (err) {
    return res.status(500).json({ error: err.message });
  }
});

if (require.main === module) {
  const PORT = process.env.PORT || 3000;
  app.listen(PORT, () => {
    console.log(`Server running on http://localhost:${PORT}`);
  });
}

module.exports = app;
module.exports.extractJSON = extractJSON;
module.exports.normalizeApiUrl = normalizeApiUrl;