274 lines
11 KiB
TypeScript
274 lines
11 KiB
TypeScript
import type { BrowserContext } from 'playwright';
|
||
import { prisma } from '@/lib/prisma';
|
||
import { uploadAvatarFromUrl } from './uploader';
|
||
import { firstUrl } from './utils';
|
||
|
||
export async function saveToDB(
|
||
context: BrowserContext,
|
||
detailResp: DouyinVideoDetailResponse,
|
||
commentResp: DouyinCommentResponse,
|
||
videoUrl?: string,
|
||
width?: number,
|
||
height?: number,
|
||
coverUrl?: string,
|
||
fps?: number
|
||
) {
|
||
if (!detailResp?.aweme_detail) throw new Error('视频详情为空');
|
||
const d = detailResp.aweme_detail;
|
||
|
||
// 1) Upsert Author
|
||
const authorAvatarSrc = firstUrl(d.author.avatar_thumb?.url_list);
|
||
const authorAvatarUploaded = await uploadAvatarFromUrl(context, authorAvatarSrc, `authors/${d.author.sec_uid}`);
|
||
const author = await prisma.author.upsert({
|
||
where: { sec_uid: d.author.sec_uid },
|
||
create: {
|
||
sec_uid: d.author.sec_uid,
|
||
uid: d.author.uid,
|
||
nickname: d.author.nickname,
|
||
signature: d.author.signature ?? null,
|
||
avatar_url: authorAvatarUploaded ?? null,
|
||
follower_count: BigInt(d.author.follower_count || 0),
|
||
total_favorited: BigInt(d.author.total_favorited || 0),
|
||
unique_id: d.author.unique_id ?? null,
|
||
short_id: d.author.short_id ?? null,
|
||
},
|
||
update: {
|
||
uid: d.author.uid,
|
||
nickname: d.author.nickname,
|
||
signature: d.author.signature ?? null,
|
||
avatar_url: authorAvatarUploaded ?? null,
|
||
follower_count: BigInt(d.author.follower_count || 0),
|
||
total_favorited: BigInt(d.author.total_favorited || 0),
|
||
unique_id: d.author.unique_id ?? null,
|
||
short_id: d.author.short_id ?? null,
|
||
},
|
||
});
|
||
|
||
// 2) Upsert Video
|
||
const video = await prisma.video.upsert({
|
||
where: { aweme_id: d.aweme_id },
|
||
create: {
|
||
aweme_id: d.aweme_id,
|
||
desc: d.desc,
|
||
preview_title: d.preview_title ?? null,
|
||
duration_ms: d.duration,
|
||
created_at: new Date((d.create_time || 0) * 1000),
|
||
share_url: d.share_url,
|
||
digg_count: BigInt(d.statistics?.digg_count || 0),
|
||
comment_count: BigInt(d.statistics?.comment_count || 0),
|
||
share_count: BigInt(d.statistics?.share_count || 0),
|
||
collect_count: BigInt(d.statistics?.collect_count || 0),
|
||
authorId: author.sec_uid,
|
||
tags: (d.tags?.map(t => t.tag_name) ?? []),
|
||
video_url: videoUrl ?? '',
|
||
width: width ?? null,
|
||
height: height ?? null,
|
||
cover_url: coverUrl ?? null,
|
||
fps: fps ?? null,
|
||
raw_json: detailResp as any, // 保存完整接口 JSON
|
||
},
|
||
update: {
|
||
desc: d.desc,
|
||
preview_title: d.preview_title ?? null,
|
||
duration_ms: d.duration,
|
||
created_at: new Date((d.create_time || 0) * 1000),
|
||
share_url: d.share_url,
|
||
digg_count: BigInt(d.statistics?.digg_count || 0),
|
||
comment_count: BigInt(d.statistics?.comment_count || 0),
|
||
share_count: BigInt(d.statistics?.share_count || 0),
|
||
collect_count: BigInt(d.statistics?.collect_count || 0),
|
||
authorId: author.sec_uid,
|
||
...(videoUrl ? { video_url: videoUrl } : {}),
|
||
...(width ? { width } : {}),
|
||
...(height ? { height } : {}),
|
||
...(coverUrl ? { cover_url: coverUrl } : {}),
|
||
...(fps ? { fps } : {}),
|
||
raw_json: detailResp as any, // 更新完整接口 JSON
|
||
},
|
||
});
|
||
|
||
// 3) Upsert Comments + CommentUser
|
||
const comments = commentResp?.comments ?? [];
|
||
for (const c of comments) {
|
||
const origAvatar: string | null = firstUrl(c.user?.avatar_thumb?.url_list) ?? null;
|
||
const nameHint = `comment-users/${(c.user?.nickname || 'unknown').replace(/\s+/g, '_')}-${c.cid}`;
|
||
const uploadedAvatar = await uploadAvatarFromUrl(context, origAvatar ?? undefined, nameHint);
|
||
const finalAvatar = uploadedAvatar ?? origAvatar; // string | null
|
||
const finalAvatarKey = finalAvatar ?? '';
|
||
const cu = await prisma.commentUser.upsert({
|
||
where: {
|
||
nickname_avatar_url: {
|
||
nickname: c.user?.nickname || '未知用户',
|
||
avatar_url: finalAvatarKey,
|
||
},
|
||
},
|
||
create: {
|
||
nickname: c.user?.nickname || '未知用户',
|
||
avatar_url: finalAvatar ?? null,
|
||
},
|
||
update: {
|
||
avatar_url: finalAvatar ?? null,
|
||
},
|
||
});
|
||
|
||
await prisma.comment.upsert({
|
||
where: { cid: c.cid },
|
||
create: {
|
||
cid: c.cid,
|
||
text: c.text,
|
||
digg_count: BigInt(c.digg_count || 0),
|
||
created_at: new Date((c.create_time || 0) * 1000),
|
||
videoId: video.aweme_id,
|
||
userId: cu.id,
|
||
},
|
||
update: {
|
||
text: c.text,
|
||
digg_count: BigInt(c.digg_count || 0),
|
||
created_at: new Date((c.create_time || 0) * 1000),
|
||
videoId: video.aweme_id,
|
||
userId: cu.id,
|
||
},
|
||
});
|
||
}
|
||
|
||
return { aweme_id: video.aweme_id, author_sec_uid: author.sec_uid, comment_count: comments.length };
|
||
}
|
||
|
||
export async function saveImagePostToDB(
|
||
context: BrowserContext,
|
||
aweme: DouyinImageAweme,
|
||
commentResp: DouyinCommentResponse,
|
||
uploads: { images: { url: string; width?: number; height?: number, video?: string }[]; musicUrl?: string },
|
||
rawJson?: any
|
||
) {
|
||
if (!aweme?.author?.sec_uid) throw new Error('作者 sec_uid 缺失');
|
||
|
||
// Upsert Author（与视频一致）
|
||
const authorAvatarSrc = firstUrl(aweme.author.avatar_thumb?.url_list);
|
||
const authorAvatarUploaded = await uploadAvatarFromUrl(context, authorAvatarSrc, `authors/${aweme.author.sec_uid}`);
|
||
const author = await prisma.author.upsert({
|
||
where: { sec_uid: aweme.author.sec_uid },
|
||
create: {
|
||
sec_uid: aweme.author.sec_uid,
|
||
uid: aweme.author.uid,
|
||
nickname: aweme.author.nickname,
|
||
signature: aweme.author.signature ?? null,
|
||
avatar_url: authorAvatarUploaded ?? null,
|
||
follower_count: BigInt((aweme.author as any).follower_count || 0),
|
||
total_favorited: BigInt((aweme.author as any).total_favorited || 0),
|
||
unique_id: (aweme.author as any).unique_id ?? null,
|
||
short_id: (aweme.author as any).short_id ?? null,
|
||
},
|
||
update: {
|
||
uid: aweme.author.uid,
|
||
nickname: aweme.author.nickname,
|
||
signature: aweme.author.signature ?? null,
|
||
avatar_url: authorAvatarUploaded ?? null,
|
||
follower_count: BigInt((aweme.author as any).follower_count || 0),
|
||
total_favorited: BigInt((aweme.author as any).total_favorited || 0),
|
||
unique_id: (aweme.author as any).unique_id ?? null,
|
||
short_id: (aweme.author as any).short_id ?? null,
|
||
},
|
||
});
|
||
|
||
// Upsert ImagePost
|
||
const imagePost = await prisma.imagePost.upsert({
|
||
where: { aweme_id: aweme.aweme_id },
|
||
create: {
|
||
aweme_id: aweme.aweme_id,
|
||
desc: aweme.desc,
|
||
created_at: new Date((aweme.create_time || 0) * 1000),
|
||
share_url: aweme.share_url || '',
|
||
digg_count: BigInt(aweme.statistics?.digg_count || 0),
|
||
comment_count: BigInt(aweme.statistics?.comment_count || 0),
|
||
share_count: BigInt(aweme.statistics?.share_count || 0),
|
||
collect_count: BigInt(aweme.statistics?.collect_count || 0),
|
||
authorId: author.sec_uid,
|
||
tags: (aweme.video_tag?.map(t => t.tag_name) ?? []),
|
||
music_url: uploads.musicUrl ?? null,
|
||
raw_json: rawJson ?? null, // 保存完整接口 JSON
|
||
},
|
||
update: {
|
||
desc: aweme.desc,
|
||
created_at: new Date((aweme.create_time || 0) * 1000),
|
||
share_url: aweme.share_url,
|
||
digg_count: BigInt(aweme.statistics?.digg_count || 0),
|
||
comment_count: BigInt(aweme.statistics?.comment_count || 0),
|
||
share_count: BigInt(aweme.statistics?.share_count || 0),
|
||
collect_count: BigInt(aweme.statistics?.collect_count || 0),
|
||
authorId: author.sec_uid,
|
||
tags: (aweme.video_tag?.map(t => t.tag_name) ?? []),
|
||
music_url: uploads.musicUrl ?? undefined,
|
||
raw_json: rawJson ?? undefined, // 更新完整接口 JSON
|
||
},
|
||
});
|
||
|
||
// Upsert ImageFiles（按顺序）
|
||
for (let i = 0; i < uploads.images.length; i++) {
|
||
const { url, width, height, video } = uploads.images[i];
|
||
await prisma.imageFile.upsert({
|
||
where: { postId_order: { postId: imagePost.aweme_id, order: i } },
|
||
create: {
|
||
postId: imagePost.aweme_id,
|
||
order: i,
|
||
url,
|
||
width: typeof width === 'number' ? width : null,
|
||
height: typeof height === 'number' ? height : null,
|
||
animated: video || null,
|
||
},
|
||
update: {
|
||
url,
|
||
width: typeof width === 'number' ? width : null,
|
||
height: typeof height === 'number' ? height : null,
|
||
animated: video || null,
|
||
},
|
||
});
|
||
}
|
||
|
||
// 评论入库：关联到 ImagePost
|
||
const comments = commentResp?.comments ?? [];
|
||
for (const c of comments) {
|
||
const origAvatar: string | null = firstUrl(c.user?.avatar_thumb?.url_list) ?? null;
|
||
const nameHint = `comment-users/${(c.user?.nickname || 'unknown').replace(/\s+/g, '_')}-${c.cid}`;
|
||
const uploadedAvatar = await uploadAvatarFromUrl(context, origAvatar ?? undefined, nameHint);
|
||
const finalAvatar = uploadedAvatar ?? origAvatar; // string | null
|
||
const finalAvatarKey = finalAvatar ?? '';
|
||
const cu = await prisma.commentUser.upsert({
|
||
where: {
|
||
nickname_avatar_url: {
|
||
nickname: c.user?.nickname || '未知用户',
|
||
avatar_url: finalAvatarKey,
|
||
},
|
||
},
|
||
create: {
|
||
nickname: c.user?.nickname || '未知用户',
|
||
avatar_url: finalAvatar ?? null,
|
||
},
|
||
update: {
|
||
avatar_url: finalAvatar ?? null,
|
||
},
|
||
});
|
||
|
||
await prisma.comment.upsert({
|
||
where: { cid: c.cid },
|
||
create: {
|
||
cid: c.cid,
|
||
text: c.text,
|
||
digg_count: BigInt(c.digg_count || 0),
|
||
created_at: new Date((c.create_time || 0) * 1000),
|
||
imagePostId: imagePost.aweme_id,
|
||
userId: cu.id,
|
||
},
|
||
update: {
|
||
text: c.text,
|
||
digg_count: BigInt(c.digg_count || 0),
|
||
created_at: new Date((c.create_time || 0) * 1000),
|
||
imagePostId: imagePost.aweme_id,
|
||
userId: cu.id,
|
||
},
|
||
});
|
||
}
|
||
|
||
return { aweme_id: imagePost.aweme_id, author_sec_uid: author.sec_uid, image_count: uploads.images.length, comment_count: comments.length };
|
||
}
|