// MongoDB 连接 URL
const uri = "mongodb+srv://<username>:<password>.@<endpoint>/?retryWrites=true&w=majority&appName=Azure-Cluster-HongKong";

// 创建 MongoDB 客户端
const client = new MongoClient(uri);

// 连接数据库
let db;
let isConnected = false; // 添加连接状态标志

// 添加根路由处理
app.get("/", (req, res) => {
  res.json({ message: "API 服务器正在运行" });
});

async function connectToDatabase() {
  try {
    await client.connect();
    db = client.db("kyle_blog_secret");
    isConnected = true; // 设置连接状态
    console.log("成功连接到 MongoDB");
  } catch (error) {
    console.error("MongoDB 连接失败:", error);
    throw error;
  }
}

// 添加中间件来检查数据库连接
app.use(async (req, res, next) => {
  if (!isConnected) {
    try {
      await connectToDatabase();
    } catch (error) {
      return res.status(500).json({ message: "数据库连接失败", error: error.message });
    }
  }
  next();
});

// 修改查询接口
app.get("/query", async (req, res) => {
  try {
    if (!db) {
      throw new Error("数据库连接未就绪");
    }
    const { item } = req.query;
    const filter = item ? { item } : {};

    const result = await db.collection("kyle").find(filter).toArray();
    res.json(result);
  } catch (error) {
    console.error("查询失败:", error);
    res.status(500).json({ message: "查询失败", error: error.message });
  }
});


// 心跳接口
app.get("/health", (req, res) => {
  res.json({ status: "ok" });
});

// 错误处理中间件
app.use((err, req, res, next) => {
  console.error(err.stack);
  res.status(500).json({ message: "服务器内部错误", error: err.message });
});

// 启动服务器
if (process.env.NODE_ENV !== 'production') {
  app.listen(port, () => {
    console.log(`服务器运行在端口 ${port}`);
  });
}

module.exports = app;