89 lines
3.1 KiB
TypeScript
89 lines
3.1 KiB
TypeScript
|
import "server-only";
|
|||
|
|
|
|||
|
|
import { Prisma, PrismaClient } from "@prisma/client";
|
|||
|
|
import { buildStyle8DPromptBlock } from "@/lib/types/style-profile";
|
|||
|
|
|
|||
|
|
const globalForPrisma = globalThis as unknown as {
|
|||
|
|
prisma?: PrismaClient & { __schemaFingerprint?: string };
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/** 與 schema 同步；dmmf 不可用時用固定字串讓 dev 快取仍能失效。 */
|
|||
|
|
const SCHEMA_FINGERPRINT_FALLBACK =
|
|||
|
|
"ProductProfile:accountId,brand,brandId,context,createdAt,id,label,matchTags,topics,updatedAt|Topic:accountId,active,brief,brandProfileId,createdAt,id,label,productContext,productProfileId,query,researchMap,selectedTags,topicGoal";
|
|||
|
|
|
|||
|
|
let schemaFingerprint: string | null = null;
|
|||
|
|
|
|||
|
|
function getSchemaFingerprint(): string {
|
|||
|
|
if (schemaFingerprint) return schemaFingerprint;
|
|||
|
|
try {
|
|||
|
|
const models = Prisma.dmmf?.datamodel?.models;
|
|||
|
|
if (models) {
|
|||
|
|
const account = models.find((m) => m.name === "Account");
|
|||
|
|
const product = models.find((m) => m.name === "ProductProfile");
|
|||
|
|
const topic = models.find((m) => m.name === "Topic");
|
|||
|
|
const productFields = product?.fields.map((f) => f.name).sort().join(",") ?? "";
|
|||
|
|
const topicFields = topic?.fields.map((f) => f.name).sort().join(",") ?? "";
|
|||
|
|
const accountFields = account?.fields.map((f) => f.name).sort().join(",") ?? "";
|
|||
|
|
schemaFingerprint = `Account:${accountFields}|ProductProfile:${productFields}|Topic:${topicFields}`;
|
|||
|
|
return schemaFingerprint;
|
|||
|
|
}
|
|||
|
|
} catch {
|
|||
|
|
// Prisma.dmmf 僅在 server runtime 可用
|
|||
|
|
}
|
|||
|
|
schemaFingerprint = SCHEMA_FINGERPRINT_FALLBACK;
|
|||
|
|
return schemaFingerprint;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function createPrismaClient() {
|
|||
|
|
const fingerprint = getSchemaFingerprint();
|
|||
|
|
const client = new PrismaClient({
|
|||
|
|
log: process.env.NODE_ENV === "development" ? ["error", "warn"] : ["error"],
|
|||
|
|
}) as PrismaClient & { __schemaFingerprint?: string };
|
|||
|
|
client.__schemaFingerprint = fingerprint;
|
|||
|
|
return client;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function isPrismaClientCurrent(
|
|||
|
|
client: PrismaClient & { __schemaFingerprint?: string }
|
|||
|
|
): boolean {
|
|||
|
|
return (
|
|||
|
|
"backgroundJob" in client &&
|
|||
|
|
"user" in client &&
|
|||
|
|
"productProfile" in client &&
|
|||
|
|
"brandProfile" in client &&
|
|||
|
|
client.__schemaFingerprint === getSchemaFingerprint()
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function getPrismaClient() {
|
|||
|
|
const cached = globalForPrisma.prisma;
|
|||
|
|
if (cached && isPrismaClientCurrent(cached)) return cached;
|
|||
|
|
|
|||
|
|
const client = createPrismaClient();
|
|||
|
|
if (process.env.NODE_ENV !== "production") {
|
|||
|
|
globalForPrisma.prisma = client;
|
|||
|
|
}
|
|||
|
|
return client;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export const prisma = getPrismaClient();
|
|||
|
|
|
|||
|
|
export function resolvePersona(
|
|||
|
|
settings: { persona: string | null },
|
|||
|
|
account?: {
|
|||
|
|
persona: string | null;
|
|||
|
|
styleProfile?: string | null;
|
|||
|
|
targetAudience?: string | null;
|
|||
|
|
goals?: string | null;
|
|||
|
|
} | null
|
|||
|
|
) {
|
|||
|
|
const blocks = [
|
|||
|
|
account?.persona?.trim() || settings.persona?.trim(),
|
|||
|
|
buildStyle8DPromptBlock(account?.styleProfile),
|
|||
|
|
account?.targetAudience?.trim() ? `目標受眾：${account.targetAudience.trim()}` : null,
|
|||
|
|
account?.goals?.trim() ? `經營目標：${account.goals.trim()}` : null,
|
|||
|
|
].filter(Boolean);
|
|||
|
|
|
|||
|
|
return blocks.length > 0 ? blocks.join("\n\n") : null;
|
|||
|
|
}
|