mirror of https://gitlab.com/litecord/litecord.git
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
"""
|
|
|
|
Litecord
|
|
Copyright (C) 2018-2019 Luna Mendes
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, version 3 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
|
|
import urllib.parse
|
|
from litecord.gateway.websocket import GatewayWebsocket
|
|
|
|
|
|
async def websocket_handler(app, ws, url):
|
|
"""Main websocket handler, checks query arguments when connecting to
|
|
the gateway and spawns a GatewayWebsocket instance for the connection."""
|
|
args = urllib.parse.parse_qs(
|
|
urllib.parse.urlparse(url).query
|
|
)
|
|
|
|
# pull a dict.get but in a really bad way.
|
|
try:
|
|
gw_version = args['v'][0]
|
|
except (KeyError, IndexError):
|
|
gw_version = '6'
|
|
|
|
try:
|
|
gw_encoding = args['encoding'][0]
|
|
except (KeyError, IndexError):
|
|
gw_encoding = 'json'
|
|
|
|
if gw_version not in ('6', '7'):
|
|
return await ws.close(1000, 'Invalid gateway version')
|
|
|
|
if gw_encoding not in ('json', 'etf'):
|
|
return await ws.close(1000, 'Invalid gateway encoding')
|
|
|
|
try:
|
|
gw_compress = args['compress'][0]
|
|
except (KeyError, IndexError):
|
|
gw_compress = None
|
|
|
|
if gw_compress and gw_compress not in ('zlib-stream',):
|
|
return await ws.close(1000, 'Invalid gateway compress')
|
|
|
|
gws = GatewayWebsocket(
|
|
ws, app, v=gw_version, encoding=gw_encoding, compress=gw_compress)
|
|
|
|
# this can be run with a single await since this whole coroutine
|
|
# is already running in the background.
|
|
await gws.run()
|