GitAutoDeploy.py

#!/usr/bin/env python

import json, urlparse, sys, os
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
from subprocess import call

class GitAutoDeploy(BaseHTTPRequestHandler):

    CONFIG_FILEPATH = './GitAutoDeploy.conf.json'
    config = None
    quiet = False
    daemon = False

    @classmethod
    def getConfig(myClass):
        if(myClass.config == None):
            try:
                configString = open(myClass.CONFIG_FILEPATH).read()
            except:
                sys.exit('Could not load ' + myClass.CONFIG_FILEPATH + ' file')

            try:
                myClass.config = json.loads(configString)
            except:
                sys.exit(myClass.CONFIG_FILEPATH + ' file is not valid json')

            for repository in myClass.config['repositories']:
                if(not os.path.isdir(repository['path'])):
                    sys.exit('Directory ' + repository['path'] + ' not found')
                if(not os.path.isdir(repository['path'] + '/.git')):
                    sys.exit('Directory ' + repository['path'] + ' is not a Git repository')

        return myClass.config

    def do_POST(self):
        if(not GitAutoDeploy.quiet):
            print "\nPost request received"
        url_refs = self.parseRequest()
        deployed = 0
        for url, ref in url_refs:
            paths = self.getMatchingPaths(url, ref)
            if(not GitAutoDeploy.quiet):
                print "\nFound ",len(paths)," matching path(s)"
            for path in paths:
                self.pull(path)
                self.deploy(path)
                deployed += 1
        if(deployed > 0):
            self.respond()

    def parseRequest(self):
        length = int(self.headers.getheader('content-length'))
        body = self.rfile.read(length)
        post = urlparse.parse_qs(body)
        items = []
        for itemString in post['payload']:
            item = json.loads(itemString)
            items.append((item['repository']['url'], item['ref']))
        return items

    def getMatchingPaths(self, repoUrl, ref):
        res = []
        config = self.getConfig()
        for repository in config['repositories']:
            if(not GitAutoDeploy.quiet):
                print "\nComparing ", repoUrl, "with ", repository['url']
                print "Comparing ", repository.get('ref', ''), "with ", ('', ref)
            if(repository['url'] == repoUrl and repository.get('ref', '') in ('', ref)):
                res.append((repository['path'], repository.get('ref','')))
        return res

    def respond(self):
        self.send_response(200)
        self.send_header('Content-type', 'text/plain')
        self.end_headers()

    def pull(self, path):
        if(not self.quiet):
            print "\nPulling update for %s refspec %s" % path
        call(['cd "' + path[0] + '" && git pull origin "' + path[1] +'"'], shell=True)

    def deploy(self, path):
        config = self.getConfig()
        for repository in config['repositories']:
            if(repository['path'] == path[0]):
                if 'deploy' in repository:
                     if(not self.quiet):
                         print '\nExecuting deploy command'
                     call(['cd "' + path[0] + '" && ' + repository['deploy']], shell=True)
                break

def main():
    try:
        server = None
        for arg in sys.argv: 
            if(arg == '-d' or arg == '--daemon-mode'):
                GitAutoDeploy.daemon = True
                GitAutoDeploy.quiet = True
            if(arg == '-q' or arg == '--quiet'):
                GitAutoDeploy.quiet = True
                
        if(GitAutoDeploy.daemon):
            pid = os.fork()
            if(pid != 0):
                sys.exit()
            os.setsid()

        if(not GitAutoDeploy.quiet):
            print 'Github Autodeploy Service v 0.1 started'
        else:
            print 'Github Autodeploy Service v 0.1 started in daemon mode'
             
        server = HTTPServer(('', GitAutoDeploy.getConfig()['port']), GitAutoDeploy)
        server.serve_forever()
    except (KeyboardInterrupt, SystemExit) as e:
        if(e): # wtf, why is this creating a new line?
            print >> sys.stderr, e

        if(not server is None):
            server.socket.close()

        if(not GitAutoDeploy.quiet):
            print 'Goodbye'

if __name__ == '__main__':
     main()