api.ts 4.2 KB

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