from datetime import datetime, timedelta

from flask import flash, redirect, render_template, request, url_for


def computers_manage_icon_label(*, get_conn, clean_text, safe_next_url):
    computer_id_raw = clean_text(request.form.get("computer_id"))
    icon_label = clean_text(request.form.get("icon_label"))
    next_url = safe_next_url(request.form.get("next"), fallback="index")
    if not computer_id_raw.isdigit():
        return redirect(next_url)

    computer_id = int(computer_id_raw)
    conn = get_conn()
    cur = conn.cursor()
    cur.execute("SELECT 1 FROM computers WHERE id=%s", (computer_id,))
    exists = cur.fetchone() is not None
    if not exists:
        conn.close()
        return redirect(next_url)

    if icon_label:
        cur.execute(
            """
            INSERT INTO computer_manage_icons (computer_id, icon_label)
            VALUES (%s, %s)
            ON CONFLICT (computer_id)
            DO UPDATE SET icon_label = EXCLUDED.icon_label
            """,
            (computer_id, icon_label),
        )
    else:
        cur.execute("DELETE FROM computer_manage_icons WHERE computer_id=%s", (computer_id,))
    conn.commit()
    conn.close()
    return redirect(next_url)


def computers_update_cabinet(*, get_conn, log_equipment_movement):
    cid = request.form.get("id", "").strip()
    cabinet_id = request.form.get("cabinet_id", "").strip()
    if not cid.isdigit():
        flash("Некорректный компьютер")
        return redirect(request.referrer or url_for("index"))
    cab_id = int(cabinet_id) if cabinet_id.isdigit() else None
    conn = get_conn()
    cur = conn.cursor()
    cur.execute("SELECT inventory_number, cabinet_id FROM computers WHERE id=%s", (int(cid),))
    row = cur.fetchone()
    inv_num = row[0] if row else None
    old_cab_id = row[1] if row else None
    cur.execute("UPDATE computers SET cabinet_id=%s WHERE id=%s", (cab_id, int(cid)))
    conn.commit()
    conn.close()
    if inv_num and old_cab_id != cab_id:
        log_equipment_movement(inv_num, "computer", old_cab_id, cab_id)
    flash("Расположение обновлено")
    return redirect(request.referrer or url_for("index"))


def computers_list(
    *,
    get_conn,
    computer_types,
    computer_brands_source,
    cpu_brands,
    memory_types,
    os_options,
):
    conn = get_conn()
    cur = conn.cursor()
    cur.execute(
        """
        SELECT c.id, c.inventory_number, c.brand, c.model, c.serial_number, c.type,
               c.cpu_brand, c.cpu_model, c.gpu_model, c.memory_size, c.memory_type,
               c.storage_size, c.motherboard, c.os, c.rustdesk_id, c.rustdesk_password,
               c.hostname, c.ip_address, c.mac_address, c.is_online, c.last_seen,
               c.cabinet_id, cab.name AS cabinet_name,
               c.note, c.date_in_operation, c.date_added
        FROM computers c
        LEFT JOIN cabinets cab ON cab.id = c.cabinet_id
        ORDER BY
            cab.building NULLS LAST,
            CASE
                WHEN NULLIF(REGEXP_REPLACE(COALESCE(cab.cabinet, cab.name, ''), '\D', '', 'g'), '') IS NULL THEN 1
                ELSE 0
            END,
            COALESCE(NULLIF(REGEXP_REPLACE(COALESCE(cab.cabinet, cab.name, ''), '\D', '', 'g'), ''), '0')::INT,
            cab.cabinet NULLS LAST,
            cab.name NULLS LAST,
            c.inventory_number,
            c.id
        """
    )
    raw_rows = cur.fetchall()
    now = datetime.now()
    offline_after = timedelta(minutes=5)
    rows = []
    for row in raw_rows:
        last_seen = row[19]
        is_online = bool(row[18])
        if last_seen and now - last_seen > offline_after:
            is_online = False
        status_label = "Online" if is_online else "Offline"
        rows.append(tuple(row) + (status_label,))
    total_computers = len(rows)
    laptops_count = 0
    pcs_count = 0
    for row in rows:
        ctype = (row[5] or "").strip().lower()
        if ctype == "laptop":
            laptops_count += 1
        elif ctype == "pc":
            pcs_count += 1
    cur.execute("SELECT id, name FROM cabinets ORDER BY name")
    cabinets = cur.fetchall()
    cur.execute(computer_brands_source)
    computer_brands = [row[0] for row in cur.fetchall()]
    conn.close()
    return render_template(
        "computers.html",
        rows=rows,
        cabinets=cabinets,
        computer_types=computer_types,
        computer_brands=computer_brands,
        cpu_brands=cpu_brands,
        memory_types=memory_types,
        os_options=os_options,
        total_computers=total_computers,
        laptops_count=laptops_count,
        pcs_count=pcs_count,
    )


def computers_add(*, get_conn):
    inventory_number = request.form.get("inventory_number", "").strip()
    brand = request.form.get("brand", "").strip()
    model = request.form.get("model", "").strip()
    serial_number = request.form.get("serial_number", "").strip()
    ctype = request.form.get("type", "").strip()
    cpu_brand = request.form.get("cpu_brand", "").strip()
    cpu_model = request.form.get("cpu_model", "").strip()
    gpu_model = request.form.get("gpu_model", "").strip()
    memory_size = request.form.get("memory_size", "").strip()
    memory_type = request.form.get("memory_type", "").strip()
    storage_size = request.form.get("storage_size", "").strip()
    motherboard = request.form.get("motherboard", "").strip()
    os_name = request.form.get("os", "").strip()
    rustdesk_id = request.form.get("rustdesk_id", "").strip()
    rustdesk_password = request.form.get("rustdesk_password", "").strip()
    cabinet_id = request.form.get("cabinet_id", "").strip()
    note = request.form.get("note", "").strip()
    date_in_operation = request.form.get("date_in_operation", "").strip()

    if not inventory_number:
        flash("Укажите инвентарный номер")
        return redirect(url_for("computers"))
    if ctype not in ("pc", "laptop"):
        flash("Укажите тип (ПК/ноутбук)")
        return redirect(url_for("computers"))

    cab_id = int(cabinet_id) if cabinet_id.isdigit() else None
    conn = get_conn()
    cur = conn.cursor()
    try:
        cur.execute(
            """
            INSERT INTO computers (
                inventory_number, brand, model, serial_number, type,
                cpu_brand, cpu_model, gpu_model, memory_size, memory_type,
                storage_size, motherboard, os, rustdesk_id, rustdesk_password,
                cabinet_id, note, date_in_operation
            )
            VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)
            """,
            (
                inventory_number,
                brand or None,
                model or None,
                serial_number or None,
                ctype,
                cpu_brand or None,
                cpu_model or None,
                gpu_model or None,
                memory_size or None,
                memory_type or None,
                storage_size or None,
                motherboard or None,
                os_name or None,
                rustdesk_id or None,
                rustdesk_password or None,
                cab_id,
                note or None,
                date_in_operation or None,
            ),
        )
        conn.commit()
        flash("Компьютер добавлен")
    except Exception:
        conn.rollback()
        flash("Не удалось добавить компьютер")
    finally:
        conn.close()
    return redirect(url_for("computers"))


def computers_edit(*, get_conn, log_equipment_movement):
    cid = request.form.get("id", "").strip()
    inventory_number = request.form.get("inventory_number", "").strip()
    brand = request.form.get("brand", "").strip()
    model = request.form.get("model", "").strip()
    serial_number = request.form.get("serial_number", "").strip()
    ctype = request.form.get("type", "").strip()
    cpu_brand = request.form.get("cpu_brand", "").strip()
    cpu_model = request.form.get("cpu_model", "").strip()
    gpu_model = request.form.get("gpu_model", "").strip()
    memory_size = request.form.get("memory_size", "").strip()
    memory_type = request.form.get("memory_type", "").strip()
    storage_size = request.form.get("storage_size", "").strip()
    motherboard = request.form.get("motherboard", "").strip()
    os_name = request.form.get("os", "").strip()
    rustdesk_id = request.form.get("rustdesk_id", "").strip()
    rustdesk_password = request.form.get("rustdesk_password", "").strip()
    cabinet_id = request.form.get("cabinet_id", "").strip()
    note = request.form.get("note", "").strip()
    date_in_operation = request.form.get("date_in_operation", "").strip()

    if not cid.isdigit():
        flash("Некорректное устройство")
        return redirect(url_for("computers"))
    if not inventory_number:
        flash("Укажите инвентарный номер")
        return redirect(url_for("computers"))
    if ctype not in ("pc", "laptop"):
        flash("Укажите тип устройства (ПК/ноутбук)")
        return redirect(url_for("computers"))

    cab_id = int(cabinet_id) if cabinet_id.isdigit() else None
    conn = get_conn()
    cur = conn.cursor()
    try:
        cur.execute("SELECT cabinet_id FROM computers WHERE id=%s", (int(cid),))
        row = cur.fetchone()
        old_cab_id = row[0] if row else None
        cur.execute(
            """
            UPDATE computers
            SET inventory_number=%s, brand=%s, model=%s, serial_number=%s, type=%s,
                cpu_brand=%s, cpu_model=%s, gpu_model=%s, memory_size=%s, memory_type=%s,
                storage_size=%s, motherboard=%s, os=%s, rustdesk_id=%s, rustdesk_password=%s,
                cabinet_id=%s, note=%s, date_in_operation=%s
            WHERE id=%s
            """,
            (
                inventory_number,
                brand or None,
                model or None,
                serial_number or None,
                ctype,
                cpu_brand or None,
                cpu_model or None,
                gpu_model or None,
                memory_size or None,
                memory_type or None,
                storage_size or None,
                motherboard or None,
                os_name or None,
                rustdesk_id or None,
                rustdesk_password or None,
                cab_id,
                note or None,
                date_in_operation or None,
                int(cid),
            ),
        )
        conn.commit()
        if old_cab_id != cab_id:
            log_equipment_movement(inventory_number, "computer", old_cab_id, cab_id)
        flash("Компьютер обновлен")
    except Exception:
        conn.rollback()
        flash("Не удалось обновить компьютер")
    finally:
        conn.close()
    return redirect(url_for("computers"))


def computers_delete(*, get_conn):
    cid = request.form.get("id", "").strip()
    if not cid.isdigit():
        flash("Некорректное устройство")
        return redirect(url_for("computers"))
    conn = get_conn()
    cur = conn.cursor()
    cur.execute("DELETE FROM computers WHERE id=%s", (int(cid),))
    conn.commit()
    conn.close()
    flash("Компьютер удален")
    return redirect(url_for("computers"))