first commit

README.md

@@ -0,0 +1,3 @@
+# Automated Nginx reverse Proxy for Docker Webservices
+
+

nginx_proxy.py

@@ -0,0 +1,133 @@
+#!/usr/bin/python3.4
+
+import os
+import http.client
+import json
+import signal
+from string import Template
+
+target_path="/tmp/nginx"
+pid_file="/run/nginx.pid"
+non_location_template = """server {
+    listen 80;
+    server_name $name;
+    location / {
+        proxy_set_header X-Real-IP  $$remote_addr;
+        proxy_set_header X-Forwarded-For $$remote_addr;
+        proxy_set_header Host $$host;
+        proxy_pass http://$ip:$port/;      
+    }
+}
+"""
+
+location_template="""server {
+    listen 80;
+    server_name $name;
+    location / {
+        return 301 $$scheme://$name/$location;
+    }
+    location /$location {
+        proxy_set_header X-Real-IP  $$remote_addr;
+        proxy_set_header X-Forwarded-For $$remote_addr;
+        proxy_set_header Host $$host;
+        proxy_pass http://$ip:$port/;      
+    }
+}
+"""
+
+def get_pid():
+    with open(pid_file, 'r') as file:
+        return int(file.read())
+
+def get_container_data():
+    conn = http.client.HTTPConnection("localhost:2375")
+    conn.request("GET", "/containers/json")
+    response = conn.getresponse()
+    data = response.read().decode("UTF-8")
+    data = json.loads(data)
+#    print(json.dumps(data, sort_keys=True, indent=4))
+    conn.close()
+    return data
+
+def inspect_container(container_id):
+    conn = http.client.HTTPConnection("localhost:2375")
+    conn.request("GET", "/containers/"+str(container_id)+"/json")
+    response = conn.getresponse()
+    data = response.read().decode("UTF-8")
+    data = json.loads(data)
+#    print(json.dumps(data, sort_keys=True, indent=4))
+    conn.close()
+    return data
+
+def analyse_env_vars(data):
+    env_data = {}
+    for env_var in data:
+        t = env_var.split("=")
+        env_data[t[0]] = t[1]
+    return env_data
+
+def analyse_proxy_data(data):
+    proxy_data = {}
+    for proxy_var in data.split(','):
+        t = proxy_var.split(":")
+        proxy_data[t[0]] = t[1]
+    return proxy_data
+
+def handle_event(event):
+    files = os.listdir(target_path)
+    print('deleting files')
+    for file in files:
+        if file.startswith('proxy_'):
+            os.remove(target_path+'/'+file)
+    event_data = json.loads(event)
+    containers_data = get_container_data()
+    for container_data in containers_data:
+        inspect_data = inspect_container(container_data['Id'])
+        env_data = analyse_env_vars(inspect_data['Config']['Env'])
+        if 'PROXY_DATA' in env_data:
+            container_id = container_data['Id']
+            ip = inspect_data['NetworkSettings']['IPAddress']
+            proxy_data = analyse_proxy_data(env_data['PROXY_DATA'])
+
+            server_name = ''
+            if 'server_name' in proxy_data:
+                server_name = proxy_data['server_name']
+            
+            port = 0
+            if 'port' in proxy_data:
+                port = proxy_data['port']
+
+            location = ''
+            if 'location' in proxy_data:
+                location = proxy_data['location']
+
+            print('writing /tmp/nginx/proxy_'+container_id)
+            with open('/tmp/nginx/proxy_'+container_id, 'w') as file:
+                if location == '':
+                    s = Template(non_location_template)                
+                    file.write(s.substitute(name=server_name,ip=ip,port=port))
+                else:
+                    s = Template(location_template)            
+                    file.write(s.substitute(name=server_name,ip=ip,port=port,location=location))
+            print('HUPing nginx')
+            os.kill(pid, signal.SIGHUP)
+
+pid = get_pid()
+
+conn = http.client.HTTPConnection("localhost:2375")
+conn.request("GET", "/events")
+response = conn.getresponse()
+
+events = ""
+while not response.closed:
+    data = response.read(80)
+    if len(data) != 0:
+        events += data.decode("UTF-8")
+        if events.find("{") != -1:
+            open_pos = events.index("{")
+            if events.find("}", open_pos) != -1:
+                close_pos = events.index("}", open_pos)+1
+                handle_event(events[open_pos:close_pos])
+                events = events[close_pos:]
+            
+