epicyon/git.py

__filename__ = "git.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.1.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@freedombone.net"
__status__ = "Production"

import os


def getGitProjectName(baseDir: str, nickname: str, domain: str,
                      subject: str) -> str:
    """Returns the project name for a git patch
    The project name should be contained within the subject line
    and should match against a list of projects which the account
    holder wants to receive
    """
    gitProjectsFilename = \
        baseDir + '/accounts/' + nickname + '@' + domain + '/gitprojects.txt'
    if not os.path.isfile(gitProjectsFilename):
        return None
    projectName = None
    subjectLineWords = subject.lower().split(' ')
    for word in subjectLineWords:
        if word + '\n' in open(gitProjectsFilename).read():
            return word
    return projectName


def isGitPatch(baseDir: str, nickname: str, domain: str,
               subject: str, content: str) -> bool:
    """Is the given post content a git patch?
    """
    # must have a subject line
    if not subject:
        return False
    if '[PATCH]' not in content:
        return False
    if '---' not in content:
        return False
    if 'diff ' not in content:
        return False
    if 'From:' not in content:
        return False
    if 'Date:' not in content:
        return False
    if 'Subject:' not in content:
        return False
    if '<br>' not in content:
        if '<br />' not in content:
            return False
    projectName = \
        getGitProjectName(baseDir, nickname, domain,
                          subject)
    if not projectName:
        return False
    return True


def receiveGitPatch(baseDir: str, nickname: str, domain: str,
                    subject: str, content: str) -> bool:
    """Receive a git patch
    """
    if not isGitPatch(baseDir, nickname, domain,
                      subject, content):
        return False

    # replace html formatting, so that it's more
    # like the original patch file
    contentStr = content.replace('<br>','\n').replace('<br />','\n')
    contentStr = contentStr.replace('<p>','').replace('</p>','\n')

    patchLines = contentStr.split('\n')
    patchFilename = None
    patchDir = None
    # get the subject line and turn it into a filename
    for line in patchLines:
        if line.startswith('Subject:'):
            patchSubject = \
                line.replace('Subject:', '').replace('/', '|').strip()
            projectName = \
                getGitProjectName(baseDir, nickname, domain, subject)
            patchDir = \
                baseDir + '/accounts/' + nickname + '@' + domain + \
                '/patches/' + projectName
            patchFilename = \
                patchDir + '/' + patchSubject + '.patch'
            break
    if not patchFilename:
        return False
    patchFile = open(patchFilename, "w")
    if not patchFile:
        return False
    patchFile.write(contentStr)
    patchFile.close()
    return True