api.ts 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. import { OpenAPIHono } from '@hono/zod-openapi'
  2. import { swaggerUI } from '@hono/swagger-ui'
  3. import { errorHandler } from './utils/errorHandler'
  4. import usersRouter from './api/users/index'
  5. import authRoute from './api/auth/index'
  6. import rolesRoute from './api/roles/index'
  7. import fileRoutes from './api/files/index'
  8. import { activitiesRoutes as adminActivitiesRoutes } from './api/admin/activities'
  9. import { routesRoutes as adminRoutesRoutes } from './api/admin/routes'
  10. import areasRoutes from './api/admin/areas'
  11. import locationsRoutes from './api/admin/locations'
  12. import { passengersRoutes as adminPassengersRoutes } from './api/admin/passengers'
  13. import passengersRoutes from './api/passengers/index'
  14. import routesRoutes from './api/routes'
  15. import areasUserRoutes from './api/areas'
  16. import locationsUserRoutes from './api/locations'
  17. import { AuthContext } from './types/context'
  18. import { AppDataSource } from './data-source'
  19. import { Hono } from 'hono'
  20. import { databaseBackup } from './utils/backup'
  21. if(!AppDataSource.isInitialized) {
  22. await AppDataSource.initialize();
  23. console.log('数据库初始化完成')
  24. // 启动数据库备份调度
  25. databaseBackup.startScheduledBackups();
  26. console.log('数据库备份调度已启动');
  27. }
  28. const app = new Hono();
  29. const api = new OpenAPIHono<AuthContext>()
  30. api.onError(errorHandler)
  31. // Rate limiting
  32. api.use('/api/v1/*', async (_, next) => {
  33. // 实现速率限制逻辑
  34. await next()
  35. })
  36. // // 数据库初始化中间件
  37. // api.use('/api/v1/*', async (c, next) => {
  38. // await next();
  39. // })
  40. // 注册Bearer认证方案
  41. api.openAPIRegistry.registerComponent('securitySchemes','bearerAuth',{
  42. type:'http',
  43. scheme:'bearer',
  44. bearerFormat:'JWT',
  45. description:'使用JWT进行认证'
  46. })
  47. // OpenAPI documentation endpoint
  48. // !import.meta.env.PROD
  49. if(1){
  50. // api.doc31('/doc', {
  51. // openapi: '3.1.0',
  52. // info: {
  53. // title: 'API Documentation',
  54. // version: '1.0.0'
  55. // },
  56. // security: [{
  57. // bearerAuth: []
  58. // }]
  59. // // servers: [{ url: '/api/v1' }]
  60. // })
  61. api.get('/doc', (c) => {
  62. const config = {
  63. openapi: '3.1.0',
  64. info: {
  65. title: 'API Documentation',
  66. version: '1.0.0'
  67. },
  68. security: [{
  69. bearerAuth: []
  70. }]
  71. // servers: [{ url: '/api/v1' }]
  72. };
  73. try {
  74. const document = api.getOpenAPI31Document(config);
  75. return c.json(document);
  76. } catch (e: any) {
  77. return c.json({
  78. error: 'OpenAPI document generation failed',
  79. message: e.message || 'Unknown error',
  80. stack: e.stack
  81. }, 500)
  82. }
  83. })
  84. app.get('/ui', swaggerUI({
  85. url: '/doc',
  86. persistAuthorization: true,
  87. manuallySwaggerUIHtml: () => `
  88. <div>
  89. <div id="swagger-ui"></div>
  90. <link rel="stylesheet" href="https://ai-oss.d8d.fun/swagger-ui-dist/swagger-ui.css" />
  91. <script src="https://ai-oss.d8d.fun/swagger-ui-dist/swagger-ui-bundle.js" crossorigin="anonymous"></script>
  92. <script>
  93. window.onload = () => {
  94. window.ui = SwaggerUIBundle({
  95. dom_id: '#swagger-ui',
  96. url: '/doc',
  97. persistAuthorization: true
  98. })
  99. }
  100. </script>
  101. </div>
  102. `
  103. }))
  104. }
  105. export const userRoutes = api.route('/api/v1/users', usersRouter)
  106. export const authRoutes = api.route('/api/v1/auth', authRoute)
  107. export const roleRoutes = api.route('/api/v1/roles', rolesRoute)
  108. export const fileApiRoutes = api.route('/api/v1/files', fileRoutes)
  109. export const adminActivitiesRoutesExport = api.route('/api/v1/admin/activities', adminActivitiesRoutes)
  110. export const adminRoutesRoutesExport = api.route('/api/v1/admin/routes', adminRoutesRoutes)
  111. export const adminAreasRoutesExport = api.route('/api/v1/admin/areas', areasRoutes)
  112. export const adminLocationsRoutesExport = api.route('/api/v1/admin/locations', locationsRoutes)
  113. export const adminPassengersRoutesExport = api.route('/api/v1/admin/passengers', adminPassengersRoutes)
  114. export const passengersRoutesExport = api.route('/api/v1/passengers', passengersRoutes)
  115. export const routesRoutesExport = api.route('/api/v1/routes', routesRoutes)
  116. export const areasUserRoutesExport = api.route('/api/v1/areas', areasUserRoutes)
  117. export const locationsUserRoutesExport = api.route('/api/v1/locations', locationsUserRoutes)
  118. export type AuthRoutes = typeof authRoutes
  119. export type UserRoutes = typeof userRoutes
  120. export type RoleRoutes = typeof roleRoutes
  121. export type FileRoutes = typeof fileApiRoutes
  122. export type AdminActivitiesRoutes = typeof adminActivitiesRoutesExport
  123. export type AdminRoutesRoutes = typeof adminRoutesRoutesExport
  124. export type AdminAreasRoutes = typeof adminAreasRoutesExport
  125. export type AdminLocationsRoutes = typeof adminLocationsRoutesExport
  126. export type AdminPassengersRoutes = typeof adminPassengersRoutesExport
  127. export type PassengersRoutes = typeof passengersRoutesExport
  128. export type RoutesRoutes = typeof routesRoutesExport
  129. export type AreasUserRoutes = typeof areasUserRoutesExport
  130. export type LocationsUserRoutes = typeof locationsUserRoutesExport
  131. app.route('/', api)
  132. export default app