#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import time
from datetime import datetime, timedelta

import pymysql
import qbittorrentapi
import bencodepy


# ==============================
# ⚙ CONFIGURATION
# ==============================

DB_CONFIG = {
    "host": "192.168.1.76",
    "port": 3307,
    "user": "root",
    "password": "Vlado9674+",
    "database": "torrents",
    "charset": "utf8mb4",
    "autocommit": True,
}

QBT_CONFIG = {
    "host": "192.168.1.76",
    "port": 8080,
    "username": "admin",
    "password": "adminadmin",
}

MAX_ACTIVE_DOWNLOADS = 10
LOOP_SLEEP_SECONDS = 60

# Torrent označíme jako "dead" pokud nebyl nikdy "seen_complete"
# více než X minut od přidání
DEAD_TORRENT_MINUTES = 5

DEFAULT_SAVE_PATH = None



# ==============================
# 🔧 CONNECT
# ==============================

db = pymysql.connect(**DB_CONFIG)
cursor = db.cursor(pymysql.cursors.DictCursor)

qb = qbittorrentapi.Client(
    host=QBT_CONFIG["host"],
    port=QBT_CONFIG["port"],
    username=QBT_CONFIG["username"],
    password=QBT_CONFIG["password"],
)

try:
    qb.auth_log_in()
    print("✅ Connected to qBittorrent.")
except Exception as e:
    print("❌ Could not connect:", e)
    raise SystemExit(1)



# ==============================
# 🧪 TORRENT VALIDATION
# ==============================

def is_valid_torrent(blob: bytes) -> bool:
    """
    Returns True only if BLOB is a valid .torrent file.
    """
    try:
        data = bencodepy.decode(blob)
        return isinstance(data, dict) and b"info" in data
    except Exception:
        return False



# ==============================
# 🔄 SYNC FROM QB → DB
# ==============================

def sync_qb_to_db():
    torrents = qb.torrents_info()

    for t in torrents:
        completion_dt = None
        if getattr(t, "completion_on", 0):
            try:
                completion_dt = datetime.fromtimestamp(t.completion_on)
            except:
                pass

        sql = """
            UPDATE torrents
            SET qb_added = 1,
                qb_hash = COALESCE(qb_hash, %s),
                qb_state = %s,
                qb_progress = %s,
                qb_savepath = %s,
                qb_completed_datetime =
                    IF(%s IS NOT NULL AND qb_completed_datetime IS NULL, %s, qb_completed_datetime),
                qb_last_update = NOW()
            WHERE qb_hash = %s OR torrent_hash = %s
        """

        cursor.execute(sql, (
            t.hash,
            t.state,
            float(t.progress) * 100.0,
            getattr(t, "save_path", None),
            completion_dt,
            completion_dt,
            t.hash,
            t.hash
        ))



# ==============================
# 🧹 HANDLE COMPLETED + DEAD TORRENTS
# ==============================

def handle_completed_and_dead():
    torrents = qb.torrents_info()

    for t in torrents:
        t_hash = t.hash
        state = t.state
        progress = float(t.progress)

        # ==========================
        # ✔ COMPLETED
        # ==========================
        if progress >= 1.0 or state in {"completed", "uploading", "stalledUP", "queuedUP"}:
            print(f"✅ Completed torrent → remove (keep data): {t.name}")
            try:
                qb.torrents_delete(torrent_hashes=t_hash, delete_files=False)
            except Exception as e:
                print("⚠️ delete failed:", e)

            cursor.execute("""
                UPDATE torrents
                SET qb_state='completed',
                    qb_progress=100,
                    qb_completed_datetime=NOW(),
                    qb_last_update=NOW()
                WHERE qb_hash=%s OR torrent_hash=%s
            """, (t_hash, t_hash))

            continue

        # ==========================
        # ❌ DEAD TORRENT (never seen_complete)
        # ==========================

        props = qb.torrents_properties(t_hash)
        seen = getattr(props, "last_seen", 0)

        if seen == -1:  # never seen complete
            added_dt = getattr(t, "added_on", 0)
            if added_dt:
                added_time = datetime.fromtimestamp(added_dt)
                if datetime.now() - added_time > timedelta(minutes=DEAD_TORRENT_MINUTES):
                    print(f"💀 Dead torrent (> {DEAD_TORRENT_MINUTES} min unseen): {t.name}")
                    try:
                        qb.torrents_delete(torrent_hashes=t_hash, delete_files=True)
                    except:
                        pass

                    cursor.execute("""
                        UPDATE torrents
                        SET qb_state='dead',
                            qb_last_update=NOW()
                        WHERE qb_hash=%s OR torrent_hash=%s
                    """, (t_hash, t_hash))



# ==============================
# 📊 COUNT ACTIVE DOWNLOADS
# ==============================

def count_active_downloads():
    torrents = qb.torrents_info(filter="all")
    return sum(1 for t in torrents if float(t.progress) < 1.0)



# ==============================
# ➕ ENQUEUE NEW TORRENTS
# ==============================

def enqueue_new_torrents():
    active = count_active_downloads()

    print("DEBUG active =", active)

    if active >= MAX_ACTIVE_DOWNLOADS:
        print(f"📦 {active}/{MAX_ACTIVE_DOWNLOADS} active → no enqueue")
        return

    slots = MAX_ACTIVE_DOWNLOADS - active

    sql = """
        SELECT id, torrent_hash, torrent_content, torrent_filename, added_datetime
        FROM torrents
        WHERE (qb_added IS NULL OR qb_added = 0)
          AND torrent_content IS NOT NULL
        ORDER BY added_datetime DESC    -- <── take NEWEST FIRST
        LIMIT %s
    """
    cursor.execute(sql, (slots,))
    rows = cursor.fetchall()

    if not rows:
        print("ℹ️ No new torrents")
        return

    for row in rows:
        t_id = row["id"]
        t_hash = row["torrent_hash"]
        blob = row["torrent_content"]
        filename = row.get("torrent_filename", "unknown.torrent")

        if not blob:
            print("⚠️ empty blob, skip")
            continue

        # ==========================
        # 🧪 VALIDATION OF .TORRENT
        # ==========================

        if not is_valid_torrent(blob):
            print(f"❌ INVALID TORRENT id={t_id}, size={len(blob)} → deleting content")

            cursor.execute("""
                UPDATE torrents
                SET qb_state='invalid',
                    torrent_content=NULL,
                    qb_last_update=NOW()
                WHERE id=%s
            """, (t_id,))
            continue

        # ==========================
        # ➕ ADD TORRENT
        # ==========================

        print(f"➕ Adding torrent: {filename} ({t_hash})")

        try:
            qb.torrents_add(torrent_files=blob, savepath=DEFAULT_SAVE_PATH)
        except Exception as e:
            print(f"❌ Failed to add {t_hash}: {e}")
            continue

        cursor.execute("""
            UPDATE torrents
            SET qb_added=1,
                qb_hash=COALESCE(qb_hash, %s),
                qb_state='added',
                qb_last_update=NOW()
            WHERE id=%s
        """, (t_hash, t_id))



# ==============================
# 🏁 MAIN LOOP
# ==============================

print("🚀 Worker started")

try:
    while True:
        print(f"\n⏱ Loop {datetime.now():%Y-%m-%d %H:%M:%S}")

        sync_qb_to_db()
        handle_completed_and_dead()
        enqueue_new_torrents()

        print(f"🛌 Sleep {LOOP_SLEEP_SECONDS}s\n")
        time.sleep(LOOP_SLEEP_SECONDS)

except KeyboardInterrupt:
    print("🛑 Stopping worker...")

finally:
    db.close()
    print("👋 Bye.")