import 'dotenv/config';
import express from 'express';
import { proxyHandler } from './proxy.js';
import { openaiProxyHandler } from './proxy-openai.js';
import { healthHandler } from './health.js';

const app = express();
const PORT = parseInt(process.env.PORT || '80', 10);

// 解析 JSON 请求体，设置较大的限制以支持大请求
app.use(express.json({ limit: '50mb' }));

// 健康检查端点
app.get('/health', healthHandler);

// Claude API 代理路由: /claude/*
app.all('/claude/*', proxyHandler);

// OpenAI API 代理路由: /openai/*
app.all('/openai/*', openaiProxyHandler);

// 启动服务器
const server = app.listen(PORT, () => {
  console.log(`API Proxy 服务已启动`);
  console.log(`端口: ${PORT}`);
  console.log(`Claude API: ${process.env.ANTHROPIC_BASE_URL || 'https://api.anthropic.com'}`);
  console.log(`OpenAI API: ${process.env.OPENAI_BASE_URL || 'https://api.openai.com'}`);
  console.log(`健康检查: http://localhost:${PORT}/health`);
});

// 优雅关闭处理
let isShuttingDown = false;

function gracefulShutdown(signal: string) {
  if (isShuttingDown) return;
  isShuttingDown = true;

  console.log(`\n收到 ${signal} 信号，开始优雅关闭...`);

  // 停止接受新连接
  server.close(() => {
    console.log('服务器已关闭');
    process.exit(0);
  });

  // 强制退出超时（K8S 默认 30 秒宽限期）
  setTimeout(() => {
    console.log('优雅关闭超时，强制退出');
    process.exit(1);
  }, 25000);
}

process.on('SIGTERM', () => gracefulShutdown('SIGTERM'));
process.on('SIGINT', () => gracefulShutdown('SIGINT'));