import json import logging import os from datetime import datetime from src.configs.logging_setup import setup_logging setup_logging() logger = logging.getLogger(__name__) from flask import Flask, render_template, jsonify, request, redirect, url_for, send_from_directory from src.routes.api_config_routes import api_config_bp from src.routes.search_routes import search_bp from src.routes.resource_routes import resources_bp from src.routes.auth_routes import auth_bp from src.routes.system_config_routes import system_config_bp from src.routes.plugin_routes import plugin_bp from src.routes.search_sources_routes import search_sources_bp from src.routes.dashboard_routes import dashboard_bp from src.routes.api_v1_routes import api_v1_bp from src.routes.log_routes import log_bp from src.configs.app_config import SECRET_KEY from src.db.connection import init_sqlite_db from src.services.scheduler_service import start_scheduler from src.services.system_config_service import ( get_frontend_link_mode, get_frontend_display_netdisk_config, get_frontend_link_check_config, get_dynamic_transfer_netdisk_config, is_excel_download_enabled, is_pc_qr_code_enabled, is_frontend_github_enabled, get_frontend_github_config, is_frontend_enabled, is_api_only_enabled, get_active_theme, get_site_logo, get_site_name, build_frontend_seo_meta, is_resource_request_enabled, is_hot_searches_enabled, get_hot_search_words, get_frontend_disclaimer_text, get_frontend_copyright_text, get_frontend_seo_config, is_frontend_footer_text_enabled, is_frontend_disclaimer_enabled, is_frontend_copyright_enabled, ) from src.services.theme_service import theme_static_bp from src.utils.netdisk_utils import get_netdisk_drive_type import jinja2 app = Flask(__name__) app.config['TEMPLATES_AUTO_RELOAD'] = True # 配置多主题模板加载器支持 themes_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "themes") app.jinja_loader = jinja2.ChoiceLoader([ app.jinja_loader, jinja2.FileSystemLoader(themes_path), ]) app.secret_key = SECRET_KEY FRONTEND_NETDISK_ICONS = { "夸克网盘": "kuake-dark.svg", "百度网盘": "icon_baiduwangpan-dark.svg", "阿里云盘": "ali-dark.svg", "迅雷网盘": "xunlei1-dark.svg", "UC网盘": "changyonglogo31-dark.svg", "光鸭云盘": "guangya-dark.svg", "悟空网盘": "wukong-dark.svg", "移动云盘": "yidong-dark.svg", "磁力链接": "zhongzi-dark.svg", "迅雷链接": "zhongzi-dark.svg", "电驴链接": "zhongzi-dark.svg", } def get_frontend_theme_netdisks(limit=9, include_icons=True): """按后台配置顺序准备前台网盘标签；默认最多展示九项。""" enabled_netdisks = get_frontend_display_netdisk_config().get("enabled_netdisks", []) if limit is not None: enabled_netdisks = enabled_netdisks[:limit] result = [] for name in enabled_netdisks: label = name for suffix in ("网盘", "云盘", "链接"): if label.endswith(suffix): label = label[:-len(suffix)] break icon = FRONTEND_NETDISK_ICONS.get(name) if include_icons else None result.append({ "name": name, "label": label or name, "drive": get_netdisk_drive_type(name), "icon": f"/themes/aurora/assets/{icon}" if icon else "", }) return result # 初始化 SQLite 数据库与表结构 init_sqlite_db() # 初始化管理员账号认证配置（初次启动哈希落库） from src.services.admin_auth_service import init_admin_auth, is_using_default_password, get_admin_username init_admin_auth() # 注册蓝图 app.register_blueprint(theme_static_bp) app.register_blueprint(auth_bp) app.register_blueprint(dashboard_bp) app.register_blueprint(api_config_bp) app.register_blueprint(search_bp) app.register_blueprint(resources_bp) app.register_blueprint(system_config_bp) app.register_blueprint(plugin_bp) app.register_blueprint(search_sources_bp) app.register_blueprint(api_v1_bp) app.register_blueprint(log_bp) start_scheduler() # 上下文处理器，将登录状态传递给所有模板 @app.context_processor def inject_login_status(): token = request.cookies.get('token') is_logged_in = False try: if token: import jwt jwt.decode(token, app.secret_key, algorithms=['HS256']) is_logged_in = True except Exception: pass is_default_pwd = False admin_name = 'Admin' if is_logged_in: try: is_default_pwd = is_using_default_password() admin_name = get_admin_username() except Exception: pass frontend_seo_config = get_frontend_seo_config() return { 'is_logged_in': is_logged_in, 'enable_frontend_github': is_frontend_github_enabled(), 'enable_frontend_footer_text': is_frontend_footer_text_enabled(), 'frontend_footer_text': get_frontend_github_config().get('footer_text', 'Pan-Relay 发现海量网盘资源'), 'enable_frontend_disclaimer': is_frontend_disclaimer_enabled(), 'frontend_disclaimer_text': get_frontend_disclaimer_text(), 'enable_frontend_copyright': is_frontend_copyright_enabled(), 'frontend_copyright_text': get_frontend_copyright_text(), 'frontend_friend_links': frontend_seo_config.get('friend_links', {'enabled': False, 'items': []}), 'frontend_announcement': frontend_seo_config.get('announcement', {'enabled': True, 'text': '每日持续更新，热门影视与网盘资源一站搜索。'}), 'is_default_admin_password': is_default_pwd, 'admin_username': admin_name, 'site_logo': get_site_logo(), 'site_name': get_site_name(), 'current_year': datetime.now().year, 'enable_resource_request': is_resource_request_enabled(), 'enable_hot_searches': is_hot_searches_enabled(), 'hot_search_words': get_hot_search_words(), } @app.before_request def check_ui_access(): path = request.path # 若试图访问前台根路径但前台 UI 已关闭，自动重定向至后台管理登录页 if path == "/" and not is_frontend_enabled(): return redirect(url_for('auth.login')) # 首页，支持根据当前激活主题渲染不同模板 @app.route('/') def search_index(): if not is_frontend_enabled(): return redirect(url_for('auth.login')) active_theme = get_active_theme() link_mode = get_frontend_link_mode() link_check_config = get_frontend_link_check_config() dynamic_transfer_config = get_dynamic_transfer_netdisk_config() search_term = request.args.get('keyword') or request.args.get('q') or request.args.get('kw') or '' ctx = { 'seo_meta': build_frontend_seo_meta('search' if search_term.strip() else 'home', search_term), 'frontend_link_mode': link_mode, 'allow_excel_download': (link_mode != "view") and is_excel_download_enabled(), 'enable_link_check': link_check_config.get("enable_link_check", True), 'enabled_check_pans': link_check_config.get("enabled_check_pans", []), 'enabled_dynamic_transfer_pans': dynamic_transfer_config.get("enabled_netdisks", []), 'enable_pc_qr_code': is_pc_qr_code_enabled(), 'enable_frontend_github': is_frontend_github_enabled(), 'frontend_footer_text': get_frontend_github_config().get('footer_text', 'Pan-Relay 发现海量网盘资源'), 'enable_resource_request': is_resource_request_enabled(), 'enable_hot_searches': is_hot_searches_enabled(), 'hot_search_words': get_hot_search_words(), 'frontend_theme_netdisks': ( get_frontend_theme_netdisks(limit=None, include_icons=False) if active_theme == "cinema" else get_frontend_theme_netdisks() ), } if active_theme == "cinema": from src.db.resources import get_resource_hall_items ctx['resource_hall_items'] = get_resource_hall_items(limit=8) return render_template('cinema/templates/index.html', **ctx) if active_theme == "aurora": return render_template('aurora/templates/index.html', **ctx) return render_template('classic/templates/index.html', **ctx) # 搜索结果页 (主要用于 aurora 主题或独立搜索 URL) @app.route('/search') def search_page(): if not is_frontend_enabled(): return redirect(url_for('auth.login')) active_theme = get_active_theme() q = request.args.get('q') or request.args.get('keyword') or '' link_mode = get_frontend_link_mode() link_check_config = get_frontend_link_check_config() if active_theme == "cinema": # 模板三的侧栏应完整遵循后台勾选项，不受模板二的九项上限影响。 netdisks = get_frontend_theme_netdisks(limit=None, include_icons=False) return render_template( 'cinema/templates/search.html', query=q, seo_meta=build_frontend_seo_meta('search', q), frontend_link_mode=link_mode, enable_link_check=link_check_config.get("enable_link_check", True), enabled_check_pans=link_check_config.get("enabled_check_pans", []), enable_pc_qr_code=is_pc_qr_code_enabled(), enable_resource_request=is_resource_request_enabled(), enable_hot_searches=is_hot_searches_enabled(), hot_search_words=get_hot_search_words(), frontend_theme_netdisks=netdisks, ) if active_theme == "aurora": return render_template( 'aurora/templates/search.html', query=q, seo_meta=build_frontend_seo_meta('search', q), frontend_link_mode=link_mode, enable_link_check=link_check_config.get("enable_link_check", True), enabled_check_pans=link_check_config.get("enabled_check_pans", []), enable_pc_qr_code=is_pc_qr_code_enabled(), enable_resource_request=is_resource_request_enabled(), enable_hot_searches=is_hot_searches_enabled(), hot_search_words=get_hot_search_words(), frontend_theme_netdisks=get_frontend_theme_netdisks(), ) return redirect(url_for('search_index', keyword=q)) # 独立 API 接入文档网页 @app.route('/docs') def public_api_docs_page(): return render_template('api_docs.html') # 根路径 Favicon 路由 (消除浏览器自动请求 /favicon.ico 导致的 404，统一返回系统项目图标) @app.route('/favicon.ico') def favicon(): return send_from_directory( os.path.join(app.root_path, 'static', 'images'), 'favicon.ico', mimetype='image/vnd.microsoft.icon' ) if __name__ == '__main__': port = int(os.getenv("PORT", 5004)) logger.info(f"启动 Flask 应用，监听端口: {port}") app.run(host='0.0.0.0', port=port)