From 2f762db3377454c9a85a4ac5bf6516c21560ba95 Mon Sep 17 00:00:00 2001 From: yuyx <237899745@qq.com> Date: Thu, 11 Dec 2025 21:38:28 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E9=82=AE=E4=BB=B6?= =?UTF-8?q?=E5=8A=9F=E8=83=BD=E7=AC=AC=E4=B8=80=E9=98=B6=E6=AE=B5=20-=20?= =?UTF-8?q?=E9=82=AE=E4=BB=B6=E5=9F=BA=E7=A1=80=E8=AE=BE=E6=96=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 新增功能: - 创建 email_service.py 邮件服务模块 - 支持多SMTP配置（主备切换、故障转移） - 发送纯文本/HTML邮件 - 发送带附件邮件（支持ZIP压缩） - 异步发送队列（多线程工作池） - 每日发送限额控制 - 发送日志记录和统计 - 数据库表结构 - smtp_configs: 多SMTP配置表 - email_settings: 全局邮件设置 - email_tokens: 邮件验证Token - email_logs: 邮件发送日志 - email_stats: 邮件发送统计 - API接口 - GET/POST /yuyx/api/email/settings: 全局邮件设置 - CRUD /yuyx/api/smtp/configs: SMTP配置管理 - POST /yuyx/api/smtp/configs//test: 测试SMTP连接 - POST /yuyx/api/smtp/configs//primary: 设为主配置 - GET /yuyx/api/email/stats: 邮件统计 - GET /yuyx/api/email/logs: 邮件日志 - POST /yuyx/api/email/logs/cleanup: 清理日志 - 后台管理页面 - 新增"邮件配置"Tab - 全局邮件开关、故障转移开关 - SMTP配置列表管理 - 添加/编辑SMTP配置弹窗 - 邮件发送统计展示 - 邮件日志查询和清理 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 --- Dockerfile | 1 + app.py | 208 ++++++- email_service.py | 1254 ++++++++++++++++++++++++++++++++++++++++++ templates/admin.html | 666 ++++++++++++++++++++++ 4 files changed, 2128 insertions(+), 1 deletion(-) create mode 100644 email_service.py diff --git a/Dockerfile b/Dockerfile index 62d7213..01e40d7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -31,6 +31,7 @@ COPY browser_installer.py . COPY password_utils.py . COPY crypto_utils.py . COPY task_checkpoint.py . +COPY email_service.py . # 复制新的优化模块 COPY app_config.py . diff --git a/app.py b/app.py index 0370827..9674f35 100755 --- a/app.py +++ b/app.py @@ -3463,7 +3463,7 @@ def checkpoint_resume(task_id): return jsonify({'success': False, 'message': str(e)}), 500 @app.route('/yuyx/api/checkpoint//abandon', methods=['POST']) -@admin_required +@admin_required def checkpoint_abandon(task_id): try: if checkpoint_mgr.abandon_task(task_id): @@ -3472,6 +3472,198 @@ def checkpoint_abandon(task_id): except Exception as e: return jsonify({'success': False, 'message': str(e)}), 500 + +# ==================== 邮件服务API ==================== +import email_service + +@app.route('/yuyx/api/email/settings', methods=['GET']) +@admin_required +def get_email_settings_api(): + """获取全局邮件设置""" + try: + settings = email_service.get_email_settings() + return jsonify(settings) + except Exception as e: + logger.error(f"获取邮件设置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/email/settings', methods=['POST']) +@admin_required +def update_email_settings_api(): + """更新全局邮件设置""" + try: + data = request.json + enabled = data.get('enabled', False) + failover_enabled = data.get('failover_enabled', True) + + email_service.update_email_settings(enabled, failover_enabled) + return jsonify({'success': True}) + except Exception as e: + logger.error(f"更新邮件设置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs', methods=['GET']) +@admin_required +def get_smtp_configs_api(): + """获取所有SMTP配置列表""" + try: + configs = email_service.get_smtp_configs(include_password=False) + return jsonify(configs) + except Exception as e: + logger.error(f"获取SMTP配置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs', methods=['POST']) +@admin_required +def create_smtp_config_api(): + """创建SMTP配置""" + try: + data = request.json + + # 验证必填字段 + if not data.get('host'): + return jsonify({'error': 'SMTP服务器地址不能为空'}), 400 + if not data.get('username'): + return jsonify({'error': 'SMTP用户名不能为空'}), 400 + + config_id = email_service.create_smtp_config(data) + return jsonify({'success': True, 'id': config_id}) + except Exception as e: + logger.error(f"创建SMTP配置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs/', methods=['GET']) +@admin_required +def get_smtp_config_api(config_id): + """获取单个SMTP配置详情""" + try: + config = email_service.get_smtp_config(config_id, include_password=False) + if not config: + return jsonify({'error': '配置不存在'}), 404 + return jsonify(config) + except Exception as e: + logger.error(f"获取SMTP配置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs/', methods=['PUT']) +@admin_required +def update_smtp_config_api(config_id): + """更新SMTP配置""" + try: + data = request.json + + if email_service.update_smtp_config(config_id, data): + return jsonify({'success': True}) + return jsonify({'error': '更新失败'}), 400 + except Exception as e: + logger.error(f"更新SMTP配置��败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs/', methods=['DELETE']) +@admin_required +def delete_smtp_config_api(config_id): + """删除SMTP配置""" + try: + if email_service.delete_smtp_config(config_id): + return jsonify({'success': True}) + return jsonify({'error': '删除失败'}), 400 + except Exception as e: + logger.error(f"删除SMTP配置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs//test', methods=['POST']) +@admin_required +def test_smtp_config_api(config_id): + """测试SMTP配置""" + try: + data = request.json + test_email = data.get('email', '') + + if not test_email: + return jsonify({'error': '请提供测试邮箱'}), 400 + + # 简单验证邮箱格式 + import re + if not re.match(r'^[^\s@]+@[^\s@]+\.[^\s@]+$', test_email): + return jsonify({'error': '邮箱格式不正确'}), 400 + + result = email_service.test_smtp_config(config_id, test_email) + return jsonify(result) + except Exception as e: + logger.error(f"测试SMTP配置失败: {e}") + return jsonify({'success': False, 'error': str(e)}), 500 + + +@app.route('/yuyx/api/smtp/configs//primary', methods=['POST']) +@admin_required +def set_primary_smtp_config_api(config_id): + """设置主SMTP配置""" + try: + if email_service.set_primary_smtp_config(config_id): + return jsonify({'success': True}) + return jsonify({'error': '设置失败'}), 400 + except Exception as e: + logger.error(f"设置主SMTP配置失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/email/stats', methods=['GET']) +@admin_required +def get_email_stats_api(): + """获取邮件发送统计""" + try: + stats = email_service.get_email_stats() + return jsonify(stats) + except Exception as e: + logger.error(f"获取邮件统计失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/email/logs', methods=['GET']) +@admin_required +def get_email_logs_api(): + """获取邮件发送日志""" + try: + page = request.args.get('page', 1, type=int) + page_size = request.args.get('page_size', 20, type=int) + email_type = request.args.get('type', None) + status = request.args.get('status', None) + + # 限制page_size范围 + page_size = min(max(page_size, 10), 100) + + result = email_service.get_email_logs(page, page_size, email_type, status) + return jsonify(result) + except Exception as e: + logger.error(f"获取邮件日志失败: {e}") + return jsonify({'error': str(e)}), 500 + + +@app.route('/yuyx/api/email/logs/cleanup', methods=['POST']) +@admin_required +def cleanup_email_logs_api(): + """清理过期邮件日志""" + try: + data = request.json or {} + days = data.get('days', 30) + + # 限制days范围 + days = min(max(days, 7), 365) + + deleted = email_service.cleanup_email_logs(days) + return jsonify({'success': True, 'deleted': deleted}) + except Exception as e: + logger.error(f"清理邮件日志失败: {e}") + return jsonify({'error': str(e)}), 500 + + # 初始化浏览器池（在后台线程中预热，不阻塞启动） # ==================== 用户定时任务API ==================== @@ -3815,6 +4007,13 @@ def cleanup_on_exit(): except: pass + # 3.5 关闭邮件队列 + print("- 关闭邮件队列...") + try: + email_service.shutdown_email_queue() + except: + pass + # 4. 关闭数据库连接池 print("- 关闭数据库连接池...") try: @@ -3850,6 +4049,13 @@ if __name__ == '__main__': checkpoint_mgr = get_checkpoint_manager() print("✓ 任务断点管理器已初始化") + # 初始化邮件服务 + try: + email_service.init_email_service() + print("✓ 邮件服务已初始化") + except Exception as e: + print(f"警告: 邮件服务初始化失败: {e}") + # 启动内存清理调度器 start_cleanup_scheduler() diff --git a/email_service.py b/email_service.py new file mode 100644 index 0000000..2a1cf39 --- /dev/null +++ b/email_service.py @@ -0,0 +1,1254 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +""" +邮件服务模块 +功能: +1. 多SMTP配置管理（主备切换、故障转移） +2. 发送纯文本/HTML邮件 +3. 发送带附件邮件（支持ZIP压缩） +4. 异步发送队列 +5. 每日发送限额控制 +6. 发送日志记录 +""" + +import os +import smtplib +import threading +import queue +import time +import zipfile +import secrets +from datetime import datetime, timedelta +from email.mime.text import MIMEText +from email.mime.multipart import MIMEMultipart +from email.mime.base import MIMEBase +from email.mime.image import MIMEImage +from email import encoders +from email.header import Header +from email.utils import formataddr +from typing import Optional, List, Dict, Any, Callable +from io import BytesIO + +import db_pool +from crypto_utils import encrypt_password, decrypt_password, is_encrypted + + +# ============ 常量配置 ============ + +# 邮件类型 +EMAIL_TYPE_REGISTER = 'register' # 注册验证 +EMAIL_TYPE_RESET = 'reset' # 密码重置 +EMAIL_TYPE_BIND = 'bind' # 邮箱绑定 +EMAIL_TYPE_TASK_COMPLETE = 'task_complete' # 任务完成通知 + +# Token有效期（秒） +TOKEN_EXPIRE_REGISTER = 24 * 60 * 60 # 注册验证: 24小时 +TOKEN_EXPIRE_RESET = 30 * 60 # 密码重置: 30分钟 +TOKEN_EXPIRE_BIND = 60 * 60 # 邮箱绑定: 1小时 + +# 发送频率限制（秒） +RATE_LIMIT_REGISTER = 60 # 注册邮件: 1分钟 +RATE_LIMIT_RESET = 5 * 60 # 重置邮件: 5分钟 +RATE_LIMIT_BIND = 60 # 绑定邮件: 1分钟 + +# 异步队列配置 +QUEUE_WORKERS = int(os.environ.get('EMAIL_QUEUE_WORKERS', '2')) +QUEUE_MAX_SIZE = int(os.environ.get('EMAIL_QUEUE_MAX_SIZE', '100')) + + +# ============ 数据库操作 ============ + +def init_email_tables(): + """初始化邮件相关数据库表""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + # 1. SMTP配置表（支持多配置） + cursor.execute(""" + CREATE TABLE IF NOT EXISTS smtp_configs ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + name TEXT NOT NULL DEFAULT '默认配置', + enabled INTEGER DEFAULT 1, + is_primary INTEGER DEFAULT 0, + priority INTEGER DEFAULT 0, + host TEXT NOT NULL DEFAULT '', + port INTEGER DEFAULT 465, + username TEXT NOT NULL DEFAULT '', + password TEXT DEFAULT '', + use_ssl INTEGER DEFAULT 1, + use_tls INTEGER DEFAULT 0, + sender_name TEXT DEFAULT '知识管理平台', + sender_email TEXT DEFAULT '', + daily_limit INTEGER DEFAULT 0, + daily_sent INTEGER DEFAULT 0, + daily_reset_date TEXT DEFAULT '', + last_success_at TIMESTAMP, + last_error TEXT DEFAULT '', + success_count INTEGER DEFAULT 0, + fail_count INTEGER DEFAULT 0, + created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, + updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP + ) + """) + + # 创建索引 + cursor.execute(""" + CREATE INDEX IF NOT EXISTS idx_smtp_configs_enabled + ON smtp_configs(enabled, priority) + """) + cursor.execute(""" + CREATE INDEX IF NOT EXISTS idx_smtp_configs_primary + ON smtp_configs(is_primary) + """) + + # 2. 全局邮件设置表 + cursor.execute(""" + CREATE TABLE IF NOT EXISTS email_settings ( + id INTEGER PRIMARY KEY DEFAULT 1, + enabled INTEGER DEFAULT 0, + failover_enabled INTEGER DEFAULT 1, + updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP + ) + """) + + # 初始化默认设置 + cursor.execute(""" + INSERT OR IGNORE INTO email_settings (id, enabled, failover_enabled) + VALUES (1, 0, 1) + """) + + # 3. 邮件验证Token表 + cursor.execute(""" + CREATE TABLE IF NOT EXISTS email_tokens ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + user_id INTEGER, + email TEXT NOT NULL, + token TEXT UNIQUE NOT NULL, + token_type TEXT NOT NULL, + expires_at TIMESTAMP NOT NULL, + used INTEGER DEFAULT 0, + created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, + FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE + ) + """) + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_tokens_token ON email_tokens(token)") + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_tokens_email ON email_tokens(email)") + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_tokens_expires ON email_tokens(expires_at)") + + # 4. 邮件发送日志表 + cursor.execute(""" + CREATE TABLE IF NOT EXISTS email_logs ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + user_id INTEGER, + smtp_config_id INTEGER, + email_to TEXT NOT NULL, + email_type TEXT NOT NULL, + subject TEXT NOT NULL, + status TEXT NOT NULL, + error_message TEXT, + attachment_count INTEGER DEFAULT 0, + attachment_size INTEGER DEFAULT 0, + created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, + FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE SET NULL, + FOREIGN KEY (smtp_config_id) REFERENCES smtp_configs(id) ON DELETE SET NULL + ) + """) + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_logs_user ON email_logs(user_id)") + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_logs_status ON email_logs(status)") + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_logs_type ON email_logs(email_type)") + cursor.execute("CREATE INDEX IF NOT EXISTS idx_email_logs_created ON email_logs(created_at)") + + # 5. 邮件发送统计表 + cursor.execute(""" + CREATE TABLE IF NOT EXISTS email_stats ( + id INTEGER PRIMARY KEY DEFAULT 1, + total_sent INTEGER DEFAULT 0, + total_success INTEGER DEFAULT 0, + total_failed INTEGER DEFAULT 0, + register_sent INTEGER DEFAULT 0, + reset_sent INTEGER DEFAULT 0, + bind_sent INTEGER DEFAULT 0, + task_complete_sent INTEGER DEFAULT 0, + last_updated TIMESTAMP DEFAULT CURRENT_TIMESTAMP + ) + """) + + # 初始化统计记录 + cursor.execute(""" + INSERT OR IGNORE INTO email_stats (id) VALUES (1) + """) + + conn.commit() + print("[邮件服务] 数据库表初始化完成") + + +# ============ SMTP配置管理 ============ + +def get_email_settings() -> Dict[str, Any]: + """获取全局邮件设置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute("SELECT enabled, failover_enabled, updated_at FROM email_settings WHERE id = 1") + row = cursor.fetchone() + if row: + return { + 'enabled': bool(row[0]), + 'failover_enabled': bool(row[1]), + 'updated_at': row[2] + } + return {'enabled': False, 'failover_enabled': True, 'updated_at': None} + + +def update_email_settings(enabled: bool, failover_enabled: bool) -> bool: + """更新全局邮件设置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + UPDATE email_settings + SET enabled = ?, failover_enabled = ?, updated_at = CURRENT_TIMESTAMP + WHERE id = 1 + """, (int(enabled), int(failover_enabled))) + conn.commit() + return True + + +def get_smtp_configs(include_password: bool = False) -> List[Dict[str, Any]]: + """获取所有SMTP配置列表""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + SELECT id, name, enabled, is_primary, priority, host, port, + username, password, use_ssl, use_tls, sender_name, sender_email, + daily_limit, daily_sent, daily_reset_date, + last_success_at, last_error, success_count, fail_count, + created_at, updated_at + FROM smtp_configs + ORDER BY is_primary DESC, priority ASC, id ASC + """) + + configs = [] + for row in cursor.fetchall(): + config = { + 'id': row[0], + 'name': row[1], + 'enabled': bool(row[2]), + 'is_primary': bool(row[3]), + 'priority': row[4], + 'host': row[5], + 'port': row[6], + 'username': row[7], + 'password': '******' if row[8] and not include_password else (decrypt_password(row[8]) if include_password and row[8] else ''), + 'has_password': bool(row[8]), + 'use_ssl': bool(row[9]), + 'use_tls': bool(row[10]), + 'sender_name': row[11], + 'sender_email': row[12], + 'daily_limit': row[13], + 'daily_sent': row[14], + 'daily_reset_date': row[15], + 'last_success_at': row[16], + 'last_error': row[17], + 'success_count': row[18], + 'fail_count': row[19], + 'created_at': row[20], + 'updated_at': row[21] + } + + # 计算成功率 + total = config['success_count'] + config['fail_count'] + config['success_rate'] = round(config['success_count'] / total * 100, 1) if total > 0 else 0 + + configs.append(config) + + return configs + + +def get_smtp_config(config_id: int, include_password: bool = False) -> Optional[Dict[str, Any]]: + """获取单个SMTP配置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + SELECT id, name, enabled, is_primary, priority, host, port, + username, password, use_ssl, use_tls, sender_name, sender_email, + daily_limit, daily_sent, daily_reset_date, + last_success_at, last_error, success_count, fail_count, + created_at, updated_at + FROM smtp_configs WHERE id = ? + """, (config_id,)) + + row = cursor.fetchone() + if not row: + return None + + return { + 'id': row[0], + 'name': row[1], + 'enabled': bool(row[2]), + 'is_primary': bool(row[3]), + 'priority': row[4], + 'host': row[5], + 'port': row[6], + 'username': row[7], + 'password': '******' if row[8] and not include_password else (decrypt_password(row[8]) if include_password and row[8] else ''), + 'has_password': bool(row[8]), + 'use_ssl': bool(row[9]), + 'use_tls': bool(row[10]), + 'sender_name': row[11], + 'sender_email': row[12], + 'daily_limit': row[13], + 'daily_sent': row[14], + 'daily_reset_date': row[15], + 'last_success_at': row[16], + 'last_error': row[17], + 'success_count': row[18], + 'fail_count': row[19], + 'created_at': row[20], + 'updated_at': row[21] + } + + +def create_smtp_config(data: Dict[str, Any]) -> int: + """创建新的SMTP配置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + # 加密密码 + password = encrypt_password(data.get('password', '')) if data.get('password') else '' + + cursor.execute(""" + INSERT INTO smtp_configs + (name, enabled, is_primary, priority, host, port, username, password, + use_ssl, use_tls, sender_name, sender_email, daily_limit) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) + """, ( + data.get('name', '默认配置'), + int(data.get('enabled', True)), + int(data.get('is_primary', False)), + data.get('priority', 0), + data.get('host', ''), + data.get('port', 465), + data.get('username', ''), + password, + int(data.get('use_ssl', True)), + int(data.get('use_tls', False)), + data.get('sender_name', '知识管理平台'), + data.get('sender_email', ''), + data.get('daily_limit', 0) + )) + + config_id = cursor.lastrowid + conn.commit() + return config_id + + +def update_smtp_config(config_id: int, data: Dict[str, Any]) -> bool: + """更新SMTP配置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + updates = [] + params = [] + + field_mapping = { + 'name': 'name', + 'enabled': 'enabled', + 'priority': 'priority', + 'host': 'host', + 'port': 'port', + 'username': 'username', + 'use_ssl': 'use_ssl', + 'use_tls': 'use_tls', + 'sender_name': 'sender_name', + 'sender_email': 'sender_email', + 'daily_limit': 'daily_limit' + } + + for key, db_field in field_mapping.items(): + if key in data: + value = data[key] + if key in ('enabled', 'use_ssl', 'use_tls'): + value = int(value) + updates.append(f"{db_field} = ?") + params.append(value) + + # 密码特殊处理：只有当提供了新密码时才更新 + if 'password' in data and data['password'] and data['password'] != '******': + updates.append("password = ?") + params.append(encrypt_password(data['password'])) + + if not updates: + return False + + updates.append("updated_at = CURRENT_TIMESTAMP") + params.append(config_id) + + cursor.execute(f""" + UPDATE smtp_configs SET {', '.join(updates)} WHERE id = ? + """, params) + + conn.commit() + return cursor.rowcount > 0 + + +def delete_smtp_config(config_id: int) -> bool: + """删除SMTP配置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute("DELETE FROM smtp_configs WHERE id = ?", (config_id,)) + conn.commit() + return cursor.rowcount > 0 + + +def set_primary_smtp_config(config_id: int) -> bool: + """设置主SMTP配置""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + # 先取消所有主配置标记 + cursor.execute("UPDATE smtp_configs SET is_primary = 0") + + # 设置新的主配置 + cursor.execute(""" + UPDATE smtp_configs + SET is_primary = 1, updated_at = CURRENT_TIMESTAMP + WHERE id = ? + """, (config_id,)) + + conn.commit() + return cursor.rowcount > 0 + + +def _get_available_smtp_config(failover: bool = True) -> Optional[Dict[str, Any]]: + """ + 获取可用的SMTP配置 + 优先级: 主配置 > 按priority排序的启用配置 + """ + today = datetime.now().strftime('%Y-%m-%d') + + with db_pool.get_db() as conn: + cursor = conn.cursor() + + # 先重置过期的每日计数 + cursor.execute(""" + UPDATE smtp_configs + SET daily_sent = 0, daily_reset_date = ? + WHERE daily_reset_date != ? OR daily_reset_date IS NULL OR daily_reset_date = '' + """, (today, today)) + conn.commit() + + # 获取所有启用的配置，按优先级排序 + cursor.execute(""" + SELECT id, name, host, port, username, password, use_ssl, use_tls, + sender_name, sender_email, daily_limit, daily_sent, is_primary + FROM smtp_configs + WHERE enabled = 1 + ORDER BY is_primary DESC, priority ASC, id ASC + """) + + configs = cursor.fetchall() + + for row in configs: + config_id, name, host, port, username, password, use_ssl, use_tls, \ + sender_name, sender_email, daily_limit, daily_sent, is_primary = row + + # 检查每日限额 + if daily_limit > 0 and daily_sent >= daily_limit: + continue # 超过限额，跳过此配置 + + # 解密密码 + decrypted_password = decrypt_password(password) if password else '' + + return { + 'id': config_id, + 'name': name, + 'host': host, + 'port': port, + 'username': username, + 'password': decrypted_password, + 'use_ssl': bool(use_ssl), + 'use_tls': bool(use_tls), + 'sender_name': sender_name, + 'sender_email': sender_email, + 'is_primary': bool(is_primary) + } + + return None + + +def _update_smtp_stats(config_id: int, success: bool, error: str = ''): + """更新SMTP配置的统计信息""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + if success: + cursor.execute(""" + UPDATE smtp_configs + SET daily_sent = daily_sent + 1, + success_count = success_count + 1, + last_success_at = CURRENT_TIMESTAMP, + last_error = '', + updated_at = CURRENT_TIMESTAMP + WHERE id = ? + """, (config_id,)) + else: + cursor.execute(""" + UPDATE smtp_configs + SET fail_count = fail_count + 1, + last_error = ?, + updated_at = CURRENT_TIMESTAMP + WHERE id = ? + """, (error[:500], config_id)) + + conn.commit() + + +# ============ 邮件发送核心 ============ + +class EmailSender: + """邮件发送器""" + + def __init__(self, config: Dict[str, Any]): + self.config = config + self.server = None + + def connect(self) -> bool: + """连接SMTP服务器""" + try: + if self.config['use_ssl']: + self.server = smtplib.SMTP_SSL( + self.config['host'], + self.config['port'], + timeout=30 + ) + else: + self.server = smtplib.SMTP( + self.config['host'], + self.config['port'], + timeout=30 + ) + if self.config['use_tls']: + self.server.starttls() + + self.server.login(self.config['username'], self.config['password']) + return True + + except Exception as e: + print(f"[邮件服务] SMTP连接失败 [{self.config['name']}]: {e}") + self.server = None + raise + + def disconnect(self): + """断开SMTP连接""" + if self.server: + try: + self.server.quit() + except Exception: + pass + self.server = None + + def send(self, to_email: str, subject: str, body: str, + html_body: str = None, attachments: List[Dict] = None) -> bool: + """ + 发送邮件 + + Args: + to_email: 收件人邮箱 + subject: 邮件主题 + body: 纯文本正文 + html_body: HTML正文（可选） + attachments: 附件列表 [{'filename': 'xxx', 'data': bytes, 'mime_type': 'xxx'}] + """ + try: + if not self.server: + self.connect() + + # 构建邮件 + if html_body or attachments: + msg = MIMEMultipart('mixed') + + # 添加正文 + if html_body: + text_part = MIMEMultipart('alternative') + text_part.attach(MIMEText(body, 'plain', 'utf-8')) + text_part.attach(MIMEText(html_body, 'html', 'utf-8')) + msg.attach(text_part) + else: + msg.attach(MIMEText(body, 'plain', 'utf-8')) + + # 添加附件 + if attachments: + for att in attachments: + part = MIMEBase('application', 'octet-stream') + part.set_payload(att['data']) + encoders.encode_base64(part) + part.add_header( + 'Content-Disposition', + 'attachment', + filename=('utf-8', '', att['filename']) + ) + msg.attach(part) + else: + msg = MIMEText(body, 'plain', 'utf-8') + + # 设置邮件头 + msg['Subject'] = Header(subject, 'utf-8') + msg['From'] = formataddr(( + self.config['sender_name'], + self.config['sender_email'] or self.config['username'] + )) + msg['To'] = to_email + + # 发送 + self.server.sendmail( + self.config['sender_email'] or self.config['username'], + [to_email], + msg.as_string() + ) + + return True + + except Exception as e: + print(f"[邮件服务] 发送失败: {e}") + raise + + +def send_email( + to_email: str, + subject: str, + body: str, + html_body: str = None, + attachments: List[Dict] = None, + email_type: str = '', + user_id: int = None, + log_callback: Callable = None +) -> Dict[str, Any]: + """ + 发送邮件（带故障转移） + + Returns: + {'success': bool, 'error': str, 'config_id': int} + """ + # 检查全局开关 + settings = get_email_settings() + if not settings['enabled']: + return {'success': False, 'error': '邮件功能未启用', 'config_id': None} + + # 获取可用配置 + config = _get_available_smtp_config(settings['failover_enabled']) + if not config: + return {'success': False, 'error': '没有可用的SMTP配置', 'config_id': None} + + tried_configs = [] + last_error = '' + + while config: + tried_configs.append(config['id']) + sender = EmailSender(config) + + try: + sender.connect() + sender.send(to_email, subject, body, html_body, attachments) + sender.disconnect() + + # 更新统计 + _update_smtp_stats(config['id'], True) + _log_email_send(user_id, config['id'], to_email, email_type, subject, 'success', '', attachments) + _update_email_stats(email_type, True) + + if log_callback: + log_callback(f"[邮件服务] 发送成功: {to_email} (使用: {config['name']})") + + return {'success': True, 'error': '', 'config_id': config['id']} + + except Exception as e: + last_error = str(e) + sender.disconnect() + + # 更新失败统计 + _update_smtp_stats(config['id'], False, last_error) + + if log_callback: + log_callback(f"[邮件服务] 发送失败 [{config['name']}]: {e}") + + # 故障转移：尝试下一个配置 + if settings['failover_enabled']: + config = _get_next_available_smtp_config(tried_configs) + else: + config = None + + # 所有配置都失败 + _log_email_send(user_id, tried_configs[0] if tried_configs else None, + to_email, email_type, subject, 'failed', last_error, attachments) + _update_email_stats(email_type, False) + + return {'success': False, 'error': last_error, 'config_id': tried_configs[0] if tried_configs else None} + + +def _get_next_available_smtp_config(exclude_ids: List[int]) -> Optional[Dict[str, Any]]: + """获取下一个可用的SMTP配置（排除已尝试的）""" + today = datetime.now().strftime('%Y-%m-%d') + + with db_pool.get_db() as conn: + cursor = conn.cursor() + + placeholders = ','.join(['?' for _ in exclude_ids]) + cursor.execute(f""" + SELECT id, name, host, port, username, password, use_ssl, use_tls, + sender_name, sender_email, daily_limit, daily_sent, is_primary + FROM smtp_configs + WHERE enabled = 1 AND id NOT IN ({placeholders}) + ORDER BY is_primary DESC, priority ASC, id ASC + LIMIT 1 + """, exclude_ids) + + row = cursor.fetchone() + if not row: + return None + + config_id, name, host, port, username, password, use_ssl, use_tls, \ + sender_name, sender_email, daily_limit, daily_sent, is_primary = row + + # 检查每日限额 + if daily_limit > 0 and daily_sent >= daily_limit: + return _get_next_available_smtp_config(exclude_ids + [config_id]) + + return { + 'id': config_id, + 'name': name, + 'host': host, + 'port': port, + 'username': username, + 'password': decrypt_password(password) if password else '', + 'use_ssl': bool(use_ssl), + 'use_tls': bool(use_tls), + 'sender_name': sender_name, + 'sender_email': sender_email, + 'is_primary': bool(is_primary) + } + + +def test_smtp_config(config_id: int, test_email: str) -> Dict[str, Any]: + """测试SMTP配置""" + config = get_smtp_config(config_id, include_password=True) + if not config: + return {'success': False, 'error': '配置不存在'} + + sender = EmailSender({ + 'name': config['name'], + 'host': config['host'], + 'port': config['port'], + 'username': config['username'], + 'password': config['password'], + 'use_ssl': config['use_ssl'], + 'use_tls': config['use_tls'], + 'sender_name': config['sender_name'], + 'sender_email': config['sender_email'] + }) + + try: + sender.connect() + sender.send( + test_email, + '知识管理平台 - SMTP配置测试', + f'这是一封测试邮件。\n\n配置名称: {config["name"]}\nSMTP服务器: {config["host"]}:{config["port"]}\n\n如果您收到此邮件，说明SMTP配置正确。', + None, + None + ) + sender.disconnect() + + # 更新最后成功时间 + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + UPDATE smtp_configs + SET last_success_at = CURRENT_TIMESTAMP, last_error = '' + WHERE id = ? + """, (config_id,)) + conn.commit() + + return {'success': True, 'error': ''} + + except Exception as e: + # 记录错误 + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + UPDATE smtp_configs SET last_error = ? WHERE id = ? + """, (str(e)[:500], config_id)) + conn.commit() + + return {'success': False, 'error': str(e)} + + +# ============ 邮件日志 ============ + +def _log_email_send(user_id: int, smtp_config_id: int, email_to: str, + email_type: str, subject: str, status: str, + error_message: str, attachments: List[Dict] = None): + """记录邮件发送日志""" + attachment_count = len(attachments) if attachments else 0 + attachment_size = sum(len(a.get('data', b'')) for a in attachments) if attachments else 0 + + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + INSERT INTO email_logs + (user_id, smtp_config_id, email_to, email_type, subject, status, + error_message, attachment_count, attachment_size) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) + """, (user_id, smtp_config_id, email_to, email_type, subject, status, + error_message, attachment_count, attachment_size)) + conn.commit() + + +def _update_email_stats(email_type: str, success: bool): + """更新邮件统计""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + type_field_map = { + EMAIL_TYPE_REGISTER: 'register_sent', + EMAIL_TYPE_RESET: 'reset_sent', + EMAIL_TYPE_BIND: 'bind_sent', + EMAIL_TYPE_TASK_COMPLETE: 'task_complete_sent' + } + + type_field = type_field_map.get(email_type, '') + + if success: + if type_field: + cursor.execute(f""" + UPDATE email_stats + SET total_sent = total_sent + 1, + total_success = total_success + 1, + {type_field} = {type_field} + 1, + last_updated = CURRENT_TIMESTAMP + WHERE id = 1 + """) + else: + cursor.execute(""" + UPDATE email_stats + SET total_sent = total_sent + 1, + total_success = total_success + 1, + last_updated = CURRENT_TIMESTAMP + WHERE id = 1 + """) + else: + cursor.execute(""" + UPDATE email_stats + SET total_sent = total_sent + 1, + total_failed = total_failed + 1, + last_updated = CURRENT_TIMESTAMP + WHERE id = 1 + """) + + conn.commit() + + +def get_email_stats() -> Dict[str, Any]: + """获取邮件统计""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + SELECT total_sent, total_success, total_failed, + register_sent, reset_sent, bind_sent, task_complete_sent, + last_updated + FROM email_stats WHERE id = 1 + """) + row = cursor.fetchone() + if row: + return { + 'total_sent': row[0], + 'total_success': row[1], + 'total_failed': row[2], + 'register_sent': row[3], + 'reset_sent': row[4], + 'bind_sent': row[5], + 'task_complete_sent': row[6], + 'last_updated': row[7], + 'success_rate': round(row[1] / row[0] * 100, 1) if row[0] > 0 else 0 + } + return {} + + +def get_email_logs(page: int = 1, page_size: int = 20, + email_type: str = None, status: str = None) -> Dict[str, Any]: + """获取邮件日志（分页）""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + + where_clauses = [] + params = [] + + if email_type: + where_clauses.append("l.email_type = ?") + params.append(email_type) + + if status: + where_clauses.append("l.status = ?") + params.append(status) + + where_sql = f"WHERE {' AND '.join(where_clauses)}" if where_clauses else "" + + # 获取总数 + cursor.execute(f"SELECT COUNT(*) FROM email_logs l {where_sql}", params) + total = cursor.fetchone()[0] + + # 获取分页数据 + offset = (page - 1) * page_size + cursor.execute(f""" + SELECT l.id, l.user_id, l.smtp_config_id, l.email_to, l.email_type, + l.subject, l.status, l.error_message, l.attachment_count, + l.attachment_size, l.created_at, u.username, s.name as smtp_name + FROM email_logs l + LEFT JOIN users u ON l.user_id = u.id + LEFT JOIN smtp_configs s ON l.smtp_config_id = s.id + {where_sql} + ORDER BY l.created_at DESC + LIMIT ? OFFSET ? + """, params + [page_size, offset]) + + logs = [] + for row in cursor.fetchall(): + logs.append({ + 'id': row[0], + 'user_id': row[1], + 'smtp_config_id': row[2], + 'email_to': row[3], + 'email_type': row[4], + 'subject': row[5], + 'status': row[6], + 'error_message': row[7], + 'attachment_count': row[8], + 'attachment_size': row[9], + 'created_at': row[10], + 'username': row[11], + 'smtp_name': row[12] + }) + + return { + 'logs': logs, + 'total': total, + 'page': page, + 'page_size': page_size, + 'total_pages': (total + page_size - 1) // page_size + } + + +def cleanup_email_logs(days: int = 30) -> int: + """清理过期邮件日志""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + DELETE FROM email_logs + WHERE datetime(created_at) < datetime('now', '-' || ? || ' days') + """, (days,)) + deleted = cursor.rowcount + conn.commit() + return deleted + + +# ============ Token管理 ============ + +def generate_email_token(email: str, token_type: str, user_id: int = None) -> str: + """生成邮件验证Token""" + token = secrets.token_urlsafe(32) + + # 设置过期时间 + expire_seconds = { + EMAIL_TYPE_REGISTER: TOKEN_EXPIRE_REGISTER, + EMAIL_TYPE_RESET: TOKEN_EXPIRE_RESET, + EMAIL_TYPE_BIND: TOKEN_EXPIRE_BIND + }.get(token_type, TOKEN_EXPIRE_REGISTER) + + expires_at = datetime.now() + timedelta(seconds=expire_seconds) + + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + INSERT INTO email_tokens (user_id, email, token, token_type, expires_at) + VALUES (?, ?, ?, ?, ?) + """, (user_id, email, token, token_type, expires_at)) + conn.commit() + + return token + + +def verify_email_token(token: str, token_type: str) -> Optional[Dict[str, Any]]: + """ + 验证Token + + Returns: + 成功返回 {'user_id': int, 'email': str}，失败返回 None + """ + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + SELECT id, user_id, email, expires_at, used + FROM email_tokens + WHERE token = ? AND token_type = ? + """, (token, token_type)) + + row = cursor.fetchone() + if not row: + return None + + token_id, user_id, email, expires_at, used = row + + # 检查是否已使用 + if used: + return None + + # 检查是否过期 + if datetime.strptime(expires_at, '%Y-%m-%d %H:%M:%S') < datetime.now(): + return None + + # 标记为已使用 + cursor.execute(""" + UPDATE email_tokens SET used = 1 WHERE id = ? + """, (token_id,)) + conn.commit() + + return {'user_id': user_id, 'email': email} + + +def check_rate_limit(email: str, token_type: str) -> bool: + """ + 检查发送频率限制 + + Returns: + True = 可以发送, False = 受限 + """ + limit_seconds = { + EMAIL_TYPE_REGISTER: RATE_LIMIT_REGISTER, + EMAIL_TYPE_RESET: RATE_LIMIT_RESET, + EMAIL_TYPE_BIND: RATE_LIMIT_BIND + }.get(token_type, 60) + + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + SELECT created_at FROM email_tokens + WHERE email = ? AND token_type = ? + ORDER BY created_at DESC + LIMIT 1 + """, (email, token_type)) + + row = cursor.fetchone() + if not row: + return True + + last_sent = datetime.strptime(row[0], '%Y-%m-%d %H:%M:%S') + elapsed = (datetime.now() - last_sent).total_seconds() + + return elapsed >= limit_seconds + + +def cleanup_expired_tokens() -> int: + """清理过期Token""" + with db_pool.get_db() as conn: + cursor = conn.cursor() + cursor.execute(""" + DELETE FROM email_tokens + WHERE datetime(expires_at) < datetime('now') + """) + deleted = cursor.rowcount + conn.commit() + return deleted + + +# ============ 异步发送队列 ============ + +class EmailQueue: + """邮件异步发送队列""" + + def __init__(self, workers: int = QUEUE_WORKERS, max_size: int = QUEUE_MAX_SIZE): + self.queue = queue.Queue(maxsize=max_size) + self.workers = [] + self.running = False + self.worker_count = workers + + def start(self): + """启动队列工作线程""" + if self.running: + return + + self.running = True + for i in range(self.worker_count): + worker = threading.Thread(target=self._worker, daemon=True, name=f"EmailWorker-{i+1}") + worker.start() + self.workers.append(worker) + + print(f"[邮件服务] 异步队列已启动 ({self.worker_count}个工作线程)") + + def stop(self): + """停止队列""" + self.running = False + + # 发送停止信号 + for _ in self.workers: + try: + self.queue.put(None, timeout=1) + except queue.Full: + pass + + # 等待工作线程结束 + for worker in self.workers: + worker.join(timeout=5) + + self.workers.clear() + print("[邮件服务] 异步队列已停止") + + def _worker(self): + """工作线程""" + while self.running: + try: + task = self.queue.get(timeout=5) + if task is None: + break + + self._process_task(task) + + except queue.Empty: + continue + except Exception as e: + print(f"[邮件服务] 队列工作线程错误: {e}") + + def _process_task(self, task: Dict): + """处理邮件任务""" + try: + result = send_email( + to_email=task['to_email'], + subject=task['subject'], + body=task['body'], + html_body=task.get('html_body'), + attachments=task.get('attachments'), + email_type=task.get('email_type', ''), + user_id=task.get('user_id'), + log_callback=task.get('log_callback') + ) + + # 执行回调 + if task.get('callback'): + task['callback'](result) + + except Exception as e: + print(f"[邮件服务] 处理邮件任务失败: {e}") + if task.get('callback'): + task['callback']({'success': False, 'error': str(e)}) + + def enqueue(self, to_email: str, subject: str, body: str, + html_body: str = None, attachments: List[Dict] = None, + email_type: str = '', user_id: int = None, + callback: Callable = None, log_callback: Callable = None) -> bool: + """ + 将邮件任务加入队列 + + Returns: + 是否成功加入队列 + """ + try: + self.queue.put({ + 'to_email': to_email, + 'subject': subject, + 'body': body, + 'html_body': html_body, + 'attachments': attachments, + 'email_type': email_type, + 'user_id': user_id, + 'callback': callback, + 'log_callback': log_callback + }, timeout=5) + return True + except queue.Full: + print("[邮件服务] 邮件队列已满") + return False + + @property + def pending_count(self) -> int: + """队列中待处理的任务数""" + return self.queue.qsize() + + +# 全局队列实例 +_email_queue: Optional[EmailQueue] = None +_queue_lock = threading.Lock() + + +def get_email_queue() -> EmailQueue: + """获取全局邮件队列（单例）""" + global _email_queue + + with _queue_lock: + if _email_queue is None: + _email_queue = EmailQueue() + _email_queue.start() + return _email_queue + + +def shutdown_email_queue(): + """关闭邮件队列""" + global _email_queue + + with _queue_lock: + if _email_queue: + _email_queue.stop() + _email_queue = None + + +# ============ 便捷函数 ============ + +def send_email_async(to_email: str, subject: str, body: str, + html_body: str = None, attachments: List[Dict] = None, + email_type: str = '', user_id: int = None, + callback: Callable = None, log_callback: Callable = None) -> bool: + """异步发送邮件""" + queue = get_email_queue() + return queue.enqueue( + to_email=to_email, + subject=subject, + body=body, + html_body=html_body, + attachments=attachments, + email_type=email_type, + user_id=user_id, + callback=callback, + log_callback=log_callback + ) + + +def create_zip_attachment(files: List[Dict[str, Any]], zip_filename: str = 'screenshots.zip') -> Dict: + """ + 创建ZIP附件 + + Args: + files: [{'filename': 'xxx.png', 'data': bytes}] + zip_filename: ZIP文件名 + + Returns: + {'filename': 'xxx.zip', 'data': bytes} + """ + buffer = BytesIO() + with zipfile.ZipFile(buffer, 'w', zipfile.ZIP_DEFLATED) as zf: + for f in files: + zf.writestr(f['filename'], f['data']) + + return { + 'filename': zip_filename, + 'data': buffer.getvalue() + } + + +# ============ 初始化 ============ + +def init_email_service(): + """初始化邮件服务""" + init_email_tables() + get_email_queue() + print("[邮件服务] 初始化完成") + + +if __name__ == '__main__': + # 测试代码 + init_email_tables() + print("邮件服务模块测试完成") diff --git a/templates/admin.html b/templates/admin.html index ee0af9c..e6406be 100644 --- a/templates/admin.html +++ b/templates/admin.html @@ -732,6 +732,7 @@ + @@ -1199,6 +1200,111 @@ + +
+

邮件功能设置

+ + +
+
+ +
+ 开启后，系统将支持邮箱验证、密码重置邮件、任务完成通知等功能 +
+
+
+ +
+ 开启后，主SMTP配置发送失败时自动切换到备用配置 +
+
+
+ + +
+

SMTP配置列表

+ +
+ +
+
加载中...
+
+ + +

邮件发送统计

+
+
+
0
+
总发送
+
+
+
0
+
成功
+
+
+
0
+
失败
+
+
+
0%
+
成功率
+
+
+ +
+
+
0
+
注册验证
+
+
+
0
+
密码重置
+
+
+
0
+
邮箱绑定
+
+
+
0
+
任务完成
+
+
+ + +
+

邮件发送日志

+
+ + + +
+
+ +
+
加载中...
+
+ + +
+
+
+

管理员账号设置

@@ -1218,6 +1324,106 @@
+ + +
@@ -1274,6 +1480,14 @@ if (tabName === 'feedbacks') { loadFeedbacks(); } + + // 切换到邮件配置标签时加载邮件相关数据 + if (tabName === 'email') { + loadEmailSettings(); + loadSmtpConfigs(); + loadEmailStats(); + loadEmailLogs(); + } } // VIP functions @@ -2551,6 +2765,458 @@ showNotification('删除失败: ' + error.message, 'error'); } } + + // ==================== 邮件配置功能 ==================== + + let smtpConfigs = []; + let currentEmailLogPage = 1; + let totalEmailLogPages = 1; + + // 加载邮件设置 + async function loadEmailSettings() { + try { + const response = await fetch('/yuyx/api/email/settings'); + if (response.ok) { + const data = await response.json(); + document.getElementById('emailEnabled').checked = data.enabled; + document.getElementById('failoverEnabled').checked = data.failover_enabled; + } + } catch (error) { + console.error('加载邮件设置失败:', error); + } + } + + // 更新邮件设置 + async function updateEmailSettings() { + try { + const response = await fetch('/yuyx/api/email/settings', { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ + enabled: document.getElementById('emailEnabled').checked, + failover_enabled: document.getElementById('failoverEnabled').checked + }) + }); + + if (response.ok) { + showNotification('邮件设置已更新', 'success'); + } else { + const data = await response.json(); + showNotification('更新失败: ' + data.error, 'error'); + } + } catch (error) { + showNotification('更新失败: ' + error.message, 'error'); + } + } + + // 加载SMTP配置列表 + async function loadSmtpConfigs() { + try { + const response = await fetch('/yuyx/api/smtp/configs'); + if (response.ok) { + smtpConfigs = await response.json(); + renderSmtpConfigs(); + } + } catch (error) { + console.error('加载SMTP配置失败:', error); + document.getElementById('smtpConfigsList').innerHTML = + '
加载失败
'; + } + } + + // 渲染SMTP配置列表 + function renderSmtpConfigs() { + const container = document.getElementById('smtpConfigsList'); + + if (smtpConfigs.length === 0) { + container.innerHTML = '
暂无SMTP配置，请点击"添加配置"创建
'; + return; + } + + let html = '
'; + html += ''; + html += ''; + html += ''; + html += ''; + html += ''; + html += ''; + html += ''; + + smtpConfigs.forEach(config => { + const statusIcon = config.is_primary ? '⭐主' : + (config.enabled ? '✓备用' : '✗禁用'); + const statusClass = config.is_primary ? 'color: #f39c12;' : + (config.enabled ? 'color: #27ae60;' : 'color: #95a5a6;'); + + const dailyText = config.daily_limit > 0 ? + `${config.daily_sent}/${config.daily_limit}` : `${config.daily_sent}/∞`; + + html += ''; + html += ``; + html += ``; + html += ``; + html += ``; + html += ``; + html += ``; + html += ''; + }); + + html += '
状态名称服务器今日/限额成功率操作
${statusIcon}${config.name}${config.host}:${config.port}${dailyText}${config.success_rate}%
'; + container.innerHTML = html; + } + + // 显示SMTP配置弹窗 + function showSmtpModal(configId = null) { + document.getElementById('smtpModal').style.display = 'block'; + + if (configId) { + // 编辑模式 + document.getElementById('smtpModalTitle').textContent = '编辑SMTP配置'; + document.getElementById('smtpEditActions').style.display = 'block'; + + const config = smtpConfigs.find(c => c.id === configId); + if (config) { + document.getElementById('smtpConfigId').value = config.id; + document.getElementById('smtpName').value = config.name; + document.getElementById('smtpEnabled').checked = config.enabled; + document.getElementById('smtpHost').value = config.host; + document.getElementById('smtpPort').value = config.port; + document.getElementById('smtpUsername').value = config.username; + document.getElementById('smtpPassword').value = ''; + document.getElementById('smtpPassword').placeholder = config.has_password ? '留空保持不变' : 'SMTP密码或授权码'; + document.getElementById('smtpUseSsl').checked = config.use_ssl; + document.getElementById('smtpUseTls').checked = config.use_tls; + document.getElementById('smtpSenderName').value = config.sender_name; + document.getElementById('smtpSenderEmail').value = config.sender_email; + document.getElementById('smtpDailyLimit').value = config.daily_limit; + document.getElementById('smtpPriority').value = config.priority; + } + } else { + // 添加模式 + document.getElementById('smtpModalTitle').textContent = '添加SMTP配置'; + document.getElementById('smtpEditActions').style.display = 'none'; + document.getElementById('smtpConfigId').value = ''; + document.getElementById('smtpName').value = ''; + document.getElementById('smtpEnabled').checked = true; + document.getElementById('smtpHost').value = ''; + document.getElementById('smtpPort').value = 465; + document.getElementById('smtpUsername').value = ''; + document.getElementById('smtpPassword').value = ''; + document.getElementById('smtpPassword').placeholder = 'SMTP密码或授权码'; + document.getElementById('smtpUseSsl').checked = true; + document.getElementById('smtpUseTls').checked = false; + document.getElementById('smtpSenderName').value = '知识管理平台'; + document.getElementById('smtpSenderEmail').value = ''; + document.getElementById('smtpDailyLimit').value = 0; + document.getElementById('smtpPriority').value = 0; + } + } + + // 隐藏SMTP配置弹窗 + function hideSmtpModal() { + document.getElementById('smtpModal').style.display = 'none'; + } + + // 编辑SMTP配置 + function editSmtpConfig(configId) { + showSmtpModal(configId); + } + + // 保存SMTP配置 + async function saveSmtpConfig() { + const configId = document.getElementById('smtpConfigId').value; + const host = document.getElementById('smtpHost').value.trim(); + const username = document.getElementById('smtpUsername').value.trim(); + + if (!host) { + showNotification('请输入SMTP服务器地址', 'error'); + return; + } + if (!username) { + showNotification('请输入SMTP用户名', 'error'); + return; + } + + const data = { + name: document.getElementById('smtpName').value.trim() || '默认配置', + enabled: document.getElementById('smtpEnabled').checked, + host: host, + port: parseInt(document.getElementById('smtpPort').value) || 465, + username: username, + use_ssl: document.getElementById('smtpUseSsl').checked, + use_tls: document.getElementById('smtpUseTls').checked, + sender_name: document.getElementById('smtpSenderName').value.trim(), + sender_email: document.getElementById('smtpSenderEmail').value.trim(), + daily_limit: parseInt(document.getElementById('smtpDailyLimit').value) || 0, + priority: parseInt(document.getElementById('smtpPriority').value) || 0 + }; + + const password = document.getElementById('smtpPassword').value; + if (password) { + data.password = password; + } + + try { + let response; + if (configId) { + // 更新 + response = await fetch('/yuyx/api/smtp/configs/' + configId, { + method: 'PUT', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify(data) + }); + } else { + // 新增 + if (!password) { + showNotification('新建配置需要输入密码', 'error'); + return; + } + response = await fetch('/yuyx/api/smtp/configs', { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify(data) + }); + } + + const result = await response.json(); + if (response.ok) { + showNotification(configId ? '配置已更新' : '配置已添加', 'success'); + hideSmtpModal(); + loadSmtpConfigs(); + } else { + showNotification('保存失败: ' + result.error, 'error'); + } + } catch (error) { + showNotification('保存失败: ' + error.message, 'error'); + } + } + + // 测试SMTP配置 + async function testSmtpConfig() { + const configId = document.getElementById('smtpConfigId').value; + if (!configId) { + showNotification('请先保存配置再测试', 'error'); + return; + } + + const testEmail = prompt('请输入测试接收邮箱:'); + if (!testEmail) return; + + showNotification('正在发送测试邮件...', 'info'); + + try { + const response = await fetch('/yuyx/api/smtp/configs/' + configId + '/test', { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ email: testEmail }) + }); + + const result = await response.json(); + if (result.success) { + showNotification('测试邮件发送成功！请检查收件箱', 'success'); + } else { + showNotification('测试失败: ' + result.error, 'error'); + } + } catch (error) { + showNotification('测试失败: ' + error.message, 'error'); + } + } + + // 设为主配置 + async function setPrimarySmtp() { + const configId = document.getElementById('smtpConfigId').value; + if (!configId) return; + + if (!confirm('确定要将此配置设为主配置吗？')) return; + + try { + const response = await fetch('/yuyx/api/smtp/configs/' + configId + '/primary', { + method: 'POST' + }); + + if (response.ok) { + showNotification('已设为主配置', 'success'); + hideSmtpModal(); + loadSmtpConfigs(); + } else { + const result = await response.json(); + showNotification('设置失败: ' + result.error, 'error'); + } + } catch (error) { + showNotification('设置失败: ' + error.message, 'error'); + } + } + + // 删除SMTP配置 + async function deleteSmtpConfig() { + const configId = document.getElementById('smtpConfigId').value; + if (!configId) return; + + if (!confirm('确定要删除此SMTP配置吗？此操作不可恢复！')) return; + + try { + const response = await fetch('/yuyx/api/smtp/configs/' + configId, { + method: 'DELETE' + }); + + if (response.ok) { + showNotification('配置已删除', 'success'); + hideSmtpModal(); + loadSmtpConfigs(); + } else { + const result = await response.json(); + showNotification('删除失败: ' + result.error, 'error'); + } + } catch (error) { + showNotification('删除失败: ' + error.message, 'error'); + } + } + + // 加载邮件统计 + async function loadEmailStats() { + try { + const response = await fetch('/yuyx/api/email/stats'); + if (response.ok) { + const stats = await response.json(); + document.getElementById('statTotalSent').textContent = stats.total_sent || 0; + document.getElementById('statTotalSuccess').textContent = stats.total_success || 0; + document.getElementById('statTotalFailed').textContent = stats.total_failed || 0; + document.getElementById('statSuccessRate').textContent = (stats.success_rate || 0) + '%'; + document.getElementById('statRegister').textContent = stats.register_sent || 0; + document.getElementById('statReset').textContent = stats.reset_sent || 0; + document.getElementById('statBind').textContent = stats.bind_sent || 0; + document.getElementById('statTaskComplete').textContent = stats.task_complete_sent || 0; + } + } catch (error) { + console.error('加载邮件统计失败:', error); + } + } + + // 加载邮件日志 + async function loadEmailLogs(page = 1) { + currentEmailLogPage = page; + + const typeFilter = document.getElementById('emailLogTypeFilter').value; + const statusFilter = document.getElementById('emailLogStatusFilter').value; + + let url = `/yuyx/api/email/logs?page=${page}&page_size=15`; + if (typeFilter) url += `&type=${typeFilter}`; + if (statusFilter) url += `&status=${statusFilter}`; + + try { + const response = await fetch(url); + if (response.ok) { + const data = await response.json(); + totalEmailLogPages = data.total_pages; + renderEmailLogs(data.logs); + renderEmailLogsPagination(data); + } + } catch (error) { + console.error('加载邮件日志失败:', error); + document.getElementById('emailLogsList').innerHTML = + '
加载失败
'; + } + } + + // 渲染邮件日志 + function renderEmailLogs(logs) { + const container = document.getElementById('emailLogsList'); + + if (!logs || logs.length === 0) { + container.innerHTML = '
暂无邮件日志
'; + return; + } + + const typeMap = { + 'register': '注册验证', + 'reset': '密码重置', + 'bind': '邮箱绑定', + 'task_complete': '任务完成' + }; + + let html = '
'; + html += ''; + html += ''; + + logs.forEach(log => { + const statusClass = log.status === 'success' ? 'color: #27ae60;' : 'color: #e74c3c;'; + const statusText = log.status === 'success' ? '成功' : '失败'; + + html += ''; + html += ``; + html += ``; + html += ``; + html += ``; + html += ``; + html += ``; + html += ''; + }); + + html += '
时间收件人类型主题状态错误
${log.created_at}${log.email_to}${typeMap[log.email_type] || log.email_type}${log.subject}${statusText}${log.error_message || '-'}
'; + container.innerHTML = html; + } + + // 渲染邮件日志分页 + function renderEmailLogsPagination(data) { + const container = document.getElementById('emailLogsPagination'); + + if (data.total_pages <= 1) { + container.innerHTML = `共 ${data.total} 条记录`; + return; + } + + let html = ''; + html += ``; + html += ``; + html += `第 ${data.page} 页 / 共 ${data.total_pages} 页`; + html += ``; + html += ``; + html += `共 ${data.total} 条记录`; + + container.innerHTML = html; + } + + // 清理邮件日志 + async function cleanupEmailLogs() { + const days = prompt('请输入保留天数（将删除该天数之前的日志）:', '30'); + if (!days) return; + + const daysNum = parseInt(days); + if (isNaN(daysNum) || daysNum < 7) { + showNotification('天数必须大于等于7', 'error'); + return; + } + + if (!confirm(`确定要删除 ${daysNum} 天之前的邮件日志吗？`)) return; + + try { + const response = await fetch('/yuyx/api/email/logs/cleanup', { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ days: daysNum }) + }); + + const result = await response.json(); + if (response.ok) { + showNotification(`已清理 ${result.deleted} 条日志`, 'success'); + loadEmailLogs(); + } else { + showNotification('清理失败: ' + result.error, 'error'); + } + } catch (error) { + showNotification('清理失败: ' + error.message, 'error'); + } + } + + // 切换密码显示 + function togglePasswordVisibility(inputId) { + const input = document.getElementById(inputId); + if (input.type === 'password') { + input.type = 'text'; + } else { + input.type = 'password'; + } + } \ No newline at end of file