SantaSpeen/KuiToi-Server
1
0
Fork 0
mirror of https://github.com/kuitoi/kuitoi-Server.git synced 2026-04-23 00:30:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prepare for Upload mods

Browse Source
This commit is contained in:
Maxim Khomutov
2023-07-13 02:35:38 +03:00
parent d21798aaf1
commit 6e46af4c13
4 changed files with 40 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/core/tcp_server.py
View File

@@ -68,22 +68,29 @@ class TCPServer:
return True, client
async def handle_download(self, writer):
# TODO: HandleDownload
self.log.debug(f"Client: \"IP: {0!r}; ID: {0}\" - HandleDownload!")
return False
async def set_down_rw(self, reader, writer):
try:
cid = (await reader.read(1)).decode() # FIXME: wtf? 1 byte?
self.log.debug(f"Client: \"ID: {cid}\" - HandleDownload!")
if not cid.isdigit():
return False
for _client in self.Core.clients:
if _client.cid == cid:
_client.down_rw = (reader, writer)
return True
finally:
return False
async def handle_code(self, code, reader, writer):
match code:
case "C":
result, client = await self.auth_client(reader, writer)
if result:
await client.sync_resources()
# await client.kick("Authentication success! Server not ready.")
await client.looper()
return True
return False
case "D":
return await self.handle_download(writer)
return await self.set_down_rw(reader, writer)
case "P":
writer.write(b"P")
await writer.drain()