video-stream/driver/utils.py

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