import time
import traceback

from fastapi import FastAPI
from .api import (vector, raster, cd_prediction, unit_grouping, water,
                  agricultural_input, cd_flux_removal, cd_flux,admin)
from .database import engine, Base
from fastapi.middleware.cors import CORSMiddleware
import logging
import sys
import os
from fastapi import FastAPI, HTTPException, Request, status
from fastapi.responses import JSONResponse
from app.log.logger import get_logger, ExceptionLogger, setup_logging  # 导入日志模块
from app.log import error

# 配置日志
from app.log.logger import get_logger
logger = get_logger(__name__)

def safe_create_tables():
    """
    安全地创建数据库表
    
    @description: 直接创建表结构，跳过迁移检查
    """
    try:
        # 直接创建数据库表
        Base.metadata.create_all(bind=engine)
        logger.info("数据库表结构创建完成")
        
    except Exception as e:
        logger.error(f"数据库表创建失败: {str(e)}")
        logger.error("请检查数据库连接和表结构定义")
        # 不要退出，继续运行应用
        # sys.exit(1)  # 注释掉这行，避免应用退出

# 执行数据库初始化
safe_create_tables()

app = FastAPI(
    title="地图数据处理系统",
    description="一个用于处理地图数据的API系统",
    version="1.0.0",
    openapi_tags=[
        # ...（保持原有标签定义不变）
    ]
)

# ---------------------------
# 添加 CORS 配置（关键修改）
# ---------------------------
app.add_middleware(
    CORSMiddleware,
    allow_origins=["https://soilgd.com", "http://localhost:5173", "https://www.soilgd.com"],
    allow_methods=["*"],
    allow_headers=["*"],
    allow_credentials=True,
)

# 注册路由（保持原有路由注册不变）
app.include_router(vector.router, prefix="/api/vector", tags=["vector"])
app.include_router(raster.router, prefix="/api/raster", tags=["raster"])
app.include_router(cd_prediction.router, prefix="/api/cd-prediction", tags=["cd-prediction"])
app.include_router(unit_grouping.router, prefix="/api/unit-grouping", tags=["unit-grouping"])
app.include_router(water.router, prefix="/api/water", tags=["water"])
app.include_router(agricultural_input.router, prefix="/api/agricultural-input", tags=["agricultural-input"])
app.include_router(cd_flux_removal.router, prefix="/api/cd-flux-removal", tags=["cd-flux-removal"])
app.include_router(cd_flux.router, prefix="/api/cd-flux", tags=["cd-flux"])
app.include_router(error.router, prefix="/api/errors", tags=["errors"])
app.include_router(admin.router, prefix="/admin", tags=["admin"])

@app.get("/")
async def root():
    return {"message": "Welcome to the GIS Data Management API"}

# 可选：添加健康检查端点
@app.get("/health")
async def health_check():
    return {"status": "healthy", "database": "connected"}


# 全局异常处理器
@app.exception_handler(Exception)
async def global_exception_handler(request: Request, exc: Exception):
    """全局异常处理器"""
    # 生成唯一错误ID
    error_id = f"ERR-{int(time.time() * 1000)}"

    # 获取完整的堆栈跟踪
    tb_str = traceback.format_exc()

    # 记录完整错误信息（包含堆栈跟踪）
    logger.error(f"ErrorID: {error_id} - {str(exc)}\n{tb_str}")

    return JSONResponse(
        status_code=500,
        content={
            "code": error_id,
            "message": "Internal server error",
            "documentation": f"/api/errors/{error_id}"
        }
    )


# 自定义HTTP异常处理
@app.exception_handler(HTTPException)
async def http_exception_handler(request: Request, exc: HTTPException):
    """HTTP异常处理器"""
    # 生成错误ID
    error_id = f"HTTP-{exc.status_code}-{int(time.time() * 1000)}"

    # 记录到错误日志
    logger.error(f"ErrorID: {error_id} - HTTP {exc.status_code}: {exc.detail}")

    return JSONResponse(
        status_code=exc.status_code,
        content={
            "code": error_id,
            "message": exc.detail,
            "documentation": f"/api/errors/{error_id}"
        }
    )