ITStudio_OUC/OUC_Shell
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #2 from Yaosanqi137/main

Browse Source 
电费提醒和网费提醒
This commit is contained in:
Yaosanqi137
2025-12-01 22:10:48 +08:00
committed by GitHub
parent 226f170dff 2b0070ceed
commit 9ac2161879
5 changed files with 530 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4
View File
@@ -71,6 +71,10 @@ systemctl start oucshell.service
systemctl enable oucshell.service
```
# 效果演示
![img.png](show.png)
## 🤝 贡献者
<!-- readme: contributors -start -->
main.sh
+198
View File
@@ -0,0 +1,198 @@
#!/bin/bash
# ==========================================
# 主控脚本
# ==========================================
# --- 1. 路径定义 ---
BASE_DIR=$(cd $(dirname $0); pwd)
SRC_DIR="$BASE_DIR/src"
CONFIG_FILE="$BASE_DIR/config.toml"
LOG_FILE="$BASE_DIR/service.log"
# 日志
log() {
echo "[$(date '+%Y-%m-%d %H:%M:%S')] $1" | tee -a "$LOG_FILE"
}
trap "log '服务停止'; exit" SIGTERM SIGINT
# 依赖检查和安装
check_and_install_dependencies() {
local dependencies=("curl" "jq" "bc")
local missing_packages=()
for pkg in "${dependencies[@]}"; do
if ! command -v "$pkg" &> /dev/null; then
missing_packages+=("$pkg")
fi
done
if [ ${#missing_packages[@]} -eq 0 ]; then
log "环境检查通过: 依赖已就绪。"
return 0
fi
log "检测到缺失依赖: ${missing_packages[*]}，尝试自动安装..."
local install_cmd=""
if command -v apt-get &> /dev/null; then
install_cmd="apt-get update && apt-get install -y"
elif command -v yum &> /dev/null; then
install_cmd="yum install -y"
elif command -v dnf &> /dev/null; then
install_cmd="dnf install -y"
elif command -v apk &> /dev/null; then
install_cmd="apk add --no-cache"
else
log "无法自动安装，请手动安装: ${missing_packages[*]}"
exit 1
fi
if [ "$EUID" -ne 0 ] && command -v sudo &> /dev/null; then
install_cmd="sudo $install_cmd"
fi
# 这里也将安装日志通过管道传递给 log 函数，保持日志格式统一
eval "$install_cmd ${missing_packages[*]}" 2>&1 | while IFS= read -r line; do
log "[依赖安装] $line"
done
# 二次检查
for pkg in "${missing_packages[@]}"; do
if ! command -v "$pkg" &> /dev/null; then
log "依赖 $pkg 安装失败，请检查网络或源。"
exit 1
fi
done
log "依赖安装完成。"
}
# 配置文件生成与检查
check_and_create_config() {
if [ ! -f "$CONFIG_FILE" ]; then
log "配置文件不存在，正在生成默认配置..."
cat > "$CONFIG_FILE" << EOF
# ==========================================
# OUCShell配置文件
# ==========================================
[Global]
# 接收通知的邮箱
TargetEmail = "your_email@example.com"
[SMTP]
# 发件人邮箱服务器配置
Host = "smtp.qq.com"
Port = "465"
User = "your_smtp_email@qq.com"
Password = "your_smtp_auth_code"
# 电费提醒模块
[Electricity]
Enabled = true # 是否启用本模块
Campus = "xha" # 校区选择
# xha = 西海岸
[Electricity.xha]
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 "检测到第一次运行，脚本将自动退出，请编辑配置文件后重新启动"
exit 0
else
log "加载配置文件: $CONFIG_FILE"
fi
}
# 主函数
log "服务启动..."
check_and_install_dependencies
check_and_create_config
# 电费提醒任务间隔 (秒)
INTERVAL_ELEC=7200 # 2小时
LAST_RUN_ELEC=0
# 网费提醒任务间隔 (秒)
INTERVAL_NET=43200 # 半天
LAST_RUN_NET=0
log "进入循环调度模式..."
while true; do
CURRENT_TIME=$(date +%s)
# 电费监控
# 计算时间差
TIME_DIFF=$((CURRENT_TIME - LAST_RUN_ELEC))
if [ $TIME_DIFF -ge $INTERVAL_ELEC ]; then
SCRIPT_PATH="$SRC_DIR/elec_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 "[elec_monitor] $line"
done
log "任务结束: 电费监控"
else
log "警告: 找不到脚本 $SCRIPT_PATH"
fi
LAST_RUN_ELEC=$(date +%s)
fi
# 网费监控
# 计算时间差
TIME_DIFF=$((CURRENT_TIME - LAST_RUN_NET))
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
sleep 60
done
show.png
BIN
View File
Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 84 KiB

src/elec_monitor.sh
+157
View File
@@ -0,0 +1,157 @@
#!/bin/bash
# ==========================================
# 电费提醒功能脚本
# ==========================================
CONFIG_PATH="$1"
# 检查参数
if [ -z "$CONFIG_PATH" ]; then
echo "错误: 未传入配置文件路径。"
exit 1
fi
if [ ! -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() {
local sno=$(get_val "Electricity.xha" "StudentID")
local token=$(get_val "Electricity.xha" "Token")
local thresholds_raw=$(get_val "Electricity.xha" "RemindTime" | tr -d '[]' | tr ',' ' ')
local light_limit=$(echo $thresholds_raw | awk '{print $1}')
local ac_limit=$(echo $thresholds_raw | awk '{print $2}')
# 如果解析失败默认设为0
light_limit=${light_limit:-0}
ac_limit=${ac_limit:-0}
if [ -z "$sno" ] || [ -z "$token" ]; then
echo "错误: 学号或Token未配置，请修改 config.toml"
return
fi
echo "查询西海岸校区... (学号: $sno)"
local response=$(curl -s -X POST "http://10.128.13.25/hydxcas/getCadByNo" \
-H "Token: $token" \
-H "Content-Type: application/json" \
-d "{\"sno\": \"$sno\"}")
local errcode=$(echo "$response" | jq -r '.errcode')
if [ "$errcode" != "0" ]; then
echo "API 请求失败: $(echo "$response" | jq -r '.errmsg')"
return
fi
# 解析数据
local light_val=$(echo "$response" | jq -r '.value | fromjson | .eqptData[] | select(.categoryEnergyName == "照明与插座") | .buyElec')
local ac_val=$(echo "$response" | jq -r '.value | fromjson | .eqptData[] | select(.categoryEnergyName == "空调末端") | .buyElec')
light_val=${light_val:-0}
ac_val=${ac_val:-0}
echo "当前电量 -> 照明: $light_val, 空调: $ac_val"
echo "警戒阈值 -> 照明: $light_limit, 空调: $ac_limit"
local need_alert=0
local msg="<h3>电费余额不足提醒</h3><ul>"
# 使用 bc 进行比较，增加 2>/dev/null 屏蔽非数字报错
if (( $(echo "$light_val < $light_limit" | bc -l 2>/dev/null) )); then
need_alert=1
msg="${msg}<li><strong>照明:</strong> ${light_val} 度 (低于 ${light_limit})</li>"
fi
if (( $(echo "$ac_val < $ac_limit" | bc -l 2>/dev/null) )); then
need_alert=1
msg="${msg}<li><strong>空调:</strong> ${ac_val} 度 (低于 ${ac_limit})</li>"
fi
msg="${msg}</ul><p>请及时充值！</p>"
if [ $need_alert -eq 1 ]; then
send_email "【紧急】宿舍电费余额不足" "$msg"
else
echo "电量充足，无需提醒。"
fi
}
# 主函数部分
ENABLED=$(get_val "Electricity" "Enabled")
if [ "$ENABLED" != "true" ]; then
echo "电费监控模块未启用。"
exit 0
fi
CAMPUS=$(get_val "Electricity" "Campus")
case "$CAMPUS" in
"xha")
check_xha
;;
*)
echo "未知的校区配置: $CAMPUS"
;;
esac
src/internet_monitor.sh
+171
View File
@@ -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="<h3>校园网费余额预警</h3>
<ul>
<li><strong>学号:</strong> ${sno}</li>
<li><strong>当前余额:</strong> <span style='color:red'>${balance_num} 元</span></li>
<li><strong>已用流量:</strong> ${flow_str}</li>
<li><strong>警告原因:</strong> ${reason}</li>
</ul>
<p>请确认是否需要充值以避免下月断网。</p>"
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