api.ts 1.7 KB

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