177 lines
5.0 KiB
Python
177 lines
5.0 KiB
Python
import asyncio
|
||
import os
|
||
|
||
from driver.core import bot, calls, user
|
||
from driver.database.dbqueue import remove_active_chat
|
||
from driver.queues import (
|
||
QUEUE,
|
||
clear_queue,
|
||
get_queue,
|
||
pop_an_item,
|
||
clean_trash,
|
||
)
|
||
|
||
from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
|
||
from pytgcalls.types import Update
|
||
from pytgcalls.types.input_stream import AudioPiped, AudioVideoPiped
|
||
from pytgcalls.types.input_stream.quality import (
|
||
HighQualityAudio,
|
||
HighQualityVideo,
|
||
LowQualityVideo,
|
||
MediumQualityVideo,
|
||
)
|
||
from pytgcalls.types.stream import StreamAudioEnded
|
||
|
||
|
||
keyboard = InlineKeyboardMarkup(
|
||
[
|
||
[
|
||
InlineKeyboardButton(text="• Mᴇɴᴜ", callback_data="stream_menu_panel"),
|
||
InlineKeyboardButton(text="• Cʟᴏsᴇ", callback_data="set_close"),
|
||
]
|
||
]
|
||
)
|
||
|
||
|
||
async def skip_current_song(chat_id):
|
||
if chat_id in QUEUE:
|
||
chat_queue = get_queue(chat_id)
|
||
if "t.me" in chat_queue[0][2]:
|
||
clean_trash(chat_queue[0][1], chat_id)
|
||
if len(chat_queue) == 1:
|
||
await calls.leave_group_call(chat_id)
|
||
await remove_active_chat(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]
|
||
Q = chat_queue[1][4]
|
||
if type == "music":
|
||
await calls.change_stream(
|
||
chat_id,
|
||
AudioPiped(
|
||
url,
|
||
HighQualityAudio(),
|
||
),
|
||
)
|
||
elif type == "video":
|
||
if Q == 720:
|
||
hm = HighQualityVideo()
|
||
elif Q == 480:
|
||
hm = MediumQualityVideo()
|
||
elif Q == 360:
|
||
hm = LowQualityVideo()
|
||
await calls.change_stream(
|
||
chat_id,
|
||
AudioVideoPiped(
|
||
url,
|
||
HighQualityAudio(),
|
||
hm
|
||
)
|
||
)
|
||
pop_an_item(chat_id)
|
||
return [songname, link, type]
|
||
except BaseException as error:
|
||
print(error)
|
||
await calls.leave_group_call(chat_id)
|
||
await remove_active_chat(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
|
||
|
||
|
||
@calls.on_kicked()
|
||
async def kicked_handler(_, chat_id: int):
|
||
if chat_id in QUEUE:
|
||
await remove_active_chat(chat_id)
|
||
clear_queue(chat_id)
|
||
|
||
|
||
@calls.on_closed_voice_chat()
|
||
async def closed_voice_chat_handler(_, chat_id: int):
|
||
if chat_id in QUEUE:
|
||
await remove_active_chat(chat_id)
|
||
clear_queue(chat_id)
|
||
|
||
|
||
@calls.on_left()
|
||
async def left_handler(_, chat_id: int):
|
||
if chat_id in QUEUE:
|
||
await remove_active_chat(chat_id)
|
||
clear_queue(chat_id)
|
||
|
||
|
||
@calls.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 remove_active_chat(chat_id)
|
||
return
|
||
elif op == 2:
|
||
await bot.send_message(
|
||
chat_id,
|
||
"❌ 出现错误\n\n» **已经清空** __队列__ 并且退出了 __语音聊天__",
|
||
)
|
||
else:
|
||
await bot.send_message(
|
||
chat_id,
|
||
f"💡 **播放下一首**\n\n"
|
||
f"🗂 **名称：** [{op[0]}]({op[1]}) | `{op[2].replace('Audio', '音乐').replace('Video', '视频')}`\n"
|
||
f"💭 **会话：** `{chat_id}`",
|
||
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
|
||
|
||
|
||
def remove_if_exists(path):
|
||
if os.path.exists(path):
|
||
os.remove(path)
|
||
|
||
|
||
async def from_tg_get_msg(url: str):
|
||
data = url.split('/')[-2:]
|
||
if len(data) == 2:
|
||
cid = data[0]
|
||
if cid.isdigit():
|
||
cid = int('-100' + cid)
|
||
mid = int(data[1])
|
||
return await user.get_messages(cid, message_ids=mid)
|
||
return None
|