import os
import asyncio
import platform
import subprocess
import time
import random
from datetime import datetime
from secrets import choice
from telethon import Button, events
from telethon.errors.rpcerrorlist import MessageDeleteForbiddenError
from telethon.utils import get_display_name
from telethon.errors.rpcerrorlist import MessageDeleteForbiddenError
from telethon.utils import get_display_name
from telethon.tl.types import InputMessagesFilterVideo, InputMessagesFilterVoice, InputMessagesFilterPhotos

from xteam._misc import SUDO_M, owner_and_sudos
from xteam.dB.base import KeyManager
from xteam.fns.helper import inline_mention
from strings import get_string
from platform import python_version as pyver
from pyrogram import __version__ as pver
from telegram import __version__ as lver
from telethon import __version__ as tver
from pytgcalls import __version__ as pytver
from pyrogram import filters
from pyrogram.types import Message
from telethon import TelegramClient, events
from telethon.tl.custom import Button
from . import *
from . import ultroid_bot as client
import resources
from xteam.fns.helper import inline_mention
from . import (
    OWNER_NAME,
    OWNER_ID,
    BOT_NAME,
    OWNER_USERNAME,
    asst,
    start_time,
    time_formatter,
    udB,
    ultroid_cmd as xteam_cmd,
    get_string,
    ultroid_bot as client,
    eor,
    ultroid_bot,
    call_back,
    callback,
)


def format_message_text(uptime):
    return f"<blockquote expandable><b>✰ xᴛᴇᴀᴍ ᴜʀʙᴏᴛ ɪꜱ ᴀʟɪᴠᴇ ✰</b>\n\n" \
                     f"✵ Owner : {OWNER_NAME}\n" \
                     f"✵ Dc id : {ultroid_bot.dc_id}\n" \
                     f"✵ Library : {lver}\n" \
                     f"✵ Uptime : {uptime}\n" \
                     f"✵ Telethon : {tver}\n" \
                     f"✵ Pyrogram :  {pver}\n" \
                     f"✵ Python : {pyver()}\n</blockquote>"

@xteam_cmd(pattern="aline$")
async def alive(event):
    start = time.time()
    pro = await event.eor("🔥")
    await asyncio.sleep(1)
    end = round((time.time() - start) * 1000)
    uptime = time_formatter((time.time() - start_time) * 1000)
    message_text = format_message_text(uptime)
    await pro.edit(f"<blockquote expandable><b>✰ xᴛᴇᴀᴍ ᴜʀʙᴏᴛ ɪꜱ ᴀʟɪᴠᴇ ✰</b>\n\n" \
                     f"✵ Owner : {OWNER_NAME}\n" \
                     f"✵ Dc id : {ultroid_bot.dc_id}\n" \
                     f"✵ Library : {lver}\n" \
                     f"✵ Uptime : {uptime}\n" \
                     f"✵ Telethon : {tver}\n" \
                     f"✵ Pyrogram :  {pver}\n" \
                     f"✵ Python : {pyver()}\n</blockquote>",
                   parse_mode="html",
                  )

@xteam_cmd(pattern="Alive$")
async def alive_video(event):
    try:
        asupannya = [
            asupan
            async for asupan in event.client.iter_messages(
                "@xcryasupan", filter=InputMessagesFilterVideo
            )
        ]

        if not asupannya:
            await event.respond("No video found in @xcryasupan.")
            return

        pro = await event.eor("⚡")
        await asyncio.sleep(1)
        await pro.delete()
        uptime = time_formatter((time.time() - start_time) * 1000000)
        message_text = format_message_text(uptime)

        selected_video_message = random.choice(asupannya)

        # Print information about the selected video message for debugging
        print(f"Selected video message ID: {selected_video_message.id}")
        if selected_video_message.video:
            print(f"Selected video attributes: {selected_video_message.video}")
            # Try to send the video directly
            await client.send_file(
                event.chat.id,
                file=selected_video_message.video, # Pass the video object
                caption=message_text,
                parse_mode="html",
            )
        else:
            await event.respond("Selected message does not contain a video.")


    except Exception as e:
        await event.respond(f"An error occurred: {e}")


@xteam_cmd(pattern="live$")
async def live_video(event):
    try:
        asupannya = [
            asupan
            async for asupan in event.client.iter_messages(
                "@xcryasupan", filter=InputMessagesFilterVideo
            )
        ]

        if not asupannya:
            await event.respond("No video found in @xcryasupan.")
            return

        pro = await event.eor("⚡")
        await asyncio.sleep(1)
        await pro.delete()

        # Assuming start_time and time_formatter are defined
        uptime = time_formatter((time.time() - start_time) * 1000) # Assuming time_formatter expects milliseconds
        message_text = format_message_text(uptime)

        selected_video_message = random.choice(asupannya)

        # Create the inline keyboard with a "Close" button
        buttons = [
            [Button.inline("Close", data="close_alive_message")]
        ]

        if selected_video_message.video:
            await client.send_message(
                event.chat.id,
                caption=message_text,
                parse_mode="html",
                buttons=buttons # Add the buttons here
            )
        else:
            await event.respond("Selected message does not contain a video.")

    except Exception as e:
        await event.respond(f"An error occurred: {e}")


@in_pattern("Alive", owner=False)
async def alive_inline_handler(ult):
    start = time.time()

    end = round((time.time() - start) * 1000)
    uptime = time_formatter((time.time() - start_time) * 1000)
    
    message_text = (
        f"<blockquote><b>✰ xᴛᴇᴀᴍ ᴜʀʙᴏᴛ ɪꜱ ᴀʟɪᴠᴇ ✰</b>\n\n"
        f"✵ Owner : {OWNER_NAME}\n"
        f"✵ Dc id : {ultroid_bot.dc_id}\n"
        f"✵ Library : {lver}\n"
        f"✵ Uptime : {uptime}\n"
        f"✵ Telethon : {tver}\n"
        f"✵ Pyrogram :  {pver}\n"
        f"✵ Python : {pyver()}</blockquote>"
    )

    # Menggabungkan tombol dalam list of lists untuk 'buttons' parameter.
    buttons_markup = [
    [
        Button.inline("🏡 Modules 🏡",data="ownr"), # New data for modules
    ],
    ]
    

    result = await ult.builder.article(
        title="Bot Status",
        text=message_text,
        parse_mode="html",
        buttons=buttons_markup # Menyertakan tombol "Close"
    )
    await ult.answer([result], cache_time=0)


@client.on(events.CallbackQuery(b'delete_this_message'))
async def handle_delete_callback(event):
    """
    Menangani callback dari tombol 'Hapus Pesan Ini'.
    """
    print(f"Callback diterima dari chat ID: {event.chat_id}, pesan ID: {event.message_id}")

    # Dapatkan ID pesan dan ID chat dari callback query
    message_id_to_delete = event.message_id
    chat_id_of_message = event.chat_id

    try:
        # Hapus pesan yang terkait dengan callback query
        await client.delete_messages(chat_id_of_message, message_id_to_delete)
        print(f"Pesan dengan ID {message_id_to_delete} di chat {chat_id_of_message} telah dihapus.")
        # Opsional: Beri tahu pengguna bahwa pesan telah dihapus
        await event.answer("Pesan berhasil dihapus!", alert=False)
    except Exception as e:
        print(f"Gagal menghapus pesan: {e}")
        await event.answer("Gagal menghapus pesan.", alert=True)



@xteam_cmd(pattern="alive$")
async def alive_message_handler(ult):
    start = time.time()

    # The 'end' variable is still unused if you just want to display uptime.
    # end = round((time.time() - start) * 1000)
    
    uptime = time_formatter((time.time() - start_time) * 1000)
    
    message_text = (
        f"<blockquote><b>✰ xᴛᴇᴀᴍ ᴜʀʙᴏᴛ ɪꜱ ᴀʟɪᴠᴇ ✰</b>\n\n"
        f"✵ Owner : {OWNER_NAME}\n"
        f"✵ Dc id : {ultroid_bot.dc_id}\n"
        f"✵ Library : {lver}\n"
        f"✵ Uptime : {uptime}\n"
        f"✵ Telethon : {tver}\n"
        f"✵ Pyrogram :  {pver}\n"
        f"✵ Python : {pyver()}</blockquote>"
    )

    buttons_markup = [
        [
            #Button.url("🏡 Modules 🏡", url="https://example.com/modules"), # Use Button.url for external links
            # Or Button.inline if you have an inline query handler for 'ownr'
            Button.inline("🏡 Modules 🏡", data="ownr"), 
        ],
    ]
    
    await ult.reply(
        message_text,
        parse_mode="html",
        buttons=buttons_markup
    )


@xteam_cmd(pattern="alivee$")
async def alive(event):
    start = time.time()
    pro = await event.eor("♥️")
    await asyncio.sleep(2) # Simulate some delay
    end = round((time.time() - start) * 1000)
    uptime = time_formatter((time.time() - start_time) * 1000)

    # Reformat message using markdown
    message_text = (
        f"```\n" # Use a code block for the border to keep it preformatted
        f"┏──────────────────┓\n"
        f"❍─┫**ᴜʀʙᴏᴛ ɪꜱ ɴᴏᴡ ᴀʟɪᴠᴇ!**┣─❍\n"
        f"┣──────────────────┫\n"
        f"❍ ᴏᴡɴᴇʀ : `{OWNER_NAME}`\n"
        f"❍ ʟɪʙʀᴀʀʏ : `{lver}`\n"
        f"❍ ᴜᴘᴛɪᴍᴇ : `{uptime}`\n"
        f"❍ ᴛᴇʟᴇᴛʜᴏɴ : `{tver}`\n"
        f"❍ ᴘʏʀᴏɢʀᴀᴍ : `{pver}`\n"
        f"❍ ᴘʏᴛʜᴏɴ : `{pyver()}`\n"
        f"┗──────────────────┛\n"
        f"```"
    )
    # The parse_mode would likely default to "markdown" or "md" in Telethon if not specified
    await pro.edit(message_text, parse_mode="markdown") # Explicitly setting parse_mode="markdown"