Matrix-Mirrors/pantalaimon
1
0
Fork 0
mirror of https://github.com/matrix-org/pantalaimon.git synced 2025-02-07 02:25:23 -05:00
Code Issues Packages Projects Releases Wiki Activity

daemon: Refactor out the error responses.

Browse Source
This commit is contained in:
Damir Jelić 2019-03-26 12:59:46 +01:00
parent 1e71a945a8
commit ea9a30367a
1 changed files with 38 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
pantalaimon/daemon.py
View File

@ -10,9 +10,12 @@ from aiohttp import web, ClientSession
from nio import (
AsyncClient,
LoginResponse,
KeysQueryResponse,
GroupEncryptionError,
RoomEncryptedEvent,
MegolmEvent,
EncryptionError
EncryptionError,
SyncResponse
)
from appdirs import user_data_dir
from json import JSONDecodeError
@ -145,28 +148,46 @@ class ProxyDaemon:
text=await response.transport_response.text()
)
async def sync(self, request):
access_token = self.get_access_token(request)
@property
def _missing_token(self):
return web.Response(
status=401,
text=json.dumps({
"errcode": "M_MISSING_TOKEN",
"error": "Missing access token."
})
)
if not access_token:
return web.Response(
status=401,
text=json.dumps({
"errcode": "M_MISSING_TOKEN",
"error": "Missing access token."
})
)
try:
client = self.client_sessions[access_token]
except KeyError:
return web.Response(
@property
def _uknown_token(self):
return web.Response(
status=401,
text=json.dumps({
"errcode": "M_UNKNOWN_TOKEN",
"error": "Unrecognised access token."
})
)
)
@property
def _not_json(self):
return web.Response(
status=400,
text=json.dumps({
"errcode": "M_NOT_JSON",
"error": "Request did not contain valid JSON."
})
)
async def sync(self, request):
access_token = self.get_access_token(request)
if not access_token:
return self._missing_token
try:
client = self.client_sessions[access_token]
except KeyError:
return self._uknown_token
sync_filter = request.query.get("filter", None)
timeout = request.query.get("timeout", None)