api.ts 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. import { OpenAPIHono } from '@hono/zod-openapi'
  2. import { errorHandler } from './utils/errorHandler'
  3. import usersRouter from './api/users/index'
  4. import authRoute from './api/auth/index'
  5. import rolesRoute from './api/roles/index'
  6. // 新实体路由导入
  7. import areaRoutes from './api/areas/index'
  8. import clientRoutes from './api/clients/index'
  9. import expenseRoutes from './api/expenses/index'
  10. import fileRoutes from './api/files/index'
  11. import hetongRoutes from './api/contracts/index'
  12. import hetongRenewRoutes from './api/contracts-renew/index'
  13. import linkmanRoutes from './api/contacts/index'
  14. import logfileRoutes from './api/logs/index'
  15. import orderRecordRoutes from './api/order-records/index'
  16. import followUpRecordRoutes from './api/follow-up-records/index'
  17. import departmentsRoute from './api/departments/index'
  18. import userDepartmentsRoute from './api/user-departments/index'
  19. import permissionRoutes from './api/permissions/index'
  20. import rolePermissionRoutes from './api/role-permissions/index'
  21. import { AuthContext } from './types/context'
  22. import { AppDataSource } from './data-source'
  23. const api = new OpenAPIHono<AuthContext>()
  24. api.onError(errorHandler)
  25. // Rate limiting
  26. api.use('/api/v1/*', async (c, next) => {
  27. const ip = c.req.header('x-forwarded-for') || c.req.header('cf-connecting-ip')
  28. // 实现速率限制逻辑
  29. await next()
  30. })
  31. // 数据库初始化中间件
  32. api.use('/api/v1/*', async (c, next) => {
  33. if(!AppDataSource.isInitialized) {
  34. await AppDataSource.initialize();
  35. }
  36. await next();
  37. })
  38. // 注册Bearer认证方案
  39. api.openAPIRegistry.registerComponent('securitySchemes','bearerAuth',{
  40. type:'http',
  41. scheme:'bearer',
  42. bearerFormat:'JWT',
  43. description:'使用JWT进行认证'
  44. })
  45. // OpenAPI documentation endpoint
  46. if(!import.meta.env.PROD){
  47. api.doc31('/doc', {
  48. openapi: '3.1.0',
  49. info: {
  50. title: 'API Documentation',
  51. version: '1.0.0'
  52. },
  53. security: [{
  54. bearerAuth: []
  55. }]
  56. // servers: [{ url: '/api/v1' }]
  57. })
  58. }
  59. const userRoutes = api.route('/api/v1/users', usersRouter)
  60. const authRoutes = api.route('/api/v1/auth', authRoute)
  61. const roleRoutes = api.route('/api/v1/roles', rolesRoute)
  62. // 新实体路由注册
  63. const areaApiRoutes = api.route('/api/v1/areas', areaRoutes)
  64. const clientApiRoutes = api.route('/api/v1/clients', clientRoutes)
  65. const expenseApiRoutes = api.route('/api/v1/expenses', expenseRoutes)
  66. const fileApiRoutes = api.route('/api/v1/files', fileRoutes)
  67. const hetongApiRoutes = api.route('/api/v1/contracts', hetongRoutes)
  68. const hetongRenewApiRoutes = api.route('/api/v1/contracts-renew', hetongRenewRoutes)
  69. const linkmanApiRoutes = api.route('/api/v1/contacts', linkmanRoutes)
  70. const logfileApiRoutes = api.route('/api/v1/logs', logfileRoutes)
  71. const orderRecordApiRoutes = api.route('/api/v1/order-records', orderRecordRoutes)
  72. const followUpRecordApiRoutes = api.route('/api/v1/follow-up-records', followUpRecordRoutes)
  73. const departmentsApiRoutes = api.route('/api/v1/departments', departmentsRoute)
  74. const userDepartmentsApiRoutes = api.route('/api/v1/user-departments', userDepartmentsRoute)
  75. const permissionsApiRoutes = api.route('/api/v1/permissions', permissionRoutes)
  76. const rolePermissionsApiRoutes = api.route('/api/v1/role-permissions', rolePermissionRoutes)
  77. export type AuthRoutes = typeof authRoutes
  78. export type UserRoutes = typeof userRoutes
  79. export type RoleRoutes = typeof roleRoutes
  80. // 新实体路由类型导出
  81. export type AreaRoutes = typeof areaApiRoutes
  82. export type ClientRoutes = typeof clientApiRoutes
  83. export type ExpenseRoutes = typeof expenseApiRoutes
  84. export type FileRoutes = typeof fileApiRoutes
  85. export type HetongRoutes = typeof hetongApiRoutes
  86. export type HetongRenewRoutes = typeof hetongRenewApiRoutes
  87. export type LinkmanRoutes = typeof linkmanApiRoutes
  88. export type LogfileRoutes = typeof logfileApiRoutes
  89. export type OrderRecordRoutes = typeof orderRecordApiRoutes
  90. export type FollowUpRecordRoutes = typeof followUpRecordApiRoutes
  91. export type DepartmentRoutes = typeof departmentsApiRoutes
  92. export type UserDepartmentRoutes = typeof userDepartmentsApiRoutes
  93. export type PermissionRoutes = typeof permissionsApiRoutes
  94. export type RolePermissionRoutes = typeof rolePermissionsApiRoutes
  95. export default api