from fabric.core import Service, Property, Signal
from pydbus import SystemBus
from loguru import logger

DeviceState = {
    0: "UNKNOWN",
    1: "CHARGING",
    2: "DISCHARGING",
    3: "EMPTY",
    4: "FULLY_CHARGED",
    5: "PENDING_CHARGE",
    6: "PENDING_DISCHARGE",
}


class Battery(Service):
    @staticmethod
    def seconds_to_hours_minutes(seconds):
        hours = seconds // 3600
        minutes = (seconds % 3600) // 60
        return f"{hours}h {minutes}m" if hours else f"{minutes}m"

    @Signal
    def changed(self) -> None: ...

    @Signal
    def profile_changed(self, value: str) -> None: ...

    @Property(int, "readable")
    def percentage(self):
        return int(self._battery.Percentage)

    @Property(str, "readable")
    def temperature(self):
        return f"{self._battery.Temperature}°C" if hasattr(self._battery, "Temperature") else "N/A"

    @Property(str, "readable")
    def time_to_empty(self):
        return self.seconds_to_hours_minutes(getattr(self._battery, "TimeToEmpty", 0))

    @Property(str, "readable")
    def time_to_full(self):
        return self.seconds_to_hours_minutes(getattr(self._battery, "TimeToFull", 0))

    @Property(str, "readable")
    def icon_name(self):
        return self._battery.IconName

    @Property(str, "readable")
    def state(self):
        return DeviceState.get(self._battery.State, "UNKNOWN")

    @Property(str, "readable")
    def capacity(self):
        return f"{self._battery.Capacity}%"

    @Property(bool, "readable", default_value=False)
    def is_present(self):
        return self._battery.IsPresent

    @Property(str, "readable")
    def power_profile(self):
        return self._profile_proxy.ActiveProfile

    def set_power_profile(self, profile: str) -> bool:
        if profile not in [p.Profile for p in self._profile_proxy.Profiles]:
            logger.warning(f"[Battery] Invalid profile: {profile}")
            return False
        try:
            self._profile_proxy.ActiveProfile = profile
            logger.info(f"[Battery] Power profile set to {profile}")
            self.profile_changed.emit(profile)
            self.changed.emit()
            return True
        except Exception as e:
            logger.error(f"[Battery] Failed to set profile: {e}")
            return False

    def __init__(self):
        super().__init__()
        self._bus = SystemBus()

        # Battery device
        try:
            self._battery = self._bus.get("org.freedesktop.UPower", "/org/freedesktop/UPower/devices/battery_BAT0")
            self._battery.onPropertiesChanged = self.handle_battery_change
            logger.info("[Battery] UPower device connected.")
        except Exception as e:
            logger.error(f"[Battery] Failed to connect to UPower battery: {e}")
            return

        # PowerProfiles
        try:
            self._profile_proxy = self._bus.get("net.hadess.PowerProfiles", "/net/hadess/PowerProfiles")
            self._profile_proxy.onPropertiesChanged = self.handle_profile_change
            logger.info("[Battery] PowerProfiles daemon connected.")
        except Exception as e:
            logger.error(f"[Battery] Failed to connect to PowerProfiles daemon: {e}")
            self._profile_proxy = None

        self.changed.emit()

    def handle_battery_change(self, iface, changed, invalidated):
        logger.debug(f"[Battery] UPower properties changed: {changed}")
        self.changed.emit()

    def handle_profile_change(self, iface, changed, invalidated):
        if "ActiveProfile" in changed:
            new_profile = changed["ActiveProfile"]
            logger.info(f"[Battery] Power profile changed: {new_profile}")
            self.profile_changed.emit(new_profile)
            self.changed.emit()