From ca420b50701342c50d49a87d3b20b74719e0b791 Mon Sep 17 00:00:00 2001
From: Jonatan Rek <gamerclassn7@gmail.com>
Date: Tue, 21 Jan 2025 11:13:51 +0100
Subject: [PATCH] Initial commit

---
 Dockerfile       | 25 ++++++++++++++
 app.py           | 86 ++++++++++++++++++++++++++++++++++++++++++++++++
 config.yml       |  8 +++++
 requirements.txt |  1 +
 4 files changed, 120 insertions(+)
 create mode 100644 Dockerfile
 create mode 100644 app.py
 create mode 100644 config.yml
 create mode 100644 requirements.txt

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..5e33fc6
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,25 @@
+# Use PHP with Apache as the base image
+FROM python:3.9
+
+ENV PYTHONUNBUFFERED=1
+
+# Install Additional System Dependencies
+RUN apt-get update && apt-get upgrade -y
+RUN apt-get install -y \
+    libgl1 \
+    nano
+
+# Clear cache
+RUN apt-get autoremove
+RUN apt-get clean && rm -rf /var/lib/apt/lists/*
+
+#Copy Project
+COPY . /app
+
+# Install any needed dependencies specified in requirements.txt
+RUN pip install --no-cache-dir -r /app/requirements.txt
+
+# Start Anonimization
+CMD ["python", "-u", "/app/app.py"]
+
+WORKDIR /app
\ No newline at end of file
diff --git a/app.py b/app.py
new file mode 100644
index 0000000..a800558
--- /dev/null
+++ b/app.py
@@ -0,0 +1,86 @@
+import http.server
+import http.client
+import yaml
+import docker
+from socketserver import ThreadingMixIn
+import threading
+
+# Define the target server to proxy requests to
+class ProxyHandler(http.server.BaseHTTPRequestHandler):
+    def __init__(self, configuration, docker_client):
+        self.configuration = configuration
+        self.docker_client = docker_client
+
+    def __call__(self, *args, **kwargs):
+        """Handle a request."""
+        super().__init__(*args, **kwargs)
+
+    def log_message(self, format, *args):
+        pass
+
+    def do_GET(self):
+        self.handle_request('GET')
+
+    def do_POST(self):
+        self.handle_request('POST')
+
+    def do_PUT(self):
+        self.handle_request('PUT')
+
+    def do_DELETE(self):
+        self.handle_request('DELETE')
+
+    def do_HEAD(self):
+        self.handle_request('HEAD')
+
+    def handle_request(self, method):
+        #print(self.headers.get('Host').split(":")[0])
+        proxy_host_configuration  = next(filter(lambda host: host['domain'] == self.headers.get('Host').split(":")[0], self.configuration['proxy_hosts']))
+        
+        starting = False
+        for container in proxy_host_configuration['containers']:
+            container_object = self.docker_client.containers.list(all=True, filters = { 'name' : container['container_name'] })[0]
+            if (container_object.status != 'running'):
+                print("starting container: {0}".format(container['container_name']))
+                container_object.start()
+                starting = True
+
+        if (starting == True):
+            self.send_response(201)
+            self.end_headers()
+            self.wfile.write(str.encode("starting container: {0}".format(container['container_name'])))
+            conn.close()
+
+        # Open a connection to the target server
+        conn = http.client.HTTPConnection(proxy_host_configuration['proxy_host'], proxy_host_configuration['proxy_port'])
+        conn.request(method, self.path, headers=self.headers)
+        response = conn.getresponse()
+        
+        #print(response.status)
+        self.send_response(response.status)
+        #self.send_header('host', proxy_host_configuration['container_name'])
+        for header, value in response.getheaders():
+            self.send_header(header, value)
+
+        self.end_headers()
+
+        if (response.status != 301):
+            self.wfile.write(response.getcode())
+
+        conn.close()
+
+class ThreadedHTTPServer(ThreadingMixIn, http.server.HTTPServer):
+    """Handle requests in a separate thread."""
+
+if __name__ == '__main__':
+    with open('config.yml', 'r') as file:
+        configuration = yaml.safe_load(file)
+
+    docker_client = docker.from_env()
+
+    # Start the reverse proxy server on port 8888
+    server_address = ('', configuration['proxy_port'])
+    proxy_handler = ProxyHandler(configuration, docker_client)
+    httpd = ThreadedHTTPServer(server_address, proxy_handler)
+    print('Reverse proxy server running on port 8888...')
+    httpd.serve_forever()
\ No newline at end of file
diff --git a/config.yml b/config.yml
new file mode 100644
index 0000000..1b08a56
--- /dev/null
+++ b/config.yml
@@ -0,0 +1,8 @@
+proxy_port: 8888
+proxy_hosts:
+  - domain: wp.local
+    containers:
+      - container_name: wp-dev-db-1
+      - container_name: wp-dev-wordpress-1
+    proxy_host: localhost
+    proxy_port: 80
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..d0174d3
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+docker-py
\ No newline at end of file