diff --git a/main.sh b/main.sh index 02ad623..ba978a4 100644 --- a/main.sh +++ b/main.sh @@ -88,7 +88,7 @@ Port = "465" User = "your_smtp_email@qq.com" Password = "your_smtp_auth_code" -# --- 功能模块: 电费提醒 --- +# 电费提醒模块 [Electricity] Enabled = true # 是否启用本模块 Campus = "xha" # 校区选择 @@ -99,6 +99,21 @@ StudentID = "XXX" # 学号 Token = "9f7c6e76979c4cb9dd3828f8cc44a5ef" # MD5(Sd1234) 居然加密这么简单吗? # [照明警戒值, 空调警戒值] RemindTime = [30.0, 30.0] + +# 网费提醒模块 +[Internet] +Enabled = true # 是否启用本模块 +Campus = "xha" # 校区选择 +# xha = 西海岸 + +[Internet.xha] +StudentID = "XXX" # 学号 +# [最低余额, 触发天数] +# 触发天数: 离下个月1号还有几天时开始检测。 +# e.g. +# 填 -1 表示忽略日期,只要余额低就提醒。 +# 填 5 表示只有余额低 且 离月底少于5天时才提醒。 +RemindTime = [10, -1] EOF log "配置文件已生成: $CONFIG_FILE" log "检测到第一次运行,脚本将自动退出,请编辑配置文件后重新启动" @@ -116,7 +131,7 @@ check_and_install_dependencies check_and_create_config # 任务间隔 (秒) -INTERVAL_ELEC=1800 # 半小时 +INTERVAL_ELEC=7200 # 2小时 LAST_RUN_ELEC=0 log "进入循环调度模式..." @@ -124,8 +139,9 @@ log "进入循环调度模式..." while true; do CURRENT_TIME=$(date +%s) - # 电费监控 # 计算时间差 + + # 电费监控 TIME_DIFF=$((CURRENT_TIME - LAST_RUN_ELEC)) if [ $TIME_DIFF -ge $INTERVAL_ELEC ]; then @@ -147,6 +163,26 @@ while true; do LAST_RUN_ELEC=$(date +%s) fi + # 电费监控 + if [ $TIME_DIFF -ge $INTERVAL_NET ]; then + SCRIPT_PATH="$SRC_DIR/internet_monitor.sh" + + if [ -f "$SCRIPT_PATH" ]; then + chmod +x "$SCRIPT_PATH" + log "调度任务: 网费监控..." + + /bin/bash "$SCRIPT_PATH" "$CONFIG_FILE" 2>&1 | while IFS= read -r line; do + log "[net_monitor] $line" + done + + log "任务结束: 网费监控" + else + log "警告: 找不到脚本 $SCRIPT_PATH" + fi + + LAST_RUN_NET=$(date +%s) + fi + # 其他任务(预留) # TODO diff --git a/src/internet_monitor.sh b/src/internet_monitor.sh new file mode 100644 index 0000000..6d390b0 --- /dev/null +++ b/src/internet_monitor.sh @@ -0,0 +1,171 @@ +#!/bin/bash + +# ========================================== +# 网费提醒功能脚本 +# ========================================== + +CONFIG_PATH="$1" + +# 基础检查 +if [ -z "$CONFIG_PATH" ] || [ ! -f "$CONFIG_PATH" ]; then + echo "错误: 配置文件未找到 ($CONFIG_PATH)" + exit 1 +fi + +# 读取配置 +get_val() { + local section=$1 + local key=$2 + local safe_section=$(echo "$section" | sed 's/\./\\./g') + + sed -n "/^\[$safe_section\]/,/^\[/p" "$CONFIG_PATH" \ + | grep "^$key" \ + | head -n 1 \ + | awk -F'=' '{print $2}' \ + | tr -d ' "' \ + | sed 's/#.*//' \ + | tr -d '\r' +} + +# 邮件发送函数 +send_email() { + local subject=$1 + local content=$2 + + local smtp_host=$(get_val "SMTP" "Host") + local smtp_port=$(get_val "SMTP" "Port") + local smtp_user=$(get_val "SMTP" "User") + local smtp_pass=$(get_val "SMTP" "Password") + local target_email=$(get_val "Global" "TargetEmail") + + if [ -z "$smtp_user" ] || [ -z "$target_email" ]; then + echo "错误: 邮箱配置不完整,跳过发送。" + return + fi + + echo "正在发送邮件到 $target_email ..." + + local curl_url="smtp://$smtp_host:$smtp_port" + if [ "$smtp_port" == "465" ]; then curl_url="smtps://$smtp_host:$smtp_port"; fi + + local mail_data="From: \"校园助手\" <$smtp_user> +To: <$target_email> +Subject: $subject +MIME-Version: 1.0 +Content-Type: text/html; charset=utf-8 + +$content" + + echo "$mail_data" | curl --silent --ssl-reqd --url "$curl_url" --user "$smtp_user:$smtp_pass" --mail-from "$smtp_user" --mail-rcpt "$target_email" --upload-file - +} + +# 西海岸网费检查 +check_xha_net() { + local sno=$(get_val "Internet.xha" "StudentID") + + # 解析 RemindTime = [10, 5] -> money_limit=10, day_limit=5 + local config_raw=$(get_val "Internet.xha" "RemindTime" | tr -d '[]' | tr ',' ' ') + local money_limit=$(echo $config_raw | awk '{print $1}') + local day_limit=$(echo $config_raw | awk '{print $2}') + + # 默认值保护 + money_limit=${money_limit:-10} + day_limit=${day_limit:--1} + + if [ -z "$sno" ]; then + echo "错误: 未配置学号 (StudentID)" + return + fi + + echo "查询西海岸校区网费... (学号: $sno)" + + # 1. 发送请求 + local url="https://xha.ouc.edu.cn:802/eportal/portal/page/loadUserInfo?callback=dr1002&lang=zh-CN&program_index=ctshNw1713845951&page_index=V5fmKw1713845966&user_account=${sno}&wlan_user_ip=0.0.0.0&wlan_user_mac=000000000000&jsVersion=4.1&v=3015&lang=zh" + + local response=$(curl -k -s --http1.1 -A "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36" "$url") + + # 2. 解析 JSONP + # 返回格式: dr1002({ ... }); 需要去掉 dr1002( 和 ); + local json_body=$(echo "$response" | sed -E 's/^[^(]*\(//;s/\)[^)]*$//') + + # 检查 parsed JSON 是否有效 + local code=$(echo "$json_body" | jq -r '.code') + if [ "$code" != "1" ]; then + echo "API 请求失败或未登录: $(echo "$json_body" | jq -r '.msg')" + return + fi + + # 3. 提取余额 + # 原始值可能是 "0 元" 或 "12.50 元" + local balance_str=$(echo "$json_body" | jq -r '.user_info.balance') + # 只保留数字和小数点 + local balance_num=$(echo "$balance_str" | sed 's/[^0-9.]//g') + + # 提取已用流量用于展示 + local flow_str=$(echo "$json_body" | jq -r '.user_info.use_flow') + + balance_num=${balance_num:-0} + echo "当前余额: ${balance_num} 元, 已用流量: ${flow_str}" + + # 4. 判断是否需要提醒 + local need_alert=0 + local reason="" + + # 只有当余额低于阈值才考虑发信 + if (( $(echo "$balance_num < $money_limit" | bc -l 2>/dev/null) )); then + + # 4.1 如果 day_limit 是 -1,不看日期,直接发 + if [ "$day_limit" == "-1" ]; then + need_alert=1 + reason="余额低于 ${money_limit} 元" + else + # 4.2 如果配置了日期限制,检查是否接近月底 + # 计算距离下个月1号还有几天 + local next_month_ts=$(date -d "$(date +%Y-%m-01) +1 month" +%s) + local now_ts=$(date +%s) + local diff_sec=$((next_month_ts - now_ts)) + local diff_days=$((diff_sec / 86400)) + + echo "距离月底结算还有: ${diff_days} 天 (配置阈值: ${day_limit} 天)" + + if [ "$diff_days" -le "$day_limit" ]; then + need_alert=1 + reason="临近月底且余额低于 ${money_limit} 元" + else + echo "余额虽然不足,但未到提醒日期 (当前 ${diff_days} > 阈值 ${day_limit}),跳过。" + fi + fi + fi + + # 5. 发送通知 + if [ $need_alert -eq 1 ]; then + local msg="
请确认是否需要充值以避免下月断网。
" + + send_email "【提醒】校园网费余额不足 (${balance_num}元)" "$msg" + fi +} + +# 主函数 +ENABLED=$(get_val "Internet" "Enabled") +if [ "$ENABLED" != "true" ]; then + echo "网费监控模块未启用。" + exit 0 +fi + +CAMPUS=$(get_val "Internet" "Campus") + +case "$CAMPUS" in + "xha") + check_xha_net + ;; + *) + echo "未知的校区配置: $CAMPUS" + ;; +esac \ No newline at end of file