import os
import re
import asyncio
from driver.veez import call_py
from pytgcalls import StreamType
from driver.filters import command, other_filters
from pyrogram import Client, filters
from youtubesearchpython import VideosSearch
from config import BOT_USERNAME, IMG_1, IMG_2
from driver.queues import QUEUE, add_to_queue
from pytgcalls.types.input_stream import AudioPiped
from pyrogram.types import Message, InlineKeyboardButton, InlineKeyboardMarkup


def ytsearch(query):
   try:
      search = VideosSearch(query, limit=1)
      for r in search.result()["result"]:
         ytid = r['id']
         if len(r['title']) > 34:
            songname = r['title'][:45] + "..."
         else:
            songname = r['title']
         url = f"https://www.youtube.com/watch?v={ytid}"
      return [songname, url]
   except Exception as e:
      print(e)
      return 0


async def ytdl(link):
   proc = await asyncio.create_subprocess_exec(
       'yt-dlp',
       '-g',
       '-f',
       'bestaudio',
       f'{link}',
       stdout=asyncio.subprocess.PIPE,
       stderr=asyncio.subprocess.PIPE,
   )
   stdout, stderr = await proc.communicate()
   if stdout:
      return 1, stdout.decode().split('\n')[0]
   else:
      return 0, stderr.decode()


@Client.on_message(command(["play", f"play@{BOT_USERNAME}"]) & other_filters)
async def play(client, m: Message):
   
   keyboard = InlineKeyboardMarkup(
            [
                [
                    InlineKeyboardButton(
                        text="✨ ɢʀᴏᴜᴘ",
                        url=f"https://t.me/{GROUP_SUPPORT}"),
                    InlineKeyboardButton(
                        text="🌻 ᴄʜᴀɴɴᴇʟ",
                        url=f"https://t.me/{UPDATES_CHANNEL}")
                ]
            ]
        )
   
   replied = m.reply_to_message
   chat_id = m.chat.id
   if replied:
      if replied.audio or replied.voice:
         suhu = await replied.reply("📥 **downloading audio...**")
         dl = await replied.download()
         link = replied.link
         if replied.audio:
            if replied.audio.title:
               songname = replied.audio.title[:45] + "..."
            else:
               songname = replied.audio.file_name[:45] + "..."
         elif replied.voice:
            songname = "Voice Note"
         if chat_id in QUEUE:
            pos = add_to_queue(chat_id, songname, dl, link, "Audio", 0)
            await suhu.delete()
            await m.reply_photo(
               photo=f"{IMG_1}",
               caption=f"💡 **Track added to the queue**\n\n💭 **Chat:** `{chat_id}`\n🎧 **Request by:** {m.from_user.mention()}\n🔢 **At position »** `{pos}`",
               reply_markup=keyboard,
            )
         else:
            await call_py.join_group_call(
               chat_id,
               AudioPiped(
                  dl,
               ),
               stream_type=StreamType().pulse_stream,
            )
            add_to_queue(chat_id, songname, dl, link, "Audio", 0)
            await m.reply_photo(
               photo=f"{IMG_2}",
               caption=f"💡 **music streaming started.**\n\n🏷 **Name:** [{songname}]({link})\n💭 **Chat:** `{chat_id}`\n💡 **Status:** `Playing`\n🎧 **Request by:** {m.from_user.mention()}",
               disable_web_page_preview=True,
               reply_markup=keyboard,
            )
      else:
         if len(m.command) < 2:
            await m.reply("» reply to an **audio file** or **give something to search.**")
         else:
            suhu = await m.reply("🔎 **searching...**")
            query = m.text.split(None, 1)[1]
            search = ytsearch(query)
            if search==0:
               await suhu.edit("❌ **no results found.**")
            else:
               songname = search[0]
               url = search[1]
               veez, ytlink = await ytdl(url)
               if veez==0:
                  await suhu.edit(f"❌ youtube-dl issues detected\n\n» `{ytlink}`")
               else:
                  if chat_id in QUEUE:
                     pos = add_to_queue(chat_id, songname, ytlink, url, "Audio", 0)
                     await suhu.delete()
                     await m.reply_photo(
                        photo=f"{IMG_1}",
                        caption=f"💡 **Track added to the queue**\n\n💭 **Chat:** `{chat_id}`\n🎧 **Request by:** {m.from_user.mention()}\n🔢 **At position »** `{pos}`",
                        reply_markup=keyboard,
                     )
                  else:
                     try:
                        await call_py.join_group_call(
                           chat_id,
                           AudioPiped(
                              ytlink,
                           ),
                           stream_type=StreamType().pulse_stream,
                        )
                        add_to_queue(chat_id, songname, ytlink, url, "Audio", 0)
                        await suhu.delete()
                        await m.reply_photo(
                           photo=f"{IMG_2}",
                           caption=f"💡 **music streaming started.**\n\n🏷 **Name:** [{songname}]({url})\n💭 **Chat:** `{chat_id}`\n💡 **Status:** `Playing`\n🎧 **Request by:** {m.from_user.mention()}",
                           disable_web_page_preview=True,
                           reply_markup=keyboard,
                        )
                     except Exception as ep:
                        await suhu.edit(f"❌ issues: `{ep}`")
            
   else:
         if len(m.command) < 2:
            await m.reply("» reply to an **audio file** or **give something to search.**")
         else:
            suhu = await m.reply("🔎 **searching...**")
            query = m.text.split(None, 1)[1]
            search = ytsearch(query)
            if search==0:
               await suhu.edit("❌ **no results found.**")
            else:
               songname = search[0]
               url = search[1]
               veez, ytlink = await ytdl(url)
               if veez==0:
                  await suhu.edit(f"❌ youtube-dl issues detected\n\n» `{ytlink}`")
               else:
                  if chat_id in QUEUE:
                     pos = add_to_queue(chat_id, songname, ytlink, url, "Audio", 0)
                     await suhu.delete()
                     await m.reply_photo(
                        photo=f"{IMG_1}",
                        caption=f"💡 **Track added to the queue**\n\n💭 **Chat:** `{chat_id}`\n🎧 **Request by:** {m.from_user.mention()}\n🔢 **At position »** `{pos}`",
                        reply_markup=keyboard,
                     )
                  else:
                     try:
                        await call_py.join_group_call(
                           chat_id,
                           AudioPiped(
                              ytlink,
                           ),
                           stream_type=StreamType().pulse_stream,
                        )
                        add_to_queue(chat_id, songname, ytlink, url, "Audio", 0)
                        await suhu.delete()
                        await m.reply_photo(
                           photo=f"{IMG_2}",
                           caption=f"💡 **music streaming started.**\n\n🏷 **Name:** [{songname}]({url})\n💭 **Chat:** `{chat_id}`\n💡 **Status:** `Playing`\n🎧 **Request by:** {m.from_user.mention()}",
                           disable_web_page_preview=True,
                           reply_markup=keyboard,
                        )
                     except Exception as ep:
                        await suhu.edit(f"❌ issues: `{ep}`")


@Client.on_message(command(["stream", f"stream@{BOT_USERNAME}"]) & other_filters)
async def stream(client, m: Message):
   
   keyboard = InlineKeyboardMarkup(
            [
                [
                    InlineKeyboardButton(
                        text="✨ ɢʀᴏᴜᴘ",
                        url=f"https://t.me/{GROUP_SUPPORT}"),
                    InlineKeyboardButton(
                        text="🌻 ᴄʜᴀɴɴᴇʟ",
                        url=f"https://t.me/{UPDATES_CHANNEL}")
                ]
            ]
        )
   
   chat_id = m.chat.id
   if len(m.command) < 2:
      await m.reply("» give me a live-link/m3u8 url/youtube link to stream.")
   else: 
      link = m.text.split(None, 1)[1]
      suhu = await m.reply("🔄 **processing stream...**")

      regex = r"^(https?\:\/\/)?(www\.youtube\.com|youtu\.?be)\/.+"
      match = re.match(regex,link)
      if match:
         veez, livelink = await ytdl(link)
      else:
         livelink = link
         veez = 1
      
      if veez==0:
         await suhu.edit(f"❌ youtube-dl issues detected\n\n» `{ytlink}`")
      else:
         if chat_id in QUEUE:
            pos = add_to_queue(chat_id, "Radio", livelink, link, "Audio", 0)
            await suhu.delete()
            await m.reply_photo(
               photo=f"{IMG_1}",
               caption=f"💡 **Track added to the queue**\n\n💭 **Chat:** `{chat_id}`\n🎧 **Request by:** {m.from_user.mention()}\n🔢 **At position »** `{pos}`",
               reply_markup=keyboard,
            )
         else:
            try:
               await call_py.join_group_call(
                  chat_id,
                  AudioPiped(
                     livelink,
                  ),
                  stream_type=StreamType().pulse_stream,
               )
               add_to_queue(chat_id, "Radio", livelink, link, "Audio", 0)
               await suhu.delete()
               await m.reply_photo(
                  photo=f"{IMG_2}",
                  caption=f"💡 **[Radio Live]({link}) stream started.**\n\n💭 **Chat:** `{chat_id}`\n💡 **Status:** `Playing`\n🎧 **Request by:** {m.from_user.mention()}",
                  disable_web_page_preview=True,
                  reply_markup=keyboard,
               )
            except Exception as ep:
               await suhu.edit(f"❌ issues: `{ep}`")