Wharfie

#!/usr/bin/python
#
# Copyright 2017 Ingo Hornberger <ingo_@gmx.net>
#
# This software is licensed under the MIT License
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of this
# software and associated documentation files (the "Software"), to deal in the Software
# without restriction, including without limitation the rights to use, copy, modify,
# merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to the following
# conditions:
#
# The above copyright notice and this permission notice shall be included in all copies
# or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
# PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
# OR OTHER DEALINGS IN THE SOFTWARE.
#
################################################################################

import re
from lib import makefile as make
from lib import files

g_cmdUnitCount = 0
g_workingDir = "/"

def run(name, dep, cmd):
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'trgcmd': cmd[1]
    };
    return makeTarget;

def run_host(name, dep, cmd):
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'hostcmd': cmd[1]
    };
    return makeTarget;

def copy(name, dep, cmd):
    if " " not in cmd[1].lstrip():
        print ("parse error: COPY needs two arguments. (%s)" % cmd[1].lstrip())
        exit(1)
    src,dst = cmd[1].lstrip().split(" ")
    isUrl = src.startswith("http://") or src.startswith("https://") or src.startswith("ftp://")
    src = files.input_file(src)
    # Overwrite potential implicite make rules for file
    if not isUrl:
        make.makeTargets.append({'name': src, 'dep': '', 'simplecmd': 'touch $@'});
    dep.append(src)
    myHostCmd = '''
        destdir="$(pwd)";
        src="%s";
        dst="%s";
        cd ../;
        case "${dst}" in
            */)
                mkdir -p "${destdir}/${dst}";
                ;;
            *)
                mkdir -p $(dirname "${destdir}/${dst}");
                ;;
        esac;
        cp -R --preserve=mode ${src} "${destdir}/${dst}";
''' % (src, dst)
    
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'hostcmd': myHostCmd
    };
    return makeTarget;

def add(name, dep, cmd):
    if " " not in cmd[1].lstrip():
        print ("parse error: ADD needs two arguments. (%s)" % cmd[1].lstrip())
        exit(1)
    src,dst = cmd[1].lstrip().split(" ")
    isUrl = src.startswith("http://") or src.startswith("https://") or src.startswith("ftp://")
    src = files.input_file(src)
    # Overwrite potential implicite make rules for file
    if not isUrl:
        make.makeTargets.append({'name': src, 'dep': '', 'simplecmd': 'touch $@'});
    dep.append(src)
    myHostCmd = '''
        destdir="$(pwd)";
        src="%s";
        dst="%s";
        cd ../;
        case "${dst}" in
            */)
                mkdir -p "${destdir}/${dst}";
                ;;
            *)
                mkdir -p $(dirname "${destdir}/${dst}");
                ;;
        esac;
        case "${src}" in
            *.tar)
                cat ${src} | (cd ${destdir}/${dst}; tar -xf -);
                ;;
            *.tar.gz)
                cat ${src} | (cd ${destdir}/${dst}; tar -xzf -);
                ;;
            *.tar.bz2)
                cat ${src} | (cd ${destdir}/${dst}; tar -xjf -);
                ;;
            *)
                cp -R --preserve=mode ${src} "${destdir}/${dst}";
                ;;
        esac;
''' % (src, dst)
    
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'hostcmd': myHostCmd
    };
    return makeTarget;

def workingdir(name, dep, cmd):
    global g_workingDir;
    g_workingDir = cmd[1]

def entrypoint(name, dep, cmd):
    global g_cmdUnitCount;
    global g_workingDir;

    args = eval(cmd[1])
    unitCmd = "%s %s" % (cmd[0], " ".join(args).replace("'", "'\"'\"'"))
    unitName = "wharfie%u.service" % g_cmdUnitCount
    myTrgCmd = '''
        echo '[Unit]\nDescription=Wharfie Unit\n\n[Service]\nType=simple\nWorkingDirectory=%s\nExecStart=%s\n\n[Install]\nWantedBy=multi-user.target\n' > /etc/systemd/system/%s;
        systemctl enable %s;
    ''' % (g_workingDir, unitCmd, unitName, unitName)
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'trgcmd': myTrgCmd
    };
    g_cmdUnitCount = g_cmdUnitCount + 1;
    return makeTarget;

def source(name, dep, cmd):
    myTrgCmd = '''
        mkdir /sources;
        cd /sources;
	apt-get update;
        [ ! -f .pkg.list ] && 
            dpkg-query -f \'${binary:Package}\\n\' -W > .pkg.list;
        apt-get install -y dpkg-dev;
        cat .pkg.list | xargs -n 1 apt-get source'''
    myHostCmd = 'tar -cf ../' + cmd[1].lstrip() + ' sources;'
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'trgcmd': myTrgCmd,
        'hostcmd': myHostCmd,
        'temporary': True
    };
    return makeTarget;

def license(name, dep, cmd):
    myTrgCmd = '''
        mkdir /licenses;
        for i in /usr/share/doc/*/;
        do [ -f $i/copyright ] &&
            cp $i/copyright licenses/$(basename $i).txt ||
            echo $(basename $i) >> licenses/___MISSING___.txt;
        done'''
    myHostCmd = 'tar -cf ../' + cmd[1].lstrip() + ' licenses;'
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'trgcmd': myTrgCmd,
        'hostcmd': myHostCmd,
        'temporary': True
    };
    return makeTarget;

def toolchain(name, dep, cmd):
    myTrgCmd = '''
        (apt-get update; apt-get install -y libc6-dev;
        [ -d ./usr/lib/arm-linux-gnueabi*/ ] &&
            (cd ./usr/lib/arm-linux-gnueabi*/; ln -s crt1.o crt0.o) || true);'''
    if " " in cmd[1].lstrip():
        params = re.findall('^([^ ]+)[ \t]([^\n]*)', cmd[1].lstrip(), flags=re.MULTILINE)
        if len(params) > 0:
            myTrgCmd += params[0][1]

        myHostCmd = 'files=$(ls -1); mkdir target; mv ${files} target; mkdir host; (cd host; tar -xf ../../debian_toolchain.tar); cp ../debian_toolchain_env.sh env.sh; tar -cf ../' + params[0][0] + ' .;'
    else:
        myHostCmd = 'files=$(ls -1); mkdir target; mv ${files} target; mkdir host; (cd host; tar -xf ../../debian_toolchain.tar); cp ../debian_toolchain_env.sh env.sh; tar -cf ../' + cmd[1].lstrip() + ' .;'
        
    dep.append('debian_toolchain.tar')
    dep.append('debian_toolchain_env.sh')
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': name,
        'dep': dep,
        'trgcmd': myTrgCmd,
        'hostcmd': myHostCmd,
        'temporary': True
    };
    return makeTarget;

def to(name, dep, cmd):
    makeTarget = {
        'comment' : "%s %s" % (cmd[0], cmd[1]),
        'name': cmd[1].lstrip(),
        'dep': dep,
        'temporary': True
    };
    return makeTarget;