import os

from pyrogram import Client, enums, filters
from pyrogram.types import Message
from utils.misc import modules_help, prefix
from utils.scripts import import_library

lottie = import_library('lottie')
from lottie.exporters import exporters  # noqa: E402
from lottie.importers import importers  # noqa: E402


@Client.on_message(filters.command('destroy', prefix) & filters.me)
async def destroy_sticker(client: Client, message: Message):
    """Destroy animated stickers by modifying their animation properties"""
    try:
        reply = message.reply_to_message
        if not reply or not reply.sticker or not reply.sticker.is_animated:
            return await message.edit(
                '**Please reply to an animated sticker!**',
                parse_mode=enums.ParseMode.MARKDOWN,
            )

        edit_msg = await message.edit('**🔄 Destroying sticker...**', parse_mode=enums.ParseMode.MARKDOWN)

        # Download sticker
        tgs_path = await reply.download()
        if not tgs_path or not os.path.exists(tgs_path):
            return await edit_msg.edit('**❌ Download failed!**', parse_mode=enums.ParseMode.MARKDOWN)

        # Conversion process
        json_path = 'temp.json'
        output_path = 'MoonUB.tgs'

        importer = importers.get_from_filename(tgs_path)
        if not importer:
            return await edit_msg.edit('**❌ JSON conversion failed!**', parse_mode=enums.ParseMode.MARKDOWN)

        animation = importer.process(tgs_path)
        exporter = exporters.get_from_filename(json_path)
        exporter.process(animation, json_path)

        # Modify JSON data
        with open(json_path, 'r+') as f:
            content = f.read()
            modified = (
                content.replace('[1]', '[2]')
                .replace('[2]', '[3]')
                .replace('[3]', '[4]')
                .replace('[4]', '[5]')
                .replace('[5]', '[6]')
            )
            f.seek(0)
            f.write(modified)
            f.truncate()

        importer = importers.get_from_filename(json_path)
        animation = importer.process(json_path)
        exporter = exporters.get_from_filename(output_path)
        exporter.process(animation, output_path)

        # Send result
        await message.reply_document(output_path, reply_to_message_id=reply.id)
        await edit_msg.delete()

    except Exception as e:
        await message.edit(f'**❌ Error:** `{e}`', parse_mode=enums.ParseMode.MARKDOWN)
    finally:
        # Cleanup temporary files
        for file_path in [tgs_path, json_path, output_path]:
            if file_path and os.path.exists(file_path):
                try:
                    os.remove(file_path)
                except Exception as clean_error:
                    print(f'Cleanup error: {clean_error}')


modules_help['destroy'] = {'destroy [reply]': 'Modify and destroy animated stickers'}