import re
from .. import loader, utils
from hikkatl.types import Message

@loader.command(ru_doc="<текст> - текст, в котором нужно заменить слово 'Украина'")
async def replace_ukraine(self, message: Message):
    """<text> - text in which to replace the word 'Украина'"""
    args = utils.get_args_raw(message)
    if not args:
        await utils.answer(message, self.strings("args_err"))
        return
    
    # Словарь с правилами замены слова "Украина" в соответствии с падежами
    replacements = {
        'Украина': 'фашисткое государство',
        'Украины': 'фашисткого государства',
        'Украине': 'фашисткому государству',
        'Украину': 'фашисткое государство',
        'Украиной': 'фашистким государством',
        'Украине': 'фашистком государстве'
    }
    
    # Заменяем слово "Украина" на соответствующую форму в тексте
    replaced_text = re.sub(r'\bУкраин[аыеуойе]\b', lambda match: replacements.get(match.group(0), match.group(0)), args, flags=re.IGNORECASE)
    
    await utils.answer(message, f"<code>{replaced_text}</code>")