"""User UI preferences — dashboard layout and visualization modes."""
from __future__ import annotations

import json
from typing import Any

from app.db import execute, fetch_one

DEFAULT_LAYOUT = ["kpis", "executive", "briefing", "retail", "analytics", "approvals", "feed"]
DEFAULT_VIZ = "neo-bars"

VIZ_MODES = [
    {"id": "neo-bars", "label": "Neo staafdiagram", "icon": "📊"},
    {"id": "neo-rings", "label": "Neo ringen", "icon": "🍩"},
    {"id": "neo-equalizer", "label": "Neo equalizer", "icon": "🎚️"},
    {"id": "neo-cards", "label": "Neo kaarten", "icon": "🃏"},
    {"id": "neo-table", "label": "Neo tabel", "icon": "📋"},
]


def get_preferences(user_key: str = "ceo") -> dict[str, Any]:
    row = fetch_one("SELECT * FROM user_ui_preferences WHERE user_key = %s", (user_key,))
    if not row:
        return {
            "user_key": user_key,
            "dashboard_layout": DEFAULT_LAYOUT,
            "viz_modes": {},
            "global_viz_mode": DEFAULT_VIZ,
            "locale": "nl",
            "viz_options": VIZ_MODES,
        }
    layout = row.get("dashboard_layout") or DEFAULT_LAYOUT
    if isinstance(layout, str):
        try:
            layout = json.loads(layout)
        except Exception:
            layout = DEFAULT_LAYOUT
    viz_modes = row.get("viz_modes") or {}
    if isinstance(viz_modes, str):
        try:
            viz_modes = json.loads(viz_modes)
        except Exception:
            viz_modes = {}
    return {
        "user_key": user_key,
        "dashboard_layout": layout,
        "viz_modes": viz_modes,
        "global_viz_mode": row.get("global_viz_mode") or DEFAULT_VIZ,
        "locale": row.get("locale") or "nl",
        "viz_options": VIZ_MODES,
    }


def save_preferences(
    user_key: str,
    dashboard_layout: list[str] | None = None,
    global_viz_mode: str | None = None,
    viz_modes: dict[str, str] | None = None,
    locale: str | None = None,
) -> dict[str, Any]:
    current = get_preferences(user_key)
    layout = dashboard_layout if dashboard_layout is not None else current["dashboard_layout"]
    gviz = global_viz_mode if global_viz_mode is not None else current["global_viz_mode"]
    vmodes = viz_modes if viz_modes is not None else current["viz_modes"]
    loc = locale if locale is not None else current.get("locale", "nl")
    if loc not in ("nl", "en"):
        loc = "nl"
    fetch_one(
        """
        INSERT INTO user_ui_preferences (user_key, dashboard_layout, global_viz_mode, viz_modes, locale, updated_at)
        VALUES (%s, %s::jsonb, %s, %s::jsonb, %s, NOW())
        ON CONFLICT (user_key) DO UPDATE SET
            dashboard_layout = EXCLUDED.dashboard_layout,
            global_viz_mode = EXCLUDED.global_viz_mode,
            viz_modes = EXCLUDED.viz_modes,
            locale = EXCLUDED.locale,
            updated_at = NOW()
        RETURNING user_key
        """,
        (user_key, json.dumps(layout), gviz, json.dumps(vmodes), loc),
    )
    return get_preferences(user_key)