356 lines
9.7 KiB
TypeScript
356 lines
9.7 KiB
TypeScript
import type { DatabaseClient } from "./database";
|
||
import { getRuntimeEnv } from "./runtime-env";
|
||
import {
|
||
sendWecomAppMessage,
|
||
type WecomConfig,
|
||
} from "./wecom-client";
|
||
import { pushProjectTask, type ProjectPushOutcome } from "./project-push-service";
|
||
import type { TaskPushRow } from "./wecom-group-push-service";
|
||
|
||
type FetchLike = typeof fetch;
|
||
|
||
export class OneClickPushError extends Error {
|
||
status: number;
|
||
|
||
constructor(message: string, status = 400) {
|
||
super(message);
|
||
this.name = "OneClickPushError";
|
||
this.status = status;
|
||
}
|
||
}
|
||
|
||
export type BoundWecomPartner = {
|
||
id: string;
|
||
name: string;
|
||
wecomExternalUserId: string;
|
||
};
|
||
|
||
export type KocPushResult = {
|
||
partnerId: string;
|
||
partnerName: string;
|
||
externalUserId: string;
|
||
ok: boolean;
|
||
error: string;
|
||
};
|
||
|
||
export type TaskPushLogRow = {
|
||
id: string;
|
||
task_id: string;
|
||
project_id: string;
|
||
push_kind: string;
|
||
main_group_count: number;
|
||
koc_userids: string;
|
||
koc_ok_count: number;
|
||
koc_fail_list: string;
|
||
group_push_id: string;
|
||
detail: string;
|
||
created_at: string;
|
||
};
|
||
|
||
export type OneClickPushOutcome = {
|
||
linkUrl: string;
|
||
projectOutcome: ProjectPushOutcome | null;
|
||
kocResults: KocPushResult[];
|
||
kocSent: number;
|
||
kocFailed: number;
|
||
skippedReasons: string[];
|
||
logId: string;
|
||
};
|
||
|
||
function text(value: unknown) {
|
||
return String(value ?? "").trim();
|
||
}
|
||
|
||
export function buildKocDirectMessage(
|
||
task: Pick<
|
||
TaskPushRow,
|
||
| "name"
|
||
| "brand"
|
||
| "quantity"
|
||
| "due_at"
|
||
| "task_type"
|
||
| "platform"
|
||
| "content_format"
|
||
| "share_token"
|
||
>,
|
||
portalUrl: string,
|
||
): string {
|
||
const portal = text(portalUrl).replace(/\/$/, "");
|
||
const claimUrl =
|
||
portal && task.share_token ? `${portal}/?task=${task.share_token}` : "";
|
||
const formatLabel =
|
||
task.task_type === "screenshot_collect"
|
||
? "截图回收"
|
||
: task.content_format === "video"
|
||
? "视频"
|
||
: "图文";
|
||
return [
|
||
`【新任务】${task.name}`,
|
||
`品牌：${task.brand}｜数量：${task.quantity} 份｜截止：${task.due_at}`,
|
||
`平台：${task.platform}｜形式：${formatLabel}`,
|
||
"数量有限，先到先得，点击链接直接领取（企微内打开免输身份）：",
|
||
claimUrl || "领取链接生成失败，请联系管理员",
|
||
].join("\n");
|
||
}
|
||
|
||
export async function listBoundWecomPartners(
|
||
db: DatabaseClient,
|
||
): Promise<BoundWecomPartner[]> {
|
||
const result = await db
|
||
.prepare(
|
||
`SELECT id, name, wecom_external_user_id
|
||
FROM partners
|
||
WHERE wecom_external_user_id IS NOT NULL
|
||
AND wecom_external_user_id != ''
|
||
ORDER BY name ASC`,
|
||
)
|
||
.all<{
|
||
id: string;
|
||
name: string;
|
||
wecom_external_user_id: string;
|
||
}>();
|
||
return result.results.map((row) => ({
|
||
id: row.id,
|
||
name: row.name,
|
||
wecomExternalUserId: row.wecom_external_user_id,
|
||
}));
|
||
}
|
||
|
||
export async function findTodayTaskPushLog(
|
||
db: DatabaseClient,
|
||
taskId: string,
|
||
pushKind = "task_open",
|
||
): Promise<TaskPushLogRow | null> {
|
||
const result = await db
|
||
.prepare(
|
||
`SELECT id, task_id, project_id, push_kind, main_group_count,
|
||
koc_userids, koc_ok_count, koc_fail_list, group_push_id,
|
||
detail, created_at
|
||
FROM wecom_push_logs
|
||
WHERE task_id = ? AND push_kind = ?
|
||
AND DATE(created_at) = CURDATE()
|
||
ORDER BY created_at DESC
|
||
LIMIT 1`,
|
||
)
|
||
.bind(taskId, pushKind)
|
||
.first<TaskPushLogRow>();
|
||
return result ?? null;
|
||
}
|
||
|
||
export async function listTaskPushLogs(
|
||
db: DatabaseClient,
|
||
taskId: string,
|
||
): Promise<TaskPushLogRow[]> {
|
||
const filter = taskId ? "WHERE task_id = ?" : "";
|
||
const statement = db
|
||
.prepare(
|
||
`SELECT id, task_id, project_id, push_kind, main_group_count,
|
||
koc_userids, koc_ok_count, koc_fail_list, group_push_id,
|
||
detail, created_at
|
||
FROM wecom_push_logs
|
||
${filter}
|
||
ORDER BY created_at DESC
|
||
LIMIT 50`,
|
||
)
|
||
.bind(...(taskId ? [taskId] : []));
|
||
const result = await statement.all<TaskPushLogRow>();
|
||
return result.results;
|
||
}
|
||
|
||
export async function recordCardAck(
|
||
db: DatabaseClient,
|
||
params: { cardTaskId: string; btnKey: string; operator: string },
|
||
) {
|
||
const taskId = /^task:([^:]+):/.exec(params.cardTaskId)?.[1] ?? "";
|
||
await db
|
||
.prepare(
|
||
`INSERT INTO wecom_push_logs
|
||
(id, task_id, project_id, push_kind, main_group_count,
|
||
koc_userids, koc_ok_count, koc_fail_list, group_push_id, detail)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||
)
|
||
.bind(
|
||
`wpl-${crypto.randomUUID().slice(0, 12)}`,
|
||
taskId || params.cardTaskId,
|
||
"",
|
||
"card_ack",
|
||
0,
|
||
"[]",
|
||
0,
|
||
"[]",
|
||
"",
|
||
JSON.stringify({
|
||
cardTaskId: params.cardTaskId,
|
||
btnKey: params.btnKey,
|
||
operator: params.operator,
|
||
}),
|
||
)
|
||
.run();
|
||
}
|
||
|
||
export async function oneClickPushTask(
|
||
db: DatabaseClient,
|
||
params: {
|
||
taskId: unknown;
|
||
sendMainGroup?: unknown;
|
||
sendKoc?: unknown;
|
||
createGroupTemplate?: unknown;
|
||
selectedKocPartnerIds?: unknown;
|
||
force?: unknown;
|
||
text?: unknown;
|
||
},
|
||
config: WecomConfig,
|
||
fetchImpl: FetchLike = fetch,
|
||
): Promise<OneClickPushOutcome> {
|
||
const taskId = text(params.taskId);
|
||
if (!taskId) throw new OneClickPushError("缺少 taskId");
|
||
|
||
const sendMainGroup = params.sendMainGroup !== false;
|
||
const sendKoc = params.sendKoc !== false;
|
||
const createGroupTemplate = params.createGroupTemplate === true;
|
||
if (!sendMainGroup && !sendKoc && !createGroupTemplate) {
|
||
throw new OneClickPushError("请至少选择一个推送目标");
|
||
}
|
||
|
||
const task = await db
|
||
.prepare(
|
||
`SELECT id, name, brand, quantity, due_at, task_type, platform,
|
||
content_format, share_token, project_id
|
||
FROM tasks WHERE id = ?`,
|
||
)
|
||
.bind(taskId)
|
||
.first<TaskPushRow & { project_id: string }>();
|
||
if (!task) throw new OneClickPushError("任务不存在", 404);
|
||
if (!text(task.share_token)) {
|
||
throw new OneClickPushError("任务尚未生成领取链接，请先开放领取", 400);
|
||
}
|
||
|
||
if (params.force !== true) {
|
||
const existing = await findTodayTaskPushLog(db, taskId, "task_open");
|
||
if (existing) {
|
||
throw new OneClickPushError(
|
||
"今天已推送过该任务；如确需再次推送，请勾选「仍要推送」",
|
||
409,
|
||
);
|
||
}
|
||
}
|
||
|
||
const skippedReasons: string[] = [];
|
||
const wantsProjectTargets = sendMainGroup || createGroupTemplate;
|
||
let projectOutcome: ProjectPushOutcome | null = null;
|
||
if (wantsProjectTargets) {
|
||
if (!text(task.project_id)) {
|
||
skippedReasons.push("任务未挂项目，已跳过项目总群与任务群");
|
||
} else {
|
||
projectOutcome = await pushProjectTask(
|
||
db,
|
||
{
|
||
taskId,
|
||
sendMainGroup,
|
||
sendGroupChats: createGroupTemplate,
|
||
text: params.text,
|
||
},
|
||
config,
|
||
fetchImpl,
|
||
);
|
||
skippedReasons.push(...projectOutcome.skippedReasons);
|
||
}
|
||
}
|
||
|
||
const portalUrl = text(getRuntimeEnv().KOC_PORTAL_URL);
|
||
const linkUrl = task.share_token
|
||
? `${portalUrl.replace(/\/$/, "")}/?task=${task.share_token}`
|
||
: "";
|
||
|
||
const kocResults: KocPushResult[] = [];
|
||
if (sendKoc) {
|
||
const bound = await listBoundWecomPartners(db);
|
||
const selected = Array.isArray(params.selectedKocPartnerIds)
|
||
? params.selectedKocPartnerIds.map((id) => text(id)).filter(Boolean)
|
||
: null;
|
||
const targets = selected
|
||
? bound.filter((partner) => selected.includes(partner.id))
|
||
: bound;
|
||
if (targets.length === 0) {
|
||
skippedReasons.push("没有可直达的合作方（未勾选或均未绑定企微身份）");
|
||
} else {
|
||
const message = buildKocDirectMessage(task, portalUrl);
|
||
for (const partner of targets) {
|
||
try {
|
||
const sent = await sendWecomAppMessage(
|
||
[partner.wecomExternalUserId],
|
||
message,
|
||
config,
|
||
fetchImpl,
|
||
);
|
||
const failed = sent.sent === 0;
|
||
kocResults.push({
|
||
partnerId: partner.id,
|
||
partnerName: partner.name,
|
||
externalUserId: partner.wecomExternalUserId,
|
||
ok: !failed,
|
||
error: failed ? "企业微信未送达（invaliduser）" : "",
|
||
});
|
||
} catch (error) {
|
||
kocResults.push({
|
||
partnerId: partner.id,
|
||
partnerName: partner.name,
|
||
externalUserId: partner.wecomExternalUserId,
|
||
ok: false,
|
||
error: error instanceof Error ? error.message : "发送失败",
|
||
});
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
const kocSent = kocResults.filter((item) => item.ok).length;
|
||
const kocFailed = kocResults.length - kocSent;
|
||
const logId = `wpl-${crypto.randomUUID().slice(0, 12)}`;
|
||
const groupPushId =
|
||
projectOutcome?.groupResults.map((item) => item.msgid).join(",") ?? "";
|
||
|
||
await db
|
||
.prepare(
|
||
`INSERT INTO wecom_push_logs
|
||
(id, task_id, project_id, push_kind, main_group_count,
|
||
koc_userids, koc_ok_count, koc_fail_list, group_push_id, detail)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||
)
|
||
.bind(
|
||
logId,
|
||
taskId,
|
||
text(task.project_id),
|
||
"task_open",
|
||
projectOutcome ? projectOutcome.appResults.length : 0,
|
||
JSON.stringify(
|
||
kocResults.map((item) => ({
|
||
partnerId: item.partnerId,
|
||
name: item.partnerName,
|
||
externalUserId: item.externalUserId,
|
||
ok: item.ok,
|
||
error: item.error,
|
||
})),
|
||
),
|
||
kocSent,
|
||
JSON.stringify(kocResults.filter((item) => !item.ok)),
|
||
groupPushId,
|
||
JSON.stringify({
|
||
skippedReasons,
|
||
createGroupTemplate,
|
||
linkUrl,
|
||
}),
|
||
)
|
||
.run();
|
||
|
||
return {
|
||
linkUrl,
|
||
projectOutcome,
|
||
kocResults,
|
||
kocSent,
|
||
kocFailed,
|
||
skippedReasons,
|
||
logId,
|
||
};
|
||
}
|