from datetime import datetime

import aiohttp
from pyrogram import Client, filters
from pyrogram.types import Message
from utils.misc import modules_help, prefix

# Aladhan API credentials
ALADHAN_API_URL = 'https://api.aladhan.com/v1/timingsByCity'
DEFAULT_METHOD = 2  # Islamic Society of North America
DEFAULT_CITY = 'Lahore'
DEFAULT_COUNTRY = 'PK'


async def fetch_namaz_times(city_name: str, country_name: str) -> dict:
    params = {'city': city_name, 'country': country_name, 'method': DEFAULT_METHOD}
    async with aiohttp.ClientSession() as session:
        try:
            async with session.get(ALADHAN_API_URL, params=params) as response:
                response.raise_for_status()
                return await response.json()
        except Exception as e:
            return {'error': str(e)}


def format_time_12hr(time_str: str) -> str:
    try:
        # Split time into hours and minutes
        hours, minutes = map(int, time_str.split(':'))
        # Convert to 12-hour format
        period = 'AM' if hours < 12 else 'PM'
        if hours == 0:
            hours = 12
        elif hours > 12:
            hours -= 12
        elif hours == 12:
            period = 'PM'
        return f'{hours}:{minutes:02d} {period}'
    except Exception as e:
        return f'Error formatting time: {str(e)}'


@Client.on_message(filters.command('prayer', prefix) & filters.me)
async def namaz_times(_client: Client, message: Message):
    if message.reply_to_message:
        city_name = message.reply_to_message.text
        country_name = DEFAULT_COUNTRY  # Default to Pakistan if no country is provided
    else:
        args = message.text.split(maxsplit=2)
        if len(args) < 2:
            city_name = DEFAULT_CITY
            country_name = DEFAULT_COUNTRY
        elif len(args) == 2:
            city_name = args[1]
            country_name = DEFAULT_COUNTRY
        else:
            city_name = args[1]
            country_name = args[2]

    data = await fetch_namaz_times(city_name, country_name)

    if 'error' in data:
        result = f'<b>Error:</b> <i>{data["error"]}</i>'
    elif 'data' in data:
        timings = data['data']['timings']
        today = datetime.now().strftime('%Y-%m-%d')
        formatted_timings = {
            'Fajr': format_time_12hr(timings['Fajr']),
            'Dhuhr': format_time_12hr(timings['Dhuhr']),
            'Asr': format_time_12hr(timings['Asr']),
            'Maghrib': format_time_12hr(timings['Maghrib']),
            'Isha': format_time_12hr(timings['Isha']),
        }
        result = (
            f'<b>Prayer Times for {city_name}, {country_name} on {today}:</b>\n\n'
            f'<b>Fajr:</b> {formatted_timings["Fajr"]}\n'
            f'<b>Dhuhr:</b> {formatted_timings["Dhuhr"]}\n'
            f'<b>Asr:</b> {formatted_timings["Asr"]}\n'
            f'<b>Maghrib:</b> {formatted_timings["Maghrib"]}\n'
            f'<b>Isha:</b> {formatted_timings["Isha"]}\n'
        )
    else:
        result = '<b>Error:</b> <i>Unable to get prayer times for the specified location.</i>'

    await message.edit_text(result)


modules_help['namaz'] = {
    'prayer [city_name] [country_name]': 'Shows the prayer times. Default to Pakistan if no country is provided'
}