145 lines
4.1 KiB
Python
145 lines
4.1 KiB
Python
import os
|
||
import asyncio
|
||
from driver.veez import bot, call_py
|
||
from pytgcalls.types import Update
|
||
from pytgcalls.types.input_stream import AudioPiped, AudioVideoPiped
|
||
from driver.queues import (
|
||
QUEUE,
|
||
clear_queue,
|
||
get_queue,
|
||
pop_an_item,
|
||
)
|
||
from pytgcalls.types.input_stream.quality import (
|
||
HighQualityAudio,
|
||
HighQualityVideo,
|
||
LowQualityVideo,
|
||
MediumQualityVideo,
|
||
)
|
||
from pyrogram.types import (
|
||
CallbackQuery,
|
||
InlineKeyboardButton,
|
||
InlineKeyboardMarkup,
|
||
Message,
|
||
)
|
||
from pyrogram import Client, filters
|
||
from pytgcalls.types.stream import StreamAudioEnded, StreamVideoEnded
|
||
|
||
|
||
keyboard = InlineKeyboardMarkup(
|
||
[
|
||
[
|
||
InlineKeyboardButton(text="• Mᴇɴᴜ", callback_data="cbmenu"),
|
||
InlineKeyboardButton(text="• Cʟᴏsᴇ", callback_data="cls"),
|
||
]
|
||
]
|
||
)
|
||
|
||
|
||
async def skip_current_song(chat_id):
|
||
if chat_id in QUEUE:
|
||
chat_queue = get_queue(chat_id)
|
||
if len(chat_queue) == 1:
|
||
await call_py.leave_group_call(chat_id)
|
||
clear_queue(chat_id)
|
||
return 1
|
||
else:
|
||
try:
|
||
songname = chat_queue[1][0]
|
||
url = chat_queue[1][1]
|
||
link = chat_queue[1][2]
|
||
type = chat_queue[1][3]
|
||
resol = chat_queue[1][4]
|
||
image = chat_queue[1][5]
|
||
duration = chat_queue[1][6]
|
||
if type == "Audio":
|
||
await call_py.change_stream(
|
||
chat_id,
|
||
AudioPiped(
|
||
url,
|
||
),
|
||
)
|
||
elif type == "Video":
|
||
if resol == 720:
|
||
hm = HighQualityVideo()
|
||
elif resol == 480:
|
||
hm = MediumQualityVideo()
|
||
elif resol == 360:
|
||
hm = LowQualityVideo()
|
||
await call_py.change_stream(
|
||
chat_id,
|
||
AudioVideoPiped(
|
||
url,
|
||
HighQualityAudio(),
|
||
hm
|
||
)
|
||
)
|
||
pop_an_item(chat_id)
|
||
return [songname, link, type, image, duration]
|
||
except:
|
||
await call_py.leave_group_call(chat_id)
|
||
clear_queue(chat_id)
|
||
return 2
|
||
else:
|
||
return 0
|
||
|
||
|
||
async def skip_item(chat_id, h):
|
||
if chat_id in QUEUE:
|
||
chat_queue = get_queue(chat_id)
|
||
try:
|
||
x = int(h)
|
||
songname = chat_queue[x][0]
|
||
chat_queue.pop(x)
|
||
return songname
|
||
except Exception as e:
|
||
print(e)
|
||
return 0
|
||
else:
|
||
return 0
|
||
|
||
|
||
@call_py.on_kicked()
|
||
async def kicked_handler(_, chat_id: int):
|
||
if chat_id in QUEUE:
|
||
clear_queue(chat_id)
|
||
|
||
|
||
@call_py.on_closed_voice_chat()
|
||
async def closed_voice_chat_handler(_, chat_id: int):
|
||
if chat_id in QUEUE:
|
||
clear_queue(chat_id)
|
||
|
||
|
||
@call_py.on_left()
|
||
async def left_handler(_, chat_id: int):
|
||
if chat_id in QUEUE:
|
||
clear_queue(chat_id)
|
||
|
||
|
||
@call_py.on_stream_end()
|
||
async def stream_end_handler(_, u: Update):
|
||
if isinstance(u, StreamAudioEnded):
|
||
chat_id = u.chat_id
|
||
print(chat_id)
|
||
op = await skip_current_song(chat_id)
|
||
if op==1:
|
||
await bot.send_message(chat_id, "✅ streaming end")
|
||
elif op==2:
|
||
await bot.send_message(chat_id, "❌ an error occurred\n\n» **Clearing** __Queues__ and leaving video chat.")
|
||
else:
|
||
await bot.send_message(chat_id, f"💡 **Streaming next track**\n\n🏷 **Name:** [{op[0]}]({op[1]}) | `{op[2]}`\n⏱ **Duration:** `{op[3]}`", disable_web_page_preview=True, reply_markup=keyboard)
|
||
else:
|
||
pass
|
||
|
||
|
||
async def bash(cmd):
|
||
process = await asyncio.create_subprocess_shell(
|
||
cmd,
|
||
stdout=asyncio.subprocess.PIPE,
|
||
stderr=asyncio.subprocess.PIPE,
|
||
)
|
||
stdout, stderr = await process.communicate()
|
||
err = stderr.decode().strip()
|
||
out = stdout.decode().strip()
|
||
return out, err
|