const cron = require('node-cron') const db = require('./db') const wx = require('./wx') const occurrence = require('./occurrence') const quota = require('./quota') const { importanceRank } = require('./importance') const TEMPLATE_ID = process.env.WX_TEMPLATE_ID const MINIPROGRAM_STATE = process.env.WX_MINIPROGRAM_STATE || 'formal' const TYPE_NAMES = { birthday: '生日', // 老数据 type=lunar_birthday 兼容回显(公历/农历由 isLunar 决定) lunar_birthday: '生日', wedding: '结婚纪念日', engagement: '订婚纪念日', other: '其他纪念日' } function getTypeName(type, customName) { if (type === 'other' && customName) return customName return TYPE_NAMES[type] || '纪念日' } function formatDate(date) { const y = date.getFullYear() const m = String(date.getMonth() + 1).padStart(2, '0') const d = String(date.getDate()).padStart(2, '0') return `${y}年${m}月${d}日` } // 检查今天是否已经给这条纪念日发过提醒 function alreadySentToday(anniversaryId) { const start = new Date() start.setHours(0, 0, 0, 0) const row = db.prepare( 'SELECT COUNT(*) AS n FROM remind_logs WHERE anniversaryId = ? AND sendDate >= ? AND status = ?' ).get(anniversaryId, start.getTime(), 'success') return row.n > 0 } const insertLog = db.prepare(` INSERT INTO remind_logs (anniversaryId, personName, typeName, daysUntil, sendDate, status, error) VALUES (@anniversaryId, @personName, @typeName, @daysUntil, @sendDate, @status, @error) `) // 43101 = 用户拒收或下发次数不足,是我们与微信侧对账的唯一信号 function isQuotaError(err) { return err && err.errcode === 43101 } /** * 处理单个用户的当日提醒 * * 按「当天 > 提前」、同级「重要程度降序」排序后依次发送。 * 记账余额耗尽时不会直接全部放弃,而是发一条「探针」向微信求证(见下方 probe 注释)。 */ async function runForUser(openid, items, today = new Date()) { const due = [] for (const anniv of items) { const target = occurrence.getNextOccurrence(anniv, today) const daysUntil = occurrence.daysBetween(target, today) const remindDays = anniv.remindDays || 0 const isOnDay = daysUntil === 0 const isAhead = remindDays > 0 && daysUntil === remindDays if (!isOnDay && !isAhead) continue if (alreadySentToday(anniv.id)) continue due.push({ anniv, target, daysUntil, kind: isOnDay ? 'onDay' : 'ahead' }) } due.sort((x, y) => { if (x.kind !== y.kind) return x.kind === 'onDay' ? -1 : 1 return importanceRank(x.anniv.importance) - importanceRank(y.anniv.importance) }) let balance = quota.getBalance(openid) let halted = false let ok = 0, fail = 0, skipped = 0 // ---- 探针机制 ---- // balance 只有「前端授权后上报 +1」这一条上升通道。没升级小程序的老用户在微信侧 // 仍在真实累积额度(每存一条纪念日就授权一次)却永远不会上报,于是我们的记账会 // 系统性低估。如果低估到 0 就一条都不发,反而比改造前更糟——改造前至少还会试一下。 // // 所以余额为 0 时仍然把「优先级最高的那一条」发出去,拿它当探针问微信要个答案: // 探针成功 → 证实微信侧确实还有额度、是我们低估了,本轮继续往下发, // 直到遇到 43101 或发完 // 探针返回 43101 → 证实确实没额度,立即中止,剩余全部记 skipped(与原行为一致) // 探针遇非额度错误 → 什么都没证明(网络/配置问题),记 failed、不动余额、不中止 // 每个用户每轮最多探一次:在「有额度」这个结论被证实之前,不值得再赌第二个请求。 let probeUsed = false // 本轮已经用掉探针机会 let quotaProven = false // 探针已证实微信侧有额度,后续不必再受记账余额约束 // errorMsg 默认是「根本没发请求」的兜底文案;真正打了请求并被 43101 拒绝的那一条 // 会传入微信返回的原始错误信息,两者都记为 skipped,但 error 字段要能区分开 const logSkip = (item, errorMsg = 'quota_exhausted') => insertLog.run({ anniversaryId: item.anniv.id, personName: item.anniv.personName, typeName: getTypeName(item.anniv.type, item.anniv.customTypeName), daysUntil: item.daysUntil, sendDate: Date.now(), status: 'skipped', error: errorMsg }) for (const item of due) { if (halted) { logSkip(item); skipped++ continue } // 记账余额为 0 且尚未证实微信侧有额度时,只允许发一条探针 let isProbe = false if (balance <= 0 && !quotaProven) { if (probeUsed) { logSkip(item); skipped++ continue } isProbe = true probeUsed = true } const anniv = item.anniv const typeName = getTypeName(anniv.type, anniv.customTypeName) try { await wx.sendSubscribeMessage({ touser: openid, page: 'pages/index/index', templateId: TEMPLATE_ID, miniprogramState: MINIPROGRAM_STATE, data: { name1: { value: anniv.personName }, thing2: { value: item.daysUntil === 0 ? '今天' : `还有${item.daysUntil}天` }, thing6: { value: formatDate(item.target) }, thing5: { value: anniv.remark || '别忘了准备一份礼物哦!' } } }) // 探针成功时余额怎么记:照常 consume。 // consume 内部是 MAX(0, balance - 1),余额本来就是 0,扣不出负数, // 净效果是「balance 保持 0、sentTotal +1」。 // 之所以不趁机把 balance 调高:微信不提供余额查询,探针只证明「至少还有 1 条」, // 凭空补一个猜出来的数字会让首页的风险预告变成乐观的假话。宁可让 balance 保持 // 保守的 0,靠每轮的探针去发现真实额度——账面继续低估是安全的,因为低估不再等于停发。 quota.consume(openid, 1) balance = Math.max(0, balance - 1) if (isProbe) { quotaProven = true console.log(`[reminder] ${openid} 探针发送成功,微信侧仍有额度,记账偏低,本轮继续发送`) } insertLog.run({ anniversaryId: anniv.id, personName: anniv.personName, typeName, daysUntil: item.daysUntil, sendDate: Date.now(), status: 'success', error: null }) ok++ console.log(`[reminder] 发送成功: ${anniv.personName} (${typeName}, ${item.daysUntil}天)`) } catch (err) { if (isQuotaError(err)) { // 微信侧实际已无额度(或用户关了通知总开关),归零并停止骚扰 quota.reset(openid) balance = 0 halted = true logSkip(item, err.message); skipped++ console.warn(`[reminder] ${openid} 额度已耗尽,本轮剩余全部跳过: ${err.message}`) } else { insertLog.run({ anniversaryId: anniv.id, personName: anniv.personName, typeName: null, daysUntil: null, sendDate: Date.now(), status: 'failed', error: err.message }) fail++ console.error(`[reminder] 发送失败: ${anniv.personName}`, err.message) } } } return { ok, fail, skipped } } async function runOnce() { console.log('[reminder] 开始扫描纪念日...') // ORDER BY id 不是为了排序好看,而是为了「平局可复现」: // 同一天、同 kind、同 importance 的条目在 sort 里比不出先后(JS sort 是稳定的), // 最终取舍就取决于 SQLite 的返回顺序。这里和 index.js 的 getQuotaStatus 必须用同一个 // 兜底顺序,否则首页预告「谁有风险」的人名会和实际被跳过的人对不上。 const list = db.prepare('SELECT * FROM anniversaries WHERE remindEnabled = 1 ORDER BY id').all() console.log(`[reminder] 启用提醒的纪念日 ${list.length} 条`) // 额度是按用户算的,所以必须分组处理 const byOpenid = new Map() for (const a of list) { if (!byOpenid.has(a.openid)) byOpenid.set(a.openid, []) byOpenid.get(a.openid).push(a) } let ok = 0, fail = 0, skipped = 0 for (const [openid, items] of byOpenid) { const r = await runForUser(openid, items) ok += r.ok; fail += r.fail; skipped += r.skipped } console.log(`[reminder] 完成: 成功 ${ok}, 失败 ${fail}, 跳过 ${skipped}`) return { total: list.length, ok, fail, skipped } } function start() { const expr = process.env.REMINDER_CRON || '0 9 * * *' if (!cron.validate(expr)) { console.error(`[reminder] 无效的 cron 表达式: ${expr},定时任务未启动`) return } cron.schedule(expr, runOnce, { timezone: 'Asia/Shanghai' }) console.log(`[reminder] 定时任务已注册: ${expr} (Asia/Shanghai)`) } module.exports = { start, runOnce, runForUser }