#!/usr/bin/bash
errors_terminate=$2

extra=$3
if [ 0$extra -eq 1 ]; then
    echo -e "%_python_bytecompile_extra is discontinued, use %py_byte_compile instead." >/dev/stderr
    exit 1
fi

if [ -z "$RPM_BUILD_ROOT" -o "$RPM_BUILD_ROOT" = "/" ]; then
	exit 0
fi

function python_bytecompile()
{
    local options=$1
    local python_binary=$2
    local exclude=$3
    local python_libdir="$4"

    python_version=$($python_binary -c "import sys; sys.stdout.write('{0.major}{0.minor}'.format(sys.version_info))")

    if [ "$python_version" -ge 34 ]; then

        if [ -n "$options" ]; then
            return
        fi

        if [ "$python_version" -ge 39 ]; then
            compileall_module=compileall
        else
            compileall_module=compileall2
        fi

        [ ! -z $exclude ] && exclude="-x '$exclude'"

        # PYTHONPATH is needed for compileall2, but doesn't hurt for the stdlib
        # -o 0 -o 1 are the optimization levels
        # -q disables verbose output
        # -f forces the process to overwrite existing compiled files
        # -x excludes paths defined by regex
        # -e excludes symbolic links pointing outside the build root
        # -x and -e together implements the same functionality as the Filter class below
        # -s strips $RPM_BUILD_ROOT from the path
        # -p prepends the leading slash to the path to make it absolute
        PYTHONPATH=/usr/lib/rpm/TencentOS/ $python_binary -B -m $compileall_module -o 0 -o 1 -q -f $exclude -s "$RPM_BUILD_ROOT" -p / --hardlink-dupes -e "$RPM_BUILD_ROOT" "$python_libdir"

    else
#
# Python 3.3 and lower (incl. Python 2)
#

local real_libdir=${python_libdir/$RPM_BUILD_ROOT/}

cat << EOF | $python_binary $options
import compileall, sys, os, re

python_libdir = "$python_libdir"
depth = sys.getrecursionlimit()
real_libdir = "$real_libdir"
build_root = "$RPM_BUILD_ROOT"
exclude = r"$exclude"

class Filter:
    def search(self, path):
        ret = not os.path.realpath(path).startswith(build_root)
        if exclude:
            ret = ret or re.search(exclude, path)
        return ret

sys.exit(not compileall.compile_dir(python_libdir, depth, real_libdir, force=1, rx=Filter(), quiet=1))
EOF

fi
}

export PYTHONHASHSEED=0

shopt -s nullglob
find "$RPM_BUILD_ROOT" -type d -print0|grep -z -E "/(usr|app)/lib(64)?/python[0-9]\.[0-9]+$" | while read -d "" python_libdir;
do
	python_binary=$(basename "$python_libdir")
	echo "Bytecompiling .py files below $python_libdir using $python_binary"

	python_bytecompile "" "$python_binary" "" "$python_libdir"
	if [ $? -ne 0 -a 0$errors_terminate -ne 0 ]; then
		exit 1
	fi

	python_bytecompile "-O" "$python_binary" "" "$python_libdir"
	if [ $? -ne 0 -a 0$errors_terminate -ne 0 ]; then
		exit 1
	fi
done
