Compare commits
2 Commits
main
...
49017650ba
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
49017650ba | ||
|
|
582c26e57e |
@@ -24,6 +24,15 @@ FEISHU_APP_SECRET=
|
|||||||
AI_TOOL_CENTER_MCP_URL=
|
AI_TOOL_CENTER_MCP_URL=
|
||||||
AI_TOOL_CENTER_MCP_KEY=
|
AI_TOOL_CENTER_MCP_KEY=
|
||||||
|
|
||||||
|
# 企业微信通知(临期催办 + 管理员汇总)。群机器人只需 webhook;KOC 侧催办还需 corp/agent/secret,
|
||||||
|
# 并在后台 partners 编辑里把 wecom_external_user_id 填好。
|
||||||
|
WECOM_ROBOT_WEBHOOK=
|
||||||
|
WECOM_CORP_ID=
|
||||||
|
WECOM_AGENT_ID=
|
||||||
|
WECOM_SECRET=
|
||||||
|
WECOM_NOTIFY_DUE_DAYS=3
|
||||||
|
WECOM_NOTIFY_ENABLED=true
|
||||||
|
|
||||||
# 每天北京时间 09:00 自动执行采集计划。
|
# 每天北京时间 09:00 自动执行采集计划。
|
||||||
ENABLE_SCHEDULER=true
|
ENABLE_SCHEDULER=true
|
||||||
SEED_DEMO_DATA=false
|
SEED_DEMO_DATA=false
|
||||||
|
|||||||
@@ -60,6 +60,7 @@ type Account = {
|
|||||||
post_count: number;
|
post_count: number;
|
||||||
avg_views: number;
|
avg_views: number;
|
||||||
cooperation_source: string;
|
cooperation_source: string;
|
||||||
|
tags: string;
|
||||||
first_seen_at: string;
|
first_seen_at: string;
|
||||||
last_seen_at: string;
|
last_seen_at: string;
|
||||||
};
|
};
|
||||||
@@ -74,6 +75,7 @@ type ResourceImportPreview = {
|
|||||||
ipLocation: string;
|
ipLocation: string;
|
||||||
followers: number;
|
followers: number;
|
||||||
cooperationSource: string;
|
cooperationSource: string;
|
||||||
|
tags: string;
|
||||||
action: "create" | "update" | "error";
|
action: "create" | "update" | "error";
|
||||||
errors: string[];
|
errors: string[];
|
||||||
}>;
|
}>;
|
||||||
@@ -1991,6 +1993,14 @@ function ResourcesPage({
|
|||||||
].filter(Boolean),
|
].filter(Boolean),
|
||||||
),
|
),
|
||||||
].sort((left, right) => left.localeCompare(right, "zh-CN")),
|
].sort((left, right) => left.localeCompare(right, "zh-CN")),
|
||||||
|
tags: [
|
||||||
|
...new Set(
|
||||||
|
(account.tags || "")
|
||||||
|
.split(/[、,,;;|]/)
|
||||||
|
.map((item) => item.trim())
|
||||||
|
.filter(Boolean),
|
||||||
|
),
|
||||||
|
],
|
||||||
partnerManagedOnly:
|
partnerManagedOnly:
|
||||||
deliveries.some((item) => item.delegation_bundle_id) &&
|
deliveries.some((item) => item.delegation_bundle_id) &&
|
||||||
deliveries.every((item) => item.delegation_bundle_id),
|
deliveries.every((item) => item.delegation_bundle_id),
|
||||||
@@ -2203,7 +2213,7 @@ function ResourcesPage({
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div className="resource-grid">
|
<div className="resource-grid">
|
||||||
{filteredAccounts.map(({ account, sources, partnerManagedOnly }) => (
|
{filteredAccounts.map(({ account, sources, tags: tagList, partnerManagedOnly }) => (
|
||||||
<article className="resource-card" key={account.id}>
|
<article className="resource-card" key={account.id}>
|
||||||
<div className="resource-card-head">
|
<div className="resource-card-head">
|
||||||
<div className={`avatar xlarge ${avatarColor(account.nickname)}`}>{account.nickname.slice(0, 1)}</div>
|
<div className={`avatar xlarge ${avatarColor(account.nickname)}`}>{account.nickname.slice(0, 1)}</div>
|
||||||
@@ -2227,6 +2237,14 @@ function ResourcesPage({
|
|||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
{tagList.length > 0 && (
|
||||||
|
<div className="resource-tags">
|
||||||
|
<span>账号标签</span>
|
||||||
|
<div>
|
||||||
|
{tagList.map((tag) => <b key={tag}>{tag}</b>)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
<div className="resource-card-foot">
|
<div className="resource-card-foot">
|
||||||
<span>最近合作 {formatDate(account.last_seen_at)}</span>
|
<span>最近合作 {formatDate(account.last_seen_at)}</span>
|
||||||
{account.profile_url && <a href={account.profile_url} target="_blank" rel="noreferrer">查看主页 ↗</a>}
|
{account.profile_url && <a href={account.profile_url} target="_blank" rel="noreferrer">查看主页 ↗</a>}
|
||||||
@@ -2282,7 +2300,7 @@ function ResourcesPage({
|
|||||||
</label>
|
</label>
|
||||||
{!importPreview && (
|
{!importPreview && (
|
||||||
<div className="import-template-note">
|
<div className="import-template-note">
|
||||||
<div><strong>还没有模板?</strong><span>只有账号链接必填;已有昵称、账号ID、IP属地、粉丝数和合作来源可直接填写,系统只补全空缺字段。</span></div>
|
<div><strong>还没有模板?</strong><span>只有账号链接必填;已有昵称、账号ID、IP属地、粉丝数、合作来源和标签可直接填写,系统只补全空缺字段。</span></div>
|
||||||
<a className="ghost-button" download href="/KOC资源导入模板.xlsx">下载模板</a>
|
<a className="ghost-button" download href="/KOC资源导入模板.xlsx">下载模板</a>
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
@@ -2303,7 +2321,7 @@ function ResourcesPage({
|
|||||||
<span>{row.rowNumber}</span>
|
<span>{row.rowNumber}</span>
|
||||||
<span><strong>{row.nickname || "—"}</strong><small>{row.platform || "未填写平台"}</small></span>
|
<span><strong>{row.nickname || "—"}</strong><small>{row.platform || "未填写平台"}</small></span>
|
||||||
<span>{row.publicAccountId || "主页识别"}</span>
|
<span>{row.publicAccountId || "主页识别"}</span>
|
||||||
<span><strong>{row.ipLocation}</strong><small>{row.cooperationSource || "未填写来源"}</small></span>
|
<span><strong>{row.ipLocation}</strong><small>{row.cooperationSource || "未填写来源"}</small>{row.tags && <em>{row.tags}</em>}</span>
|
||||||
<span className={`import-result ${row.action}`}>
|
<span className={`import-result ${row.action}`}>
|
||||||
{row.action === "create" ? "新增" : row.action === "update" ? "更新" : row.errors.join(";")}
|
{row.action === "create" ? "新增" : row.action === "update" ? "更新" : row.errors.join(";")}
|
||||||
</span>
|
</span>
|
||||||
|
|||||||
@@ -34,6 +34,13 @@ import {
|
|||||||
DistributionReleaseError,
|
DistributionReleaseError,
|
||||||
releaseUnfinishedDistribution,
|
releaseUnfinishedDistribution,
|
||||||
} from "../../../lib/distribution-release-service";
|
} from "../../../lib/distribution-release-service";
|
||||||
|
import {
|
||||||
|
resolveWecomConfig,
|
||||||
|
sendWecomAppMessage,
|
||||||
|
sendWecomRobotMessage,
|
||||||
|
WecomClientError,
|
||||||
|
type WecomBindings,
|
||||||
|
} from "../../../lib/wecom-client";
|
||||||
import { isManagerRequest } from "../../../lib/user-auth";
|
import { isManagerRequest } from "../../../lib/user-auth";
|
||||||
|
|
||||||
type ActionBody = {
|
type ActionBody = {
|
||||||
@@ -365,6 +372,66 @@ export async function POST(request: Request) {
|
|||||||
)
|
)
|
||||||
.bind(exposure, views, distributionId)
|
.bind(exposure, views, distributionId)
|
||||||
.run();
|
.run();
|
||||||
|
} else if (body.action === "bind_wecom_external_id") {
|
||||||
|
const partnerId = String(body.partnerId ?? "").trim().slice(0, 80);
|
||||||
|
const externalId = String(body.wecomExternalUserId ?? "")
|
||||||
|
.trim()
|
||||||
|
.slice(0, 128);
|
||||||
|
if (!partnerId) {
|
||||||
|
return Response.json(
|
||||||
|
{ error: "缺少 partnerId" },
|
||||||
|
{ status: 400 },
|
||||||
|
);
|
||||||
|
}
|
||||||
|
await db
|
||||||
|
.prepare(
|
||||||
|
"UPDATE partners SET wecom_external_user_id = ? WHERE id = ?",
|
||||||
|
)
|
||||||
|
.bind(externalId || null, partnerId)
|
||||||
|
.run();
|
||||||
|
return Response.json({
|
||||||
|
partnerId,
|
||||||
|
wecomExternalUserId: externalId || null,
|
||||||
|
});
|
||||||
|
} else if (body.action === "send_test_wecom") {
|
||||||
|
const wecomConfig = resolveWecomConfig(
|
||||||
|
env as unknown as WecomBindings,
|
||||||
|
);
|
||||||
|
const partnerId = String(body.partnerId ?? "").trim();
|
||||||
|
let partnerExternalId: string | null = null;
|
||||||
|
if (partnerId) {
|
||||||
|
const row = await db
|
||||||
|
.prepare(
|
||||||
|
"SELECT wecom_external_user_id FROM partners WHERE id = ?",
|
||||||
|
)
|
||||||
|
.bind(partnerId)
|
||||||
|
.first<{ wecom_external_user_id: string | null }>();
|
||||||
|
partnerExternalId = row?.wecom_external_user_id ?? null;
|
||||||
|
}
|
||||||
|
const testContent = `[KOC LOOP 测试] 群机器人连通性正常,时间 ${new Date().toISOString()}`;
|
||||||
|
let robotStatus: "ok" | "skipped" = "skipped";
|
||||||
|
if (wecomConfig.robotWebhook) {
|
||||||
|
await sendWecomRobotMessage(testContent, wecomConfig);
|
||||||
|
robotStatus = "ok";
|
||||||
|
}
|
||||||
|
let appStatus: "ok" | "skipped" | "failed" = "skipped";
|
||||||
|
if (
|
||||||
|
partnerExternalId &&
|
||||||
|
wecomConfig.corpId &&
|
||||||
|
wecomConfig.agentId &&
|
||||||
|
wecomConfig.secret
|
||||||
|
) {
|
||||||
|
const result = await sendWecomAppMessage(
|
||||||
|
[partnerExternalId],
|
||||||
|
testContent,
|
||||||
|
wecomConfig,
|
||||||
|
);
|
||||||
|
appStatus = result.failed > 0 ? "failed" : "ok";
|
||||||
|
}
|
||||||
|
return Response.json({
|
||||||
|
robot: robotStatus,
|
||||||
|
app: appStatus,
|
||||||
|
});
|
||||||
} else {
|
} else {
|
||||||
return Response.json({ error: "不支持的操作" }, { status: 400 });
|
return Response.json({ error: "不支持的操作" }, { status: 400 });
|
||||||
}
|
}
|
||||||
@@ -376,7 +443,8 @@ export async function POST(request: Request) {
|
|||||||
{
|
{
|
||||||
status:
|
status:
|
||||||
error instanceof FeishuSourceError ||
|
error instanceof FeishuSourceError ||
|
||||||
error instanceof DistributionReleaseError
|
error instanceof DistributionReleaseError ||
|
||||||
|
error instanceof WecomClientError
|
||||||
? error.status
|
? error.status
|
||||||
: 500,
|
: 500,
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ type AccountRow = {
|
|||||||
followers: number;
|
followers: number;
|
||||||
post_count: number;
|
post_count: number;
|
||||||
cooperation_source: string;
|
cooperation_source: string;
|
||||||
|
tags: string;
|
||||||
first_seen_at: string;
|
first_seen_at: string;
|
||||||
last_seen_at: string;
|
last_seen_at: string;
|
||||||
};
|
};
|
||||||
@@ -111,6 +112,7 @@ async function exportAccounts(accountIds: string[]) {
|
|||||||
"粉丝数",
|
"粉丝数",
|
||||||
"合作发布数",
|
"合作发布数",
|
||||||
"历史合作来源",
|
"历史合作来源",
|
||||||
|
"标签",
|
||||||
"资源归属",
|
"资源归属",
|
||||||
"首次合作时间",
|
"首次合作时间",
|
||||||
"最近合作时间",
|
"最近合作时间",
|
||||||
@@ -140,6 +142,7 @@ async function exportAccounts(accountIds: string[]) {
|
|||||||
account.followers,
|
account.followers,
|
||||||
account.post_count,
|
account.post_count,
|
||||||
sources.join("、"),
|
sources.join("、"),
|
||||||
|
account.tags || "",
|
||||||
partnerManagedOnly ? "合作社资源 · 不可直联" : "可直联",
|
partnerManagedOnly ? "合作社资源 · 不可直联" : "可直联",
|
||||||
formatExportDate(account.first_seen_at),
|
formatExportDate(account.first_seen_at),
|
||||||
formatExportDate(account.last_seen_at),
|
formatExportDate(account.last_seen_at),
|
||||||
@@ -163,6 +166,7 @@ async function exportAccounts(accountIds: string[]) {
|
|||||||
14,
|
14,
|
||||||
14,
|
14,
|
||||||
32,
|
32,
|
||||||
|
28,
|
||||||
22,
|
22,
|
||||||
21,
|
21,
|
||||||
21,
|
21,
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ import {
|
|||||||
import { getRuntimeEnv } from "../../../lib/runtime-env";
|
import { getRuntimeEnv } from "../../../lib/runtime-env";
|
||||||
import {
|
import {
|
||||||
mergeCooperationSources,
|
mergeCooperationSources,
|
||||||
|
mergeTags,
|
||||||
normalizeProfileUrl,
|
normalizeProfileUrl,
|
||||||
parseResourceImportFile,
|
parseResourceImportFile,
|
||||||
RESOURCE_IMPORT_MAX_BYTES,
|
RESOURCE_IMPORT_MAX_BYTES,
|
||||||
@@ -27,6 +28,7 @@ type AccountRow = {
|
|||||||
ip_location: string;
|
ip_location: string;
|
||||||
followers: number;
|
followers: number;
|
||||||
cooperation_source: string;
|
cooperation_source: string;
|
||||||
|
tags: string;
|
||||||
};
|
};
|
||||||
|
|
||||||
type AnalyzedRow = ResourceImportRow & {
|
type AnalyzedRow = ResourceImportRow & {
|
||||||
@@ -34,6 +36,7 @@ type AnalyzedRow = ResourceImportRow & {
|
|||||||
accountId: string;
|
accountId: string;
|
||||||
platformUid: string;
|
platformUid: string;
|
||||||
cooperationSource: string;
|
cooperationSource: string;
|
||||||
|
tags: string;
|
||||||
};
|
};
|
||||||
|
|
||||||
function identityKey(platform: string, value: string) {
|
function identityKey(platform: string, value: string) {
|
||||||
@@ -46,7 +49,7 @@ async function loadAccounts() {
|
|||||||
return getRawDb()
|
return getRawDb()
|
||||||
.prepare(
|
.prepare(
|
||||||
`SELECT id, platform, platform_uid, public_account_id, nickname,
|
`SELECT id, platform, platform_uid, public_account_id, nickname,
|
||||||
profile_url, ip_location, followers, cooperation_source
|
profile_url, ip_location, followers, cooperation_source, tags
|
||||||
FROM accounts`,
|
FROM accounts`,
|
||||||
)
|
)
|
||||||
.all<AccountRow>();
|
.all<AccountRow>();
|
||||||
@@ -187,6 +190,7 @@ function analyzeRows(rows: ResourceImportRow[], accountRows: AccountRow[]) {
|
|||||||
existing?.cooperation_source ?? "",
|
existing?.cooperation_source ?? "",
|
||||||
row.cooperationSource,
|
row.cooperationSource,
|
||||||
),
|
),
|
||||||
|
tags: mergeTags(existing?.tags ?? "", row.tags),
|
||||||
};
|
};
|
||||||
if (analyzed.action !== "error") {
|
if (analyzed.action !== "error") {
|
||||||
const virtual: AccountRow = {
|
const virtual: AccountRow = {
|
||||||
@@ -199,6 +203,7 @@ function analyzeRows(rows: ResourceImportRow[], accountRows: AccountRow[]) {
|
|||||||
ip_location: row.ipLocation || existing?.ip_location || "待识别",
|
ip_location: row.ipLocation || existing?.ip_location || "待识别",
|
||||||
followers: row.followers || existing?.followers || 0,
|
followers: row.followers || existing?.followers || 0,
|
||||||
cooperation_source: analyzed.cooperationSource,
|
cooperation_source: analyzed.cooperationSource,
|
||||||
|
tags: analyzed.tags,
|
||||||
};
|
};
|
||||||
if (row.profileUrl) profileMap.set(identityKey(row.platform, row.profileUrl), virtual);
|
if (row.profileUrl) profileMap.set(identityKey(row.platform, row.profileUrl), virtual);
|
||||||
if (row.publicAccountId) {
|
if (row.publicAccountId) {
|
||||||
@@ -249,6 +254,7 @@ export async function POST(request: Request) {
|
|||||||
ipLocation: row.ipLocation,
|
ipLocation: row.ipLocation,
|
||||||
followers: row.followers,
|
followers: row.followers,
|
||||||
cooperationSource: row.cooperationSource,
|
cooperationSource: row.cooperationSource,
|
||||||
|
tags: row.tags,
|
||||||
action: row.action,
|
action: row.action,
|
||||||
errors: row.errors,
|
errors: row.errors,
|
||||||
})),
|
})),
|
||||||
@@ -275,6 +281,7 @@ export async function POST(request: Request) {
|
|||||||
WHEN ? != '' AND ? != '待识别' THEN ? ELSE ip_location END,
|
WHEN ? != '' AND ? != '待识别' THEN ? ELSE ip_location END,
|
||||||
followers = CASE WHEN ? = 1 THEN ? ELSE followers END,
|
followers = CASE WHEN ? = 1 THEN ? ELSE followers END,
|
||||||
cooperation_source = ?,
|
cooperation_source = ?,
|
||||||
|
tags = ?,
|
||||||
last_seen_at = CURRENT_TIMESTAMP
|
last_seen_at = CURRENT_TIMESTAMP
|
||||||
WHERE id = ?`,
|
WHERE id = ?`,
|
||||||
)
|
)
|
||||||
@@ -290,6 +297,7 @@ export async function POST(request: Request) {
|
|||||||
row.followersResolved ? 1 : 0,
|
row.followersResolved ? 1 : 0,
|
||||||
row.followers,
|
row.followers,
|
||||||
row.cooperationSource,
|
row.cooperationSource,
|
||||||
|
row.tags,
|
||||||
row.accountId,
|
row.accountId,
|
||||||
)
|
)
|
||||||
: db
|
: db
|
||||||
@@ -297,8 +305,8 @@ export async function POST(request: Request) {
|
|||||||
`INSERT INTO accounts
|
`INSERT INTO accounts
|
||||||
(id, platform, platform_uid, public_account_id, nickname,
|
(id, platform, platform_uid, public_account_id, nickname,
|
||||||
profile_url, ip_location, followers, post_count, avg_views,
|
profile_url, ip_location, followers, post_count, avg_views,
|
||||||
cooperation_source)
|
cooperation_source, tags)
|
||||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, 0, 0, ?)`,
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, 0, 0, ?, ?)`,
|
||||||
)
|
)
|
||||||
.bind(
|
.bind(
|
||||||
row.accountId,
|
row.accountId,
|
||||||
@@ -310,6 +318,7 @@ export async function POST(request: Request) {
|
|||||||
row.ipLocation,
|
row.ipLocation,
|
||||||
row.followers,
|
row.followers,
|
||||||
row.cooperationSource,
|
row.cooperationSource,
|
||||||
|
row.tags,
|
||||||
),
|
),
|
||||||
);
|
);
|
||||||
if (statements.length > 0) await db.batch(statements);
|
if (statements.length > 0) await db.batch(statements);
|
||||||
|
|||||||
@@ -2008,6 +2008,18 @@ a {
|
|||||||
font-size: 8px;
|
font-size: 8px;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.import-preview-row em {
|
||||||
|
display: block;
|
||||||
|
margin-top: 3px;
|
||||||
|
color: #4a5575;
|
||||||
|
background: #eef1f6;
|
||||||
|
border-radius: 4px;
|
||||||
|
padding: 2px 6px;
|
||||||
|
font-size: 8px;
|
||||||
|
font-style: normal;
|
||||||
|
font-weight: 580;
|
||||||
|
}
|
||||||
|
|
||||||
.import-result {
|
.import-result {
|
||||||
color: #557269;
|
color: #557269;
|
||||||
font-weight: 650;
|
font-weight: 650;
|
||||||
@@ -2175,6 +2187,31 @@ a {
|
|||||||
background: #fff3e8;
|
background: #fff3e8;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.resource-tags {
|
||||||
|
margin-top: 13px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.resource-tags > span {
|
||||||
|
color: #9ba4a1;
|
||||||
|
font-size: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.resource-tags > div {
|
||||||
|
display: flex;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
gap: 5px;
|
||||||
|
margin-top: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.resource-tags b {
|
||||||
|
padding: 5px 7px;
|
||||||
|
border-radius: 6px;
|
||||||
|
color: #4a5575;
|
||||||
|
background: #eef1f6;
|
||||||
|
font-size: 8px;
|
||||||
|
font-weight: 580;
|
||||||
|
}
|
||||||
|
|
||||||
.resource-card-foot {
|
.resource-card-foot {
|
||||||
display: flex;
|
display: flex;
|
||||||
justify-content: space-between;
|
justify-content: space-between;
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ export const partners = mysqlTable("partners", {
|
|||||||
name: varchar("name", { length: 255 }).notNull(),
|
name: varchar("name", { length: 255 }).notNull(),
|
||||||
wecomName: varchar("wecom_name", { length: 255 }).notNull(),
|
wecomName: varchar("wecom_name", { length: 255 }).notNull(),
|
||||||
owner: varchar("owner", { length: 255 }).notNull().default("运营组"),
|
owner: varchar("owner", { length: 255 }).notNull().default("运营组"),
|
||||||
|
wecomExternalUserId: varchar("wecom_external_user_id", { length: 128 }),
|
||||||
claimedTotal: int("claimed_total").notNull().default(0),
|
claimedTotal: int("claimed_total").notNull().default(0),
|
||||||
completedTotal: int("completed_total").notNull().default(0),
|
completedTotal: int("completed_total").notNull().default(0),
|
||||||
createdAt: timestamp("created_at"),
|
createdAt: timestamp("created_at"),
|
||||||
@@ -81,6 +82,7 @@ export const accounts = mysqlTable(
|
|||||||
cooperationSource: varchar("cooperation_source", { length: 500 })
|
cooperationSource: varchar("cooperation_source", { length: 500 })
|
||||||
.notNull()
|
.notNull()
|
||||||
.default(""),
|
.default(""),
|
||||||
|
tags: varchar("tags", { length: 500 }).notNull().default(""),
|
||||||
firstSeenAt: timestamp("first_seen_at"),
|
firstSeenAt: timestamp("first_seen_at"),
|
||||||
lastSeenAt: timestamp("last_seen_at"),
|
lastSeenAt: timestamp("last_seen_at"),
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ export function getUploadBucket() {
|
|||||||
|
|
||||||
export async function ensureSchema(database?: DatabaseClient) {
|
export async function ensureSchema(database?: DatabaseClient) {
|
||||||
const db = database ?? getRawDb();
|
const db = database ?? getRawDb();
|
||||||
|
await ensurePartnersWecomColumn(db);
|
||||||
{
|
{
|
||||||
await db.prepare("SELECT id FROM tasks LIMIT 1").all();
|
await db.prepare("SELECT id FROM tasks LIMIT 1").all();
|
||||||
|
|
||||||
@@ -81,6 +82,7 @@ export async function ensureSchema(database?: DatabaseClient) {
|
|||||||
name TEXT NOT NULL,
|
name TEXT NOT NULL,
|
||||||
wecom_name TEXT NOT NULL,
|
wecom_name TEXT NOT NULL,
|
||||||
owner TEXT NOT NULL DEFAULT '运营组',
|
owner TEXT NOT NULL DEFAULT '运营组',
|
||||||
|
wecom_external_user_id TEXT,
|
||||||
claimed_total INTEGER NOT NULL DEFAULT 0,
|
claimed_total INTEGER NOT NULL DEFAULT 0,
|
||||||
completed_total INTEGER NOT NULL DEFAULT 0,
|
completed_total INTEGER NOT NULL DEFAULT 0,
|
||||||
created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP
|
created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP
|
||||||
@@ -127,6 +129,7 @@ export async function ensureSchema(database?: DatabaseClient) {
|
|||||||
post_count INTEGER NOT NULL DEFAULT 0,
|
post_count INTEGER NOT NULL DEFAULT 0,
|
||||||
avg_views INTEGER NOT NULL DEFAULT 0,
|
avg_views INTEGER NOT NULL DEFAULT 0,
|
||||||
cooperation_source TEXT NOT NULL DEFAULT '',
|
cooperation_source TEXT NOT NULL DEFAULT '',
|
||||||
|
tags TEXT NOT NULL DEFAULT '',
|
||||||
first_seen_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
first_seen_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
last_seen_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP
|
last_seen_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP
|
||||||
)`,
|
)`,
|
||||||
@@ -293,6 +296,7 @@ export async function ensureSchema(database?: DatabaseClient) {
|
|||||||
"public_account_id",
|
"public_account_id",
|
||||||
"public_account_id TEXT NOT NULL DEFAULT ''",
|
"public_account_id TEXT NOT NULL DEFAULT ''",
|
||||||
);
|
);
|
||||||
|
await ensureColumn("accounts", "tags", "tags TEXT NOT NULL DEFAULT ''");
|
||||||
await ensureColumn("distributions", "claim_id", "claim_id TEXT");
|
await ensureColumn("distributions", "claim_id", "claim_id TEXT");
|
||||||
await ensureColumn(
|
await ensureColumn(
|
||||||
"distributions",
|
"distributions",
|
||||||
@@ -832,6 +836,32 @@ export function uid(prefix: string) {
|
|||||||
return `${prefix}-${crypto.randomUUID().slice(0, 8)}`;
|
return `${prefix}-${crypto.randomUUID().slice(0, 8)}`;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function ensurePartnersWecomColumn(db: DatabaseClient) {
|
||||||
|
try {
|
||||||
|
const rows = await db
|
||||||
|
.prepare(
|
||||||
|
`SELECT COLUMN_NAME
|
||||||
|
FROM INFORMATION_SCHEMA.COLUMNS
|
||||||
|
WHERE TABLE_SCHEMA = DATABASE()
|
||||||
|
AND TABLE_NAME = 'partners'
|
||||||
|
AND COLUMN_NAME = 'wecom_external_user_id'`,
|
||||||
|
)
|
||||||
|
.all<{ COLUMN_NAME: string }>();
|
||||||
|
if (rows.results.length === 0) {
|
||||||
|
await db
|
||||||
|
.prepare(
|
||||||
|
"ALTER TABLE partners ADD COLUMN wecom_external_user_id TEXT",
|
||||||
|
)
|
||||||
|
.run();
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.warn(
|
||||||
|
"[KOC LOOP] failed to ensure partners.wecom_external_user_id column",
|
||||||
|
error,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
export function hashText(value: string) {
|
export function hashText(value: string) {
|
||||||
let hash = 2166136261;
|
let hash = 2166136261;
|
||||||
for (let index = 0; index < value.length; index += 1) {
|
for (let index = 0; index < value.length; index += 1) {
|
||||||
|
|||||||
@@ -13,6 +13,7 @@ export type ResourceImportRow = {
|
|||||||
followers: number;
|
followers: number;
|
||||||
followersResolved: boolean;
|
followersResolved: boolean;
|
||||||
cooperationSource: string;
|
cooperationSource: string;
|
||||||
|
tags: string;
|
||||||
errors: string[];
|
errors: string[];
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -23,6 +24,7 @@ const HEADER_ALIASES = {
|
|||||||
ipLocation: ["IP属地", "IP地址", "IP地区", "IP所在地"],
|
ipLocation: ["IP属地", "IP地址", "IP地区", "IP所在地"],
|
||||||
followers: ["粉丝数", "粉丝", "粉丝量"],
|
followers: ["粉丝数", "粉丝", "粉丝量"],
|
||||||
cooperationSource: ["合作来源", "资源来源", "历史合作来源"],
|
cooperationSource: ["合作来源", "资源来源", "历史合作来源"],
|
||||||
|
tags: ["标签", "账号标签", "人设标签"],
|
||||||
} as const;
|
} as const;
|
||||||
|
|
||||||
type CanonicalHeader = keyof typeof HEADER_ALIASES;
|
type CanonicalHeader = keyof typeof HEADER_ALIASES;
|
||||||
@@ -60,9 +62,11 @@ function parseWorksheet(xml: string, sharedStrings: string[]) {
|
|||||||
const rowAttributes = rowMatch[1];
|
const rowAttributes = rowMatch[1];
|
||||||
const rowNumber = Number(rowAttributes.match(/\br="(\d+)"/)?.[1] ?? rows.length + 1);
|
const rowNumber = Number(rowAttributes.match(/\br="(\d+)"/)?.[1] ?? rows.length + 1);
|
||||||
const values: string[] = [];
|
const values: string[] = [];
|
||||||
for (const cellMatch of rowMatch[2].matchAll(/<c\b([^>]*)>([\s\S]*?)<\/c>/g)) {
|
for (const cellMatch of rowMatch[2].matchAll(
|
||||||
|
/<c\b([^>]*?)(?:\/>|>([\s\S]*?)<\/c>)/g,
|
||||||
|
)) {
|
||||||
const attributes = cellMatch[1];
|
const attributes = cellMatch[1];
|
||||||
const body = cellMatch[2];
|
const body = cellMatch[2] ?? "";
|
||||||
const reference = attributes.match(/\br="([A-Z]+\d+)"/i)?.[1] ?? "A1";
|
const reference = attributes.match(/\br="([A-Z]+\d+)"/i)?.[1] ?? "A1";
|
||||||
const type = attributes.match(/\bt="([^"]+)"/)?.[1] ?? "";
|
const type = attributes.match(/\bt="([^"]+)"/)?.[1] ?? "";
|
||||||
const rawValue = body.match(/<v>([\s\S]*?)<\/v>/)?.[1] ?? "";
|
const rawValue = body.match(/<v>([\s\S]*?)<\/v>/)?.[1] ?? "";
|
||||||
@@ -259,6 +263,7 @@ function normalizeRows(rows: string[][]) {
|
|||||||
followers: parsedFollowers.value,
|
followers: parsedFollowers.value,
|
||||||
followersResolved: parsedFollowers.resolved,
|
followersResolved: parsedFollowers.resolved,
|
||||||
cooperationSource: valueAt(source, header.mapping, "cooperationSource"),
|
cooperationSource: valueAt(source, header.mapping, "cooperationSource"),
|
||||||
|
tags: normalizeTags(valueAt(source, header.mapping, "tags")),
|
||||||
errors,
|
errors,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -320,3 +325,18 @@ export function mergeCooperationSources(existing: string, incoming: string) {
|
|||||||
),
|
),
|
||||||
].join("、");
|
].join("、");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function normalizeTags(value: string) {
|
||||||
|
return [
|
||||||
|
...new Set(
|
||||||
|
value
|
||||||
|
.split(/[、,,;;|]/)
|
||||||
|
.map((item) => item.trim())
|
||||||
|
.filter(Boolean),
|
||||||
|
),
|
||||||
|
].join("、");
|
||||||
|
}
|
||||||
|
|
||||||
|
export function mergeTags(existing: string, incoming: string) {
|
||||||
|
return normalizeTags(`${existing}、${incoming}`);
|
||||||
|
}
|
||||||
|
|||||||
@@ -20,6 +20,12 @@ export type RuntimeEnv = {
|
|||||||
AI_TOOL_CENTER_MCP_KEY?: string;
|
AI_TOOL_CENTER_MCP_KEY?: string;
|
||||||
COLLECTION_MCP_URL?: string;
|
COLLECTION_MCP_URL?: string;
|
||||||
COLLECTION_MCP_KEY?: string;
|
COLLECTION_MCP_KEY?: string;
|
||||||
|
WECOM_CORP_ID?: string;
|
||||||
|
WECOM_AGENT_ID?: string;
|
||||||
|
WECOM_SECRET?: string;
|
||||||
|
WECOM_ROBOT_WEBHOOK?: string;
|
||||||
|
WECOM_NOTIFY_DUE_DAYS?: string;
|
||||||
|
WECOM_NOTIFY_ENABLED?: string;
|
||||||
SEED_DEMO_DATA?: string;
|
SEED_DEMO_DATA?: string;
|
||||||
ENABLE_SCHEDULER?: string;
|
ENABLE_SCHEDULER?: string;
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -8,6 +8,11 @@ import {
|
|||||||
} from "./mcp-collection-client";
|
} from "./mcp-collection-client";
|
||||||
import { ensureSchema, getRawDb } from "./mvp-db";
|
import { ensureSchema, getRawDb } from "./mvp-db";
|
||||||
import { getRuntimeEnv, isEnabled } from "./runtime-env";
|
import { getRuntimeEnv, isEnabled } from "./runtime-env";
|
||||||
|
import {
|
||||||
|
resolveWecomConfig,
|
||||||
|
type WecomBindings,
|
||||||
|
} from "./wecom-client";
|
||||||
|
import { runDueSoonWecomNotifications } from "./wecom-notifier-service";
|
||||||
|
|
||||||
declare global {
|
declare global {
|
||||||
var __kocLoopScheduler: ScheduledTask | undefined;
|
var __kocLoopScheduler: ScheduledTask | undefined;
|
||||||
@@ -17,14 +22,31 @@ async function runDailyJob() {
|
|||||||
await withDatabaseLock("koc-loop-daily-collection", 0, async () => {
|
await withDatabaseLock("koc-loop-daily-collection", 0, async () => {
|
||||||
await ensureSchema();
|
await ensureSchema();
|
||||||
const db = getRawDb();
|
const db = getRawDb();
|
||||||
|
const env = getRuntimeEnv();
|
||||||
const config = resolveCollectionMcpConfig(
|
const config = resolveCollectionMcpConfig(
|
||||||
getRuntimeEnv() as unknown as CollectionMcpBindings,
|
env as unknown as CollectionMcpBindings,
|
||||||
);
|
);
|
||||||
const collections = await runScheduledCollections(db, Date.now(), config);
|
const collections = await runScheduledCollections(db, Date.now(), config);
|
||||||
const accounts = await backfillAccountProfiles(db, config, 10);
|
const accounts = await backfillAccountProfiles(db, config, 10);
|
||||||
|
let wecom: Awaited<ReturnType<typeof runDueSoonWecomNotifications>> | null =
|
||||||
|
null;
|
||||||
|
if (isEnabled(env.WECOM_NOTIFY_ENABLED, true)) {
|
||||||
|
const wecomConfig = resolveWecomConfig(env as unknown as WecomBindings);
|
||||||
|
if (
|
||||||
|
wecomConfig.robotWebhook ||
|
||||||
|
(wecomConfig.corpId && wecomConfig.agentId && wecomConfig.secret)
|
||||||
|
) {
|
||||||
|
try {
|
||||||
|
wecom = await runDueSoonWecomNotifications(db, wecomConfig);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("[KOC LOOP] wecom notify failed", error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
console.info("[KOC LOOP] daily scheduler completed", {
|
console.info("[KOC LOOP] daily scheduler completed", {
|
||||||
collections,
|
collections,
|
||||||
accounts,
|
accounts,
|
||||||
|
wecom,
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
219
lib/wecom-client.ts
Normal file
219
lib/wecom-client.ts
Normal file
@@ -0,0 +1,219 @@
|
|||||||
|
const WECOM_API_ORIGIN = "https://qyapi.weixin.qq.com";
|
||||||
|
const DEFAULT_DUE_DAYS = 3;
|
||||||
|
|
||||||
|
export type WecomBindings = {
|
||||||
|
WECOM_CORP_ID?: string;
|
||||||
|
WECOM_AGENT_ID?: string;
|
||||||
|
WECOM_SECRET?: string;
|
||||||
|
WECOM_ROBOT_WEBHOOK?: string;
|
||||||
|
WECOM_NOTIFY_DUE_DAYS?: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type WecomConfig = {
|
||||||
|
corpId: string;
|
||||||
|
agentId: string;
|
||||||
|
secret: string;
|
||||||
|
robotWebhook: string;
|
||||||
|
dueDays: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
type FetchLike = typeof fetch;
|
||||||
|
|
||||||
|
type WecomEnvelope = {
|
||||||
|
errcode?: number;
|
||||||
|
errmsg?: string;
|
||||||
|
access_token?: string;
|
||||||
|
expires_in?: number;
|
||||||
|
invaliduser?: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
type CachedAccessToken = {
|
||||||
|
corpId: string;
|
||||||
|
secret: string;
|
||||||
|
token: string;
|
||||||
|
expiresAt: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
let cachedAccessToken: CachedAccessToken | null = null;
|
||||||
|
|
||||||
|
export class WecomClientError extends Error {
|
||||||
|
status: number;
|
||||||
|
|
||||||
|
constructor(message: string, status = 502) {
|
||||||
|
super(message);
|
||||||
|
this.name = "WecomClientError";
|
||||||
|
this.status = status;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function bindingValue(value: unknown) {
|
||||||
|
return String(value ?? "").trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
function safeMessage(value: unknown) {
|
||||||
|
return String(value ?? "").trim().slice(0, 240);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function resolveWecomConfig(bindings: WecomBindings): WecomConfig {
|
||||||
|
return {
|
||||||
|
corpId: bindingValue(bindings.WECOM_CORP_ID),
|
||||||
|
agentId: bindingValue(bindings.WECOM_AGENT_ID),
|
||||||
|
secret: bindingValue(bindings.WECOM_SECRET),
|
||||||
|
robotWebhook: bindingValue(bindings.WECOM_ROBOT_WEBHOOK),
|
||||||
|
dueDays: parseDueDays(bindings.WECOM_NOTIFY_DUE_DAYS),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseDueDays(value: string | undefined) {
|
||||||
|
const parsed = Number(bindingValue(value));
|
||||||
|
if (!Number.isFinite(parsed) || parsed < 1) return DEFAULT_DUE_DAYS;
|
||||||
|
return Math.min(30, Math.floor(parsed));
|
||||||
|
}
|
||||||
|
|
||||||
|
function hasAppCredentials(config: WecomConfig) {
|
||||||
|
return Boolean(config.corpId && config.agentId && config.secret);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function readEnvelope(
|
||||||
|
response: Response,
|
||||||
|
fallbackMessage: string,
|
||||||
|
): Promise<WecomEnvelope> {
|
||||||
|
const text = await response.text();
|
||||||
|
try {
|
||||||
|
return JSON.parse(text) as WecomEnvelope;
|
||||||
|
} catch {
|
||||||
|
throw new WecomClientError(
|
||||||
|
`${fallbackMessage}(企业微信返回了非 JSON 响应)`,
|
||||||
|
502,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function ensureOk(
|
||||||
|
payload: WecomEnvelope,
|
||||||
|
fallbackMessage: string,
|
||||||
|
) {
|
||||||
|
const code = Number(payload.errcode ?? 0);
|
||||||
|
if (code === 0) return;
|
||||||
|
const message = safeMessage(payload.errmsg) || fallbackMessage;
|
||||||
|
if (code === 40014 || code === 42001) {
|
||||||
|
throw new WecomClientError(`企业微信 access_token 无效:${message}`, 401);
|
||||||
|
}
|
||||||
|
throw new WecomClientError(`${fallbackMessage}(${message})`, 502);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function fetchAccessToken(
|
||||||
|
config: WecomConfig,
|
||||||
|
fetchImpl: FetchLike,
|
||||||
|
) {
|
||||||
|
if (
|
||||||
|
cachedAccessToken?.corpId === config.corpId &&
|
||||||
|
cachedAccessToken?.secret === config.secret &&
|
||||||
|
cachedAccessToken.expiresAt > Date.now() + 60_000
|
||||||
|
) {
|
||||||
|
return cachedAccessToken.token;
|
||||||
|
}
|
||||||
|
const url = new URL(`${WECOM_API_ORIGIN}/cgi-bin/gettoken`);
|
||||||
|
url.searchParams.set("corpid", config.corpId);
|
||||||
|
url.searchParams.set("corpsecret", config.secret);
|
||||||
|
const response = await fetchImpl(url.toString(), {
|
||||||
|
method: "GET",
|
||||||
|
signal: AbortSignal.timeout(12_000),
|
||||||
|
});
|
||||||
|
const payload = await readEnvelope(response, "获取企业微信 access_token 失败");
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new WecomClientError(
|
||||||
|
`获取企业微信 access_token 失败(HTTP ${response.status})`,
|
||||||
|
502,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
ensureOk(payload, "获取企业微信 access_token 失败");
|
||||||
|
const token = bindingValue(payload.access_token);
|
||||||
|
if (!token) {
|
||||||
|
throw new WecomClientError("企业微信未返回有效 access_token", 502);
|
||||||
|
}
|
||||||
|
cachedAccessToken = {
|
||||||
|
corpId: config.corpId,
|
||||||
|
secret: config.secret,
|
||||||
|
token,
|
||||||
|
expiresAt:
|
||||||
|
Date.now() + Math.max(300, Number(payload.expires_in) || 7_200) * 1_000,
|
||||||
|
};
|
||||||
|
return token;
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function sendWecomRobotMessage(
|
||||||
|
content: string,
|
||||||
|
config: WecomConfig,
|
||||||
|
fetchImpl: FetchLike = fetch,
|
||||||
|
): Promise<void> {
|
||||||
|
if (!config.robotWebhook) {
|
||||||
|
console.warn("[KOC LOOP] wecom robot webhook not configured, skipping");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const response = await fetchImpl(config.robotWebhook, {
|
||||||
|
method: "POST",
|
||||||
|
headers: { "Content-Type": "application/json; charset=utf-8" },
|
||||||
|
body: JSON.stringify({
|
||||||
|
msgtype: "text",
|
||||||
|
text: { content },
|
||||||
|
}),
|
||||||
|
signal: AbortSignal.timeout(10_000),
|
||||||
|
});
|
||||||
|
const payload = await readEnvelope(response, "企业微信群机器人推送失败");
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new WecomClientError(
|
||||||
|
`企业微信群机器人推送失败(HTTP ${response.status})`,
|
||||||
|
502,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
ensureOk(payload, "企业微信群机器人推送失败");
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function sendWecomAppMessage(
|
||||||
|
externalUserIds: string[],
|
||||||
|
content: string,
|
||||||
|
config: WecomConfig,
|
||||||
|
fetchImpl: FetchLike = fetch,
|
||||||
|
): Promise<{ sent: number; failed: number; skipped: boolean }> {
|
||||||
|
const normalized = externalUserIds
|
||||||
|
.map((id) => bindingValue(id))
|
||||||
|
.filter((id) => id.length > 0);
|
||||||
|
if (normalized.length === 0) {
|
||||||
|
return { sent: 0, failed: 0, skipped: true };
|
||||||
|
}
|
||||||
|
if (!hasAppCredentials(config)) {
|
||||||
|
return { sent: 0, failed: 0, skipped: true };
|
||||||
|
}
|
||||||
|
const token = await fetchAccessToken(config, fetchImpl);
|
||||||
|
const url = `${WECOM_API_ORIGIN}/cgi-bin/message/send?access_token=${encodeURIComponent(token)}`;
|
||||||
|
const response = await fetchImpl(url, {
|
||||||
|
method: "POST",
|
||||||
|
headers: { "Content-Type": "application/json; charset=utf-8" },
|
||||||
|
body: JSON.stringify({
|
||||||
|
touser: normalized.join("|"),
|
||||||
|
msgtype: "text",
|
||||||
|
agentid: Number(config.agentId),
|
||||||
|
text: { content },
|
||||||
|
}),
|
||||||
|
signal: AbortSignal.timeout(12_000),
|
||||||
|
});
|
||||||
|
const payload = await readEnvelope(response, "企业微信应用消息推送失败");
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new WecomClientError(
|
||||||
|
`企业微信应用消息推送失败(HTTP ${response.status})`,
|
||||||
|
502,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
ensureOk(payload, "企业微信应用消息推送失败");
|
||||||
|
const invalid = bindingValue(payload.invaliduser).split("|").filter(Boolean);
|
||||||
|
return {
|
||||||
|
sent: Math.max(0, normalized.length - invalid.length),
|
||||||
|
failed: invalid.length,
|
||||||
|
skipped: false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export function clearWecomAccessTokenCacheForTests() {
|
||||||
|
cachedAccessToken = null;
|
||||||
|
}
|
||||||
171
lib/wecom-notifier-service.ts
Normal file
171
lib/wecom-notifier-service.ts
Normal file
@@ -0,0 +1,171 @@
|
|||||||
|
import type { DatabaseClient } from "./database";
|
||||||
|
import { shanghaiDateFromTimestamp } from "./collection-service";
|
||||||
|
import {
|
||||||
|
sendWecomAppMessage,
|
||||||
|
sendWecomRobotMessage,
|
||||||
|
type WecomConfig,
|
||||||
|
} from "./wecom-client";
|
||||||
|
import { getRuntimeEnv } from "./runtime-env";
|
||||||
|
|
||||||
|
type FetchLike = typeof fetch;
|
||||||
|
|
||||||
|
export type WecomNotifySummary = {
|
||||||
|
dueSoonAttempted: number;
|
||||||
|
dueSoonSent: number;
|
||||||
|
dueSoonFailed: number;
|
||||||
|
dueSoonSkipped: number;
|
||||||
|
digestSent: boolean;
|
||||||
|
};
|
||||||
|
|
||||||
|
type DueSoonRow = {
|
||||||
|
distribution_id: string;
|
||||||
|
partner_id: string;
|
||||||
|
partner_name: string;
|
||||||
|
wecom_external_user_id: string | null;
|
||||||
|
task_name: string;
|
||||||
|
due_at: string;
|
||||||
|
content_title: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export function computeDueCutoff(today: string, dueDays: number) {
|
||||||
|
const match = /^(\d{4})-(\d{2})-(\d{2})$/.exec(today);
|
||||||
|
if (!match) return today;
|
||||||
|
const [, y, m, d] = match;
|
||||||
|
const date = new Date(
|
||||||
|
Date.UTC(Number(y), Number(m) - 1, Number(d)) + dueDays * 24 * 60 * 60 * 1_000,
|
||||||
|
);
|
||||||
|
return date.toISOString().slice(0, 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function runDueSoonWecomNotifications(
|
||||||
|
db: DatabaseClient,
|
||||||
|
config: WecomConfig,
|
||||||
|
fetchImpl: FetchLike = fetch,
|
||||||
|
now: number = Date.now(),
|
||||||
|
): Promise<WecomNotifySummary> {
|
||||||
|
const today = shanghaiDateFromTimestamp(now);
|
||||||
|
const cutoff = computeDueCutoff(today, config.dueDays);
|
||||||
|
const portalUrl = bindingValue(getRuntimeEnv().KOC_PORTAL_URL);
|
||||||
|
|
||||||
|
const result = await db
|
||||||
|
.prepare(
|
||||||
|
`SELECT
|
||||||
|
d.id AS distribution_id,
|
||||||
|
d.partner_id,
|
||||||
|
p.name AS partner_name,
|
||||||
|
p.wecom_external_user_id,
|
||||||
|
t.name AS task_name,
|
||||||
|
t.due_at,
|
||||||
|
c.title AS content_title
|
||||||
|
FROM distributions d
|
||||||
|
JOIN tasks t ON t.id = d.task_id
|
||||||
|
JOIN partners p ON p.id = d.partner_id
|
||||||
|
JOIN contents c ON c.id = d.content_id
|
||||||
|
WHERE (d.publish_url IS NULL OR d.publish_url = '')
|
||||||
|
AND t.due_at IS NOT NULL AND t.due_at != ''
|
||||||
|
AND t.due_at <= ?
|
||||||
|
ORDER BY t.due_at ASC, p.name ASC`,
|
||||||
|
)
|
||||||
|
.bind(cutoff)
|
||||||
|
.all<DueSoonRow>();
|
||||||
|
|
||||||
|
const grouped = new Map<
|
||||||
|
string,
|
||||||
|
{
|
||||||
|
partnerName: string;
|
||||||
|
externalUserId: string | null;
|
||||||
|
rows: DueSoonRow[];
|
||||||
|
}
|
||||||
|
>();
|
||||||
|
for (const row of result.results) {
|
||||||
|
const entry = grouped.get(row.partner_id) ?? {
|
||||||
|
partnerName: row.partner_name,
|
||||||
|
externalUserId: row.wecom_external_user_id,
|
||||||
|
rows: [],
|
||||||
|
};
|
||||||
|
entry.rows.push(row);
|
||||||
|
if (!entry.externalUserId && row.wecom_external_user_id) {
|
||||||
|
entry.externalUserId = row.wecom_external_user_id;
|
||||||
|
}
|
||||||
|
grouped.set(row.partner_id, entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
let dueSoonAttempted = 0;
|
||||||
|
let dueSoonSent = 0;
|
||||||
|
let dueSoonFailed = 0;
|
||||||
|
let dueSoonSkipped = 0;
|
||||||
|
const digestTasks: string[] = [];
|
||||||
|
|
||||||
|
for (const [, entry] of grouped) {
|
||||||
|
dueSoonAttempted += 1;
|
||||||
|
const external = entry.externalUserId
|
||||||
|
? [entry.externalUserId]
|
||||||
|
: [];
|
||||||
|
const lines = entry.rows.slice(0, 5).map((row) => {
|
||||||
|
return `· 《${truncate(row.task_name, 24)}》— ${truncate(row.content_title, 24)}(截止 ${row.due_at})`;
|
||||||
|
});
|
||||||
|
const overflow =
|
||||||
|
entry.rows.length > 5 ? `\n…还有 ${entry.rows.length - 5} 条` : "";
|
||||||
|
const link = portalUrl ? `\n领取链接:${portalUrl}` : "";
|
||||||
|
const content =
|
||||||
|
`${entry.partnerName},你有 ${entry.rows.length} 条内容待发布:\n${lines.join("\n")}${overflow}${link}`;
|
||||||
|
const appResult = await sendWecomAppMessage(
|
||||||
|
external,
|
||||||
|
content,
|
||||||
|
config,
|
||||||
|
fetchImpl,
|
||||||
|
).catch((error: unknown) => {
|
||||||
|
console.warn(
|
||||||
|
"[KOC LOOP] wecom app message failed",
|
||||||
|
{ partner: entry.partnerName, error: safeError(error) },
|
||||||
|
);
|
||||||
|
return null;
|
||||||
|
});
|
||||||
|
if (appResult === null) {
|
||||||
|
dueSoonFailed += 1;
|
||||||
|
} else if (appResult.skipped) {
|
||||||
|
dueSoonSkipped += 1;
|
||||||
|
} else {
|
||||||
|
dueSoonSent += 1;
|
||||||
|
}
|
||||||
|
const earliestDue = entry.rows[0]?.due_at ?? "";
|
||||||
|
digestTasks.push(
|
||||||
|
`· ${entry.partnerName}(${entry.rows.length} 条,最近截止 ${earliestDue})`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let digestSent = false;
|
||||||
|
if (config.robotWebhook && digestTasks.length > 0) {
|
||||||
|
const digest =
|
||||||
|
`今日待发布催办(${today},截止 ≤ ${cutoff}):\n${digestTasks.join("\n")}`;
|
||||||
|
try {
|
||||||
|
await sendWecomRobotMessage(digest, config, fetchImpl);
|
||||||
|
digestSent = true;
|
||||||
|
} catch (error) {
|
||||||
|
console.warn(
|
||||||
|
"[KOC LOOP] wecom robot digest failed",
|
||||||
|
{ error: safeError(error) },
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
dueSoonAttempted,
|
||||||
|
dueSoonSent,
|
||||||
|
dueSoonFailed,
|
||||||
|
dueSoonSkipped,
|
||||||
|
digestSent,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function bindingValue(value: unknown) {
|
||||||
|
return String(value ?? "").trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
function truncate(value: string, max: number) {
|
||||||
|
return value.length > max ? `${value.slice(0, max)}…` : value;
|
||||||
|
}
|
||||||
|
|
||||||
|
function safeError(error: unknown) {
|
||||||
|
return error instanceof Error ? error.message : String(error);
|
||||||
|
}
|
||||||
2
mysql/0005_account_tags.sql
Normal file
2
mysql/0005_account_tags.sql
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
ALTER TABLE accounts
|
||||||
|
ADD COLUMN tags VARCHAR(500) NOT NULL DEFAULT '' AFTER cooperation_source;
|
||||||
@@ -3,6 +3,7 @@ import type { NextConfig } from "next";
|
|||||||
const nextConfig: NextConfig = {
|
const nextConfig: NextConfig = {
|
||||||
output: "standalone",
|
output: "standalone",
|
||||||
serverExternalPackages: ["mysql2"],
|
serverExternalPackages: ["mysql2"],
|
||||||
|
allowedDevOrigins: ["192.168.30.90"],
|
||||||
};
|
};
|
||||||
|
|
||||||
export default nextConfig;
|
export default nextConfig;
|
||||||
|
|||||||
Binary file not shown.
@@ -3,6 +3,7 @@ import test from "node:test";
|
|||||||
import { buildRecoveryWorkbook } from "../lib/recovery-workbook.ts";
|
import { buildRecoveryWorkbook } from "../lib/recovery-workbook.ts";
|
||||||
import {
|
import {
|
||||||
mergeCooperationSources,
|
mergeCooperationSources,
|
||||||
|
mergeTags,
|
||||||
normalizeProfileUrl,
|
normalizeProfileUrl,
|
||||||
parseResourceFollowers,
|
parseResourceFollowers,
|
||||||
parseResourceImportFile,
|
parseResourceImportFile,
|
||||||
@@ -27,6 +28,7 @@ test("parses CSV resources and normalizes public profile data", () => {
|
|||||||
followers: 0,
|
followers: 0,
|
||||||
followersResolved: false,
|
followersResolved: false,
|
||||||
cooperationSource: "林林KOC社群",
|
cooperationSource: "林林KOC社群",
|
||||||
|
tags: "",
|
||||||
errors: [],
|
errors: [],
|
||||||
});
|
});
|
||||||
assert.equal(resourcePlatformUid(rows[0]), "abc123");
|
assert.equal(resourcePlatformUid(rows[0]), "abc123");
|
||||||
@@ -115,3 +117,16 @@ test("normalizes profile URLs and merges cooperation sources", () => {
|
|||||||
"林林社群、木子、历史表格",
|
"林林社群、木子、历史表格",
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
test("parses and normalizes the optional tags column", () => {
|
||||||
|
const csv = [
|
||||||
|
"账号链接,标签",
|
||||||
|
'"https://www.xiaohongshu.com/user/profile/abc123","美食探店, 旅游出行, 美食探店"',
|
||||||
|
].join("\n");
|
||||||
|
const [row] = parseResourceImportFile(
|
||||||
|
"resources.csv",
|
||||||
|
new TextEncoder().encode(csv),
|
||||||
|
);
|
||||||
|
assert.equal(row.tags, "美食探店、旅游出行");
|
||||||
|
assert.equal(mergeTags("美食探店", "旅游出行;数码汽车"), "美食探店、旅游出行、数码汽车");
|
||||||
|
});
|
||||||
|
|||||||
217
tests/wecom-client.test.mjs
Normal file
217
tests/wecom-client.test.mjs
Normal file
@@ -0,0 +1,217 @@
|
|||||||
|
import assert from "node:assert/strict";
|
||||||
|
import test from "node:test";
|
||||||
|
import {
|
||||||
|
clearWecomAccessTokenCacheForTests,
|
||||||
|
resolveWecomConfig,
|
||||||
|
sendWecomAppMessage,
|
||||||
|
sendWecomRobotMessage,
|
||||||
|
WecomClientError,
|
||||||
|
} from "../lib/wecom-client.ts";
|
||||||
|
|
||||||
|
const baseBindings = {
|
||||||
|
WECOM_CORP_ID: "corp-test",
|
||||||
|
WECOM_AGENT_ID: "1000001",
|
||||||
|
WECOM_SECRET: "secret-test",
|
||||||
|
WECOM_ROBOT_WEBHOOK:
|
||||||
|
"https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=robot-test",
|
||||||
|
WECOM_NOTIFY_DUE_DAYS: "3",
|
||||||
|
};
|
||||||
|
|
||||||
|
function fakeWecom(options = {}) {
|
||||||
|
const calls = [];
|
||||||
|
const fetchImpl = async (input, init = {}) => {
|
||||||
|
const url = new URL(String(input));
|
||||||
|
calls.push({ url, init });
|
||||||
|
if (url.pathname.endsWith("/cgi-bin/gettoken")) {
|
||||||
|
return Response.json({
|
||||||
|
errcode: 0,
|
||||||
|
errmsg: "ok",
|
||||||
|
access_token: "token-test",
|
||||||
|
expires_in: 7200,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
if (url.pathname.endsWith("/cgi-bin/webhook/send")) {
|
||||||
|
return Response.json({
|
||||||
|
errcode: options.robotErrcode ?? 0,
|
||||||
|
errmsg: options.robotErrmsg ?? "ok",
|
||||||
|
});
|
||||||
|
}
|
||||||
|
if (url.pathname.endsWith("/cgi-bin/message/send")) {
|
||||||
|
const body = JSON.parse(String(init.body ?? "{}"));
|
||||||
|
const invalid = options.invalidUser
|
||||||
|
? body.touser.split("|").filter((u) => u === options.invalidUser)
|
||||||
|
: [];
|
||||||
|
return Response.json({
|
||||||
|
errcode: 0,
|
||||||
|
errmsg: "ok",
|
||||||
|
invaliduser: invalid.join("|"),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return new Response("not found", { status: 404 });
|
||||||
|
};
|
||||||
|
return { calls, fetchImpl };
|
||||||
|
}
|
||||||
|
|
||||||
|
test.beforeEach(() => {
|
||||||
|
clearWecomAccessTokenCacheForTests();
|
||||||
|
});
|
||||||
|
|
||||||
|
test("resolveWecomConfig applies dueDays and trims values", () => {
|
||||||
|
const config = resolveWecomConfig({
|
||||||
|
WECOM_CORP_ID: " corp ",
|
||||||
|
WECOM_AGENT_ID: " 10 ",
|
||||||
|
WECOM_SECRET: " s ",
|
||||||
|
WECOM_ROBOT_WEBHOOK: " https://hook ",
|
||||||
|
WECOM_NOTIFY_DUE_DAYS: "5",
|
||||||
|
});
|
||||||
|
assert.equal(config.corpId, "corp");
|
||||||
|
assert.equal(config.agentId, "10");
|
||||||
|
assert.equal(config.secret, "s");
|
||||||
|
assert.equal(config.robotWebhook, "https://hook");
|
||||||
|
assert.equal(config.dueDays, 5);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("resolveWecomConfig falls back to default dueDays for bad input", () => {
|
||||||
|
assert.equal(resolveWecomConfig({ WECOM_NOTIFY_DUE_DAYS: "0" }).dueDays, 3);
|
||||||
|
assert.equal(resolveWecomConfig({ WECOM_NOTIFY_DUE_DAYS: "abc" }).dueDays, 3);
|
||||||
|
assert.equal(
|
||||||
|
resolveWecomConfig({ WECOM_NOTIFY_DUE_DAYS: "100" }).dueDays,
|
||||||
|
30,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomRobotMessage posts text payload and returns void", async () => {
|
||||||
|
const { calls, fetchImpl } = fakeWecom();
|
||||||
|
await sendWecomRobotMessage(
|
||||||
|
"hello",
|
||||||
|
resolveWecomConfig(baseBindings),
|
||||||
|
fetchImpl,
|
||||||
|
);
|
||||||
|
const hookCall = calls.find((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/webhook/send"),
|
||||||
|
);
|
||||||
|
assert.ok(hookCall, "robot webhook was called");
|
||||||
|
const body = JSON.parse(String(hookCall.init.body));
|
||||||
|
assert.equal(body.msgtype, "text");
|
||||||
|
assert.equal(body.text.content, "hello");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomRobotMessage skips silently when webhook is empty", async () => {
|
||||||
|
const { calls, fetchImpl } = fakeWecom();
|
||||||
|
const config = resolveWecomConfig({ ...baseBindings, WECOM_ROBOT_WEBHOOK: "" });
|
||||||
|
await sendWecomRobotMessage("hello", config, fetchImpl);
|
||||||
|
assert.equal(
|
||||||
|
calls.filter((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/webhook/send"),
|
||||||
|
).length,
|
||||||
|
0,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomRobotMessage throws WecomClientError on provider error", async () => {
|
||||||
|
const { fetchImpl } = fakeWecom({
|
||||||
|
robotErrcode: 93000,
|
||||||
|
robotErrmsg: "invalid webhook url",
|
||||||
|
});
|
||||||
|
await assert.rejects(
|
||||||
|
() =>
|
||||||
|
sendWecomRobotMessage(
|
||||||
|
"hello",
|
||||||
|
resolveWecomConfig(baseBindings),
|
||||||
|
fetchImpl,
|
||||||
|
),
|
||||||
|
(error) => {
|
||||||
|
assert.ok(error instanceof WecomClientError);
|
||||||
|
assert.equal(error.status, 502);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomAppMessage fetches access_token then sends to message/send", async () => {
|
||||||
|
const { calls, fetchImpl } = fakeWecom();
|
||||||
|
const result = await sendWecomAppMessage(
|
||||||
|
["user-a", "user-b"],
|
||||||
|
"催办内容",
|
||||||
|
resolveWecomConfig(baseBindings),
|
||||||
|
fetchImpl,
|
||||||
|
);
|
||||||
|
assert.equal(result.sent, 2);
|
||||||
|
assert.equal(result.failed, 0);
|
||||||
|
assert.equal(result.skipped, false);
|
||||||
|
const tokenCall = calls.find((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/gettoken"),
|
||||||
|
);
|
||||||
|
assert.ok(tokenCall, "gettoken was called");
|
||||||
|
const sendCall = calls.find((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/message/send"),
|
||||||
|
);
|
||||||
|
assert.ok(sendCall, "message/send was called");
|
||||||
|
assert.equal(sendCall.url.searchParams.get("access_token"), "token-test");
|
||||||
|
const body = JSON.parse(String(sendCall.init.body));
|
||||||
|
assert.equal(body.touser, "user-a|user-b");
|
||||||
|
assert.equal(body.agentid, 1000001);
|
||||||
|
assert.equal(body.text.content, "催办内容");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomAppMessage skips when no external user ids", async () => {
|
||||||
|
const { calls, fetchImpl } = fakeWecom();
|
||||||
|
const result = await sendWecomAppMessage(
|
||||||
|
[],
|
||||||
|
"催办",
|
||||||
|
resolveWecomConfig(baseBindings),
|
||||||
|
fetchImpl,
|
||||||
|
);
|
||||||
|
assert.equal(result.skipped, true);
|
||||||
|
assert.equal(
|
||||||
|
calls.filter((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/message/send"),
|
||||||
|
).length,
|
||||||
|
0,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomAppMessage skips when corp credentials are missing", async () => {
|
||||||
|
const { calls, fetchImpl } = fakeWecom();
|
||||||
|
const result = await sendWecomAppMessage(
|
||||||
|
["user-a"],
|
||||||
|
"催办",
|
||||||
|
resolveWecomConfig({
|
||||||
|
...baseBindings,
|
||||||
|
WECOM_CORP_ID: "",
|
||||||
|
WECOM_AGENT_ID: "",
|
||||||
|
WECOM_SECRET: "",
|
||||||
|
}),
|
||||||
|
fetchImpl,
|
||||||
|
);
|
||||||
|
assert.equal(result.skipped, true);
|
||||||
|
assert.equal(
|
||||||
|
calls.filter((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/gettoken"),
|
||||||
|
).length,
|
||||||
|
0,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("sendWecomAppMessage reports failed when provider marks invaliduser", async () => {
|
||||||
|
const { fetchImpl } = fakeWecom({ invalidUser: "user-a" });
|
||||||
|
const result = await sendWecomAppMessage(
|
||||||
|
["user-a", "user-b"],
|
||||||
|
"催办",
|
||||||
|
resolveWecomConfig(baseBindings),
|
||||||
|
fetchImpl,
|
||||||
|
);
|
||||||
|
assert.equal(result.sent, 1);
|
||||||
|
assert.equal(result.failed, 1);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("access_token is cached across calls", async () => {
|
||||||
|
const { calls, fetchImpl } = fakeWecom();
|
||||||
|
const config = resolveWecomConfig(baseBindings);
|
||||||
|
await sendWecomAppMessage(["u1"], "a", config, fetchImpl);
|
||||||
|
await sendWecomAppMessage(["u2"], "b", config, fetchImpl);
|
||||||
|
const tokenCalls = calls.filter((c) =>
|
||||||
|
c.url.pathname.endsWith("/cgi-bin/gettoken"),
|
||||||
|
);
|
||||||
|
assert.equal(tokenCalls.length, 1, "access_token cached for second call");
|
||||||
|
});
|
||||||
15
tests/wecom-notifier.test.mjs
Normal file
15
tests/wecom-notifier.test.mjs
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
import assert from "node:assert/strict";
|
||||||
|
import test from "node:test";
|
||||||
|
import { computeDueCutoff } from "../lib/wecom-notifier-service.ts";
|
||||||
|
|
||||||
|
test("computeDueCutoff advances by dueDays and crosses month/year", () => {
|
||||||
|
assert.equal(computeDueCutoff("2026-08-18", 3), "2026-08-21");
|
||||||
|
assert.equal(computeDueCutoff("2026-08-30", 3), "2026-09-02");
|
||||||
|
assert.equal(computeDueCutoff("2026-12-30", 3), "2027-01-02");
|
||||||
|
assert.equal(computeDueCutoff("2026-08-18", 0), "2026-08-18");
|
||||||
|
assert.equal(computeDueCutoff("2026-08-18", 10), "2026-08-28");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("computeDueCutoff returns input unchanged when malformed", () => {
|
||||||
|
assert.equal(computeDueCutoff("not-a-date", 3), "not-a-date");
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user