Main.py

import socket
import json
import datetime
import os
from subprocess import call
import git

baseDir = "/tmp/UmonsITListener"
targetDir = baseDir+"/CompiledPDFs"
pdfRepoUrl= "https://github.com/UMonsIT/CompiledPDFs.git"

def mainLoop(serversocket):
    while True:
        connection, address = serversocket.accept()
        buf = connection.recv(500000)
        if len(buf) > 0:
            #isolate JSON from message
            isJson = False
            js = ""
            for line in buf:
                if line[0]=="{":
                    isJson = True
                if isJson:
                    js += line
            #extract repo name
            jsdoc = json.JSONDecoder().decode(js)
            repository = jsdoc["repository"]["name"]
            repoUrl = jsdoc["repository"]["git_url"]
            print "Detected changes in", repository, "("+repoUrl+")"
            #launch the update routine
            #first check if repo has already been cloned
            dirName = baseDir+"/"+repository
            if not (os.path.exists(dirName) and os.path.isdir(dirName)):
                clone(repoUrl, dirName)
            pull(repository, dirName)

            if not (os.path.exists(targetDir) and os.path.isdir(targetDir)):
                clone(pdfRepoUrl, targetDir)
            compileAndMove(dirName, repository, targetDir)
            commitAndPush(targetDir)

def clone(url, dirName):
    print "Cloning", url, "in", dirName
    repo = git.Repo.clone_from(url, dirName)
    print "Done cloning..."

def pull(repo, dirName):
    print "Pulling", repo, "in", dirName
    git.Repo(dirName).remote().pull()


def hasMakefile(compileDir):
    for fl in os.listdir(compileDir):
        if os.path.isfile(compileDir+"/"+fl) and fl == "Makefile":
            return True


def compileAndMove(compileDir, repoName, targetDir):
    if hasMakefile(compileDir):
        print "Compiling", repoName, "in", compileDir
        call("cd "+compileDir+" && make")
        move(compileDir, repoName, targetDir)
    else:
        for subFolder in os.listdir(compileDir):
            if os.path.isdir(compileDir + "/" + subFolder):
                compileAndMove(subFolder, repoName, targetDir)

def move(compileDir, repoName, targetDir):
    print "Copying PDFs in", compileDir, "to", targetDir+repoName
    call("mv "+compileDir+"/*.pdf "+targetDir+"")

def commitAndPush(target):
    message = "Auto-compiled on " + datetime.datetime.now().isoformat()
    print "Commiting", target, "with message", message
    git.Repo(target).commit(message)
    print "Pushing", target
    git.Repo(target).remote().push()


if __name__ == "__main__":
    port = 4242
    print "Starting server on port ", port, "..."
    serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    serversocket.bind(('localhost', port))
    serversocket.listen(5)  # become a server socket, maximum 5 connections
    try:
        while True :
            mainLoop(serversocket)
    finally:
        serversocket.close()