47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from discord.ext import commands
|
|
import requests
|
|
|
|
from utils.config import LAVALINK_HOST, LAVALINK_PORT, LAVALINK_PASSWORD, LOG
|
|
|
|
|
|
class POToken(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command()
|
|
@commands.dm_only()
|
|
@commands.is_owner()
|
|
async def potoken(self, ctx, token: str = None, visitor_data: str = None):
|
|
"""Update the poToken for lavalink youtube support."""
|
|
if not token or not visitor_data:
|
|
return await ctx.send(
|
|
"Missing token and/or visitor data. Format as"
|
|
f" `{self.bot.command_prefix}potoken <token> <visitor"
|
|
" data>`\n\nTo generate a poToken, see"
|
|
" [this](https://github.com/iv-org/youtube-trusted-session-generator)"
|
|
)
|
|
|
|
url = f"http://{LAVALINK_HOST}:{LAVALINK_PORT}/youtube"
|
|
request = requests.post(
|
|
url,
|
|
json={"poToken": token, "visitorData": visitor_data},
|
|
headers={"Authorization": LAVALINK_PASSWORD},
|
|
)
|
|
|
|
if request.status_code != 204:
|
|
LOG.error("Error updating poToken")
|
|
return await ctx.send(
|
|
"Error setting poToken, YouTube source plugin is likely not"
|
|
" enabled. Read the Guava docs and look"
|
|
" [here](https://github.com/lavalink-devs/youtube-source)."
|
|
)
|
|
|
|
LOG.info("poToken successfully updated")
|
|
await ctx.send(
|
|
"Successfully posted the token and visitor data to lavalink."
|
|
)
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(POToken(bot))
|