From 2bda0b16501eacb933fc33dcf146ccedda104703 Mon Sep 17 00:00:00 2001 From: levina <82658782+levina-lab@users.noreply.github.com> Date: Tue, 26 Oct 2021 14:00:00 +0700 Subject: [PATCH] prepare --- program/music.py | 178 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 178 insertions(+) create mode 100644 program/music.py diff --git a/program/music.py b/program/music.py new file mode 100644 index 0000000..07f14b5 --- /dev/null +++ b/program/music.py @@ -0,0 +1,178 @@ +import os +import re +import asyncio +from config import BOT_USERNAME +from pytgcalls import StreamType +from driver.filters import command +from pyrogram.types import Message +from pyrogram import Client, filters +from driver.veez import call_py, bot +from youtubesearchpython import VideosSearch +from program.queues import QUEUE, add_to_queue +from pytgcalls.types.input_stream import AudioPiped + + +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'][:35] + "..." + 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): + 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[:35] + "..." + else: + songname = replied.audio.file_name[:35] + "..." + elif replied.voice: + songname = "Voice Note" + if chat_id in QUEUE: + pos = add_to_queue(chat_id, songname, dl, link, "Audio", 0) + await suhu.edit(f"๐Ÿ’ก **Track added to the queue**\n๐Ÿ”ข At position ยป `{pos}`") + 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 suhu.edit(f"๐Ÿ’ก **music streaming started.**\n\n๐Ÿท **Name:** [{songname}]({link})\n๐Ÿ’ฌ **Chat:** `{chat_id}`", disable_web_page_preview=True) + 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.edit(f"๐Ÿ’ก **Track added to the queue**\n๐Ÿ”ข At position ยป `{pos}`") + 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.edit(f"๐Ÿ’ก **music streaming started.**\n\n๐Ÿท **Name:** [{songname}]({url})\n๐Ÿ’ฌ **Chat:** `{chat_id}`", disable_web_page_preview=True) + 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.edit(f"๐Ÿ’ก **Track added to the queue**\n๐Ÿ”ข At position ยป `{pos}`") + 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.edit(f"๐Ÿ’ก **music streaming started.**\n\n๐Ÿท **Name:** [{songname}]({url})\n๐Ÿ’ฌ **Chat:** `{chat_id}`", disable_web_page_preview=True) + 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): + 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.edit(f"๐Ÿ’ก **Track added to the queue**\n๐Ÿ”ข At position ยป `{pos}`") + 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.edit(f"๐Ÿ’ก **[Radio]({link}) live stream started.**\n๐Ÿ’ฌ Chat: `{chat_id}`", disable_web_page_preview=True) + except Exception as ep: + await suhu.edit(f"โŒ issues: `{ep}`")