api.ts 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  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 classroomDataRoutes from './api/classroom-data/index'
  7. import submissionRecordsRoutes from './api/submission-records/index'
  8. import stockDataRoutes from './api/stock-data/index'
  9. import stockXunlianCodesRoutes from './api/stock-xunlian-codes/index'
  10. import dateNotesRoutes from './api/date-notes/index'
  11. import aliyunRoute from './api/aliyun/index'
  12. import { AuthContext } from './types/context'
  13. import { AppDataSource } from './data-source'
  14. const api = new OpenAPIHono<AuthContext>()
  15. api.onError(errorHandler)
  16. // Rate limiting
  17. api.use('/api/v1/*', async (c, next) => {
  18. const ip = c.req.header('x-forwarded-for') || c.req.header('cf-connecting-ip')
  19. // 实现速率限制逻辑
  20. await next()
  21. })
  22. // 数据库初始化中间件
  23. api.use('/api/v1/*', async (c, next) => {
  24. if(!AppDataSource.isInitialized) {
  25. await AppDataSource.initialize();
  26. }
  27. await next();
  28. })
  29. // 注册Bearer认证方案
  30. api.openAPIRegistry.registerComponent('securitySchemes','bearerAuth',{
  31. type:'http',
  32. scheme:'bearer',
  33. bearerFormat:'JWT',
  34. description:'使用JWT进行认证'
  35. })
  36. // OpenAPI documentation endpoint
  37. if(!import.meta.env.PROD){
  38. api.doc31('/doc', {
  39. openapi: '3.1.0',
  40. info: {
  41. title: 'API Documentation',
  42. version: '1.0.0'
  43. },
  44. security: [{
  45. bearerAuth: []
  46. }]
  47. // servers: [{ url: '/api/v1' }]
  48. })
  49. }
  50. const userRoutes = api.route('/api/v1/users', usersRouter)
  51. const authRoutes = api.route('/api/v1/auth', authRoute)
  52. const roleRoutes = api.route('/api/v1/roles', rolesRoute)
  53. const classroomDataApi = api.route('/api/v1/classroom-data', classroomDataRoutes)
  54. const submissionRecordsApi = api.route('/api/v1/submission-records', submissionRecordsRoutes)
  55. const stockDataApi = api.route('/api/v1/stock-data', stockDataRoutes)
  56. const stockXunlianCodesApi = api.route('/api/v1/stock-xunlian-codes', stockXunlianCodesRoutes)
  57. const dateNotesApi = api.route('/api/v1/date-notes', dateNotesRoutes)
  58. const aliyunApi = api.route('/api/v1/aliyun', aliyunRoute)
  59. export type AuthRoutes = typeof authRoutes
  60. export type UserRoutes = typeof userRoutes
  61. export type RoleRoutes = typeof roleRoutes
  62. export type ClassroomDataRoutes = typeof classroomDataApi
  63. export type SubmissionRecordsRoutes = typeof submissionRecordsApi
  64. export type StockDataRoutes = typeof stockDataApi
  65. export type StockXunlianCodesRoutes = typeof stockXunlianCodesApi
  66. export type DateNotesRoutes = typeof dateNotesApi
  67. export type AliyunRoutes = typeof aliyunApi
  68. export default api