mirror of
https://github.com/DerTyp7/docker_minecraft_server_auto_starter.git
synced 2025-10-29 12:42:09 +01:00
71 lines
2.8 KiB
Python
71 lines
2.8 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
|
|
# Create a TCP/IP socket
|
|
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
# Bind the socket to the port
|
|
server_address = ('localhost', self.port)
|
|
print('starting up on {} port {}'.format(*server_address))
|
|
self.sock.bind(server_address)
|
|
# Listen for incoming connections
|
|
self.sock.listen(1)
|
|
|
|
def run(self):
|
|
while True:
|
|
print('waiting for a connection on port {}'.format(self.port))
|
|
self.connection, self.client_address = self.sock.accept()
|
|
try:
|
|
print('connection from', self.client_address)
|
|
self.handle_request()
|
|
finally:
|
|
self.connection.close()
|
|
|
|
def handle_request(self):
|
|
print('handling request on port {}'.format(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)
|
|
if request[0] == 0x10:
|
|
if b'\x01' in request:
|
|
print('ping')
|
|
self.forward_request_to_server(request, isStarting)
|
|
elif b'\x02' in request:
|
|
print('join')
|
|
if isStarting:
|
|
print('container is starting, waiting for 5 seconds')
|
|
self.forward_request_to_server(request, isStarting)
|
|
else:
|
|
print('starting docker container {}'.format(container_ip))
|
|
container.start()
|
|
|
|
elif request[0] == 0xFE:
|
|
print('legacy server list ping')
|
|
self.forward_request_to_server(request, isStarting)
|
|
|
|
else:
|
|
print('no docker container mapped to this port')
|
|
|
|
def forward_request_to_server(self, request, isStarting=False):
|
|
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, waiting for 5 seconds')
|
|
ip = os.environ.get('PLACEHOLDER_SERVER_STARTING_IP')
|
|
|
|
server_socket.connect(
|
|
(ip, 25565))
|
|
server_socket.sendall(request)
|
|
response = server_socket.recv(1024)
|
|
self.connection.sendall(response)
|