119 lines
4.1 KiB
Python
119 lines
4.1 KiB
Python
from giants.map import Map
|
|
from giants.player import Player
|
|
from dpnet.netserver import Netserver
|
|
from dpnet.session import Session
|
|
import socket
|
|
import importlib
|
|
import os
|
|
from giants import Teams, GameTypes
|
|
from utils.logger import setup_logger
|
|
import traceback
|
|
import asyncio
|
|
|
|
logger = setup_logger(__name__)
|
|
|
|
|
|
class Server:
|
|
def __init__(self, **kwargs):
|
|
self.listen_ip = kwargs.get("ip", "0.0.0.0")
|
|
self.listen_port = kwargs.get("port", 19711)
|
|
self.register_with_ms = kwargs.get("register", False)
|
|
self.teams = kwargs.get("teams", Teams.MvM)
|
|
self.game_type = kwargs.get("gametype", GameTypes.TeamDeathmatchWithFullBase)
|
|
self.currentmap = kwargs.get("map", Map("Three Way Island - Canyons.gck"))
|
|
self.maxplayers = kwargs.get("maxplayers", 20)
|
|
self.name = kwargs.get("name", "Default Server Name")
|
|
|
|
fake_session = Session(socket.socket(socket.AF_INET, socket.SOCK_DGRAM))
|
|
fake_session.ip = "127.0.0.1"
|
|
fake_session.port = 3333
|
|
self.players = [Player("[Server]", fake_session)]
|
|
|
|
self.accept_new_players = True
|
|
self.version = 1.497
|
|
self.points_per_kill = 1
|
|
self.points_per_capture = 5
|
|
self.detente_time = 0 # minutes
|
|
self._plugins = []
|
|
self.tempplayers = []
|
|
self.running = True
|
|
self.ticks = 60
|
|
|
|
def update(self):
|
|
#logger.debug("Calling update")
|
|
pass
|
|
|
|
async def add_player(self, player):
|
|
self.tempplayers.remove(player)
|
|
self.players.append(player)
|
|
await self._broadcast_event("on_player_join", player)
|
|
|
|
# todo: remove
|
|
def create_temp_player(self, player):
|
|
self.tempplayers.append(player)
|
|
|
|
async def remove_player(self, player):
|
|
self.players.remove(player)
|
|
await self._broadcast_event("on_player_left", player)
|
|
|
|
def add_plugin(self, plugin):
|
|
self._plugins.append(plugin)
|
|
|
|
async def change_map(self, mappath):
|
|
try:
|
|
self.currentmap = Map(mappath)
|
|
await self._broadcast_event("on_map_change", self.currentmap)
|
|
except Exception:
|
|
logger.error("Could not change map")
|
|
|
|
async def _broadcast_event(self, event, *args):
|
|
logger.debug("Broadcasting event "+event)
|
|
for plugin in self._plugins:
|
|
if hasattr(plugin, event):
|
|
func = getattr(plugin, event)
|
|
if callable(func):
|
|
try:
|
|
await func(*args)
|
|
except Exception:
|
|
logger.error("Could not call plugin function: "+plugin.__class__.__name__+"."+event)
|
|
traceback.print_exc()
|
|
|
|
def load_plugins(self):
|
|
plugins = os.listdir("plugins")
|
|
for plugin in plugins:
|
|
if os.path.isdir("plugins/"+plugin) or plugin == "__init__.py":
|
|
continue
|
|
if not plugin.endswith(".py"):
|
|
continue
|
|
pluginname = plugin.split(".py")[0]
|
|
try:
|
|
module = importlib.import_module("plugins."+pluginname)
|
|
if hasattr(module, "setup"):
|
|
setup = getattr(module, "setup")
|
|
if callable(setup):
|
|
logger.info("Loading plugin "+module.__name__)
|
|
module.setup(self)
|
|
except Exception:
|
|
logger.warning("Could not load plugin "+plugin)
|
|
traceback.print_exc()
|
|
|
|
async def broadcast_message(self, text):
|
|
for player in self.players:
|
|
if player.name == "[Server]":
|
|
continue
|
|
await player.session.send_gamedata(b'\x35\x80\x81'+text.encode("ascii")+b'\x00\x00')
|
|
|
|
if __name__ == '__main__':
|
|
server = Server(name="giantsd", maxplayers=20, register=False)
|
|
server.load_plugins()
|
|
|
|
loop = asyncio.get_event_loop()
|
|
listen = loop.create_datagram_endpoint(lambda: Netserver(server), local_addr=(server.listen_ip, server.listen_port))
|
|
transport, protocol = loop.run_until_complete(listen)
|
|
try:
|
|
loop.run_forever()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
transport.close()
|
|
loop.close()
|