# Copyright (C) 2003-2007 John Goerzen

# <jgoerzen@complete.org>

#

#    This program is free software; you can redistribute it and/or modify

#    it under the terms of the GNU General Public License as published by

#    the Free Software Foundation; either version 2 of the License, or

#    (at your option) any later version.

#

#    This program is distributed in the hope that it will be useful,

#    but WITHOUT ANY WARRANTY; without even the implied warranty of

#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

#    GNU General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with this program; if not, write to the Free Software

#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import os, sys, re

nulldev = None

verbose = 0

class ExecProblem(Exception):

    pass

def mainexec(program, args = [], child_stdout = None,

             child_stdin = None, child_stderr = None, wait = 1, closefds = []):

    """Runs the program as a sub-process, passing to it args if specified.

    The sub-process has its file descriptors adjusted as per the arguments.

    If wait is 1, wait until the child exits, then return the result code from

    os.waitpid().

    If wait is 0, return the PID immediately."""

    global verbose

    def setfds(source, dest):

        if source != None:

            if hasattr(source, 'fileno'):

                source = source.fileno()

            os.dup2(source, dest)

    pid = os.fork()

    if not pid:

        if verbose:

            print "Running: ", program, args

        setfds(child_stdin, 0)

        setfds(child_stdout, 1)

        setfds(child_stderr, 2)

        for fd in closefds:

            os.close(fd)

        os.execvp(program, (program,) + tuple(args))

        sys.exit(255)

    else:

        if wait:

            return os.waitpid(pid, 0)[1]

        else:

            return pid

def safeexec(program, args = [], child_stdout = None,

             child_stdin = None, child_stderr = None,

             expected = 0):

    """Calls mainexec() with the appropriate arguments, and raises

    an exception if the program died with a signal or returned an

    error code other than expected.  This function will always wait."""

    result = mainexec(program, args, child_stdout, child_stdin, child_stderr)

    return checkresult(result, expected)

def getstdoutsafeexec(program, args, expected = 0):

    pipes = os.pipe()

    pid = mainexec(program, args, child_stdout = pipes[1], wait  = 0)

    os.close(pipes[1])

    fd = os.fdopen(pipes[0], 'r')

    retval = fd.readlines()

    checkpid(pid, expected)

    os.close(pipes[0])

    return retval

def silentsafeexec(program, args, expected = 0):

    """Silently runs the specified program."""

    null = getnull()

    result = mainexec(program, args, null, null, null)

    return checkresult(result, expected)

def checkresult(result, expected):

    info = []

    if os.WIFSIGNALED(result):

        info.append("got signal %d" % os.WTERMSIG(result))

    if os.WIFEXITED(result):

        info.append("exited with code %d" % os.WEXITSTATUS(result))

    info = ",".join(info)

    if not os.WIFEXITED(result):

        raise ExecProblem, info

    if os.WEXITSTATUS(result) != expected:

        raise ExecProblem, info + " (expected exit code %d)" % expected

    return result

def checkpid(pid, expected):

    return checkresult(os.waitpid(pid, 0)[1], expected)

def getnull():

    global nulldev

    if not nulldev:

        nulldev = open("/dev/null", "w+")

    return nulldev

def chdircmd(newdir, func, *args, **kwargs):

    cwd = os.getcwd()

    os.chdir(newdir)

    try:

        return apply(func, args, kwargs)

    finally:

        os.chdir(cwd)

def maketree(path, addpath = None, ignore = [], res = None):

    thisdir = os.listdir(path)

    retval = []

    others = []

    if res == None:

        res = [re.compile(x) for x in ignore]

    for item in thisdir:

        skip = 0

        for retest in res:

            if retest.search(item):

                skip = 1

                break

        if skip:

            continue

        dirname = os.path.join(path, item)

        if os.path.isdir(dirname) and not os.path.islink(dirname):

            if addpath:

                retval.append(os.path.join(addpath, item) + '/')

            else:

                retval.append(item + '/')

            if addpath:

                newaddpath = os.path.join(addpath, item)

            else:

                newaddpath = item

            others.extend(maketree(dirname, newaddpath, res = res))

        else:

            if addpath:

                retval.append(os.path.join(addpath, item))

            else:

                retval.append(item)

    return sorttree(retval + others)

def sorttree(srctree, filesfirst = False):

    retval = []

    dirs = [x for x in srctree if x.endswith('/')]

    files = [x for x in srctree if not x.endswith('/')]

    dirs.sort()

    files.sort()

    if filesfirst:

        return files + dirs

    else:

        return dirs + files

def copyfrom(srcdir, destdir):

    pipes = os.pipe()

    verbargs = []

    #if verbose:

    #    verbargs.append('-v')

    readerpid = chdircmd(srcdir, mainexec, "tar", ["-cSpf", "-", "."],

                         child_stdout = pipes[1], wait = 0,

                         closefds = [pipes[0]])

    writerpid = chdircmd(destdir, mainexec, "tar", ["-xSpf", "-"] + verbargs,

                         child_stdin = pipes[0], wait = 0, closefds = [pipes[1]])

    os.close(pipes[0])

    os.close(pipes[1])

    checkpid(readerpid, 0)

    checkpid(writerpid, 0)