52 lines
1.8 KiB
TypeScript
52 lines
1.8 KiB
TypeScript
import { defineConfig } from 'vite'
|
|
import react from '@vitejs/plugin-react'
|
|
import fs from 'fs'
|
|
import path from 'path'
|
|
|
|
const ortDistDir = path.resolve(__dirname, 'node_modules/onnxruntime-web/dist')
|
|
const vadDistDir = path.resolve(__dirname, 'node_modules/@ricky0123/vad-web/dist')
|
|
const publicDir = path.resolve(__dirname, 'public')
|
|
|
|
// 需要复制到 public 的静态资源
|
|
const staticFiles = [
|
|
{ src: vadDistDir, name: 'silero_vad_legacy.onnx' },
|
|
{ src: vadDistDir, name: 'silero_vad_v5.onnx' },
|
|
{ src: vadDistDir, name: 'vad.worklet.bundle.min.js' },
|
|
]
|
|
|
|
export default defineConfig({
|
|
plugins: [
|
|
react(),
|
|
{
|
|
name: 'serve-vad-assets',
|
|
// 1. 启动时将 VAD 模型/工作线程复制到 public
|
|
configResolved() {
|
|
for (const { src, name } of staticFiles) {
|
|
const dest = path.join(publicDir, name)
|
|
if (!fs.existsSync(dest)) {
|
|
fs.copyFileSync(path.join(src, name), dest)
|
|
}
|
|
}
|
|
},
|
|
// 2. 中间件拦截所有 WASM 相关请求，从原始路径提供文件
|
|
configureServer(server) {
|
|
server.middlewares.use((req, res, next) => {
|
|
const url = req.url?.split('?')[0] ?? ''
|
|
// 匹配任意路径下的 ort-wasm 文件（包括 .vite/deps/ 和根路径）
|
|
const wasmMatch = url.match(/\/(ort-wasm-simd-threaded\.(mjs|wasm))$/)
|
|
if (wasmMatch) {
|
|
const fileName = wasmMatch[1]
|
|
const filePath = path.join(ortDistDir, fileName)
|
|
if (fs.existsSync(filePath)) {
|
|
res.setHeader('Content-Type', fileName.endsWith('.wasm') ? 'application/wasm' : 'application/javascript')
|
|
res.setHeader('Cache-Control', 'no-cache')
|
|
res.end(fs.readFileSync(filePath))
|
|
return
|
|
}
|
|
}
|
|
next()
|
|
})
|
|
},
|
|
},
|
|
],
|
|
}) |