first commit

app/app.py

@@ -0,0 +1,134 @@
+import socket
+import threading
+import os
+import docker
+
+client = docker.DockerClient(base_url='unix://var/run/docker.sock')
+
+
+def setup_nginx_config():
+    # Stop nginx
+    os.system('nginx -s stop')
+    # open nginx.conf for write create if not exist
+    nginx_conf = open('/etc/nginx/nginx.conf', 'w+')
+    nginx_conf.truncate()
+    nginx_conf.write('events { }\n')
+    nginx_conf.write('stream {\n')
+
+    port_ip_map = docker_container_mapping()
+    for port in port_ip_map:
+        nginx_conf.write('    upstream upstream_{} {{\n'.format(port))
+        nginx_conf.write('        server {}:{};\n'.format(
+            port_ip_map[port], port))
+        nginx_conf.write('        server 127.0.0.1:{} backup;\n'.format(port))
+        nginx_conf.write('    }\n')
+
+        nginx_conf.write('    server {\n')
+        nginx_conf.write('        listen {}:{};\n'.format(
+            get_current_container_ip(), port))
+        nginx_conf.write('        proxy_pass upstream_{};\n'.format(port))
+        nginx_conf.write('    }\n')
+    nginx_conf.write('}\n')
+    nginx_conf.close()
+    os.system('nginx > /dev/null 2>&1 &')
+
+
+def get_current_container_ip():
+    # Get IP of current container
+    current_container_name = os.environ.get('HOSTNAME')
+    current_container = client.containers.get(current_container_name)
+    networks = current_container.attrs['NetworkSettings']['Networks']
+    current_network = list(networks.keys())[0]
+    current_container_ip = networks[current_network]['IPAddress']
+    return current_container_ip
+
+
+def get_current_network():
+    # Get network of current container
+    current_container_name = os.environ.get('HOSTNAME')
+    current_container = client.containers.get(current_container_name)
+    networks = current_container.attrs['NetworkSettings']['Networks']
+    current_network = list(networks.keys())[0]
+    return current_network
+
+
+def get_docker_container_by_ip(ip):
+    current_network = get_current_network()
+    print('getting docker container by ip {} in network {}'.format(
+        ip, current_network))
+    containers = client.containers.list(all=True)
+
+    for container in containers:
+        print('current network: {}'.format(current_network))
+        print('checking container {}'.format(container.name))
+        print('container networks: {}'.format(
+            container.attrs['NetworkSettings']['Networks']))
+        print('container ip: {}'.format(
+            container.attrs['NetworkSettings']['Networks'][current_network]['IPAMConfig']['IPv4Address']))
+
+        networks = container.attrs['NetworkSettings']['Networks']
+        if current_network in networks and networks[current_network]['IPAMConfig']['IPv4Address'] == ip:
+            print('found docker container {} with ip {} in network {}'.format(
+                container.name, ip, current_network))
+            return container
+    print('no docker container found with ip {} in network {}'.format(
+        ip, current_network))
+    return None
+
+
+def docker_container_mapping():
+    # Get the PORT_IP_MAP environment variable
+    port_ip_map_str = os.getenv('PORT_IP_MAP')
+
+    # Convert the environment variable to a Python dictionary
+    port_ip_map = {}
+    for line in port_ip_map_str.split('\n'):
+        if line:  # ignore empty lines
+            port, ip = line.split(':')
+            port_ip_map[port.strip()] = ip.strip()
+
+    return port_ip_map
+
+
+class RequestHandler(threading.Thread):
+    def __init__(self, port):
+        super().__init__()
+        self.port = port
+        # 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))
+        # Get the Docker container name for this port
+        container_ip = docker_container_mapping().get(str(self.port))
+        if container_ip:
+            # Start the Docker container
+            print('starting docker container {}'.format(container_ip))
+            get_docker_container_by_ip(container_ip).start()
+            # Send a response
+            self.connection.sendall(b'Request handled')
+        else:
+            print('no docker container mapped to this port')
+
+
+for port in range(25560, 25571):
+    handler = RequestHandler(port)
+    handler.start()
+
+setup_nginx_config()