mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 11:18:54 -05:00
df922a0e3e
* More robust cleanup for audio and streams Signed-off-by: Toby Harradine <tobyharradine@gmail.com> * Remove copied code from streams Signed-off-by: Toby Harradine <tobyharradine@gmail.com>
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
from pathlib import Path
|
|
from aiohttp import ClientSession
|
|
import shutil
|
|
import logging
|
|
|
|
from .audio import Audio
|
|
from .manager import start_lavalink_server
|
|
from redbot.core import commands
|
|
from redbot.core.data_manager import cog_data_path
|
|
import redbot.core
|
|
|
|
log = logging.getLogger("red.audio")
|
|
|
|
LAVALINK_DOWNLOAD_URL = (
|
|
"https://github.com/Cog-Creators/Red-DiscordBot/releases/download/{}/Lavalink.jar"
|
|
).format(redbot.core.__version__)
|
|
|
|
LAVALINK_DOWNLOAD_DIR = cog_data_path(raw_name="Audio")
|
|
LAVALINK_JAR_FILE = LAVALINK_DOWNLOAD_DIR / "Lavalink.jar"
|
|
|
|
APP_YML_FILE = LAVALINK_DOWNLOAD_DIR / "application.yml"
|
|
BUNDLED_APP_YML_FILE = Path(__file__).parent / "data/application.yml"
|
|
|
|
|
|
async def download_lavalink(session):
|
|
with LAVALINK_JAR_FILE.open(mode="wb") as f:
|
|
async with session.get(LAVALINK_DOWNLOAD_URL) as resp:
|
|
while True:
|
|
chunk = await resp.content.read(512)
|
|
if not chunk:
|
|
break
|
|
f.write(chunk)
|
|
|
|
|
|
async def maybe_download_lavalink(loop, cog):
|
|
jar_exists = LAVALINK_JAR_FILE.exists()
|
|
current_build = redbot.core.VersionInfo(*await cog.config.current_build())
|
|
|
|
if not jar_exists or current_build < redbot.core.version_info:
|
|
log.info("Downloading Lavalink.jar")
|
|
LAVALINK_DOWNLOAD_DIR.mkdir(parents=True, exist_ok=True)
|
|
async with ClientSession(loop=loop) as session:
|
|
await download_lavalink(session)
|
|
await cog.config.current_build.set(redbot.core.version_info.to_json())
|
|
|
|
shutil.copyfile(str(BUNDLED_APP_YML_FILE), str(APP_YML_FILE))
|
|
|
|
|
|
async def setup(bot: commands.Bot):
|
|
cog = Audio(bot)
|
|
if not await cog.config.use_external_lavalink():
|
|
await maybe_download_lavalink(bot.loop, cog)
|
|
await start_lavalink_server(bot.loop)
|
|
|
|
await cog.initialize()
|
|
|
|
bot.add_cog(cog)
|