79 lines
3.4 KiB
TypeScript
79 lines
3.4 KiB
TypeScript
import "server-only";
|
||
|
||
import { analyzeStyle8D, evaluateBenchmarkEngagement, serialize8DPersona } from "@/lib/ai/analyze-style-8d";
|
||
import { parseProviderApiKeys } from "@/lib/ai/keys";
|
||
import { prisma } from "@/lib/db";
|
||
import { setTaskStatus } from "@/lib/jobs/progress";
|
||
import { updateJobProgress } from "@/lib/jobs/progress-server";
|
||
import type { JobProgressDetail } from "@/lib/jobs/types";
|
||
import { ensureActiveSession, getProfilePosts } from "@/lib/threads-browser";
|
||
import { consumeCrawlerPageQuota } from "@/lib/threads-browser/safety";
|
||
import { getOrCreateSettings } from "@/lib/user-settings";
|
||
import { assertJobNotCancelled } from "@/lib/jobs/cancel";
|
||
|
||
export async function runStyle8DAnalysis(accountId: string, benchmarkUsername: string, jobId: string) {
|
||
const username = benchmarkUsername.replace(/^@/, "").trim();
|
||
const detail: JobProgressDetail = {
|
||
summary: "確認 Threads 連線…",
|
||
phase: "session",
|
||
tasks: [
|
||
{ id: "session", label: "確認 Extension 連線", status: "running", startedAt: Date.now() },
|
||
{ id: "samples", label: "抓取近期貼文", status: "pending" },
|
||
{ id: "style", label: "AI 8D 分析", status: "pending" },
|
||
{ id: "store", label: "儲存帳號策略", status: "pending" },
|
||
],
|
||
};
|
||
await updateJobProgress(jobId, detail);
|
||
await assertJobNotCancelled(jobId);
|
||
|
||
const session = await ensureActiveSession();
|
||
consumeCrawlerPageQuota(session.id);
|
||
setTaskStatus(detail, "session", { status: "done" });
|
||
setTaskStatus(detail, "samples", { status: "running", startedAt: Date.now() });
|
||
detail.phase = "samples";
|
||
detail.summary = `正在抓取 @${username} 的近期貼文…`;
|
||
await updateJobProgress(jobId, detail);
|
||
|
||
const posts = await getProfilePosts(session.storageState, username, 12);
|
||
await assertJobNotCancelled(jobId);
|
||
if (posts.length < 4) throw new Error("可讀取的近期貼文不足 4 篇，無法做可靠的 8D 分析");
|
||
setTaskStatus(detail, "samples", { status: "done", found: posts.length });
|
||
setTaskStatus(detail, "style", { status: "running", startedAt: Date.now() });
|
||
detail.phase = "style";
|
||
detail.summary = `已取得 ${posts.length} 篇，AI 正在分析 D1–D8（通常 1～3 分鐘）…`;
|
||
await updateJobProgress(jobId, detail);
|
||
|
||
const settings = await getOrCreateSettings();
|
||
const analysis = await analyzeStyle8D({
|
||
username,
|
||
posts,
|
||
aiProvider: settings.researchAiProvider ?? settings.aiProvider,
|
||
aiModel: settings.researchAiModel ?? settings.aiModel,
|
||
apiKeys: parseProviderApiKeys(settings.providerApiKeys),
|
||
});
|
||
await assertJobNotCancelled(jobId);
|
||
setTaskStatus(detail, "style", { status: "done" });
|
||
setTaskStatus(detail, "store", { status: "running", startedAt: Date.now() });
|
||
detail.phase = "store";
|
||
detail.summary = "8D 分析完成，正在寫入帳號策略…";
|
||
await updateJobProgress(jobId, detail);
|
||
|
||
const profile = {
|
||
username,
|
||
analyzedAt: new Date().toISOString(),
|
||
postCount: posts.length,
|
||
engagement: evaluateBenchmarkEngagement(posts),
|
||
analysis,
|
||
personaDraft: serialize8DPersona(analysis.personaDraft),
|
||
};
|
||
await assertJobNotCancelled(jobId);
|
||
await prisma.account.update({
|
||
where: { id: accountId },
|
||
data: { styleProfile: JSON.stringify(profile), styleBenchmark: username },
|
||
});
|
||
setTaskStatus(detail, "store", { status: "done" });
|
||
detail.summary = `完成 · @${username} · ${posts.length} 篇樣本 · 8D 策略已套用`;
|
||
await updateJobProgress(jobId, detail);
|
||
return profile;
|
||
}
|