app.py

import os
import sys
import threading
import webbrowser
import subprocess
import re
import time
import logging
from flask import Flask, send_from_directory, jsonify

# Отключаем мусорные логи Flask и ngrok
log = logging.getLogger('werkzeug')
log.setLevel(logging.ERROR)
logging.getLogger('pyngrok').setLevel(logging.ERROR)

from backend.settings import settings
from backend.routes import api_bp

# Пытаемся импортировать библиотеку для консольного QR
try:
    import qrcode
    HAS_QR = True
except ImportError:
    HAS_QR = False

settings.load_config()
settings.load_data()

app = Flask(__name__, static_folder=str(settings.APP_ROOT / "static"), static_url_path="/static")
app.register_blueprint(api_bp)

public_url = None
local_url = f"http://{settings.CONFIG['host']}:{settings.CONFIG['port']}"

@app.route("/")
def index(): 
    return send_from_directory(app.static_folder, "index.html")

@app.route('/favicon.png')
def favicon(): 
    return send_from_directory(app.root_path, 'favicon.png', mimetype='image/png')

@app.route('/api/tunnel')
def get_tunnel():
    """Возвращает публичный URL туннеля для QR-кода на фронтенде"""
    for _ in range(10):
        if public_url:
            break
        time.sleep(0.5)
    return jsonify({"url": public_url})

def print_dashboard():
    """Отрисовка красивого терминального интерфейса"""
    os.system('cls' if os.name == 'nt' else 'clear')
    print("\033[96m" + "╔" + "═"*60 + "╗")
    print("║ \033[1m🌟 Comfy File Sorter is Running!\033[0m\033[96m                           ║")
    print("╠" + "═"*60 + "╣\033[0m")
    print(f"\033[96m║\033[0m \033[92m🏠 Локально:\033[0m {local_url:<45} \033[96m║\033[0m")
    
    if public_url:
        print(f"\033[96m║\033[0m \033[94m🌍 Удаленно:\033[0m {public_url:<45} \033[96m║\033[0m")
        print("\033[96m╚" + "═"*60 + "╝\033[0m")
        
        if HAS_QR:
            print("\n\033[95m📱 Отсканируйте QR-код для доступа с телефона:\033[0m\n")
            qr = qrcode.QRCode(version=1, box_size=1, border=2)
            qr.add_data(public_url)
            qr.make(fit=True)
            qr.print_ascii(invert=True)
    else:
        print("\033[96m╚" + "═"*60 + "╝\033[0m")
        if settings.CONFIG.get('enable_tunnel', True):
            print("\n\033[93m⏳ Установка защищенного туннеля...\033[0m")
        else:
            print("\n\033[90m[Туннель отключен в настройках]\033[0m")

    print("\n\033[90m" + "-"*62)
    print(" Журнал (отображаются только ошибки):")
    print("-" * 62 + "\033[0m\n")

def start_fallback_tunnel(port):
    """Возвращен старый туннель localhost.run, так как ngrok не работает в РФ"""
    global public_url
    while True:
        try:
            process = subprocess.Popen(
                [
                    "ssh", "-o", "StrictHostKeyChecking=no", "-o", "ServerAliveInterval=30", 
                    "-o", "ServerAliveCountMax=3", "-R", f"80:127.0.0.1:{port}", "nokey@localhost.run"
                ],
                stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1
            )
            for line in iter(process.stdout.readline, ''):
                if not line: break
                if "https://" in line and "lhr.life" in line:
                    match = re.search(r'(https://[a-zA-Z0-9-]+\.lhr\.life)', line)
                    if match:
                        new_url = match.group(1)
                        if public_url != new_url:
                            public_url = new_url
                            print_dashboard() # Обновляем UI после получения ссылки
            process.wait()
        except Exception:
            pass
        
        public_url = None
        print("\n\033[91m[Туннель] Разорвано. Переподключение...\033[0m")
        time.sleep(5)

if __name__ == "__main__":
    print_dashboard()
    
    if settings.CONFIG.get('enable_tunnel', True):
        if settings.CONFIG.get('use_ngrok'):
            try:
                from pyngrok import ngrok
                if settings.CONFIG.get('ngrok_token'):
                    ngrok.set_auth_token(settings.CONFIG['ngrok_token'])
                tunnel = ngrok.connect(settings.CONFIG['port'])
                public_url = tunnel.public_url
                print_dashboard()
            except Exception as e:
                print(f"\033[91m[Ошибка] Ngrok не удался ({e}). Запуск резервного туннеля...\033[0m")
                threading.Thread(target=start_fallback_tunnel, args=(settings.CONFIG['port'],), daemon=True).start()
        else:
            threading.Thread(target=start_fallback_tunnel, args=(settings.CONFIG['port'],), daemon=True).start()

    if os.environ.get("OPEN_BROWSER", "1") == "1": 
        threading.Timer(2.5, lambda: webbrowser.open(local_url)).start()
        
    cli = sys.modules['flask.cli']
    cli.show_server_banner = lambda *x: None
    app.run(host=settings.CONFIG['host'], port=settings.CONFIG['port'], threaded=True, debug=False)