bot.py

from asyncio import sleep
from os import getenv

from discord import Intents, Activity, ActivityType, Interaction, app_commands, FFmpegPCMAudio, ui, ButtonStyle
from discord.ext import commands
from dotenv import load_dotenv
from requests import get

load_dotenv()
TOKEN = getenv("TOKEN")
NCM_API = getenv("NCM_API")
UNM_API = getenv("UNM_API")
bot = commands.Bot("/", intents=Intents.all(), proxy="http://127.0.0.1:10808")

ids = []
names = []
playing = False
playlist_id = []
playlist_name = []


@bot.event
async def on_ready():
    activity = Activity(name="Type / and select this bot to start music player.", type=ActivityType.playing)
    await bot.change_presence(activity=activity)
    await bot.tree.sync()


# player = app_commands.Group(name="player", description="A simple player to play music into Voice Channel.")
# bot.tree.add_command(player)


@bot.tree.command(name="search", description="Search music by name.")
@app_commands.describe(music_name="Music Name")
async def search(interaction: Interaction, music_name: str):
    await interaction.response.send_message(f"Searching for `{music_name}`...")
    global ids
    global names
    ids = []
    names = []
    content = await do_search(music_name, 0, False)
    view = Control(music_name)
    await interaction.edit_original_response(content=content, view=view)


offset = 0


class Control(ui.View):
    def __init__(self, music_name):
        global m_name
        m_name = music_name
        super().__init__()

    @ui.button(label="Previous", style=ButtonStyle.danger, emoji="👈")
    async def approve(self, interaction: Interaction, button: ui.Button):
        global offset
        await interaction.response.defer()
        if not offset == 0:
            offset -= 30
            content = await do_search(m_name, offset, True)
            view = Control(m_name)
            await interaction.edit_original_response(content=content, view=view)

    @ui.button(label="Next", style=ButtonStyle.success, emoji="👉")
    async def reject(self, interaction: Interaction, button: ui.Button):
        global offset
        await interaction.response.defer()
        offset += 30
        content = await do_search(m_name, offset, False)
        view = Control(m_name)
        await interaction.edit_original_response(content=content, view=view)


num = 0


async def do_search(music_name: str, offset: int, previous: bool) -> str:
    global num
    j_search = get(f"{NCM_API}/cloudsearch?keywords={music_name}&limit=30&offset={offset}&type=1")
    search = j_search.json()
    messages = "**Search result:**\n"
    if search["code"] == 200:
        for songs in search["result"]["songs"]:
            ids.append(songs["id"])
            name = songs["ar"][0]["name"] + " - " + songs["name"]
            names.append(name)
            messages += str(num) + ". " + name + "\n"
            if not previous:
                num += 1
            else:
                num -= 1
        messages += "**Type /play <music_num> to start player.**"
        return messages
    else:
        return "**No search result. Check your music name and try again**"


@bot.tree.command(name="list", description="See the playlist.")
async def list(interaction: Interaction):
    global playlist_id
    global playlist_name
    if not playlist_id or not playlist_name:
        await interaction.response.send_message("**No items in the playlist**")
        return
    messages = "**Playlist:**\n"
    num = 0
    for playlist in playlist_name:
        ids.append(playlist)
        messages += str(num) + ". " + playlist + "\n"
        num += 1
    await interaction.response.send_message(messages)


@bot.tree.command(name="play", description="Play a music into Voice Channel.")
@app_commands.describe(music_num="Music Num")
async def play(interaction: Interaction, music_num: int):
    global ids
    global names
    global playing
    if not ids or not names:
        await interaction.response.send_message("**Please search a music first!**")
        return
    music_id = ids[music_num - 1]
    music_name = names[music_num - 1]
    try:
        voice_cha = interaction.user.voice.channel
    except AttributeError:
        await interaction.response.send_message("**Please join the Voice Channel first!**")
        return
    playlist_id.append(music_id)
    playlist_name.append(music_name)
    if playing:
        await interaction.response.send_message(f"Added `{music_name}` to playlist.")
        return
    await interaction.response.send_message("Starting player...")
    j_lyric = get(f"{NCM_API}/lyric/new?id={music_id}")
    lyric = j_lyric.json()
    if lyric["code"] == 200:
        music_lrc = lyric["lrc"]["lyric"]
    voice_cli = await voice_cha.connect()
    while True:
        if playlist_id:
            await start_play(playlist_id.pop(0), playlist_name.pop(0), interaction, voice_cli)
        else:
            playing = False
            await voice_cli.disconnect()
            break


async def start_play(music_id, music_name, interaction: Interaction, voice_cli):
    global playing
    j_match = get(f"{UNM_API}/match?id={music_id}&server=qq,pyncmd")
    match = j_match.json()
    if match["code"] == 200 and match["message"] == "匹配成功":
        audio = FFmpegPCMAudio(match["data"]["url"])
        voice_cli.play(audio)
        playing = True
        await interaction.edit_original_response(content=f"Player started.")
        await interaction.channel.send(content=f"Now playing: {music_name}")
        while voice_cli.is_playing():
            await sleep(1)
        await interaction.edit_original_response(content="Player stopped.")


bot.run(TOKEN)