import psutil
from fabric.widgets.box import Box
from fabric.widgets.label import Label
from fabric.widgets.overlay import Overlay
from fabric.widgets.datetime import DateTime
from fabric.widgets.circularprogressbar import CircularProgressBar
from widgets.wayland import WaylandWindow as Window
from fabric.utils import invoke_repeater
import requests
import urllib.parse
import datetime
from gi.repository import GLib
from concurrent.futures import ThreadPoolExecutor
import time
import config.data as data
from config.data import load_config
import subprocess
import calendar

executor = ThreadPoolExecutor(max_workers=4)


config = load_config()


def margin():
    return (
        data.DOCK_ICON_SIZE + 10
        if not data.DOCK_ALWAYS_OCCLUDED and data.DOCK_ENABLED
        else 0
    )


def get_location():
    """Fetch location from config file or IP API asynchronously."""
    for attempt in range(5):
        try:
            response = requests.get("https://ipinfo.io/json", timeout=3)
            if response.status_code == 200:
                return response.json().get("city", "").replace(" ", "")
        except requests.RequestException as e:
            print(f"Error getting location: {e}")
            if attempt < 4:
                time.sleep(10)
            else:
                return ""


def get_location_async(callback):
    """Fetch location asynchronously to prevent UI freeze."""
    executor.submit(lambda: GLib.idle_add(callback, get_location()))


def get_weather(callback):
    """Fetch weather data asynchronously and update UI."""

    def fetch_weather():
        location = get_location()
        if not location:
            return GLib.idle_add(callback, None)

        encoded_location = urllib.parse.quote(location)
        url = f"https://wttr.in/{encoded_location}?format=j1"
        urlemoji = f"https://wttr.in/{encoded_location}?format=%c"

        for attempt in range(5):
            try:
                response = requests.get(urlemoji, timeout=3)
                responseinfo = requests.get(url, timeout=3).json()

                if response.status_code == 200:
                    temp_unit = "C"
                    temp = (
                        responseinfo["current_condition"][0][f"temp_{temp_unit}"] + "°"
                    )
                    condition = responseinfo["current_condition"][0]["weatherDesc"][0][
                        "value"
                    ]
                    maxtemp = responseinfo["weather"][0]["maxtempC"] + "°"
                    mintemp = responseinfo["weather"][0]["mintempC"] + "°"
                    location = responseinfo["nearest_area"][0]["areaName"][0]["value"]
                    emoji = response.text.strip()

                    GLib.idle_add(
                        callback,
                        [
                            emoji,
                            temp,
                            condition,
                            location,
                            maxtemp,
                            mintemp,
                        ],
                    )
                    return
            except requests.RequestException as e:
                print(f"Error fetching weather (attempt {attempt + 1}): {e}")
                if attempt < 4:
                    time.sleep(10)
                else:
                    GLib.idle_add(callback, None)
                    return

    executor.submit(fetch_weather)


def update_weather(widget):
    def fetch_and_update():
        get_weather(lambda weather_info: update_widget(widget, weather_info))
        return True

    GLib.timeout_add_seconds(600, fetch_and_update)
    fetch_and_update()


def update_widget(widget, weather_info):
    if weather_info:
        widget.weatherinfo = weather_info
        widget.update_labels(weather_info)


class Sysinfo(Box):
    @staticmethod
    def bake_progress_bar(name: str = "progress-bar", size: int = 45, **kwargs):
        return CircularProgressBar(
            name=name,
            start_angle=180,
            end_angle=540,
            min_value=0,
            max_value=100,
            size=size,
            **kwargs,
        )

    @staticmethod
    def bake_progress_icon(**kwargs):
        return Label(**kwargs).build().add_style_class("progress-icon").unwrap()

    def __init__(self, **kwargs):
        super().__init__(
            layer="bottom",
            title="sysinfo",
            name="sysinfo",
            visible=False,
            all_visible=False,
            **kwargs,
        )

        self.cpu_progress = self.bake_progress_bar()
        self.ram_progress = self.bake_progress_bar()
        self.bat_circular = self.bake_progress_bar().build().set_value(42).unwrap()

        self.progress_container = Box(
            name="progress-bar-container",
            spacing=12,
            children=[
                Box(
                    children=[
                        Overlay(
                            child=self.cpu_progress,
                            tooltip_text="",
                            overlays=[
                                self.bake_progress_icon(
                                    label="",
                                    name="progress-icon-cpu",
                                )
                            ],
                        ),
                    ],
                ),
                Box(
                    children=[
                        Overlay(
                            child=self.ram_progress,
                            tooltip_text="",
                            overlays=[
                                self.bake_progress_icon(
                                    name="progress-icon-ram",
                                    label="󰘚",
                                )
                            ],
                        )
                    ]
                ),
                Box(
                    children=[
                        Overlay(
                            child=self.bat_circular,
                            tooltip_text="",
                            overlays=[
                                self.bake_progress_icon(
                                    label="󱊣",
                                    name="progress-icon-bat",
                                )
                            ],
                        ),
                    ],
                ),
            ],
        )

        self.update_status()
        invoke_repeater(1000, self.update_status)

        self.add(
            Box(
                name="progress-bar-container-main",
                orientation="v",
                spacing=24,
                children=[self.progress_container],
            ),
        )
        self.show_all()

    def update_status(self):
        """Update system info asynchronously to prevent UI lag."""

        def update():
            cpu = psutil.cpu_percent()
            ram = psutil.virtual_memory().percent
            battery = (
                psutil.sensors_battery().percent if psutil.sensors_battery() else 80
            )
            GLib.idle_add(self.cpu_progress.set_value, cpu)
            GLib.idle_add(self.ram_progress.set_value, ram)
            GLib.idle_add(self.bat_circular.set_value, battery)

            GLib.idle_add(self.cpu_progress.set_tooltip_text, f"{str(round(cpu))}%")
            GLib.idle_add(self.ram_progress.set_tooltip_text, f"{str(round(ram))}%")
            GLib.idle_add(self.bat_circular.set_tooltip_text, f"{str(round(battery))}%")

        executor.submit(update)
        return True


def fetch_quote(callback):
    """Fetch quotes asynchronously."""

    def fetch():
        quotes_type = "stoic"

        url = (
            "https://stoic-quotes.com/api/quote"
            if quotes_type == "stoic"
            else "https://zenquotes.io/api/random"
        )

        for attempt in range(5):
            try:
                response = requests.get(url, timeout=3)
                response.raise_for_status()
                respdata = response.json()
                quote = (
                    f"{respdata[0]['q']} - {respdata[0]['a']}"
                    if quotes_type == "zen"
                    else f"{respdata['text']} - {respdata['author']}"
                )
                break
            except requests.RequestException as e:
                print(f"Error fetching quote: {e}")
                if attempt < 4:
                    time.sleep(10)
                else:
                    try:
                        result = subprocess.run(
                            ["hyprctl", "splash"],
                            capture_output=True,
                            text=True,
                            check=True,
                        )
                        quote = result.stdout.strip() + " - Team Hyprland"
                    except subprocess.CalledProcessError as e:
                        print(f"Error fetching quote from hyprctl: {e}")
                        quote = "I learn from the mistakes of people who take my advice - Trix"

        GLib.idle_add(callback, quote)

    executor.submit(fetch)


def fetch_quote_async(callback):
    GLib.idle_add(lambda: fetch_quote(callback))


class QuoteWidget(Label):
    def __init__(self, **kwargs):
        super().__init__(
            name="quote",
            label="",
            anchor="center",
            h_align="center",
            v_align="center",
            h_expand=True,
            justification="center",
            v_expand=True,
            visible=False,
        )
        fetch_quote_async(self.update_label)

    def update_label(self, quote):
        """Update quote asynchronously."""
        max_width = 150  # Set the maximum width for the quote
        if len(quote) > max_width:
            words = quote.split()
            line1, line2 = "", ""
            for word in words:
                if len(line1) + len(word) + 1 <= max_width:
                    line1 += word + " "
                else:
                    line2 += word + " "
            quote = line1.strip() + "\n" + line2.strip()
        self.set_label(quote)
        self.set_visible(True)


class ActivationMainText(Label):
    def __init__(self, **kwargs):
        super().__init__(
            name="activation-main-text",
            label="",
            anchor="bottom right",
            justification="left",
            v_align="start",
            h_align="start",
            h_expand=True,
            v_expand=True,
            visible=False,
        )
        self.set_label("Activate Linux")


class ActivationSubText(Label):
    def __init__(self, **kwargs):
        super().__init__(
            name="activation-sub-text",
            label="",
            anchor="bottom right",
            justification="left",
            v_align="start",
            h_align="start",
            h_expand=True,
            v_expand=True,
            visible=False,
        )
        self.set_label("Go to Settings to activate Linux")


def create_widgets(config):
    widgets = []
    if config.get("widgets_displaytype_visible", True):
        if config.get("widgets_clock_visible", True):
            widgets.append(
                DateTime(formatters=["%A, %d %B"], interval=10000, name="date")
            )
        if config.get("widgets_date_visible", True):
            widgets.append(DateTime(formatters=["%I:%M %p"], name="clock"))
        if config.get("widgets_quote_visible", True):
            widgets.append(QuoteWidget())
        if config.get("widgets_weatherwid_visible", True):
            widgets.append(WeatherContainer())
    else:
        widgets.append(DateTime(formatters=["%I:%M %p"], name="clock"))
        widgets.append(DateTime(formatters=["%A. %d %B"], interval=10000, name="date"))
    return widgets


class Weather(Box):
    def __init__(self, parent, **kwargs):
        super().__init__(
            name="weather-widget",
            h_expand=True,
            v_expand=True,
            justification="right",
            orientation="v",
            all_visible=False,
            **kwargs,
        )

        self.parent = parent
        self.city = Label(
            name="city", label="New York", justification="right", h_align="start"
        )
        self.temperature = Label(name="temperature", label="36", h_align="start")
        self.condition_em = Label(name="condition-emoji", label="☁️", h_align="start")
        self.condition = Label(name="condition", label="Cloudy", h_align="start")
        self.feels_like = Label(name="feels-like", label="H:28 L:20", h_align="start")
        self.add(self.city)
        self.add(self.temperature)
        self.add(self.condition_em)
        self.add(self.condition)
        self.add(self.feels_like)
        update_weather(self)

    def update_labels(self, weather_info):
        if not self.weatherinfo:
            return

        # Unpack weather info into variables for better readability
        emoji, temp, condition, location, maxtemp, mintemp = self.weatherinfo
        maxmin = f"H:{maxtemp} L:{mintemp}"

        # Store references to deeply nested children to avoid repeated lookups

        self.city.set_label(location)
        self.feels_like.set_label(maxmin)
        self.condition.set_label(condition)
        self.temperature.set_label(temp)
        self.condition_em.set_label(emoji)
        self.parent.set_visible(True)


class WeatherContainer(Box):
    def __init__(self, **kwargs):
        super().__init__(
            orientation="v",
            name="box-widget-2",
            v_expand=True,
            v_align="center",
            size=(170, 170),
            visible=True,
            h_align="center",
            children=[
                Weather(self),
            ],
            **kwargs,
        )
        # self.set_style("background-color: rgba(0, 0, 0, 0.5);")
        # self.add(Weather())


class Date(Box):
    def __init__(self, **kwargs):
        super().__init__(
            name="date-widget",
            h_expand=True,
            v_expand=True,
            justification="center",
            h_align="center",
            v_align="start",
            orientation="v",
            **kwargs,
        )

        self.top = Box(orientation="h", name="date-top", h_expand=True)
        self.dateone = DateTime(formatters=["%a"], interval=10000, name="day")
        self.datetwo = DateTime(formatters=["%b"], interval=10000, name="month")
        self.datethree = DateTime(formatters=["%-d"], interval=10000, name="date")
        self.top.add(self.dateone)
        self.top.add(self.datetwo)
        self.add(self.top)
        self.add(self.datethree)


class DateContainer(Box):
    def __init__(self, **kwargs):
        super().__init__(
            orientation="v",
            name="box-widget",
            v_expand=True,
            size=(170, 170),
            v_align="center",
            h_align="center",
            children=[Date()],
            **kwargs,
        )


class Calendar(Box):
    def __init__(self, **kwargs):
        super().__init__(
            name="calendar-widget",
            h_expand=True,
            v_expand=True,
            orientation="v",
            **kwargs,
        )

        # Create current date for calendar
        now = datetime.datetime.now()
        self.current_month = now.month
        self.current_year = now.year
        self.current_day = now.day

        # Month and year header
        self.month_label = Label(
            name="calendar-month",
            label=f"{calendar.month_name[self.current_month]}",
            h_align="start",
            justification="left",
        )

        # Day abbreviations
        days_header = Box(
            name="calendar-days-header",
            orientation="h",
            h_expand=True,
            spacing=2
        )

        day_names = ["S", "M", "T", "W", "T", "F", "S"]
        for i, day_name in enumerate(day_names):
            # Saturday (6) and Sunday (0) get weekend styling
            is_weekend = i == 0 or i == 6
            day_label = Label(
                name=(
                    "calendar-day-header-weekend"
                    if is_weekend
                    else "calendar-day-header"
                ),
                label=day_name,
                h_align="center",
                h_expand=True,
            )
            days_header.add(day_label)

        # Calendar grid
        self.calendar_grid = Box(
            name="calendar-grid",
            orientation="v",
            spacing=1
        )

        self.update_calendar()

        self.add(self.month_label)
        self.add(days_header)
        self.add(self.calendar_grid)

        # Update calendar daily
        invoke_repeater(86400000, self.update_calendar_if_needed)  # 24 hours

    def update_calendar_if_needed(self):
        """Check if date changed and update calendar if needed."""
        now = datetime.datetime.now()
        if (
            now.month != self.current_month
            or now.year != self.current_year
            or now.day != self.current_day
        ):
            self.current_month = now.month
            self.current_year = now.year
            self.current_day = now.day
            self.update_calendar()
        return True

    def update_calendar(self):
        """Update the calendar grid with current month."""
        # Clear existing calendar
        for child in self.calendar_grid.get_children():
            self.calendar_grid.remove(child)

        # Update month label
        self.month_label.set_label(f"{calendar.month_name[self.current_month]}")

        # Get calendar for current month
        cal = calendar.monthcalendar(self.current_year, self.current_month)

        for week in cal:
            week_box = Box(orientation="h", spacing=2, h_expand=True)

            for day_index, day in enumerate(week):
                if day == 0:
                    # Empty day
                    day_label = Label(
                        name="calendar-day-empty",
                        label="",
                        h_align="center",
                        h_expand=True,
                    )
                else:
                    # Regular day
                    is_today = (
                        day == self.current_day
                        and self.current_month == datetime.datetime.now().month
                        and self.current_year == datetime.datetime.now().year
                    )

                    # Saturday (6) and Sunday (0) get weekend styling
                    is_weekend = day_index == 0 or day_index == 6

                    if is_today:
                        name = "calendar-day-today"
                    elif is_weekend:
                        name = "calendar-day-weekend"
                    else:
                        name = "calendar-day"

                    day_label = Label(
                        name=name, label=str(day), h_align="center", h_expand=True
                    )

                week_box.add(day_label)

            self.calendar_grid.add(week_box)


class CalendarContainer(Box):
    def __init__(self, **kwargs):
        super().__init__(
            orientation="v",
            name="calendar-box-widget",
            v_expand=True,
            size=(170, 170),
            v_align="center",
            h_align="center",
            children=[Calendar()],
            **kwargs,
        )


class RamInfo(Box):
    @staticmethod
    def bake_progress_bar(name: str = "progress-bar", size: int = 80, **kwargs):
        return CircularProgressBar(
            name=name,
            start_angle=270,
            end_angle=630,
            min_value=0,
            max_value=100,
            size=size,
            **kwargs,
        )

    @staticmethod
    def bake_progress_label(text, **kwargs):
        return Label(label=text, **kwargs)

    def __init__(self, **kwargs):
        super().__init__(
            layer="bottom",
            title="sysinfo",
            name="info-box-widget",
            visible=True,
            size=(170, 170),
            h_expand=True,
            v_expand=True,
            all_visible=True,
            **kwargs,
        )

        self.ram_progress = self.bake_progress_bar(name="progress")
        self.ram_label = Label(
            label="15%\nRam", justification="center", name="progress-label"
        )

        self.ram_info_label = Label(
            label="ram info", name="info", justification="center", h_align="center"
        )
        self.progress_container = Box(
            name="progress-bar-container",
            h_expand=True,
            v_expand=True,
            orientation="v",
            spacing=12,
            h_align="center",
            v_align="center",
            children=[
                Box(
                    children=[
                        Overlay(
                            child=self.ram_progress,
                            tooltip_text="",
                            overlays=self.ram_label,
                        ),
                    ],
                ),
                Box(
                    h_align="center",
                    justification="centre",
                    orientation="v",
                    children=[
                        self.ram_info_label,
                    ],
                ),
            ],
        )
        self.add(self.progress_container)
        invoke_repeater(1000, self.update)

    def update(self):
        mem = psutil.virtual_memory()
        self.ram_label.set_label(f" {round(mem.percent):<2} %\nRAM")

        used_gb = mem.used / (1024**3)
        free_gb = mem.available / (1024**3)

        self.ram_info_label.set_markup(
            f"<span foreground='#8E8E8E'>Used</span>      {round(used_gb, 1)}GB\n<span foreground='#8E8E8E'>Free</span>      {round(free_gb, 1)}GB"
        )
        GLib.idle_add(self.ram_progress.set_value, mem.percent)
        # executor.submit(update)
        return True


class CpuInfo(Box):
    @staticmethod
    def bake_progress_bar(name: str = "progress-bar", size: int = 80, **kwargs):
        return CircularProgressBar(
            name=name,
            start_angle=270,
            end_angle=630,
            min_value=0,
            max_value=100,
            size=size,
            **kwargs,
        )

    @staticmethod
    def bake_progress_label(text, **kwargs):
        return Label(label=text, **kwargs)

    def __init__(self, **kwargs):
        super().__init__(
            layer="bottom",
            title="sysinfo",
            name="info-box-widget",
            visible=True,
            size=(170, 170),
            h_expand=True,
            v_expand=True,
            all_visible=True,
            **kwargs,
        )

        self.cpu_progress = self.bake_progress_bar(name="progress")
        self.cpu_label = Label(
            label="15%\nCPU", justification="center", name="progress-label"
        )

        self.cpu_info_label = Label(
            label="CPU Info", name="info", justification="center", h_align="center"
        )

        self.progress_container = Box(
            name="progress-bar-container",
            h_expand=True,
            v_expand=True,
            orientation="v",
            spacing=12,
            h_align="center",
            v_align="center",
            children=[
                Box(
                    children=[
                        Overlay(
                            child=self.cpu_progress,
                            tooltip_text="",
                            overlays=self.cpu_label,
                        ),
                    ],
                ),
                Box(
                    h_align="center",
                    justification="centre",
                    orientation="v",
                    children=[
                        self.cpu_info_label,
                    ],
                ),
            ],
        )
        self.add(self.progress_container)
        invoke_repeater(1000, self.update)

    def get_cpu_temp(self):
        temps = psutil.sensors_temperatures()
        if not temps:
            return None

        for name, entries in temps.items():
            if (
                "coretemp" in name.lower()
                or "k10temp" in name.lower()
                or "cpu" in name.lower()
            ):
                for entry in entries:
                    # Match common CPU temp labels
                    if (
                        "package id 0" in (entry.label or "").lower()
                        or "core 0" in (entry.label or "").lower()
                        or (entry.label is None or entry.label.strip() == "")
                    ):
                        return round(entry.current, 1)

        return None

    def update(self):
        cpu = psutil.cpu_percent()
        self.cpu_label.set_label(f" {round(cpu):<2} %\nCPU")

        self.cpu_info_label.set_markup(
            f"<span foreground='#8E8E8E'>Temp</span>      {self.get_cpu_temp()}°C"
        )
        GLib.idle_add(self.cpu_progress.set_value, cpu)
        # executor.submit(update)
        return True


# FIX: GTK ERRORS


class Deskwidgets(Window):
    config = load_config()

    def __init__(self, **kwargs):
        top_left = Window(
            anchor="top left",
            orientation="h",
            layer="bottom",
            child=Box(
                name="desktop-widgets-container",
                children=[
                    DateContainer(),
                    WeatherContainer(),
                    CalendarContainer(),
                ],
            ),
        )

        bottom_left = Window(
            anchor="bottom right",
            orientation="h",
            layer="bottom",
            child=Box(
                name="desktop-widgets-container",
                children=[
                    CpuInfo(),
                    RamInfo(),
                ],
            ),
        )

        container = Box(
            orientation="v",
            children=[
                top_left,
                bottom_left,
            ],
        )

        super().__init__(
            name="desktop",
            layer="bottom",
            title="desktop-widgets",
            orientation="v",
            exclusivity="none",
            child=container,
            **kwargs,
        )


#     else:
#
#         class Deskwidgets(Window):
#             def __init__(self, **kwargs):
#                 config = load_config()
#                 super().__init__(name="desktop", **kwargs)
#                 desktop_widget = Window(
#                     layer="bottom",
#                     anchor="bottom left",
#                     v_align="start",
#                     h_align="start",
#                     h_expand=True,
#                     v_expand=True,
#                     justification="left",
#                     exclusivity="none",
#                     child=Box(
#                         orientation="v",
#                         children=create_widgets(config),
#                     ),
#                     all_visible=True,
#                 )
#                 if config.get("widgets_activation_visible", True):
#                     activationnag = Window(
#                         name="activation",
#                         anchor="bottom right",
#                         layer="top",
#                         justification="left",
#                         v_align="start",
#                         h_align="start",
#                         h_expand=True,
#                         v_expand=True,
#                         child=Box(
#                             orientation="v",
#                             children=[
#                                 ActivationMainText(),
#                                 ActivationSubText(),
#                             ],
#                         ),
#                         all_visible=True,
#                     )
#                 else:
#                     activationnag = None
#
# else:
#
#     class Deskwidgets(Window):
#         def __init__(self, **kwargs):
#             super().__init__(name="desktop", **kwargs)
#             self.set_visible(False)