#!/usr/bin/env python3
"""PiCopy v2 - USB Copy Service mit WiFi-Fallback AP"""

import urllib.request as _urlreq
from pathlib import Path

from flask import Flask, render_template, send_file

from picopy.config import VERSION, BASE_DIR, RAW_BASE

app = Flask(__name__, template_folder='templates')

from routes import register_routes
register_routes(app)


@app.route('/')
def index():
    return render_template('index.html', version=VERSION)


@app.route('/logo.png')
def r_logo():
    logo = BASE_DIR / 'PiCopy_Logo.png'
    if not logo.exists():
        try:
            data = _urlreq.urlopen(f'{RAW_BASE}/PiCopy_Logo.png', timeout=15).read()
            logo.write_bytes(data)
        except Exception:
            return '', 404
    return send_file(logo, mimetype='image/png')


@app.route('/favicon.ico')
def r_favicon():
    logo = BASE_DIR / 'PiCopy_Logo.png'
    if logo.exists():
        return send_file(logo, mimetype='image/png')
    return '', 404


if __name__ == '__main__':
    import threading

    from picopy.usb import cleanup_stale_mounts
    from picopy.state import load_state
    from picopy.wireguard import wg_update_state, wg_connect, WG_CONF
    from picopy.copy_engine import usb_monitor
    from picopy.wifi import wifi_monitor
    from picopy.wireguard import wg_monitor
    from picopy.system import update_check_loop
    from picopy.config import log, load_cfg

    cleanup_stale_mounts()
    load_state()
    wg_update_state()
    threading.Thread(target=usb_monitor, daemon=True).start()
    threading.Thread(target=wifi_monitor, daemon=True).start()
    threading.Thread(target=wg_monitor, daemon=True).start()
    threading.Thread(target=update_check_loop, daemon=True).start()
    if load_cfg().get('wireguard_auto') and WG_CONF.exists():
        threading.Thread(target=wg_connect, daemon=True).start()
    log.info(f'PiCopy v{VERSION} läuft auf http://0.0.0.0:8080')
    app.run(host='0.0.0.0', port=8080, debug=False, use_reloader=False)