mirror of
https://github.com/DerTyp7/docker_minecraft_server_auto_starter.git
synced 2025-10-29 12:42:09 +01:00
98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
import os
|
|
import socket
|
|
import threading
|
|
from utils import docker_container_mapping
|
|
|
|
|
|
class RequestHandler(threading.Thread):
|
|
def __init__(self, port, docker_handler):
|
|
super().__init__()
|
|
self.port = port
|
|
self.docker_handler = docker_handler
|
|
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
server_address = ('localhost', self.port)
|
|
print(f'Starting up on {server_address[0]} port {server_address[1]}')
|
|
self.sock.bind(server_address)
|
|
self.sock.listen(1)
|
|
|
|
def restart(self):
|
|
print(f'Restarting request handler for port {self.port}')
|
|
self.sock.close()
|
|
self.__init__(self.port, self.docker_handler)
|
|
|
|
def run(self):
|
|
while True:
|
|
try:
|
|
print(f'Waiting for a connection on port {self.port}')
|
|
self.connection, self.client_address = self.sock.accept()
|
|
try:
|
|
print('Connection from', self.client_address)
|
|
self.handle_request()
|
|
except Exception as e:
|
|
print(
|
|
f'Error in request handler for port {self.port}: {e}')
|
|
print('Restarting request handler...')
|
|
self.restart()
|
|
finally:
|
|
self.connection.close()
|
|
self.restart()
|
|
except Exception as e:
|
|
print(f'Error in request handler for port {self.port}: {e}')
|
|
print('Restarting request handler...')
|
|
self.restart()
|
|
|
|
def handle_request(self):
|
|
print(f'Handling request on port {self.port}')
|
|
container_ip = docker_container_mapping().get(str(self.port))
|
|
if container_ip:
|
|
container = self.docker_handler.get_container_by_ip(
|
|
container_ip)
|
|
isStarting = self.docker_handler.is_container_starting(container)
|
|
request = self.connection.recv(1024)
|
|
print(f'Received request: {request}')
|
|
if request[0] == 0x10 or request[0] == 0x15:
|
|
if b'\x01' in request:
|
|
print(f'Detected ping request for {container_ip}')
|
|
self.forward_request_to_placeholder(request, isStarting)
|
|
elif b'\x02' in request:
|
|
print(f'Detected join/login request for {container_ip}')
|
|
if isStarting:
|
|
print(
|
|
f'Container {container_ip} is already starting...')
|
|
self.forward_request_to_placeholder(
|
|
request, isStarting)
|
|
else:
|
|
print(f'Starting container {container_ip}')
|
|
container.start()
|
|
|
|
elif request[0] == 0xFE:
|
|
print(f'Detected legacy ping request for {container_ip}')
|
|
self.forward_request_to_placeholder(request, isStarting)
|
|
else:
|
|
print(f'Detected unknown request for {container_ip}')
|
|
self.forward_request_to_placeholder(request, isStarting)
|
|
|
|
else:
|
|
print(f'No container mapped to port {self.port}')
|
|
|
|
def forward_request_to_placeholder(self, request, isStarting=False):
|
|
print('Forwarding request to placeholder server')
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:
|
|
ip = os.environ.get('PLACEHOLDER_SERVER_SLEEPING_IP')
|
|
if isStarting:
|
|
print('Container is starting. Using starting placeholder IP')
|
|
ip = os.environ.get('PLACEHOLDER_SERVER_STARTING_IP')
|
|
|
|
if not ip:
|
|
print('No placeholder server IP found')
|
|
return
|
|
try:
|
|
server_socket.connect((ip, 25565))
|
|
server_socket.sendall(request)
|
|
response = server_socket.recv(1024)
|
|
self.connection.sendall(response)
|
|
except Exception as e:
|
|
print(f'Error while handling request on port {self.port}: {e}')
|
|
self.restart()
|