diff --git a/app.py b/app.py index 01319cc..8141e37 100644 --- a/app.py +++ b/app.py @@ -851,6 +851,69 @@ def get_next_run_time(expression): return '无法计算' +# ==================== 邮件通知 API ==================== + +@app.route('/api/email/send', methods=['POST']) +def api_email_send(): + """发送邮件通知""" + data = request.get_json() + if not data: + return jsonify({'error': '无效数据'}), 400 + + to_email = data.get('to') + subject = data.get('subject') + body = data.get('body') + + if not to_email or not subject or not body: + return jsonify({'error': '缺少必要参数'}), 400 + + try: + import smtplib + from email.mime.text import MIMEText + from email.mime.multipart import MIMEMultipart + from datetime import datetime + + # SMTP配置 (从环境变量或默认值) + smtp_host = os.environ.get('SMTP_HOST', 'mail.tphai.com') + smtp_port = int(os.environ.get('SMTP_PORT', 587)) + smtp_user = os.environ.get('SMTP_USER', 'favor@tphai.com') + smtp_pass = os.environ.get('SMTP_PASS', 'favor@!') + + server = smtplib.SMTP(smtp_host, smtp_port) + server.ehlo() + server.login(smtp_user, smtp_pass) + + msg = MIMEMultipart() + msg['From'] = smtp_user + msg['To'] = to_email + msg['Subject'] = subject + msg['Date'] = datetime.now().strftime('%a, %d %b %Y %H:%M:%S +0800') + msg['Reply-To'] = to_email + msg.attach(MIMEText(body, 'plain', 'utf-8')) + + server.sendmail(smtp_user, to_email, msg.as_string()) + server.quit() + + return jsonify({'success': True, 'message': '邮件已发送'}) + except Exception as e: + return jsonify({'error': str(e)}), 500 + + +@app.route('/api/email/config', methods=['GET', 'POST']) +def api_email_config(): + """获取/设置邮件配置""" + if request.method == 'GET': + return jsonify({ + 'host': os.environ.get('SMTP_HOST', 'mail.tphai.com'), + 'port': int(os.environ.get('SMTP_PORT', 587)), + 'user': os.environ.get('SMTP_USER', 'favor@tphai.com') + }) + else: + data = request.get_json() + # 只保存到环境变量说明，实际使用需要配置环境变量 + return jsonify({'message': '邮件配置需要通过环境变量设置'}) + + @app.route('/api/crons') def api_crons(): """获取系统 cron 列表（兼容旧接口）""" @@ -1040,7 +1103,7 @@ HTML_TEMPLATE = ''' - 项目服务管理面板 v2.9 + 项目服务管理面板 v3.0 @@ -1367,6 +1430,21 @@ HTML_TEMPLATE = ''' + + +
+
+

+ 邮件通知规则 +

+ +
+
+
暂无邮件通知规则
+
+
@@ -1627,6 +1705,65 @@ HTML_TEMPLATE = ''' + + +
@@ -1676,6 +1813,7 @@ HTML_TEMPLATE = ''' if (tab === 'system') { loadThresholds(); loadDesktopNotifySetting(); + renderEmailRules(); loadSystemStats(); } } @@ -1804,6 +1942,9 @@ HTML_TEMPLATE = ''' lastAlertTime = now; showAlertPopup(warnings); } + + // 检查邮件通知规则 + checkEmailRules(data); } // 显示警告弹窗（同时发送桌面通知） @@ -2233,6 +2374,157 @@ HTML_TEMPLATE = ''' localStorage.setItem('customServices', JSON.stringify(services)); } + // ==================== 邮件通知规则管理 ==================== + + function getEmailRules() { + const stored = localStorage.getItem('emailNotifyRules'); + return stored ? JSON.parse(stored) : []; + } + + function saveEmailRules(rules) { + localStorage.setItem('emailNotifyRules', JSON.stringify(rules)); + } + + function renderEmailRules() { + const rules = getEmailRules(); + const list = document.getElementById('emailRulesList'); + + if (rules.length === 0) { + list.innerHTML = '
暂无邮件通知规则
'; + return; + } + + const resourceNames = { 'cpu': 'CPU', 'memory': '内存', 'disk': '磁盘' }; + + list.innerHTML = rules.map(rule => ` +
+
+ ${rule.enabled ? '启用' : '禁用'} + ${rule.name} + ${resourceNames[rule.resource]} ≥ ${rule.threshold}% + → ${rule.email} +
+
+ + +
+
+ `).join(''); + } + + function showAddEmailRuleModal() { + document.getElementById('emailRuleModalTitle').textContent = '添加邮件通知规则'; + document.getElementById('emailRuleId').value = ''; + document.getElementById('emailRuleName').value = ''; + document.getElementById('emailRuleResource').value = 'cpu'; + document.getElementById('emailRuleThreshold').value = 80; + document.getElementById('emailRuleAddress').value = ''; + document.getElementById('emailRuleInterval').value = 300; + document.getElementById('emailRuleEnabled').checked = true; + document.getElementById('emailRuleModal').classList.remove('hidden'); + } + + function closeEmailRuleModal() { + document.getElementById('emailRuleModal').classList.add('hidden'); + } + + function editEmailRule(ruleId) { + const rules = getEmailRules(); + const rule = rules.find(r => r.id === ruleId); + if (!rule) return; + + document.getElementById('emailRuleModalTitle').textContent = '编辑邮件通知规则'; + document.getElementById('emailRuleId').value = rule.id; + document.getElementById('emailRuleName').value = rule.name; + document.getElementById('emailRuleResource').value = rule.resource; + document.getElementById('emailRuleThreshold').value = rule.threshold; + document.getElementById('emailRuleAddress').value = rule.email; + document.getElementById('emailRuleInterval').value = rule.interval; + document.getElementById('emailRuleEnabled').checked = rule.enabled; + document.getElementById('emailRuleModal').classList.remove('hidden'); + } + + function saveEmailRule(event) { + event.preventDefault(); + + const ruleId = document.getElementById('emailRuleId').value; + const name = document.getElementById('emailRuleName').value.trim(); + const resource = document.getElementById('emailRuleResource').value; + const threshold = parseInt(document.getElementById('emailRuleThreshold').value); + const email = document.getElementById('emailRuleAddress').value.trim(); + const interval = parseInt(document.getElementById('emailRuleInterval').value) || 300; + const enabled = document.getElementById('emailRuleEnabled').checked; + + if (!name || !email) { + alert('请填写完整信息'); + return; + } + + const rules = getEmailRules(); + + if (ruleId) { + // 编辑 + const idx = rules.findIndex(r => r.id === ruleId); + if (idx >= 0) { + rules[idx] = { ...rules[idx], name, resource, threshold, email, interval, enabled }; + } + } else { + // 新增 + rules.push({ + id: 'rule_' + Date.now(), + name, resource, threshold, email, interval, enabled, + lastSent: 0 + }); + } + + saveEmailRules(rules); + closeEmailRuleModal(); + renderEmailRules(); + } + + function deleteEmailRule(ruleId) { + if (!confirm('确定删除此邮件通知规则？')) return; + const rules = getEmailRules().filter(r => r.id !== ruleId); + saveEmailRules(rules); + renderEmailRules(); + } + + // 检查邮件规则并发送通知 + async function checkEmailRules(data) { + const rules = getEmailRules().filter(r => r.enabled); + const now = Date.now(); + + for (const rule of rules) { + // 检查间隔 + if (now - rule.lastSent < rule.interval * 1000) continue; + + // 检查阈值 + const value = data[rule.resource]?.percent; + if (value === undefined || value < rule.threshold) continue; + + // 发送邮件 + try { + const res = await fetch('/api/email/send', { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ + to: rule.email, + subject: `[系统告警] ${rule.name}`, + body: `${rule.name} 触发告警\n\n${rule.resource.toUpperCase()} 使用率: ${value.toFixed(1)}%\n阈值: ${rule.threshold}%\n时间: ${new Date().toLocaleString()}\n\n请及时处理。` + }) + }); + + if (res.ok) { + // 更新发送时间 + rule.lastSent = now; + saveEmailRules(getEmailRules().map(r => r.id === rule.id ? rule : r)); + } + } catch (e) { + console.error('邮件发送失败:', e); + } + } + } + function showAddServiceModal() { document.getElementById('serviceName').value = ''; document.getElementById('servicePort').value = ''; diff --git a/logs/app.log b/logs/app.log index b0e5b9a..6c1531a 100644 Binary files a/logs/app.log and b/logs/app.log differ