blob: 2eba7351eecbff5db32f9039a36ba4b3808635bb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
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))
|
