look out for the github event type

2 files changed, 16 insertions, 4 deletions

diff --git a/webhook-core.py b/webhook-core.py
index a9418bc..d4c1ab7 100755
--- a/webhook-core.py
+++ b/webhook-core.py
@@ -28,22 +28,28 @@ import sys, traceback
 from git_mirror import *
 
 if __name__ == "__main__":
+    # call this with: <reponame> <event name> <signature>
     repo = None # we will try to use this during exception handling
     try:
         repos = load_repos()
-        reponame = sys.argv[1] if len(sys.argv) > 1 else None
+        if len(sys.argv) < 4:
+            raise Exception("Usage: {0} <reponame> <event name> <signature>".format(os.path.basename(sys.argv[0])))
+        reponame = sys.argv[1]
+        githubEvent = sys.argv[2]
+        githubSignature = sys.argv[3]
         if reponame not in repos:
             raise Exception("Repository missing or not found.")
         repo = repos[reponame]
         
         # now sync this repository
         data = get_github_payload()
-        if 'zen' in data:
+        if githubEvent == 'ping':
             # github sends this initially
             print("Content-Type: text/plain")
             print()
-            print("Welcome!")
+            print("Pong!")
             sys.exit(0)
+        elif githubEvent == 'push':
             ref = data["ref"]
             oldsha = data["before"]
             newsha = data["after"]
@@ -62,6 +68,8 @@ if __name__ == "__main__":
             print("Content-Type: text/plain")
             print()
             print("Updated {0}:{1} from mirror {2} from {3} to {4}".format(reponame, ref, mirror, oldsha, newsha))
+        else:
+            raise Exception("Unexpected github event {0}.".format(githubEvent))
     except Exception as e:
         if repo is not None:
             repo.mail_owner("There was a problem running the git-mirror webhook:\n\n{0}".format(traceback.format_exc()))
diff --git a/webhook.py b/webhook.py
index 7935d81..33cae39 100755
--- a/webhook.py
+++ b/webhook.py
@@ -46,5 +46,9 @@ query = urllib.parse.parse_qs(query)
 repository = query.get('repository', [])
 repository = repository[0] if len(repository) else ''
 
+# get GitHub metadata
+githubEvent = os.getenv('HTTP_X_GITHUB_EVENT')
+githubSignature = os.getenv('HTTP_X_HUB_SIGNATURE')
+
 # execute the actual script
-os.execlp("sudo", "sudo", "-n", "-u", "git", webhook_core, repository)
+os.execlp("sudo", "sudo", "-n", "-u", "git", webhook_core, repository, str(githubEvent), str(githubSignature))