from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from datetime import datetime
from zoneinfo import ZoneInfo
from .const import DOMAIN
from .coordinator import get_timezone

# Fonction pour convertir la date dans le bon format et ajouter un fuseau horaire
def convert_to_iso_format(date_str, location):
    # Créer un dictionnaire pour les mois en français
    mois = {
        "janvier": "01", "février": "02", "mars": "03", "avril": "04", "mai": "05", "juin": "06",
        "juillet": "07", "août": "08", "septembre": "09", "octobre": "10", "novembre": "11", "décembre": "12"
    }

    # Extraire les parties de la date
    day, month_str, year, _, time_str, _ = date_str.split(" ")

    # Convertir le mois en chiffre
    month = mois[month_str.lower()]

    # Créer la chaîne avec le format ISO attendu
    iso_format_date = f"{year}-{month}-{day}T{time_str}"

    # Retourner la date au format ISO
    return iso_format_date

class VacancesScolairesCalendar(CoordinatorEntity, CalendarEntity):
    """Vacances Scolaires Calendar class."""

    def __init__(self, coordinator, config_entry):
        super().__init__(coordinator)
        self._attr_name = f"Vacances Scolaires {config_entry.title}"
        self._attr_unique_id = f"{config_entry.entry_id}_calendar"
        self.entry_id = config_entry.entry_id

    @property
    def event(self):
        """Return the next upcoming event."""
        if self.coordinator.data["on_vacation"]:
            start_date_str = self.coordinator.data["start_date"]
            end_date_str = self.coordinator.data["end_date"]
            location = self.coordinator.data["location"]  # Localisation pour déterminer le fuseau horaire

            # Récupérer le fuseau horaire correspondant à la localisation
            timezone = get_timezone(location)  # Utiliser le fuseau horaire dynamique

            # Convertir les dates en datetime et appliquer le fuseau horaire
            start_date = datetime.fromisoformat(convert_to_iso_format(start_date_str, location)).astimezone(ZoneInfo(timezone))
            end_date = datetime.fromisoformat(convert_to_iso_format(end_date_str, location)).astimezone(ZoneInfo(timezone))

            return CalendarEvent(
                start=start_date,
                end=end_date,
                summary=self.coordinator.data["description"],
            )
        return None

    async def async_get_events(self, hass, start_date, end_date):
        """Get all events in a specific time frame."""
        events = []
        if self.coordinator.data["on_vacation"]:
            start_date_str = self.coordinator.data["start_date"]
            end_date_str = self.coordinator.data["end_date"]
            location = self.coordinator.data["location"]  # Localisation pour déterminer le fuseau horaire

            # Récupérer le fuseau horaire correspondant à la localisation
            timezone = get_timezone(location)  # Utiliser le fuseau horaire dynamique

            # Convertir les dates en datetime et appliquer le fuseau horaire
            event_start = datetime.fromisoformat(convert_to_iso_format(start_date_str, location)).astimezone(ZoneInfo(timezone))
            event_end = datetime.fromisoformat(convert_to_iso_format(end_date_str, location)).astimezone(ZoneInfo(timezone))

            if start_date <= event_end and end_date >= event_start:
                event = CalendarEvent(
                    start=event_start,
                    end=event_end,
                    summary=self.coordinator.data["description"],
                )
                events.append(event)
        return events
        
    @property
    def device_info(self):
        return {
            "identifiers": {(DOMAIN, self.entry_id)},
            "name": "Vacances Scolaires",
            "manufacturer": "Master13011",
            "model": "API",
        }

async def async_setup_entry(hass, config_entry, async_add_entities):
    """Set up the Vacances Scolaires Calendar platform."""
    coordinator = hass.data[DOMAIN][config_entry.entry_id]
    async_add_entities([VacancesScolairesCalendar(coordinator, config_entry)], True)