Initial commit

config.py

@@ -0,0 +1,6 @@
+GITLAB_HOST = "leviathan.local"
+GITLAB_KEY = "Escape-Anything-These-Company-3"
+GITLAB_SSH_PORT = 2200
+
+LISTENER_ADDRESS = "0.0.0.0"
+LISTENER_PORT = 9001

gitlab-listener

@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+import os
+import sys
+import time
+from subprocess import call
+
+import config
+
+MY_DIR = os.path.dirname(os.path.abspath(__file__))
+
+try:
+    from flask import Flask, request
+except ImportError:
+    call(". " + os.path.join(MY_DIR, "venv", "bin", "activate") + "; " + " ".join(sys.argv), shell=True)
+
+app = Flask(__name__)
+
+COMMAND = os.path.join(MY_DIR, "update-keys")
+
+@app.route("/", methods=['POST'])
+def commit ():
+    key = request.args.get('key', '')
+    if not key == config.GITLAB_KEY:
+        return "not ok"
+
+    time.sleep(1)
+    call([COMMAND])
+    return "ok"
+
+if __name__ == "__main__":
+    app.run(host=config.LISTENER_ADDRESS, port=config.LISTENER_PORT)

gitlab-shell

@@ -0,0 +1,14 @@
+#!/usr/bin/env python
+
+import os
+import sys
+import subprocess
+
+import config
+
+command = " ".join(sys.argv)
+
+if 'SSH_ORIGINAL_COMMAND' in os.environ:
+    command = "SSH_ORIGINAL_COMMAND='" + os.environ['SSH_ORIGINAL_COMMAND'] + "' " + command
+
+subprocess.call(["ssh", "git@{}".format(config.GITLAB_HOST), "-p", "{}".format(config.GITLAB_SSH_PORT), command])

setup

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+virtualenv venv
+. venv/bin/activate
+pip install Flask

update-keys

@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+
+import os
+import sys
+import subprocess
+
+import config
+
+keys = subprocess.check_output(["ssh", "git@{}".format(config.GITLAB_HOST), "-p", "{}".format(config.GITLAB_SSH_PORT), "cat", ".ssh/authorized_keys"])
+
+with open("/home/git/.ssh/authorized_keys", "w") as ak:
+    ak.write(keys)