557 lines
24 KiB
Python
557 lines
24 KiB
Python
"""
|
|
Video + Music Stream Telegram Bot
|
|
Copyright (c) 2022-present levina=lab <https://github.com/levina-lab>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but without any warranty; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/licenses.html>
|
|
"""
|
|
|
|
|
|
import re
|
|
import asyncio
|
|
|
|
from config import BOT_USERNAME, IMG_1, IMG_2, IMG_5
|
|
from driver.decorators import require_admin, check_blacklist
|
|
from program.utils.inline import stream_markup
|
|
from driver.design.thumbnail import thumb
|
|
from driver.design.chatname import CHAT_TITLE
|
|
from driver.filters import command, other_filters
|
|
from driver.queues import QUEUE, add_to_queue
|
|
from driver.core import calls, user, me_user
|
|
from driver.database.dbqueue import add_active_chat, remove_active_chat, music_on
|
|
from driver.utils import remove_if_exists, from_tg_get_msg
|
|
|
|
from pyrogram import Client
|
|
from pyrogram.errors import UserAlreadyParticipant, UserNotParticipant
|
|
from pyrogram.types import InlineKeyboardMarkup, Message
|
|
|
|
from pytgcalls import StreamType
|
|
from pytgcalls.types.input_stream import AudioVideoPiped
|
|
from pytgcalls.types.input_stream.quality import (
|
|
HighQualityAudio,
|
|
HighQualityVideo,
|
|
LowQualityVideo,
|
|
MediumQualityVideo,
|
|
)
|
|
from pytgcalls.exceptions import NoVideoSourceFound, NoActiveGroupCall, GroupCallNotFound
|
|
|
|
from youtubesearchpython import VideosSearch
|
|
|
|
|
|
def ytsearch(query: str):
|
|
try:
|
|
search = VideosSearch(query, limit=1).result()
|
|
data = search["result"][0]
|
|
songname = data["title"]
|
|
url = data["link"]
|
|
duration = data["duration"]
|
|
thumbnail = data["thumbnails"][0]["url"]
|
|
return [songname, url, duration, thumbnail]
|
|
except Exception as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
return 0
|
|
|
|
|
|
async def ytdl(link):
|
|
proc = await asyncio.create_subprocess_exec(
|
|
"yt-dlp",
|
|
"--geo-bypass",
|
|
"-g",
|
|
"-f",
|
|
"best[height<=?720][width<=?1280]/best",
|
|
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()
|
|
|
|
|
|
def convert_seconds(seconds):
|
|
seconds = seconds % (24 * 3600)
|
|
seconds %= 3600
|
|
minutes = seconds // 60
|
|
seconds %= 60
|
|
return "%02d:%02d" % (minutes, seconds)
|
|
|
|
|
|
async def play_tg_file(c: Client, m: Message, replied: Message = None, link: str = None):
|
|
chat_id = m.chat.id
|
|
user_id = m.from_user.id
|
|
if link:
|
|
try:
|
|
replied = await from_tg_get_msg(link)
|
|
except Exception as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
return await m.reply_text(f"🚫 error:\n\n» {e}")
|
|
if not replied:
|
|
return await m.reply(
|
|
"» reply to an **audio file** or **give something to search.**"
|
|
)
|
|
if replied.video or replied.document:
|
|
if not link:
|
|
loser = await replied.reply("📥 downloading video...")
|
|
else:
|
|
loser = await m.reply("📥 downloading video...")
|
|
dl = await replied.download()
|
|
link = replied.link
|
|
songname = "video"
|
|
duration = "00:00"
|
|
Q = 720
|
|
pq = m.text.split(None, 1)
|
|
if ("t.me" not in m.text) and len(pq) > 1:
|
|
pq = pq[1]
|
|
if pq == "720" or pq == "480" or pq == "360":
|
|
Q = int(pq)
|
|
else:
|
|
await loser.edit(
|
|
"Streaming the local video in 720p quality"
|
|
)
|
|
try:
|
|
if replied.video:
|
|
songname = replied.video.file_name[:80]
|
|
duration = convert_seconds(replied.video.duration)
|
|
elif replied.document:
|
|
songname = replied.document.file_name[:80]
|
|
except BaseException:
|
|
songname = "video"
|
|
|
|
if chat_id in QUEUE:
|
|
await loser.edit("🔄 Queueing Track...")
|
|
gcname = m.chat.title
|
|
ctitle = await CHAT_TITLE(gcname)
|
|
title = songname
|
|
userid = m.from_user.id
|
|
thumbnail = f"{IMG_5}"
|
|
image = await thumb(thumbnail, title, userid, ctitle)
|
|
pos = add_to_queue(chat_id, songname, dl, link, "video", Q)
|
|
await loser.delete()
|
|
requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=image,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"💡 **Track added to queue »** `{pos}`\n\n"
|
|
f"🗂 **Name:** [{songname}]({link}) | `video`\n"
|
|
f"⏱️ **Duration:** `{duration}`\n"
|
|
f"🧸 **Request by:** {requester}",
|
|
)
|
|
remove_if_exists(image)
|
|
else:
|
|
try:
|
|
await loser.edit("🔄 Joining Group Call...")
|
|
gcname = m.chat.title
|
|
ctitle = await CHAT_TITLE(gcname)
|
|
title = songname
|
|
userid = m.from_user.id
|
|
thumbnail = f"{IMG_5}"
|
|
image = await thumb(thumbnail, title, userid, ctitle)
|
|
if Q == 720:
|
|
amaze = HighQualityVideo()
|
|
elif Q == 480:
|
|
amaze = MediumQualityVideo()
|
|
elif Q == 360:
|
|
amaze = LowQualityVideo()
|
|
await music_on(chat_id)
|
|
await add_active_chat(chat_id)
|
|
await calls.join_group_call(
|
|
chat_id,
|
|
AudioVideoPiped(
|
|
dl,
|
|
HighQualityAudio(),
|
|
amaze,
|
|
),
|
|
stream_type=StreamType().pulse_stream,
|
|
)
|
|
add_to_queue(chat_id, songname, dl, link, "video", Q)
|
|
await loser.delete()
|
|
requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=image,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"🗂 **Name:** [{songname}]({link}) | `video`\n"
|
|
f"⏱️ **Duration:** `{duration}`\n"
|
|
f"🧸 **Request by:** {requester}",
|
|
)
|
|
remove_if_exists(image)
|
|
except (NoActiveGroupCall, GroupCallNotFound):
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The bot can't find the Group call or it's inactive.\n\n» Use /startvc command to turn on the Group call !")
|
|
except BaseException as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
else:
|
|
await m.reply(
|
|
"» reply to an **video file** or **give something to search.**"
|
|
)
|
|
|
|
|
|
@Client.on_message(command(["vplay", f"vplay@{BOT_USERNAME}"]) & other_filters)
|
|
@check_blacklist()
|
|
@require_admin(permissions=["can_manage_voice_chats", "can_delete_messages", "can_invite_users"], self=True)
|
|
async def vplay(c: Client, m: Message):
|
|
await m.delete()
|
|
replied = m.reply_to_message
|
|
chat_id = m.chat.id
|
|
user_id = m.from_user.id
|
|
if m.sender_chat:
|
|
return await m.reply_text(
|
|
"you're an __Anonymous__ user !\n\n» revert back to your real user account to use this bot."
|
|
)
|
|
try:
|
|
ubot = me_user.id
|
|
b = await c.get_chat_member(chat_id, ubot)
|
|
if b.status == "banned":
|
|
try:
|
|
await m.reply_text("❌ The userbot is banned in this chat, unban the userbot first to be able to play music !")
|
|
await remove_active_chat(chat_id)
|
|
except BaseException:
|
|
pass
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if not invitelink:
|
|
await c.export_chat_invite_link(chat_id)
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if invitelink.startswith("https://t.me/+"):
|
|
invitelink = invitelink.replace(
|
|
"https://t.me/+", "https://t.me/joinchat/"
|
|
)
|
|
await user.join_chat(invitelink)
|
|
await remove_active_chat(chat_id)
|
|
except UserNotParticipant:
|
|
try:
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if not invitelink:
|
|
await c.export_chat_invite_link(chat_id)
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if invitelink.startswith("https://t.me/+"):
|
|
invitelink = invitelink.replace(
|
|
"https://t.me/+", "https://t.me/joinchat/"
|
|
)
|
|
await user.join_chat(invitelink)
|
|
await remove_active_chat(chat_id)
|
|
except UserAlreadyParticipant:
|
|
pass
|
|
except Exception as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
return await m.reply_text(
|
|
f"❌ **userbot failed to join**\n\n**reason**: `{e}`"
|
|
)
|
|
if replied:
|
|
if replied.video or replied.document:
|
|
await play_tg_file(c, m, replied)
|
|
else:
|
|
if len(m.command) < 2:
|
|
await m.reply(
|
|
"» reply to an **video file** or **give something to search.**"
|
|
)
|
|
else:
|
|
loser = await c.send_message(chat_id, "🔍 **Loading...**")
|
|
query = m.text.split(None, 1)[1]
|
|
search = ytsearch(query)
|
|
Q = 720
|
|
amaze = HighQualityVideo()
|
|
if search == 0:
|
|
await loser.edit("❌ **no results found.**")
|
|
else:
|
|
songname = search[0]
|
|
title = search[0]
|
|
url = search[1]
|
|
duration = search[2]
|
|
thumbnail = search[3]
|
|
userid = m.from_user.id
|
|
gcname = m.chat.title
|
|
ctitle = await CHAT_TITLE(gcname)
|
|
image = await thumb(thumbnail, title, userid, ctitle)
|
|
veez, ytlink = await ytdl(url)
|
|
if veez == 0:
|
|
await loser.edit(f"❌ yt-dl issues detected\n\n» `{ytlink}`")
|
|
else:
|
|
if chat_id in QUEUE:
|
|
await loser.edit("🔄 Queueing Track...")
|
|
pos = add_to_queue(
|
|
chat_id, songname, ytlink, url, "video", Q
|
|
)
|
|
await loser.delete()
|
|
requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=image,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"💡 **Track added to queue »** `{pos}`\n\n🗂 **Name:** [{songname}]({url}) | `video`\n⏱ **Duration:** `{duration}`\n🧸 **Request by:** {requester}",
|
|
)
|
|
remove_if_exists(image)
|
|
else:
|
|
try:
|
|
await loser.edit("🔄 Joining Group Call...")
|
|
await music_on(chat_id)
|
|
await add_active_chat(chat_id)
|
|
await calls.join_group_call(
|
|
chat_id,
|
|
AudioVideoPiped(
|
|
ytlink,
|
|
HighQualityAudio(),
|
|
amaze,
|
|
),
|
|
stream_type=StreamType().local_stream,
|
|
)
|
|
add_to_queue(chat_id, songname, ytlink, url, "video", Q)
|
|
await loser.delete()
|
|
requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=image,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"🗂 **Name:** [{songname}]({url}) | `video`\n⏱ **Duration:** `{duration}`\n🧸 **Request by:** {requester}",
|
|
)
|
|
remove_if_exists(image)
|
|
except (NoActiveGroupCall, GroupCallNotFound):
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The bot can't find the Group call or it's inactive.\n\n» Use /startvc command to turn on the Group call !")
|
|
except NoVideoSourceFound:
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The content you provide to play has no video source")
|
|
except NoAudioSourceFound:
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The content you provide to play has no audio source")
|
|
except BaseException as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
|
|
else:
|
|
if len(m.command) < 2:
|
|
await m.reply(
|
|
"» reply to an **video file** or **give something to search.**"
|
|
)
|
|
elif "t.me" in m.command[1]:
|
|
for i in m.command[1:]:
|
|
if "t.me" in i:
|
|
await play_tg_file(c, m, link=i)
|
|
continue
|
|
else:
|
|
loser = await c.send_message(chat_id, "🔍 **Loading...**")
|
|
query = m.text.split(None, 1)[1]
|
|
search = ytsearch(query)
|
|
Q = 720
|
|
amaze = HighQualityVideo()
|
|
if search == 0:
|
|
await loser.edit("❌ **no results found.**")
|
|
else:
|
|
songname = search[0]
|
|
title = search[0]
|
|
url = search[1]
|
|
duration = search[2]
|
|
thumbnail = search[3]
|
|
userid = m.from_user.id
|
|
gcname = m.chat.title
|
|
ctitle = await CHAT_TITLE(gcname)
|
|
image = await thumb(thumbnail, title, userid, ctitle)
|
|
veez, ytlink = await ytdl(url)
|
|
if veez == 0:
|
|
await loser.edit(f"❌ yt-dl issues detected\n\n» `{ytlink}`")
|
|
else:
|
|
if chat_id in QUEUE:
|
|
await loser.edit("🔄 Queueing Track...")
|
|
pos = add_to_queue(chat_id, songname, ytlink, url, "video", Q)
|
|
await loser.delete()
|
|
requester = (
|
|
f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
)
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=image,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"💡 **Track added to queue »** `{pos}`\n\n🗂 **Name:** [{songname}]({url}) | `video`\n⏱ **Duration:** `{duration}`\n🧸 **Request by:** {requester}",
|
|
)
|
|
remove_if_exists(image)
|
|
else:
|
|
try:
|
|
await loser.edit("🔄 Joining Group Call...")
|
|
await music_on(chat_id)
|
|
await add_active_chat(chat_id)
|
|
await calls.join_group_call(
|
|
chat_id,
|
|
AudioVideoPiped(
|
|
ytlink,
|
|
HighQualityAudio(),
|
|
amaze,
|
|
),
|
|
stream_type=StreamType().local_stream,
|
|
)
|
|
add_to_queue(chat_id, songname, ytlink, url, "video", Q)
|
|
await loser.delete()
|
|
requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=image,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"🗂 **Name:** [{songname}]({url}) | `video`\n⏱ **Duration:** `{duration}`\n🧸 **Request by:** {requester}",
|
|
)
|
|
remove_if_exists(image)
|
|
except (NoActiveGroupCall, GroupCallNotFound):
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The bot can't find the Group call or it's inactive.\n\n» Use /startvc command to turn on the Group call !")
|
|
except NoVideoSourceFound:
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The content you provide to play has no video source")
|
|
except NoAudioSourceFound:
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The content you provide to play has no audio source")
|
|
except BaseException as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
|
|
|
|
@Client.on_message(command(["vstream", f"vstream@{BOT_USERNAME}"]) & other_filters)
|
|
@check_blacklist()
|
|
@require_admin(permissions=["can_manage_voice_chats", "can_delete_messages", "can_invite_users"], self=True)
|
|
async def vstream(c: Client, m: Message):
|
|
await m.delete()
|
|
chat_id = m.chat.id
|
|
user_id = m.from_user.id
|
|
if m.sender_chat:
|
|
return await m.reply_text(
|
|
"you're an __Anonymous__ user !\n\n» revert back to your real user account to use this bot."
|
|
)
|
|
try:
|
|
ubot = me_user.id
|
|
b = await c.get_chat_member(chat_id, ubot)
|
|
if b.status == "banned":
|
|
try:
|
|
await m.reply_text("❌ The userbot is banned in this chat, unban the userbot first to be able to play music !")
|
|
await remove_active_chat(chat_id)
|
|
except BaseException:
|
|
pass
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if not invitelink:
|
|
await c.export_chat_invite_link(chat_id)
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if invitelink.startswith("https://t.me/+"):
|
|
invitelink = invitelink.replace(
|
|
"https://t.me/+", "https://t.me/joinchat/"
|
|
)
|
|
await user.join_chat(invitelink)
|
|
await remove_active_chat(chat_id)
|
|
except UserNotParticipant:
|
|
try:
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if not invitelink:
|
|
await c.export_chat_invite_link(chat_id)
|
|
invitelink = (await c.get_chat(chat_id)).invite_link
|
|
if invitelink.startswith("https://t.me/+"):
|
|
invitelink = invitelink.replace(
|
|
"https://t.me/+", "https://t.me/joinchat/"
|
|
)
|
|
await user.join_chat(invitelink)
|
|
await remove_active_chat(chat_id)
|
|
except UserAlreadyParticipant:
|
|
pass
|
|
except Exception as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|
|
return await m.reply_text(
|
|
f"❌ **userbot failed to join**\n\n**reason**: `{e}`"
|
|
)
|
|
|
|
if len(m.command) < 2:
|
|
await m.reply("» Give me a youtube live url/m3u8 url to stream.")
|
|
else:
|
|
if len(m.command) == 2:
|
|
Q = 720
|
|
url = m.text.split(None, 1)[1]
|
|
search = ytsearch(url)
|
|
loser = await c.send_message(chat_id, "🔍 **Loading...**")
|
|
elif len(m.command) == 3:
|
|
op = m.text.split(None, 1)[1]
|
|
url = op.split(None, 1)[0]
|
|
quality = op.split(None, 1)[1]
|
|
search = ytsearch(op)
|
|
if quality == "720" or "480" or "360":
|
|
Q = int(quality)
|
|
else:
|
|
Q = 720
|
|
await m.reply(
|
|
"» Streaming the live video in 720p quality"
|
|
)
|
|
loser = await c.send_message(chat_id, "🔍 **Loading...**")
|
|
else:
|
|
pass
|
|
|
|
regex = r"^(https?\:\/\/)?(www\.youtube\.com|youtu\.?be)\/.+"
|
|
match = re.match(regex, url)
|
|
|
|
if match:
|
|
veez, livelink = await ytdl(url)
|
|
else:
|
|
livelink = url
|
|
veez = 1
|
|
|
|
if veez == 0:
|
|
await loser.edit(f"❌ yt-dl issues detected\n\n» `{livelink}`")
|
|
else:
|
|
songname = search[0]
|
|
if chat_id in QUEUE:
|
|
await loser.edit("🔄 Queueing Track...")
|
|
pos = add_to_queue(chat_id, songname, livelink, url, "video", Q)
|
|
await loser.delete()
|
|
requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=f"{IMG_1}",
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"💡 **Track added to queue »** `{pos}`\n\n🗂 **Name:** [{songname}]({url}) | `live`\n🧸 **Requested by:** {requester}",
|
|
)
|
|
else:
|
|
if Q == 720:
|
|
amaze = HighQualityVideo()
|
|
elif Q == 480:
|
|
amaze = MediumQualityVideo()
|
|
elif Q == 360:
|
|
amaze = LowQualityVideo()
|
|
try:
|
|
await loser.edit("🔄 Joining Group Call...")
|
|
await music_on(chat_id)
|
|
await add_active_chat(chat_id)
|
|
await calls.join_group_call(
|
|
chat_id,
|
|
AudioVideoPiped(
|
|
livelink,
|
|
HighQualityAudio(),
|
|
amaze,
|
|
),
|
|
stream_type=StreamType().live_stream,
|
|
)
|
|
add_to_queue(chat_id, songname, livelink, url, "video", Q)
|
|
await loser.delete()
|
|
requester = (
|
|
f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
|
|
)
|
|
buttons = stream_markup(user_id)
|
|
await m.reply_photo(
|
|
photo=f"{IMG_2}",
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
caption=f"🗂 **Name:** [{songname}]({url}) | `live`\n🧸 **Requested by:** {requester}",
|
|
)
|
|
except (NoActiveGroupCall, GroupCallNotFound):
|
|
await loser.delete()
|
|
await remove_active_chat(chat_id)
|
|
await m.reply_text("❌ The bot can't find the Group call or it's inactive.\n\n» Use /startvc command to turn on the Group call !")
|
|
except BaseException as e:
|
|
LOGS.info(f"[ERROR]: {e}")
|