docker-less support

This commit is contained in:
AF 2022-08-05 00:05:42 +03:00
parent 1d36c3e17e
commit 8a97216d88
13 changed files with 176 additions and 84 deletions

2
.gitignore vendored
View File

@ -213,3 +213,5 @@ fabric.properties
/data/ /data/
.token.txt
*.exe

View File

@ -29,7 +29,7 @@
</DockerPortBindingImpl> </DockerPortBindingImpl>
</list> </list>
</option> </option>
<option name="commandLineOptions" value="--cpus=&quot;3&quot; --memory=&quot;4000mb&quot; --network=&quot;v6d&quot;" /> <option name="commandLineOptions" value="--cpus=&quot;3&quot; --memory=&quot;4000mb&quot; --network=&quot;v6d&quot; --ip=&quot;172.18.0.30&quot;" />
<option name="showCommandPreview" value="true" /> <option name="showCommandPreview" value="true" />
<option name="volumeBindings"> <option name="volumeBindings">
<list> <list>

View File

@ -0,0 +1,24 @@
<component name="ProjectRunConfigurationManager">
<configuration default="false" name="guerilla" type="PythonConfigurationType" factoryName="Python">
<module name="v6d3music" />
<option name="INTERPRETER_OPTIONS" value="" />
<option name="PARENT_ENVS" value="true" />
<envs>
<env name="PYTHONUNBUFFERED" value="1" />
</envs>
<option name="SDK_HOME" value="$PROJECT_DIR$" />
<option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
<option name="IS_MODULE_SDK" value="false" />
<option name="ADD_CONTENT_ROOTS" value="true" />
<option name="ADD_SOURCE_ROOTS" value="true" />
<EXTENSION ID="PythonCoverageRunConfigurationExtension" runner="coverage.py" />
<option name="SCRIPT_NAME" value="v6d3music.run-bot" />
<option name="PARAMETERS" value="guerilla" />
<option name="SHOW_COMMAND_LINE" value="false" />
<option name="EMULATE_TERMINAL" value="false" />
<option name="MODULE_MODE" value="true" />
<option name="REDIRECT_INPUT" value="false" />
<option name="INPUT_FILE" value="" />
<method v="2" />
</configuration>
</component>

View File

@ -1,6 +1,7 @@
import asyncio import asyncio
import urllib.parse import urllib.parse
from pathlib import Path from pathlib import Path
from typing import Optional
import aiohttp import aiohttp
import discord import discord
@ -10,7 +11,7 @@ from v6d0auth.appfactory import AppFactory
from v6d0auth.run_app import start_app from v6d0auth.run_app import start_app
from v6d1tokens.client import request_token from v6d1tokens.client import request_token
from v6d3music.config import myroot from v6d3music.config import auth_redirect, myroot
from v6d3music.utils.bytes_hash import bytes_hash from v6d3music.utils.bytes_hash import bytes_hash
session_db = Db(myroot / 'session.db', kvrequest_type=KVJson) session_db = Db(myroot / 'session.db', kvrequest_type=KVJson)
@ -25,12 +26,17 @@ class MusicAppFactory(AppFactory):
client: discord.Client client: discord.Client
): ):
self.secret = secret self.secret = secret
self.redirect = 'https://music.parrrate.ru/auth/' self.redirect = auth_redirect
self.discord_auth = 'https://discord.com/api/oauth2/authorize?client_id=914432576926646322' \
f'&redirect_uri={urllib.parse.quote(self.redirect)}&response_type=code&scope=identify'
self.loop = asyncio.get_running_loop() self.loop = asyncio.get_running_loop()
self.client = client self.client = client
def auth_link(self):
if self.client.user is None:
return ''
else:
return f'https://discord.com/api/oauth2/authorize?client_id={self.client.user.id}' \
f'&redirect_uri={urllib.parse.quote(self.redirect)}&response_type=code&scope=identify'
def _file(self, file: str): def _file(self, file: str):
with open(self.htmlroot / file) as f: with open(self.htmlroot / file) as f:
return f.read() return f.read()
@ -46,14 +52,14 @@ class MusicAppFactory(AppFactory):
text = await self.file(f'{file}.html') text = await self.file(f'{file}.html')
text = text.replace( text = text.replace(
'$$DISCORD_AUTH$$', '$$DISCORD_AUTH$$',
self.discord_auth self.auth_link()
) )
return web.Response( return web.Response(
text=text, text=text,
content_type='text/html' content_type='text/html'
) )
async def code_token(self, code: str): async def code_token(self, code: str) -> dict:
data = { data = {
'client_id': '914432576926646322', 'client_id': '914432576926646322',
'client_secret': self.secret, 'client_secret': self.secret,
@ -68,14 +74,13 @@ class MusicAppFactory(AppFactory):
async with session.post('https://discord.com/api/oauth2/token', data=data, headers=headers) as response: async with session.post('https://discord.com/api/oauth2/token', data=data, headers=headers) as response:
return await response.json() return await response.json()
async def session_client(self, session: str): @classmethod
data = self.session_data(session) async def session_client(cls, data: dict) -> Optional[dict]:
client_token = data.get('token') match data:
if client_token is None: case {'token': {'access_token': str() as access_token}}:
return None pass
access_token = client_token.get('access_token') case _:
if access_token is None: return None
return None
headers = { headers = {
'Authorization': f'Bearer {access_token}' 'Authorization': f'Bearer {access_token}'
} }
@ -84,7 +89,7 @@ class MusicAppFactory(AppFactory):
return await response.json() return await response.json()
@classmethod @classmethod
def client_status(cls, sclient: dict): def client_status(cls, sclient: dict) -> dict:
user = cls.client_user(sclient) user = cls.client_user(sclient)
return { return {
'expires': sclient.get('expires'), 'expires': sclient.get('expires'),
@ -92,7 +97,7 @@ class MusicAppFactory(AppFactory):
} }
@classmethod @classmethod
def user_status(cls, user: dict): def user_status(cls, user: dict) -> dict:
return { return {
'avatar': cls.user_avatar_url(user), 'avatar': cls.user_avatar_url(user),
'id': cls.user_id(user), 'id': cls.user_id(user),
@ -100,37 +105,27 @@ class MusicAppFactory(AppFactory):
} }
@classmethod @classmethod
def user_username_full(cls, user: dict): def user_username_full(cls, user: dict) -> Optional[str]:
username = cls.user_username(user) match user:
if username is None: case {'username': str() as username, 'discriminator': str() as discriminator}:
return None return f'{username}#{discriminator}'
discriminator = cls.user_discriminator(user) case _:
if discriminator is None: return None
return None
return f'{username}#{discriminator}'
@classmethod @classmethod
def user_username(cls, user: dict): def client_user(cls, sclient: dict) -> Optional[dict]:
return user.get('username')
@classmethod
def user_discriminator(cls, user: dict):
return user.get('discriminator')
@classmethod
def client_user(cls, sclient: dict):
return sclient.get('user') return sclient.get('user')
@classmethod @classmethod
def user_id(cls, user: dict): def user_id(cls, user: dict) -> Optional[str | int]:
return user.get('id') return user.get('id')
@classmethod @classmethod
def user_avatar(cls, user: dict): def user_avatar(cls, user: dict) -> Optional[str]:
return user.get('avatar') return user.get('avatar')
@classmethod @classmethod
def user_avatar_url(cls, user: dict): def user_avatar_url(cls, user: dict) -> Optional[str]:
cid = cls.user_id(user) cid = cls.user_id(user)
if cid is None: if cid is None:
return None return None
@ -139,15 +134,26 @@ class MusicAppFactory(AppFactory):
return None return None
return f'https://cdn.discordapp.com/avatars/{cid}/{avatar}.png' return f'https://cdn.discordapp.com/avatars/{cid}/{avatar}.png'
async def session_status(self, session: str): async def session_status(self, session: str) -> dict:
data = self.session_data(session) data = self.session_data(session)
sclient = await self.session_client(session) sclient = await self.session_client(data)
return { return {
'code_set': data.get('code') is not None, 'code_set': data.get('code') is not None,
'token_set': data.get('token') is not None, 'token_set': data.get('token') is not None,
'client': (None if sclient is None else self.client_status(sclient)) 'client': (None if sclient is None else self.client_status(sclient))
} }
async def session_queue(self, session: str):
data = self.session_data(session)
sclient = await self.session_client(data)
if sclient is None:
return None
user = self.client_user(sclient)
if user is None:
return None
cid = self.user_id(user)
return cid
@classmethod @classmethod
def session_data(cls, session: str) -> dict: def session_data(cls, session: str) -> dict:
data = session_db.get(session, {}) data = session_db.get(session, {})
@ -198,6 +204,13 @@ class MusicAppFactory(AppFactory):
data=await self.session_status(session) data=await self.session_status(session)
) )
@routes.get('/queue/')
async def api_queue(request: web.Request) -> web.Response:
session = str(request.query.get('session'))
return web.json_response(
data=await self.session_queue(session)
)
@routes.get('/main.js') @routes.get('/main.js')
async def state(_request: web.Request) -> web.Response: async def state(_request: web.Request) -> web.Response:
return web.Response( return web.Response(
@ -212,5 +225,9 @@ class MusicAppFactory(AppFactory):
@classmethod @classmethod
async def start(cls, client: discord.Client): async def start(cls, client: discord.Client):
factory = cls(await request_token('music-client', 'token'), client) try:
await start_app(factory.app()) factory = cls(await request_token('music-client', 'token'), client)
except aiohttp.ClientConnectorError:
print('no web app (likely due to no token)')
else:
await start_app(factory.app())

View File

@ -35,7 +35,7 @@ presets: {shlex.join(allowed_presets)}
(), 'help' (), 'help'
) )
async with lock_for(ctx.guild, 'not in a guild'): async with lock_for(ctx.guild, 'not in a guild'):
queue = await queue_for(ctx, create=True) queue = await queue_for(ctx, create=True, force_play=False)
async for audio in yt_audios(ctx, args): async for audio in yt_audios(ctx, args):
queue.append(audio) queue.append(audio)
await ctx.reply('done') await ctx.reply('done')
@ -50,15 +50,15 @@ async def skip(ctx: Context, args: list[str]) -> None:
) )
match args: match args:
case []: case []:
queue = await queue_for(ctx, create=False) queue = await queue_for(ctx, create=False, force_play=False)
queue.skip_at(0, ctx.member) queue.skip_at(0, ctx.member)
case [pos] if pos.isdecimal(): case [pos] if pos.isdecimal():
pos = int(pos) pos = int(pos)
queue = await queue_for(ctx, create=False) queue = await queue_for(ctx, create=False, force_play=False)
queue.skip_at(pos, ctx.member) queue.skip_at(pos, ctx.member)
case [pos0, pos1] if pos0.isdecimal() and pos1.isdecimal(): case [pos0, pos1] if pos0.isdecimal() and pos1.isdecimal():
pos0, pos1 = int(pos0), int(pos1) pos0, pos1 = int(pos0), int(pos1)
queue = await queue_for(ctx, create=False) queue = await queue_for(ctx, create=False, force_play=False)
for i in range(pos0, pos1 + 1): for i in range(pos0, pos1 + 1):
if not queue.skip_at(pos0, ctx.member): if not queue.skip_at(pos0, ctx.member):
pos0 += 1 pos0 += 1
@ -83,7 +83,7 @@ async def skip_to(ctx: Context, args: list[str]) -> None:
seconds = int(s) seconds = int(s)
case _: case _:
raise Explicit('misformatted') raise Explicit('misformatted')
queue = await queue_for(ctx, create=False) queue = await queue_for(ctx, create=False, force_play=False)
queue.queue[0].set_seconds(seconds) queue.queue[0].set_seconds(seconds)
@ -103,7 +103,7 @@ async def effects_(ctx: Context, args: list[str]) -> None:
case _: case _:
raise Explicit('misformatted') raise Explicit('misformatted')
assert_admin(ctx.member) assert_admin(ctx.member)
queue = await queue_for(ctx, create=False) queue = await queue_for(ctx, create=False, force_play=False)
yta = queue.queue[0] yta = queue.queue[0]
seconds = yta.source_seconds() seconds = yta.source_seconds()
yta.options = options_for_effects(effects) yta.options = options_for_effects(effects)
@ -122,17 +122,19 @@ async def queue_(ctx: Context, args: list[str]) -> None:
) )
match args: match args:
case []: case []:
await ctx.long((await (await queue_for(ctx, create=False)).format()).strip() or 'no queue') await ctx.long(
(await (await queue_for(ctx, create=True, force_play=False)).format()).strip() or 'no queue'
)
case ['clear']: case ['clear']:
(await queue_for(ctx, create=False)).clear(ctx.member) (await queue_for(ctx, create=False, force_play=False)).clear(ctx.member)
await ctx.reply('done') await ctx.reply('done')
case ['resume']: case ['resume']:
async with lock_for(ctx.guild, 'not in a guild'): async with lock_for(ctx.guild, 'not in a guild'):
await queue_for(ctx, create=True) await queue_for(ctx, create=True, force_play=True)
await ctx.reply('done') await ctx.reply('done')
case ['pause']: case ['pause']:
async with lock_for(ctx.guild, 'not in a guild'): async with lock_for(ctx.guild, 'not in a guild'):
vc = await vc_for(ctx, create=True) vc = await vc_for(ctx, create=True, force_play=False)
vc.pause() vc.pause()
await ctx.reply('done') await ctx.reply('done')
case _: case _:
@ -149,7 +151,7 @@ async def swap(ctx: Context, args: list[str]) -> None:
match args: match args:
case [a, b] if a.isdecimal() and b.isdecimal(): case [a, b] if a.isdecimal() and b.isdecimal():
a, b = int(a), int(b) a, b = int(a), int(b)
(await queue_for(ctx, create=False)).swap(ctx.member, a, b) (await queue_for(ctx, create=False, force_play=False)).swap(ctx.member, a, b)
case _: case _:
raise Explicit('misformatted') raise Explicit('misformatted')
@ -164,7 +166,7 @@ async def move(ctx: Context, args: list[str]) -> None:
match args: match args:
case [a, b] if a.isdecimal() and b.isdecimal(): case [a, b] if a.isdecimal() and b.isdecimal():
a, b = int(a), int(b) a, b = int(a), int(b)
(await queue_for(ctx, create=False)).move(ctx.member, a, b) (await queue_for(ctx, create=False, force_play=False)).move(ctx.member, a, b)
case _: case _:
raise Explicit('misformatted') raise Explicit('misformatted')
@ -179,20 +181,20 @@ async def volume_(ctx: Context, args: list[str]) -> None:
match args: match args:
case [volume]: case [volume]:
volume = float(volume) volume = float(volume)
await (await main_for(ctx, create=False)).set(volume, ctx.member) await (await main_for(ctx, create=True, force_play=False)).set(volume, ctx.member)
case _: case _:
raise Explicit('misformatted') raise Explicit('misformatted')
@at('commands', 'pause') @at('commands', 'pause')
async def pause(ctx: Context, _args: list[str]) -> None: async def pause(ctx: Context, _args: list[str]) -> None:
vc = await vc_for(ctx, create=False) vc = await vc_for(ctx, create=False, force_play=False)
vc.pause() vc.pause()
@at('commands', 'resume') @at('commands', 'resume')
async def resume(ctx: Context, _args: list[str]) -> None: async def resume(ctx: Context, _args: list[str]) -> None:
vc = await vc_for(ctx, create=False) vc = await vc_for(ctx, create=False, force_play=True)
vc.resume() vc.resume()

View File

@ -3,5 +3,6 @@ import os
from v6d0auth.config import root from v6d0auth.config import root
prefix = os.getenv('v6prefix', '?/') prefix = os.getenv('v6prefix', '?/')
auth_redirect = os.getenv('v6redirect', 'https://music.parrrate.ru/auth/')
myroot = root / 'v6d3music' myroot = root / 'v6d3music'
myroot.mkdir(exist_ok=True) myroot.mkdir(exist_ok=True)

View File

@ -1,5 +1,6 @@
import shlex import shlex
import subprocess import subprocess
import time
from threading import Thread from threading import Thread
from typing import Optional from typing import Optional
@ -37,6 +38,8 @@ class FFmpegNormalAudio(discord.FFmpegAudio):
self._chunk: Optional[bytes] = None self._chunk: Optional[bytes] = None
self._generating = False self._generating = False
self._started = False self._started = False
self._loaded = False
self.loaded_at: Optional[float] = None
def _raw_read(self): def _raw_read(self):
return self._stdout.read(discord.opus.Encoder.FRAME_SIZE) return self._stdout.read(discord.opus.Encoder.FRAME_SIZE)
@ -67,6 +70,9 @@ class FFmpegNormalAudio(discord.FFmpegAudio):
self._generate() self._generate()
return chunk return chunk
def droppable(self) -> bool:
return self._loaded and time.time() - self.loaded_at < 600
def read(self): def read(self):
ret = self._raw_read() ret = self._raw_read()
if len(ret) != discord.opus.Encoder.FRAME_SIZE: if len(ret) != discord.opus.Encoder.FRAME_SIZE:
@ -74,6 +80,8 @@ class FFmpegNormalAudio(discord.FFmpegAudio):
print('poll') print('poll')
return FILL return FILL
return b'' return b''
self.loaded_at = time.time()
self._loaded = True
return ret return ret
def is_opus(self): def is_opus(self):

View File

@ -21,13 +21,15 @@ async def raw_vc_for(ctx: Context) -> discord.VoiceClient:
try: try:
vc: discord.VoiceProtocol = await vch.connect() vc: discord.VoiceProtocol = await vch.connect()
except discord.ClientException: except discord.ClientException:
vc: discord.VoiceProtocol = ctx.guild.voice_client
await ctx.guild.fetch_channels() await ctx.guild.fetch_channels()
await vc.disconnect(force=True)
raise Explicit('try again later') raise Explicit('try again later')
assert isinstance(vc, discord.VoiceClient) assert isinstance(vc, discord.VoiceClient)
return vc return vc
async def main_for_raw_vc(vc: discord.VoiceClient, *, create: bool) -> MainAudio: async def main_for_raw_vc(vc: discord.VoiceClient, *, create: bool, force_play: bool) -> MainAudio:
if vc.guild in mainasrcs: if vc.guild in mainasrcs:
source = mainasrcs[vc.guild] source = mainasrcs[vc.guild]
else: else:
@ -37,26 +39,26 @@ async def main_for_raw_vc(vc: discord.VoiceClient, *, create: bool) -> MainAudio
await MainAudio.create(vc.guild) await MainAudio.create(vc.guild)
) )
else: else:
raise Explicit('not playing') raise Explicit('not playing, use `queue pause` or `queue resume`')
if vc.source != source or create and not vc.is_playing(): if vc.source != source or create and not vc.is_playing() and (force_play or not vc.is_paused()):
vc.play(source) vc.play(source)
return source return source
async def vc_main_for(ctx: Context, *, create: bool) -> tuple[discord.VoiceClient, MainAudio]: async def vc_main_for(ctx: Context, *, create: bool, force_play: bool) -> tuple[discord.VoiceClient, MainAudio]:
vc = await raw_vc_for(ctx) vc = await raw_vc_for(ctx)
return vc, await main_for_raw_vc(vc, create=create) return vc, await main_for_raw_vc(vc, create=create, force_play=force_play)
async def vc_for(ctx: Context, *, create: bool) -> discord.VoiceClient: async def vc_for(ctx: Context, *, create: bool, force_play: bool) -> discord.VoiceClient:
vc, source = await vc_main_for(ctx, create=create) vc, source = await vc_main_for(ctx, create=create, force_play=force_play)
return vc return vc
async def main_for(ctx: Context, *, create: bool) -> MainAudio: async def main_for(ctx: Context, *, create: bool, force_play: bool) -> MainAudio:
vc, source = await vc_main_for(ctx, create=create) vc, source = await vc_main_for(ctx, create=create, force_play=force_play)
return source return source
async def queue_for(ctx: Context, *, create: bool) -> QueueAudio: async def queue_for(ctx: Context, *, create: bool, force_play: bool) -> QueueAudio:
return (await main_for(ctx, create=create)).queue return (await main_for(ctx, create=create, force_play=force_play)).queue

View File

@ -11,14 +11,23 @@ from v6d3music.utils.assert_admin import assert_admin
from v6d3music.utils.fill import FILL from v6d3music.utils.fill import FILL
queue_db = Db(myroot / 'queue.db', kvrequest_type=KVJson) queue_db = Db(myroot / 'queue.db', kvrequest_type=KVJson)
PRE_SET_LENGTH = 24
class QueueAudio(discord.AudioSource): class QueueAudio(discord.AudioSource):
def __init__(self, guild: discord.Guild, respawned: list[YTAudio]): def __init__(self, guild: discord.Guild, respawned: list[YTAudio]):
self.queue: deque[YTAudio] = deque() self.queue: deque[YTAudio] = deque()
self.queue.extend(respawned) for audio in respawned:
self.append(audio)
self.guild = guild self.guild = guild
def _update_sources(self):
for i in range(PRE_SET_LENGTH):
try:
self.queue[i].set_source_if_necessary()
except IndexError:
return
@staticmethod @staticmethod
async def respawned(guild: discord.Guild) -> list[YTAudio]: async def respawned(guild: discord.Guild) -> list[YTAudio]:
respawned = [] respawned = []
@ -45,15 +54,19 @@ class QueueAudio(discord.AudioSource):
queue_db.set_nowait(self.guild.id, hybernated) queue_db.set_nowait(self.guild.id, hybernated)
def append(self, audio: YTAudio): def append(self, audio: YTAudio):
if len(self.queue) < PRE_SET_LENGTH:
audio.set_source_if_necessary()
self.queue.append(audio) self.queue.append(audio)
def read(self) -> bytes: def read(self) -> bytes:
if not self.queue: if not self.queue:
return FILL return FILL
audio = self.queue[0] audio = self.queue[0]
audio.set_source_if_necessary()
frame = audio.read() frame = audio.read()
if len(frame) != discord.opus.Encoder.FRAME_SIZE: if len(frame) != discord.opus.Encoder.FRAME_SIZE:
self.queue.popleft().cleanup() self.queue.popleft().cleanup()
self._update_sources()
frame = FILL frame = FILL
return frame return frame
@ -72,6 +85,7 @@ class QueueAudio(discord.AudioSource):
self.queue.remove(audio) self.queue.remove(audio)
audio.cleanup() audio.cleanup()
return True return True
self._update_sources()
return False return False
def clear(self, member: discord.Member) -> None: def clear(self, member: discord.Member) -> None:
@ -83,6 +97,7 @@ class QueueAudio(discord.AudioSource):
if max(a, b) >= len(self.queue): if max(a, b) >= len(self.queue):
return return
self.queue[a], self.queue[b] = self.queue[b], self.queue[a] self.queue[a], self.queue[b] = self.queue[b], self.queue[a]
self._update_sources()
def move(self, member: discord.Member, a: int, b: int) -> None: def move(self, member: discord.Member, a: int, b: int) -> None:
assert_admin(member) assert_admin(member)
@ -91,6 +106,7 @@ class QueueAudio(discord.AudioSource):
audio = self.queue[a] audio = self.queue[a]
self.queue.remove(audio) self.queue.remove(audio)
self.queue.insert(b, audio) self.queue.insert(b, audio)
self._update_sources()
async def format(self) -> str: async def format(self) -> str:
stream = StringIO() stream = StringIO()

View File

@ -14,7 +14,7 @@ from v6d3music.utils.sparq import sparq
class YTAudio(discord.AudioSource): class YTAudio(discord.AudioSource):
source: discord.FFmpegAudio source: FFmpegNormalAudio
def __init__( def __init__(
self, self,
@ -24,7 +24,7 @@ class YTAudio(discord.AudioSource):
options: Optional[str], options: Optional[str],
rby: discord.Member, rby: discord.Member,
already_read: int, already_read: int,
tor: bool tor: bool,
): ):
self.url = url self.url = url
self.origin = origin self.origin = origin
@ -33,12 +33,15 @@ class YTAudio(discord.AudioSource):
self.rby = rby self.rby = rby
self.already_read = already_read self.already_read = already_read
self.tor = tor self.tor = tor
self.loaded = False
self.regenerating = False self.regenerating = False
self.set_source() # self.set_source()
self._durations: dict[str, str] = {} self._durations: dict[str, str] = {}
self.loop = asyncio.get_running_loop() self.loop = asyncio.get_running_loop()
def set_source_if_necessary(self):
if not hasattr(self, 'source'):
self.set_source()
def set_source(self): def set_source(self):
self.schedule_duration_update() self.schedule_duration_update()
self.source = FFmpegNormalAudio( self.source = FFmpegNormalAudio(
@ -64,8 +67,11 @@ class YTAudio(discord.AudioSource):
hours, minutes = divmod(minutes, 60) hours, minutes = divmod(minutes, 60)
return f'{hours}:{minutes:02d}:{seconds:02d}' return f'{hours}:{minutes:02d}:{seconds:02d}'
def _schedule_duration_update(self):
self.loop.create_task(self.update_duration())
def schedule_duration_update(self): def schedule_duration_update(self):
asyncio.get_running_loop().create_task(self.update_duration()) self.loop.call_soon_threadsafe(self._schedule_duration_update)
async def update_duration(self): async def update_duration(self):
url: str = self.url url: str = self.url
@ -114,9 +120,7 @@ class YTAudio(discord.AudioSource):
return FILL return FILL
self.already_read += 1 self.already_read += 1
ret: bytes = self.source.read() ret: bytes = self.source.read()
if ret: if not ret and not self.source.droppable():
self.loaded = True
elif not self.loaded:
if random.random() > .1: if random.random() > .1:
self.regenerating = True self.regenerating = True
self.loop.create_task(self.regenerate()) self.loop.create_task(self.regenerate())
@ -126,7 +130,8 @@ class YTAudio(discord.AudioSource):
return ret return ret
def cleanup(self): def cleanup(self):
self.source.cleanup() if hasattr(self, 'source'):
self.source.cleanup()
def can_be_skipped_by(self, member: discord.Member) -> bool: def can_be_skipped_by(self, member: discord.Member) -> bool:
permissions: discord.Permissions = member.guild_permissions permissions: discord.Permissions = member.guild_permissions
@ -170,7 +175,8 @@ class YTAudio(discord.AudioSource):
try: try:
print(f'regenerating {self.origin}') print(f'regenerating {self.origin}')
self.url = await real_url(self.origin, True, self.tor) self.url = await real_url(self.origin, True, self.tor)
self.source.cleanup() if hasattr(self, 'source'):
self.source.cleanup()
self.set_source() self.set_source()
print(f'regenerated {self.origin}') print(f'regenerated {self.origin}')
finally: finally:

View File

@ -81,7 +81,8 @@ const pageHome = async () => {
'div', 'div',
baseEl('div', aLogin()), baseEl('div', aLogin()),
baseEl('div', await userAvatarImg()), baseEl('div', await userAvatarImg()),
await userId() baseEl('div', await userId()),
baseEl('div', await userUsername()),
) )
}; };
let authbase; let authbase;

View File

@ -1,6 +1,7 @@
import asyncio import asyncio
import os import os
import subprocess import subprocess
import sys
import time import time
import discord import discord
@ -52,7 +53,7 @@ async def restore_vcs():
vp: discord.VoiceProtocol = await channel.connect() vp: discord.VoiceProtocol = await channel.connect()
assert isinstance(vp, discord.VoiceClient) assert isinstance(vp, discord.VoiceClient)
vc = vp vc = vp
await main_for_raw_vc(vc, create=True) await main_for_raw_vc(vc, create=True, force_play=True)
if vc_is_paused: if vc_is_paused:
vc.pause() vc.pause()
except Exception as e: except Exception as e:
@ -78,7 +79,6 @@ async def on_ready():
@client.event @client.event
async def on_message(message: discord.Message) -> None: async def on_message(message: discord.Message) -> None:
print('on message')
await handle_content(message, message.content, prefix) await handle_content(message, message.content, prefix)
@ -126,11 +126,24 @@ async def setup_tasks():
async def main(): async def main():
async with volume_db, queue_db, cache_db, session_db: async with volume_db, queue_db, cache_db, session_db:
await client.login(await request_token('music', 'token')) if 'guerilla' in sys.argv:
from pathlib import Path
tokenpath = Path('.token.txt')
if tokenpath.exists():
token = tokenpath.read_text()
else:
token = input('token:')
tokenpath.write_text(token)
else:
token = await request_token('music', 'token')
await client.login(token)
loop.create_task(setup_tasks()) loop.create_task(setup_tasks())
if os.getenv('v6monitor'): if os.getenv('v6monitor'):
loop.create_task(monitor()) loop.create_task(monitor())
subprocess.Popen('tor') try:
subprocess.Popen('tor')
except FileNotFoundError:
print('no tor')
await client.connect() await client.connect()

View File

@ -8,7 +8,7 @@ def extract(params: dict, url: str, kwargs: dict):
if 'entries' in extracted: if 'entries' in extracted:
extracted['entries'] = list(extracted['entries']) extracted['entries'] = list(extracted['entries'])
return extracted return extracted
except (youtube_dl.utils.ExtractorError, youtube_dl.utils.DownloadError) as e: except Exception as e:
msg = str(e) msg = str(e)
msg = discord.utils.escape_markdown(msg) msg = discord.utils.escape_markdown(msg)
msg = msg.replace('\x1b[0;31m', '__') msg = msg.replace('\x1b[0;31m', '__')