66 lines
2.1 KiB
TypeScript
66 lines
2.1 KiB
TypeScript
import type { AutomationRule } from "@prisma/client";
|
||
import { prisma } from "@/lib/db";
|
||
import {
|
||
AUTOMATION_TASK_TYPES,
|
||
DEFAULT_DAILY_CAP,
|
||
type AutomationMode,
|
||
type AutomationTaskType,
|
||
} from "./types";
|
||
|
||
const DEFAULT_SCHEDULE: Record<AutomationTaskType, string> = {
|
||
scan: "0 */4 * * *",
|
||
generate: "10 */4 * * *",
|
||
publish: "0 10,15,20 * * *",
|
||
outreach: "30 */2 * * *",
|
||
engagement: "*/30 * * * *",
|
||
};
|
||
|
||
/** 取得帳號的所有任務規則，缺的補上預設（manual / 未啟用）。 */
|
||
export async function getRulesForAccount(accountId: string): Promise<AutomationRule[]> {
|
||
const existing = await prisma.automationRule.findMany({ where: { accountId } });
|
||
const byType = new Map(existing.map((r) => [r.taskType, r]));
|
||
|
||
const missing = AUTOMATION_TASK_TYPES.filter((t) => !byType.has(t));
|
||
if (missing.length > 0) {
|
||
await prisma.automationRule.createMany({
|
||
data: missing.map((taskType) => ({
|
||
accountId,
|
||
taskType,
|
||
mode: "manual",
|
||
dailyCap: DEFAULT_DAILY_CAP[taskType],
|
||
schedule: DEFAULT_SCHEDULE[taskType],
|
||
enabled: false,
|
||
})),
|
||
});
|
||
}
|
||
|
||
return prisma.automationRule.findMany({
|
||
where: { accountId },
|
||
orderBy: { taskType: "asc" },
|
||
});
|
||
}
|
||
|
||
export async function upsertRule(
|
||
accountId: string,
|
||
taskType: AutomationTaskType,
|
||
data: Partial<{ mode: AutomationMode; dailyCap: number; schedule: string; enabled: boolean }>
|
||
): Promise<AutomationRule> {
|
||
return prisma.automationRule.upsert({
|
||
where: { accountId_taskType: { accountId, taskType } },
|
||
create: {
|
||
accountId,
|
||
taskType,
|
||
mode: data.mode ?? "manual",
|
||
dailyCap: data.dailyCap ?? DEFAULT_DAILY_CAP[taskType],
|
||
schedule: data.schedule ?? DEFAULT_SCHEDULE[taskType],
|
||
enabled: data.enabled ?? false,
|
||
},
|
||
update: {
|
||
...(data.mode !== undefined && { mode: data.mode }),
|
||
...(data.dailyCap !== undefined && { dailyCap: data.dailyCap }),
|
||
...(data.schedule !== undefined && { schedule: data.schedule }),
|
||
...(data.enabled !== undefined && { enabled: data.enabled }),
|
||
},
|
||
});
|
||
}
|