138 lines
3.1 KiB
JavaScript
138 lines
3.1 KiB
JavaScript
/**
|
||
* 批量迁移脚本：将旧的 query() 调用替换为 Prisma
|
||
*
|
||
* 使用方法：
|
||
* node scripts/migrate-to-prisma.js
|
||
*/
|
||
|
||
const fs = require('fs')
|
||
const path = require('path')
|
||
|
||
// 需要迁移的API文件路径列表
|
||
const API_FILES = [
|
||
'app/api/referral/bind/route.ts',
|
||
'app/api/referral/visit/route.ts',
|
||
'app/api/miniprogram/pay/route.ts',
|
||
'app/api/miniprogram/pay/notify/route.ts',
|
||
'app/api/user/check-purchased/route.ts',
|
||
'app/api/user/purchase-status/route.ts',
|
||
'app/api/user/reading-progress/route.ts',
|
||
'app/api/user/track/route.ts',
|
||
'app/api/db/config/route.ts',
|
||
'app/api/book/all-chapters/route.ts',
|
||
'app/api/book/hot/route.ts',
|
||
'app/api/book/chapter/[id]/route.ts',
|
||
'app/api/match/users/route.ts',
|
||
'app/api/match/config/route.ts',
|
||
'app/api/search/route.ts'
|
||
]
|
||
|
||
/**
|
||
* 自动替换规则
|
||
*/
|
||
const REPLACE_RULES = [
|
||
{
|
||
// 替换 import
|
||
from: /from '@\/lib\/db'/g,
|
||
to: "from '@/lib/prisma'"
|
||
},
|
||
{
|
||
// 替换 query 导入为 prisma
|
||
from: /import \{ query(.*?) \} from '@\/lib\/prisma'/g,
|
||
to: "import { prisma } from '@/lib/prisma'"
|
||
},
|
||
{
|
||
// 替换 getConfig 导入
|
||
from: /import \{ getConfig \} from '@\/lib\/db'/g,
|
||
to: "import { getPrismaConfig } from '@/lib/prisma-helpers'"
|
||
},
|
||
{
|
||
// 替换 getConfig 调用
|
||
from: /getConfig\(/g,
|
||
to: "getPrismaConfig("
|
||
}
|
||
]
|
||
|
||
/**
|
||
* 添加 Prisma 注释
|
||
*/
|
||
function addPrismaComment(content) {
|
||
if (content.includes('使用 Prisma ORM')) return content
|
||
|
||
const lines = content.split('\n')
|
||
let commentIndex = -1
|
||
|
||
// 找到文件头注释
|
||
for (let i = 0; i < lines.length; i++) {
|
||
if (lines[i].includes('*/')) {
|
||
commentIndex = i
|
||
break
|
||
}
|
||
}
|
||
|
||
if (commentIndex > 0) {
|
||
lines.splice(commentIndex, 0, ' * 使用 Prisma ORM（安全，防SQL注入）')
|
||
}
|
||
|
||
return lines.join('\n')
|
||
}
|
||
|
||
/**
|
||
* 处理单个文件
|
||
*/
|
||
function migrateFile(filePath) {
|
||
const fullPath = path.join(__dirname, '..', filePath)
|
||
|
||
if (!fs.existsSync(fullPath)) {
|
||
console.log(`❌ 文件不存在: ${filePath}`)
|
||
return false
|
||
}
|
||
|
||
let content = fs.readFileSync(fullPath, 'utf-8')
|
||
|
||
// 如果已经迁移过，跳过
|
||
if (content.includes('from \'@/lib/prisma\'')) {
|
||
console.log(`⏭️ 已迁移，跳过: ${filePath}`)
|
||
return false
|
||
}
|
||
|
||
// 应用替换规则
|
||
REPLACE_RULES.forEach(rule => {
|
||
content = content.replace(rule.from, rule.to)
|
||
})
|
||
|
||
// 添加注释
|
||
content = addPrismaComment(content)
|
||
|
||
// 写回文件
|
||
fs.writeFileSync(fullPath, content, 'utf-8')
|
||
console.log(`✅ 已迁移: ${filePath}`)
|
||
return true
|
||
}
|
||
|
||
/**
|
||
* 主函数
|
||
*/
|
||
function main() {
|
||
console.log('🚀 开始批量迁移到 Prisma ORM...\n')
|
||
|
||
let successCount = 0
|
||
let skipCount = 0
|
||
|
||
API_FILES.forEach(filePath => {
|
||
const result = migrateFile(filePath)
|
||
if (result) {
|
||
successCount++
|
||
} else {
|
||
skipCount++
|
||
}
|
||
})
|
||
|
||
console.log(`\n✨ 迁移完成！`)
|
||
console.log(` - 成功迁移: ${successCount} 个文件`)
|
||
console.log(` - 跳过: ${skipCount} 个文件`)
|
||
console.log(`\n⚠️ 注意：批量迁移只处理简单替换，复杂的SQL查询需要手动迁移！`)
|
||
}
|
||
|
||
main()
|