import os
import re
from telethon import TelegramClient, events
from telethon.tl.functions.stories import GetStoriesByIDRequest

id = 2610  # ايبي ايدي
ah = "ايبي هاش"
ABBAS = TelegramClient("ABBAS", id, ah)

td = "tmp"
os.makedirs(td, exist_ok=True)

def px(u):
    m = re.search(r't\.me/([^/?]+)/(?:s/|\?story=)(\d+)', u)
    if m:
        return m.group(1), int(m.group(2))
    return None, None

def ge(st):
    d = getattr(st.media, 'document', None)
    if d:
        for a in d.attributes:
            if hasattr(a, 'file_name'):
                ext = os.path.splitext(a.file_name)[1]
                if ext:
                    return ext
        if d.mime_type:
            return '.' + d.mime_type.split('/')[-1]
    return '.bin'

@ABBAS.on(events.NewMessage(pattern=r'^\.تحميل (.+)'))
async def dl(e):
    lnk = e.pattern_match.group(1)
    us, sid = px(lnk)

    if not us or not sid:
        await e.reply("الرابط مو صحيح")
        return

    try:
        en = await ABBAS.get_entity(us)
        rs = await ABBAS(GetStoriesByIDRequest(peer=en, id=[sid]))

        if not rs.stories:
            await e.reply("خطأ")
            return

        st = rs.stories[0]

        md = getattr(st.media, 'document', None) or getattr(st.media, 'photo', None)
        if not md:
            await e.reply("خطأ")
            return

        ex = ge(st)
        fp = os.path.join(td, f"st{sid}{ex}")

        path = await ABBAS.download_media(md, file=fp)
        if not path:
            await e.reply("خطأ")
            return

        await ABBAS.send_file(e.chat_id, path)
        os.remove(path)

    except Exception as er:
        await e.reply("صار خطأ بالتحميل")
        print("Err:", er)

ABBAS.start()
ABBAS.run_until_disconnected()