velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

voice.manager: add VST_REQ call

This commit is contained in:
Luna 2019-03-04 21:32:04 -03:00
parent e2af6b6370
commit 00100f9abb
1 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
litecord/voice/manager.py
View File

@ -17,7 +17,7 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
from typing import Tuple, Dict from typing import Tuple, Dict, List
from collections import defaultdict from collections import defaultdict
from dataclasses import fields from dataclasses import fields
@ -155,16 +155,20 @@ class VoiceManager:
await self.del_state(old_voice_key) await self.del_state(old_voice_key)
await self.create_state(old_voice_key, {'channel_id': channel_id}) await self.create_state(old_voice_key, {'channel_id': channel_id})
async def _lvsp_info_guild(self, guild_id, info_type, info_data):
hostname = await self.lvsp.get_server(guild_id)
conn = self.lvsp.get_conn(hostname)
# TODO: impl send_info
await conn.send_info(info_type, info_data)
async def _create_ctx_guild(self, guild_id, channel_id): async def _create_ctx_guild(self, guild_id, channel_id):
# get a voice server
server = await self.lvsp.get_server(guild_id)
conn = self.lvsp.get_conn(server)
chan = await self.app.storage.get_channel(channel_id) chan = await self.app.storage.get_channel(channel_id)
# TODO: this, but properly # TODO: this, but properly
# TODO: when the server sends a reply to CHAN_REQ, we need to update # TODO: when the server sends a reply to CHAN_REQ, we need to update
# LVSPManager.guild_servers. # LVSPManager.guild_servers.
await conn.send_info('CHAN_REQ', { await self._lvsp_info_guild(guild_id, 'CHAN_REQ', {
'guild_id': str(guild_id), 'guild_id': str(guild_id),
'channel_id': str(channel_id), 'channel_id': str(channel_id),
@ -180,11 +184,18 @@ class VoiceManager:
existing_states = self.states[voice_key] existing_states = self.states[voice_key]
channel_exists = any( channel_exists = any(
state.channel_id == channel_id for state in existing_states) state.channel_id == channel_id for state in existing_states
)
if not channel_exists: if not channel_exists:
await self._create_ctx_guild(guild_id, channel_id) await self._create_ctx_guild(guild_id, channel_id)
await self._lvsp_info_guild(guild_id, 'VST_REQ', {
'user_id': str(user_id),
'guild_id': str(guild_id),
'channel_id': str(channel_id),
})
async def create_state(self, voice_key: VoiceKey, data: dict): async def create_state(self, voice_key: VoiceKey, data: dict):
"""Creates (or tries to create) a voice state. """Creates (or tries to create) a voice state.
@ -208,17 +219,17 @@ class VoiceManager:
ctype = ChannelType(ctype) ctype = ChannelType(ctype)
if ctype == ChannelType.GROUP_DM: if ctype == ChannelType.GROUP_DM:
# await self._start_voice_dm() # await self._start_voice_dm(voice_key)
pass pass
elif ctype == ChannelType.DM: elif ctype == ChannelType.DM:
# await self._start_voice_gdm() # await self._start_voice_gdm(voice_key)
pass pass
return return
# if guild found, then data.channel_id exists, and we treat it # if guild found, then data.channel_id exists, and we treat it
# as a guild # as a guild
# await self._start_voice_guild() await self._start_voice_guild(voice_key, data)
async def leave_all(self, user_id: int) -> int: async def leave_all(self, user_id: int) -> int:
"""Leave all voice channels.""" """Leave all voice channels."""
@ -243,7 +254,7 @@ class VoiceManager:
"""Make a user leave a channel IN A GUILD.""" """Make a user leave a channel IN A GUILD."""
await self.del_state((guild_id, user_id)) await self.del_state((guild_id, user_id))
async def voice_server_list(self, region: str): async def voice_server_list(self, region: str) -> List[dict]:
"""Get a list of voice server objects""" """Get a list of voice server objects"""
rows = await self.app.db.fetch(""" rows = await self.app.db.fetch("""
SELECT hostname, last_health SELECT hostname, last_health