webhook-core.py

   1 #!/usr/bin/python3
   2 # Copyright (c) 2015, Ralf Jung <post@ralfj.de>
   3 # All rights reserved.
   4 #
   5 # Redistribution and use in source and binary forms, with or without
   6 # modification, are permitted provided that the following conditions are met:
   7 #
   8 # 1. Redistributions of source code must retain the above copyright notice, this
   9 #    list of conditions and the following disclaimer.
  10 # 2. Redistributions in binary form must reproduce the above copyright notice,
  11 #    this list of conditions and the following disclaimer in the documentation
  12 #    and/or other materials provided with the distribution.
  13 #
  14 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  15 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  16 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  17 # DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
  18 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  19 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  20 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  21 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  23 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24 #==============================================================================
  25
  26 # This is the hook called by GitHub as webhook. It updats the local repository, and then all the other mirrors.
  27 import sys, traceback, json
  28 from git_mirror import *
  29
  30 def get_github_payload(repo, signature):
  31     '''Return the github-style JSON encoded payload (as if we were called as a github webhook)'''
  32     data = sys.stdin.buffer.read()
  33     verify_signature = repo.compute_hmac(data)
  34     if signature != "sha1="+verify_signature:
  35         raise Exception("You are not GitHub!")
  36     try:
  37         data = json.loads(data.decode('utf-8'))
  38         return data
  39     except ValueError:
  40         return {} # nothing read
  41
  42
  43 if __name__ == "__main__":
  44     # call this with: <reponame> <event name> <signature>
  45     repo = None # we will try to use this during exception handling
  46     try:
  47         repos = load_repos()
  48         if len(sys.argv) < 4:
  49             raise Exception("Usage: {0} <reponame> <event name> <signature>".format(os.path.basename(sys.argv[0])))
  50         reponame = sys.argv[1]
  51         githubEvent = sys.argv[2]
  52         githubSignature = sys.argv[3]
  53         if reponame not in repos:
  54             raise Exception("Repository missing or not found.")
  55         repo = repos[reponame]
  56
  57         # now sync this repository
  58         data = get_github_payload(repo, githubSignature)
  59         if githubEvent == 'ping':
  60             # github sends this initially
  61             print("Content-Type: text/plain")
  62             print()
  63             print("Pong!")
  64             sys.exit(0)
  65         elif githubEvent == 'push':
  66             ref = data["ref"]
  67             oldsha = data["before"]
  68             newsha = data["after"]
  69             # validate the ref name
  70             if re.match('refs/[a-z/]+', ref) is None:
  71                 raise Exception("Invalid ref name {0}".format(ref))
  72             # collect URLs of this repository, to find the mirror name
  73             urls = []
  74             for key in ("git_url", "ssh_url", "clone_url"):
  75                 urls.append(data["repository"][key])
  76             mirror = repo.find_mirror_by_url(urls)
  77             if mirror is None:
  78                 raise Exception("Could not find the mirror.")
  79             stdout = repo.update_ref_from_mirror(ref, oldsha, newsha, mirror, suppress_stderr = True)
  80             # print an answer
  81             print("Content-Type: text/plain")
  82             print()
  83             print("Updated {0}:{1} from mirror {2} from {3} to {4}".format(reponame, ref, mirror, oldsha, newsha))
  84             print(stdout)
  85         else:
  86             raise Exception("Unexpected github event {0}.".format(githubEvent))
  87     except Exception as e:
  88         if repo is not None:
  89             repo.mail_owner("There was a problem running the git-mirror webhook:\n\n{0}".format(traceback.format_exc()))
  90         # do not print all the details
  91         print("Status: 500 Internal Server Error")
  92         print("Content-Type: text/plain")
  93         print()
  94         print("We have a problem:\n{0}".format('\n'.join(traceback.format_exception_only(type(e), e))))