64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from typing import TYPE_CHECKING, Tuple, Type
|
|
|
|
from sampy.client.player import Player
|
|
from sampy.config import Config
|
|
|
|
if TYPE_CHECKING:
|
|
from sampy.network.protocol import Protocol
|
|
|
|
|
|
class UDPProtocol:
|
|
transport: asyncio.transports.DatagramTransport
|
|
|
|
def __init__(self, protocol: Protocol, local_addr: Tuple[str, int]):
|
|
self.protocol = protocol
|
|
self.local_addr = local_addr
|
|
|
|
def start(self):
|
|
loop = asyncio.get_event_loop()
|
|
connect = loop.create_datagram_endpoint(
|
|
lambda: self,
|
|
local_addr=self.local_addr,
|
|
)
|
|
loop.run_until_complete(connect)
|
|
|
|
def stop(self): # TODO: Shutdown code
|
|
if self.transport is None:
|
|
raise Exception("Cannot stop a server that hasn't been started")
|
|
|
|
self.transport.close()
|
|
|
|
def connection_made(self, transport: asyncio.transports.DatagramTransport):
|
|
self.transport = transport
|
|
|
|
def datagram_received(self, data: bytes, addr: Tuple[str, int]):
|
|
raise NotImplementedError
|
|
|
|
def sendto(self, data: bytes | bytearray | memoryview, addr: Tuple[str, int]):
|
|
self.transport.sendto(data, addr)
|
|
|
|
|
|
class Server(UDPProtocol):
|
|
config: Config
|
|
|
|
def __init__(self, protocol: Type[Protocol], config: Config = Config()):
|
|
super().__init__(
|
|
protocol=protocol(),
|
|
local_addr=(
|
|
config.get("sampy", "host"),
|
|
config.getint("sampy", "port"),
|
|
),
|
|
)
|
|
self.config = config
|
|
|
|
def datagram_received(self, data: bytes, addr: Tuple[str, int]):
|
|
loop = asyncio.get_event_loop()
|
|
loop.create_task(self.protocol.on_packet(self, data, addr))
|
|
|
|
@property
|
|
def players(self) -> list[Player]: # TODO
|
|
return []
|