133 lines
4.5 KiB
TypeScript
133 lines
4.5 KiB
TypeScript
// scripts/fix-asset-urls.ts
|
||
import { PrismaClient } from '@prisma/client';
|
||
const prisma = new PrismaClient();
|
||
|
||
const FROM = 'douyin-archive/';
|
||
const TO = '';
|
||
|
||
function escapeForPgRegex(s: string) {
|
||
return s.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
|
||
}
|
||
const FROM_RE = `^${escapeForPgRegex(FROM)}`; // 只替换“以旧前缀开头”的字符串
|
||
const dryRun = false; // true: 只统计，不修改
|
||
|
||
async function main() {
|
||
if (dryRun) {
|
||
const rows = await prisma.$queryRawUnsafe<any[]>(`
|
||
WITH c AS (
|
||
SELECT 'Author.avatar_url' AS col, COUNT(*) AS n FROM "Author" WHERE avatar_url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'CommentUser.avatar_url' , COUNT(*) FROM "CommentUser" WHERE avatar_url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'CommentImage.url' , COUNT(*) FROM "CommentImage" WHERE url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'Video.cover_url' , COUNT(*) FROM "Video" WHERE cover_url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'Video.video_url' , COUNT(*) FROM "Video" WHERE video_url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'ImageFile.url' , COUNT(*) FROM "ImageFile" WHERE url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'ImageFile.animated' , COUNT(*) FROM "ImageFile" WHERE animated LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'ImagePost.music_url' , COUNT(*) FROM "ImagePost" WHERE music_url LIKE '${FROM}%'
|
||
UNION ALL
|
||
SELECT 'Video.raw_json' , COUNT(*) FROM "Video" WHERE raw_json::text LIKE '%${FROM}%'
|
||
UNION ALL
|
||
SELECT 'ImagePost.raw_json' , COUNT(*) FROM "ImagePost" WHERE raw_json::text LIKE '%${FROM}%'
|
||
)
|
||
SELECT * FROM c ORDER BY col;
|
||
`);
|
||
console.table(rows);
|
||
return;
|
||
}
|
||
|
||
await prisma.$transaction(async (tx) => {
|
||
// CommentUser.avatar_url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "CommentUser"
|
||
SET avatar_url = regexp_replace(avatar_url, '${FROM_RE}', '${TO}')
|
||
WHERE avatar_url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// CommentImage.url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "CommentImage"
|
||
SET url = regexp_replace(url, '${FROM_RE}', '${TO}')
|
||
WHERE url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// Author.avatar_url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "Author"
|
||
SET avatar_url = regexp_replace(avatar_url, '${FROM_RE}', '${TO}')
|
||
WHERE avatar_url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// Video.cover_url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "Video"
|
||
SET cover_url = regexp_replace(cover_url, '${FROM_RE}', '${TO}')
|
||
WHERE cover_url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// Video.video_url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "Video"
|
||
SET video_url = regexp_replace(video_url, '${FROM_RE}', '${TO}')
|
||
WHERE video_url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// ImageFile.url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "ImageFile"
|
||
SET url = regexp_replace(url, '${FROM_RE}', '${TO}')
|
||
WHERE url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// ImageFile.animated
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "ImageFile"
|
||
SET animated = regexp_replace(animated, '${FROM_RE}', '${TO}')
|
||
WHERE animated LIKE '${FROM}%'
|
||
`);
|
||
|
||
// ImagePost.music_url
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "ImagePost"
|
||
SET music_url = regexp_replace(music_url, '${FROM_RE}', '${TO}')
|
||
WHERE music_url LIKE '${FROM}%'
|
||
`);
|
||
|
||
// （可选）raw_json：简单文本整体替换；若想更精细可用递归 JSONB 方法
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "Video"
|
||
SET raw_json = to_jsonb(replace(raw_json::text, '${FROM}', '${TO}'))
|
||
WHERE raw_json::text LIKE '%${FROM}%'
|
||
`);
|
||
await tx.$executeRawUnsafe(`
|
||
UPDATE "ImagePost"
|
||
SET raw_json = to_jsonb(replace(raw_json::text, '${FROM}', '${TO}'))
|
||
WHERE raw_json::text LIKE '%${FROM}%'
|
||
`);
|
||
});
|
||
|
||
// 快速抽样
|
||
const sample = await prisma.$queryRawUnsafe<any[]>(`
|
||
(
|
||
SELECT 'CommentImage.url' AS col, url AS v FROM "CommentImage" WHERE url LIKE '${TO}%' LIMIT 2
|
||
) UNION ALL (
|
||
SELECT 'CommentUser.avatar_url', avatar_url FROM "CommentUser" WHERE avatar_url LIKE '${TO}%' LIMIT 2
|
||
) UNION ALL (
|
||
SELECT 'Video.video_url', video_url FROM "Video" WHERE video_url LIKE '${TO}%' LIMIT 2
|
||
)
|
||
`);
|
||
console.log('Sample after update:', sample);
|
||
}
|
||
|
||
main().catch((e) => {
|
||
console.error(e);
|
||
process.exit(1);
|
||
}).finally(async () => {
|
||
await prisma.$disconnect();
|
||
});
|