d4c1ab7c68bcf9a6455ed7270f1697f267b3184e
[git-mirror.git] / 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
28 from git_mirror import *
29
30 if __name__ == "__main__":
31     # call this with: <reponame> <event name> <signature>
32     repo = None # we will try to use this during exception handling
33     try:
34         repos = load_repos()
35         if len(sys.argv) < 4:
36             raise Exception("Usage: {0} <reponame> <event name> <signature>".format(os.path.basename(sys.argv[0])))
37         reponame = sys.argv[1]
38         githubEvent = sys.argv[2]
39         githubSignature = sys.argv[3]
40         if reponame not in repos:
41             raise Exception("Repository missing or not found.")
42         repo = repos[reponame]
43         
44         # now sync this repository
45         data = get_github_payload()
46         if githubEvent == 'ping':
47             # github sends this initially
48             print("Content-Type: text/plain")
49             print()
50             print("Pong!")
51             sys.exit(0)
52         elif githubEvent == 'push':
53             ref = data["ref"]
54             oldsha = data["before"]
55             newsha = data["after"]
56             # validate the ref name
57             if re.match('refs/[a-z/]+', ref) is None:
58                 raise Exception("Invalid ref name {0}".format(ref))
59             # collect URLs of this repository, to find the mirror name
60             urls = []
61             for key in ("git_url", "ssh_url", "clone_url"):
62                 urls.append(data["repository"][key])
63             mirror = repo.find_mirror_by_url(urls)
64             if mirror is None:
65                 raise Exception("Could not find the mirror.")
66             repo.update_ref_from_mirror(ref, oldsha, newsha, mirror, suppress_stderr = True)
67             # print an answer
68             print("Content-Type: text/plain")
69             print()
70             print("Updated {0}:{1} from mirror {2} from {3} to {4}".format(reponame, ref, mirror, oldsha, newsha))
71         else:
72             raise Exception("Unexpected github event {0}.".format(githubEvent))
73     except Exception as e:
74         if repo is not None:
75             repo.mail_owner("There was a problem running the git-mirror webhook:\n\n{0}".format(traceback.format_exc()))
76         # do not print all the details
77         print("Status: 500 Internal Server Error")
78         print("Content-Type: text/plain")
79         print()
80         print("We have a problem:\n{0}".format('\n'.join(traceback.format_exception_only(type(e), e))))