Mini Shell

#
# Copyright (c) 2014, Red Hat, Inc.
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the
#    distribution.
# 3. Neither the name of the Red Hat nor the names of its
#    contributors may be used to endorse or promote products derived
#    from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
# Authors:  Michal Srb <msrb@redhat.com>

import os
import signal
import sys
import six
import subprocess
import logging
import re
from optparse import OptionParser
from javapackages.common.exception import JavaPackagesToolsException
from javapackages.common.config import get_buildroot


def kill_parent_process(rpmconf):
    if not rpmconf:
        return
    # mock may kill us immediately after rpmbuild dies, before output
    # is flushed. To avoid this race condiditon we must explicitly
    # flush any pending output before trying to kill parent.
    sys.stdout.flush()
    sys.stderr.flush()
    # rpmbuild ignores non-zero exit codes, but this is bad. Make sure
    # the build fails and doesn't silently ignore problems
    try:
        os.kill(rpmconf.rpm_pid, signal.SIGTERM)
    except OSError:
        pass
    sys.exit(1)


def args_to_unicode(args):
    if six.PY2:
        for index, arg in enumerate(args):
            args[index] = arg.decode(sys.getfilesystemencoding())
    return args


def execute_command(command, input=None):
    proc = subprocess.Popen([command], shell=True,
                            stdout=subprocess.PIPE, stderr=subprocess.PIPE,
                            stdin=subprocess.PIPE, universal_newlines=True)
    stdout, stderr = proc.communicate(input=input)
    proc.wait()
    return proc.returncode, stdout, stderr


def command_exists(cmd):
    command = "type {cmd}".format(cmd=cmd)
    rc, _, _ = execute_command(command)
    return True if not rc else False


def init_rpmgen(argv):
    return _parse_rpmgen_args(argv)


def _parse_argv(argv):
    parser = OptionParser()
    parser.add_option("--cachedir", dest="cachedir")
    parser.add_option("--rpm-pid", dest="rpm_pid")
    parser.add_option("--scl", dest="scl", default=None)

    return parser.parse_args()


def _parse_rpmgen_args(argv):
    options = _parse_argv(argv)[0]

    if not options.cachedir:
        raise JavaPackagesToolsException("Missing option: --cachedir")
    options.cachedir = _get_cachedir(options.cachedir)

    if not options.rpm_pid:
        raise JavaPackagesToolsException("Missing option: --rpm-pid")
    options.rpm_pid = int(options.rpm_pid)

    return options


def _get_cachedir(path, create_if_not_exists=True):
    cachedir_path = os.path.join(path, ".javapackages_cache")
    if not os.path.exists(cachedir_path) and create_if_not_exists:
        os.makedirs(cachedir_path)
    return cachedir_path


def get_logger(name):
    logger = logging.getLogger(name)
    logger.setLevel(logging.INFO)
    handler = logging.StreamHandler()
    formatter = logging.Formatter("[%(levelname)s %(name)s] %(message)s")
    handler.setFormatter(formatter)
    logger.addHandler(handler)
    return logger


def get_buildroot_files(regexp=None,
                        inside_buildroot_only=True,
                        exclude_broken_symlinks=True,
                        buildroot_path=None):
    if not buildroot_path:
        buildroot_path = get_buildroot()
    files = set()
    for dirpath, dirnames, filenames in os.walk(buildroot_path):
        for filename in filenames:
            f = os.path.realpath(os.path.join(dirpath, filename))

            if inside_buildroot_only:
                if not f.startswith(buildroot_path):
                    continue

            if exclude_broken_symlinks:
                if not os.path.exists(f):
                    continue

            if regexp:
                match = re.search(regexp, f)
                if not match:
                    continue
            files.add(f)
    return files