api.ts 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  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. import customerRoutes from './api/customers/index'
  7. import opportunityRoutes from './api/opportunities/index'
  8. import followUpRoutes from './api/follow-ups/index'
  9. import { AuthContext } from './types/context'
  10. import { AppDataSource } from './data-source'
  11. const api = new OpenAPIHono<AuthContext>()
  12. api.onError(errorHandler)
  13. // Rate limiting
  14. api.use('/api/v1/*', async (c, next) => {
  15. const ip = c.req.header('x-forwarded-for') || c.req.header('cf-connecting-ip')
  16. // 实现速率限制逻辑
  17. await next()
  18. })
  19. // 数据库初始化中间件
  20. api.use('/api/v1/*', async (c, next) => {
  21. if(!AppDataSource.isInitialized) {
  22. await AppDataSource.initialize();
  23. }
  24. await next();
  25. })
  26. // 注册Bearer认证方案
  27. api.openAPIRegistry.registerComponent('securitySchemes','bearerAuth',{
  28. type:'http',
  29. scheme:'bearer',
  30. bearerFormat:'JWT',
  31. description:'使用JWT进行认证'
  32. })
  33. // OpenAPI documentation endpoint
  34. if(!import.meta.env.PROD){
  35. api.doc31('/doc', {
  36. openapi: '3.1.0',
  37. info: {
  38. title: 'API Documentation',
  39. version: '1.0.0'
  40. },
  41. security: [{
  42. bearerAuth: []
  43. }]
  44. // servers: [{ url: '/api/v1' }]
  45. })
  46. }
  47. const userRoutes = api.route('/api/v1/users', usersRouter)
  48. const authRoutes = api.route('/api/v1/auth', authRoute)
  49. const roleRoutes = api.route('/api/v1/roles', rolesRoute)
  50. const customerApiRoutes = api.route('/api/v1/customers', customerRoutes)
  51. const opportunityApiRoutes = api.route('/api/v1/opportunities', opportunityRoutes)
  52. const followUpApiRoutes = api.route('/api/v1/follow-ups', followUpRoutes)
  53. export type AuthRoutes = typeof authRoutes
  54. export type UserRoutes = typeof userRoutes
  55. export type RoleRoutes = typeof roleRoutes
  56. export type CustomerRoutes = typeof customerApiRoutes
  57. export type OpportunityRoutes = typeof opportunityApiRoutes
  58. export type FollowUpRoutes = typeof followUpApiRoutes
  59. export default api