init

2026-04-08 16:30:42 +08:00
commit 313c036845
131 changed files with 22393 additions and 0 deletions
@@ -0,0 +1,170 @@
+import express from 'express'
+import process from 'node:process'
+
+import { runtimeConfig } from './config/runtime.js'
+import { runDatabaseMigrations } from './db/migrate.js'
+import adminRouter from './routes/admin.js'
+import claimsRouter from './routes/claims.js'
+import webhooksRouter from './routes/webhooks.js'
+import { ensureAdminUsersBootstrapped } from './services/admin-auth-service.js'
+import { closeLocalOcrWorker, warmupLocalOcrWorker } from './services/ocr.js'
+import { closeAllTencentBrowserSessions, warmupTencentBrowser } from './services/session.js'
+import tencentRouter from './routes/tencent.js'
+
+const app = express()
+const port = Number(runtimeConfig.server.port || 3000)
+
+runDatabaseMigrations()
+await ensureAdminUsersBootstrapped()
+
+app.use((req, res, next) => {
+  res.setHeader('Access-Control-Allow-Origin', '*')
+  res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization')
+  res.setHeader('Access-Control-Allow-Methods', 'GET, POST, PUT, PATCH, DELETE, OPTIONS')
+
+  if (req.method === 'OPTIONS') {
+    res.sendStatus(204)
+    return
+  }
+
+  next()
+})
+
+app.use(express.json({ limit: '2mb' }))
+app.use(express.urlencoded({ extended: true }))
+
+app.get('/health', (_req, res) => {
+  res.json({
+    code: 0,
+    msg: 'ok',
+    time: Math.floor(Date.now() / 1000),
+  })
+})
+
+app.use('/api/v1/tencent', tencentRouter)
+app.use('/api/v1/webhooks', webhooksRouter)
+app.use('/api/v1/claim', claimsRouter)
+app.use('/api/v1/admin', adminRouter)
+
+const server = app.listen(port, () => {
+  console.log(`order-site-backend listening on http://127.0.0.1:${port}`)
+  void bootstrapBrowser()
+  void bootstrapOcr()
+})
+
+async function bootstrapBrowser() {
+  try {
+    const result = await warmupTencentBrowser()
+
+    if (!result.warmed) {
+      console.log('[startup] browser prewarm skipped')
+      return
+    }
+
+    console.log('[startup] browser prewarm ready')
+  } catch (error) {
+    if (isMissingPlaywrightBrowserError(error)) {
+      const installCommand = resolveBrowserInstallCommand()
+      console.warn(`[startup] browser prewarm skipped: ${formatErrorMessage(error)}`)
+      console.warn(`[startup] 请先安装浏览器依赖: ${installCommand}`)
+
+      if (String(runtimeConfig.browser.chromePath || '').trim()) {
+        console.warn('[startup] 当前设置了 CHROME_PATH，也请确认该路径指向的浏览器可执行文件真实存在')
+      }
+
+      return
+    }
+
+    console.error('[startup] browser prewarm failed:', error)
+  }
+}
+
+async function bootstrapOcr() {
+  try {
+    await warmupLocalOcrWorker()
+    console.log('[startup] local OCR worker ready')
+  } catch (error) {
+    console.warn(`[startup] local OCR worker skipped: ${formatStartupError(error)}`)
+  }
+}
+
+function formatStartupError(error) {
+  if (error instanceof Error && error.message) {
+    return error.message.split('\n')[0].trim()
+  }
+
+  return String(error || '未知错误').trim()
+}
+
+function isMissingPlaywrightBrowserError(error) {
+  const message = formatErrorMessage(error).toLowerCase()
+
+  return (
+    message.includes("executable doesn't exist") ||
+    message.includes('please run the following command to download new browsers') ||
+    message.includes('browserType.launch'.toLowerCase())
+      && message.includes('please run')
+  )
+}
+
+function formatErrorMessage(error) {
+  if (error instanceof Error && error.message) {
+    return error.message.split('\n')[0].trim()
+  }
+
+  return String(error || '未知错误').trim()
+}
+
+function resolveBrowserInstallCommand() {
+  if (process.platform === 'linux') {
+    return 'npm run browser:install:linux'
+  }
+
+  return 'npm run browser:install'
+}
+
+let shutdownStarted = false
+
+async function shutdown(signal) {
+  if (shutdownStarted) {
+    return
+  }
+
+  shutdownStarted = true
+  console.log(`[shutdown] received ${signal}, closing browser sessions and HTTP server`)
+
+  try {
+    await closeAllTencentBrowserSessions({ markClosed: true })
+  } catch (error) {
+    console.error('[shutdown] failed to close browser sessions:', error)
+  }
+
+  try {
+    await closeLocalOcrWorker()
+  } catch (error) {
+    console.error('[shutdown] failed to close local OCR worker:', error)
+  }
+
+  if (!server.listening) {
+    return
+  }
+
+  await new Promise((resolve) => {
+    server.close((error) => {
+      if (error) {
+        console.error('[shutdown] failed to close HTTP server:', error)
+        process.exitCode = 1
+      }
+
+      resolve()
+    })
+  })
+}
+
+process.on('SIGINT', () => {
+  void shutdown('SIGINT')
+})
+
+process.on('SIGTERM', () => {
+  void shutdown('SIGTERM')
+})