api.ts 2.4 KB

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