From e1b4a0ec5d67256e4d5efed884d8fc57d0997a55 Mon Sep 17 00:00:00 2001
From: EchoZenith <60392923+EchoZenith@users.noreply.github.com>
Date: Sat, 23 May 2026 00:48:06 +0800
Subject: [PATCH] feat: add electricity bill balance alert function
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

新增企业微信通知形式的电费余额预警功能，配置阈值环境变量ALERT_THRESHOLD，当余额低于设定值时自动发送提醒，且避免重复告警
---
 .env.example |  3 +++
 server.js    | 26 ++++++++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/.env.example b/.env.example
index d0e0fd4..97cb225 100644
--- a/.env.example
+++ b/.env.example
@@ -8,5 +8,8 @@ LOGIN_PASSWORD=你的密码
 # 选填 - 企业微信群机器人 Webhook URL
 WECOM_WEBHOOK_URL=https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=你的key
 
+# 选填 - 电费余额预警阈值（低于此值时发送企业微信通知，设为0或留空关闭）
+ALERT_THRESHOLD=20
+
 # 选填 - Cookie 签名密钥
 COOKIE_SECRET=
diff --git a/server.js b/server.js
index 55b1215..28e5806 100644
--- a/server.js
+++ b/server.js
@@ -54,6 +54,14 @@ if (!WECOM_WEBHOOK_URL) {
   console.warn('警告: 环境变量 WECOM_WEBHOOK_URL 未设置，企业微信通知功能将不可用');
 }
 
+const ALERT_THRESHOLD = parseFloat(process.env.ALERT_THRESHOLD) || 0;
+
+if (ALERT_THRESHOLD > 0) {
+  console.log(`电费预警: 当余额低于 ¥${ALERT_THRESHOLD} 时将发送企业微信通知`);
+}
+
+let lastAlertedAmount = null;
+
 function requireAuth(req, res, next) {
   const auth = req.signedCookies.auth;
   if (auth === LOGIN_USERNAME) {
@@ -163,12 +171,30 @@ async function collectData() {
     if (sameDay && sameHour) {
       updateStmt.run(data.surplus, data.amount, data.timestamp, latest.id);
       console.log(`[${new Date().toLocaleString()}] 更新本小时记录: ${data.surplus}度, ¥${data.amount}`);
+      await checkThresholdAndAlert(data.amount);
       return;
     }
   }
 
   insertStmt.run(data.surplus, data.amount, data.timestamp, data.roomName);
   console.log(`[${new Date().toLocaleString()}] 新增记录: ${data.surplus}度, ¥${data.amount}`);
+  await checkThresholdAndAlert(data.amount);
+}
+
+async function checkThresholdAndAlert(amount) {
+  if (ALERT_THRESHOLD <= 0 || amount > ALERT_THRESHOLD) {
+    lastAlertedAmount = null;
+    return;
+  }
+  if (lastAlertedAmount !== null && amount >= lastAlertedAmount) return;
+  lastAlertedAmount = amount;
+  console.log(`[预警] 余额 ¥${amount} 低于阈值 ¥${ALERT_THRESHOLD}`);
+  await sendWecomNotification(
+    '## 电费余额预警\n\n' +
+    `> 时间：${new Date().toLocaleString('zh-CN')}\n\n` +
+    `当前余额 **¥${amount.toFixed(2)}**\n\n` +
+    `已低于预警阈值 **¥${ALERT_THRESHOLD.toFixed(2)}**，请及时充值！`
+  );
 }
 
 function calculateDailyUsage(records) {