Browse Source

Add welcome message and PING handling

master
JustAnotherArchivist 4 years ago
parent
commit
a1b4969ccf
1 changed files with 13 additions and 0 deletions
  1. +13
    -0
      testserver.py

+ 13
- 0
testserver.py View File

@@ -9,6 +9,7 @@ logging.basicConfig(level = logging.INFO, format = '{asctime} {levelname} {messa
class FakeIRCServerProtocol(asyncio.Protocol):
def __init__(self, loop = None):
self.loop = loop or asyncio.get_event_loop()
self.buffer = b''

def send(self, data):
logging.info(f'Send: {data!r}')
@@ -18,6 +19,7 @@ class FakeIRCServerProtocol(asyncio.Protocol):
logging.info(f'Connection from {transport.get_extra_info("peername")}')
self.transport = transport
self.connected = True
self.send(b'001 :Hello there!\r\n')
asyncio.create_task(self.pingloop())

async def pingloop(self):
@@ -27,6 +29,17 @@ class FakeIRCServerProtocol(asyncio.Protocol):

def data_received(self, data):
logging.info(f'Data received: {data!r}')
messages = data.split(b'\r\n')
if self.buffer:
self.message_received(self.buffer + messages[0])
messages = messages[1:]
for message in messages[:-1]:
self.message_received(message)
self.buffer = messages[-1]

def message_received(self, message):
if message.startswith(b'PING '):
self.send(b'PONG ' + message[5:] + b'\r\n')

def connection_lost(self, exc):
logging.info(f'Connection to {self.transport.get_extra_info("peername")} lost')


Loading…
Cancel
Save