index.ts 3.1 KB

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