import { OpenAPIHono } from '@hono/zod-openapi'
import { errorHandler } from './utils/errorHandler'
import usersRouter from './api/users/index'
import authRoute from './api/auth/index'
import rolesRoute from './api/roles/index'
// 新实体路由导入
import areaRoutes from './api/areas/index'
import clientRoutes from './api/clients/index'
import expenseRoutes from './api/expenses/index'
import fileRoutes from './api/files/index'
import hetongRoutes from './api/contracts/index'
import hetongRenewRoutes from './api/contracts-renew/index'
import linkmanRoutes from './api/contacts/index'
import logfileRoutes from './api/logs/index'
import orderRecordRoutes from './api/order-records/index'
import followUpRecordRoutes from './api/follow-up-records/index'
import departmentsRoute from './api/departments/index'
import userDepartmentsRoute from './api/user-departments/index'
import { AuthContext } from './types/context'
import { AppDataSource } from './data-source'

const api = new OpenAPIHono<AuthContext>()

api.onError(errorHandler)

// Rate limiting
api.use('/api/v1/*', async (c, next) => {
  const ip = c.req.header('x-forwarded-for') || c.req.header('cf-connecting-ip')
  // 实现速率限制逻辑
  await next()
})

// 数据库初始化中间件
api.use('/api/v1/*', async (c, next) => {
  if(!AppDataSource.isInitialized) {
    await AppDataSource.initialize();
  }
  await next();
})

// 注册Bearer认证方案
api.openAPIRegistry.registerComponent('securitySchemes','bearerAuth',{
  type:'http',
  scheme:'bearer',
  bearerFormat:'JWT',
  description:'使用JWT进行认证'
})

// OpenAPI documentation endpoint
if(!import.meta.env.PROD){
  api.doc31('/doc', {
    openapi: '3.1.0',
    info: {
      title: 'API Documentation',
      version: '1.0.0'
    },
    security: [{
      bearerAuth: []
    }]
    // servers: [{ url: '/api/v1' }]
  })
}



const userRoutes = api.route('/api/v1/users', usersRouter)
const authRoutes = api.route('/api/v1/auth', authRoute)
const roleRoutes = api.route('/api/v1/roles', rolesRoute)
// 新实体路由注册
const areaApiRoutes = api.route('/api/v1/areas', areaRoutes)
const clientApiRoutes = api.route('/api/v1/clients', clientRoutes)
const expenseApiRoutes = api.route('/api/v1/expenses', expenseRoutes)
const fileApiRoutes = api.route('/api/v1/files', fileRoutes)
const hetongApiRoutes = api.route('/api/v1/contracts', hetongRoutes)
const hetongRenewApiRoutes = api.route('/api/v1/contracts-renew', hetongRenewRoutes)
const linkmanApiRoutes = api.route('/api/v1/contacts', linkmanRoutes)
const logfileApiRoutes = api.route('/api/v1/logs', logfileRoutes)
const orderRecordApiRoutes = api.route('/api/v1/order-records', orderRecordRoutes)
const followUpRecordApiRoutes = api.route('/api/v1/follow-up-records', followUpRecordRoutes)
const departmentsApiRoutes = api.route('/api/v1/departments', departmentsRoute)
const userDepartmentsApiRoutes = api.route('/api/v1/user-departments', userDepartmentsRoute)

export type AuthRoutes = typeof authRoutes
export type UserRoutes = typeof userRoutes
export type RoleRoutes = typeof roleRoutes
// 新实体路由类型导出
export type AreaRoutes = typeof areaApiRoutes
export type ClientRoutes = typeof clientApiRoutes
export type ExpenseRoutes = typeof expenseApiRoutes
export type FileRoutes = typeof fileApiRoutes
export type HetongRoutes = typeof hetongApiRoutes
export type HetongRenewRoutes = typeof hetongRenewApiRoutes
export type LinkmanRoutes = typeof linkmanApiRoutes
export type LogfileRoutes = typeof logfileApiRoutes
export type OrderRecordRoutes = typeof orderRecordApiRoutes
export type FollowUpRecordRoutes = typeof followUpRecordApiRoutes
export type DepartmentRoutes = typeof departmentsApiRoutes
export type UserDepartmentRoutes = typeof userDepartmentsApiRoutes

export default api