"""
title: Home Assistant
author: Andrew Tait Gehrhardt
author_url: https://github.com/atgehrhardt
funding_url: https://github.com/open-webui
version: 0.3.1
"""

import requests
from typing import Literal, Dict, Any, List
from datetime import datetime
import pytz
from difflib import get_close_matches
from pydantic import BaseModel, Field


class Tools:
    class Valves(BaseModel):
        priority: int = Field(
            default=0, description="Priority level for the filter operations."
        )
        BASE_URL: str = Field(
            default="",
        )
        AUTH_TOKEN: str = Field(
            default="",
        )

    def __init__(self):
        self.valves = self.Valves()

    def get_all_lights(self) -> Dict[str, str]:
        """
        Retrieve a dictionary of lights in the house.
        :return: A dictionary of light entity names and their IDs.
        """
        if not self.valves.BASE_URL or not self.valves.AUTH_TOKEN:
            return {
                "error": "Home Assistant URL or token not set, ask the user to set it up."
            }
        else:
            url = f"{self.valves.BASE_URL}/api/states"
            headers = {
                "Authorization": f"Bearer {self.valves.AUTH_TOKEN}",
                "Content-Type": "application/json",
            }
            response = requests.get(url, headers=headers)
            response.raise_for_status()  # Raises an HTTPError for bad responses
            data = response.json()
            lights = {
                entity["attributes"]["friendly_name"]: entity["entity_id"]
                for entity in data
                if entity["entity_id"].startswith("light.")
            }
            return lights

    def list_lights(self) -> str:
        """
        List all lights in a human-readable format.
        :return: A string listing all the lights.
        """
        lights = self.get_all_lights()
        if isinstance(lights, dict) and "error" in lights:
            return lights["error"]
        if not lights:
            return "No lights found in your home."
        light_list = "\n".join(f"- {name}" for name in lights.keys())
        return f"Here are the lights currently in your home:\n{light_list}"

    def control_lights(self, names: List[str], state: Literal["on", "off"]) -> str:
        """
        Turn multiple lights on or off based on their names.
        :param names: A list of light names.
        :param state: The desired state ('on' or 'off').
        :return: The result of the operation.
        """
        if not self.valves.BASE_URL or not self.valves.AUTH_TOKEN:
            return "Home Assistant URL or token not set, ask the user to set it up."

        lights = self.get_all_lights()
        if isinstance(lights, dict) and "error" in lights:
            return lights["error"]

        light_ids = []
        not_found = []

        for name in names:
            # Normalize the light name
            normalized_name = " ".join(name.lower().split())

            # Find the closest matching light name
            closest_matches = get_close_matches(
                normalized_name, list(lights.keys()), n=1, cutoff=0.6
            )
            if closest_matches:
                best_match = closest_matches[0]
                light_ids.append(lights[best_match])
            else:
                not_found.append(name)

        if not light_ids:
            return f"None of the specified lights were found: {', '.join(names)}"

        url = f"{self.valves.BASE_URL}/api/services/light/turn_{state}"
        headers = {
            "Authorization": f"Bearer {self.valves.AUTH_TOKEN}",
            "Content-Type": "application/json",
        }
        payload = {"entity_id": light_ids}
        response = requests.post(url, headers=headers, json=payload)

        result = []
        if response.status_code == 200:
            result.append(
                f"Successfully turned {state} the following lights: {', '.join(names)}"
            )
        else:
            result.append(
                f"Failed to turn {state} the lights. Status code: {response.status_code}"
            )

        if not_found:
            result.append(
                f"The following lights were not found: {', '.join(not_found)}"
            )

        return " ".join(result)