diff --git a/main.sh b/main.sh new file mode 100644 index 0000000..b57ce68 --- /dev/null +++ b/main.sh @@ -0,0 +1,148 @@ +#!/bin/bash + +# ========================================== +# 主控制脚本 (Main Runner) +# 功能:依赖管理、配置生成、任务调度、日志整合 +# ========================================== + +# --- 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 "依赖安装完成。" +} + +# --- 4. 配置文件生成与检查 --- +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 = "20230000" +Token = "YOUR_TOKEN_HERE" +# [照明警戒值, 空调警戒值] +RemindTime = [30.0, 30.0] +EOF + log "配置文件已生成: $CONFIG_FILE" + log "检测到第一次加载此脚本,将自动结束本脚本,请前往编辑配置文件" + exit 0 + else + log "加载配置文件: $CONFIG_FILE" + fi +} + +# 主函数 + +log "服务启动..." + +check_and_install_dependencies +check_and_create_config + +# 任务间隔 (秒) +INTERVAL_ELEC=1800 # 半小时 +LAST_RUN_ELEC=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 + + # 其他任务(预留) + # TODO + + sleep 60 +done \ No newline at end of file diff --git a/src/Elec_monitor.sh b/src/Elec_monitor.sh new file mode 100644 index 0000000..ce4c73a --- /dev/null +++ b/src/Elec_monitor.sh @@ -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="
请及时充值!
" + + 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 \ No newline at end of file