import logging
from aiogram import Router ,F ,types ,Bot
from aiogram .filters import Command
from typing import Optional ,Dict

from config .settings import Settings
from bot .services .referral_service import ReferralService
from db .database import get_db_connection_manager
from bot .keyboards .inline .user_keyboards import get_referral_link_keyboard ,get_back_to_main_menu_markup
from bot .middlewares .i18n import JsonI18n

router =Router (name ="user_referral_router")

async def referral_command_handler (
event :types .Message |types .CallbackQuery ,
settings :Settings ,
i18n_data :dict ,
referral_service :ReferralService ,
bot :Bot
):
    current_lang =i18n_data .get ("current_language",getattr (settings ,'DEFAULT_LANGUAGE','en'))
    i18n :Optional [JsonI18n ]=i18n_data .get ("i18n_instance")

    target_message =event .message if isinstance (event ,types .CallbackQuery )else event
    if not target_message :
        logging .error ("Target message is None in referral_command_handler from callback.")
        if isinstance (event ,types .CallbackQuery ):await event .answer ("Error displaying referral info.")
        return

    if not i18n or not referral_service :
        logging .error ("Deps missing in referral_command_handler")
        await target_message .answer ("Service error."if isinstance (event ,types .Message )else "Service error.",parse_mode =None )
        if isinstance (event ,types .CallbackQuery ):await event .answer ()
        return

    _ =lambda key ,**kwargs :i18n .gettext (current_lang ,key ,**kwargs )

    bot_info =await bot .get_me ()
    bot_username =bot_info .username
    inviter_user_id =event .from_user .id
    referral_link =referral_service .generate_referral_link (bot_username ,inviter_user_id )
    bonus_info_parts =[]

    if hasattr (settings ,'subscription_options')and settings .subscription_options :
        for months_period in sorted (settings .subscription_options .keys ()):
            inv_bonus =settings .referral_bonus_inviter .get (months_period )
            ref_bonus =settings .referral_bonus_referee .get (months_period )
            if inv_bonus is not None or ref_bonus is not None :
                bonus_info_parts .append (
                _ ("referral_bonus_per_period",
                months =months_period ,
                inviter_bonus_days =inv_bonus if inv_bonus is not None else _ ("no_bonus_days"),
                referee_bonus_days =ref_bonus if ref_bonus is not None else _ ("no_bonus_days")
                )
                )
    bonus_details_str ="\n".join (bonus_info_parts )if bonus_info_parts else _ ("referral_no_bonuses_configured")
    text =_ ("referral_program_info_new",referral_link =referral_link ,bonus_details =bonus_details_str )



    reply_markup_val =get_back_to_main_menu_markup (current_lang ,i18n )

    if isinstance (event ,types .Message ):
        await event .answer (text ,reply_markup =reply_markup_val ,disable_web_page_preview =True )
    elif isinstance (event ,types .CallbackQuery ):
        try :
            await event .message .edit_text (text ,reply_markup =reply_markup_val ,disable_web_page_preview =True )
        except Exception as e :
            logging .warning (f"Failed to edit message for referral info: {e}")

            await event .message .answer (text ,reply_markup =reply_markup_val ,disable_web_page_preview =True )
        await event .answer ()


@router .callback_query (F .data =="copy_referral_link_ack")
async def copy_referral_link_ack_callback_handler (callback :types .CallbackQuery ,i18n_data :dict ,settings :Settings ):
    current_lang =i18n_data .get ("current_language",getattr (settings ,'DEFAULT_LANGUAGE','en'))
    i18n :Optional [JsonI18n ]=i18n_data .get ("i18n_instance")
    if not i18n :return await callback .answer ("Language service error.",show_alert =True )
    _ =lambda key ,**kwargs :i18n .gettext (current_lang ,key ,**kwargs )
    await callback .answer (text =_ (key ="referral_link_for_copying_reminder"),show_alert =False )