#!/usr/bin/env bash
# Copyright 2014, Rackspace US, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Notes:
# To use this script you MUST move it to some path that will be called.
# I recommend that the script be stored and executed from
# "/opt/openstack-wheel-builder.sh". This script is a wrapper script that relies
# on the "openstack-wheel-builder.py" and is execute from
# "/opt/openstack-wheel-builder.py".

# Overrides:
# This script has several things that can be overriden via environment
# variables.
#     Git repository that the rcbops ansible lxc source code will be cloned from.
#     This repo should be a repo that is available via HTTP.
#     GIT_REPO=""

#     The URI for the github api. This is ONLY used when the $RELEASES variable
#     is an empty string. Which causes the script to go discover the available
#     releases.
#     GITHUB_API_ENDPOINT=""

#     Local directory to store the source code while interacting with it.
#     WORK_DIR=""

#     Local directory to store the built wheels.
#     OUTPUT_WHEEL_PATH=""

#     Space seperated list of all releases to build for. If unset the releases
#     will be discovered.
#     RELEASES=""

#     Space seperated list of all releases to exclude from building. This is
#     ONLY used when the $RELEASES variable is an empty string.
#     EXCLUDE_RELEASES=""

set -e -o -v

# Trap any errors that might happen in executing the script
trap my_trap_handler ERR

# Ensure there is a base path loaded
export PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"

# Defined variables.
GIT_REPO="${GIT_REPO:-https://github.com/stackforge/os-ansible-deployment}"
GITHUB_API_ENDPOINT="${GITHUB_API_ENDPOINT:-https://api.github.com/repos/stackforge/os-ansible-deployment}"

# Predefined working directory.
WORK_DIR="${WORK_DIR:-/tmp/openstack-ansible-deployment}"

# Output directories.
OUTPUT_WHEEL_PATH="${OUTPUT_WHEEL_PATH:-/var/www/repo/os-releases}"
LINK_PATH="${LINK_PATH:-/var/www/repo/links}"
REPORT_DIR="${REPORT_DIR:-/var/www/repo/reports}"
STORAGE_POOL="${STORAGE_POOL:-/var/www/repo/pools}"

# Additional space separated list of repos to always include in a build.
ADDON_REPOS="git+https://github.com/rcbops/horizon-extensions.git@master "

# Set the force build option to false
FORCE_BUILD="${FORCE_BUILD:-false}"

# Default is an empty string which causes the script to go discover the available
# branches from the github API.
RELEASES=${RELEASES:-""}

# Define branches that you no longer want new wheels built for or checked against.
EXCLUDE_RELEASES="${EXCLUDE_RELEASES:-v9.0.0 gh-pages revert}"

# Name of the lock file.
LOCKFILE="/tmp/wheel_builder.lock"

function my_trap_handler() {
    kill_job
}

function lock_file_remove() {
    if [ -f "${LOCKFILE}" ]; then
        rm "${LOCKFILE}"
    fi
}

function kill_job() {
    set +e
    # If the job needs killing kill the pid and unlock the file.
    if [ -f "${LOCKFILE}" ]; then
        PID="$(cat ${LOCKFILE})"
        lock_file_remove
        kill -9 "${PID}"
    fi
}

function cleanup() {
    # Ensure workspaces are cleaned up
    rm -rf /tmp/openstack_wheels*
    rm -rf /tmp/pip*
    rm -rf "${WORK_DIR}"
}

# Check for releases
if [ -z "${RELEASES}" ];then
    echo "No releases specified. Provide a space separated list branches to build for."
    exit 1
fi

# Check for system lock file.
if [ ! -f "${LOCKFILE}" ]; then
    echo $$ | tee "${LOCKFILE}"
else
    if [ "$(find ${LOCKFILE} -mmin +240)" ]; then
        logger "Stale pid found for ${LOCKFILE}."
        logger "Killing any left over processes and unlocking"
        kill_job
    else
        NOTICE="Active job already in progress. Check pid \"$(cat ${LOCKFILE})\" for status. Lock file: ${LOCKFILE}"
        echo $NOTICE
        logger ${NOTICE}
        exit 1
    fi
fi

# Iterate through the list of releases and build everything that's needed
logger "Building Python Wheels for ${RELEASES}"
for release in ${RELEASES}; do

    if [ ! -d "${OUTPUT_WHEEL_PATH}/${release}" ] || [[ "${FORCE_BUILD}" == "true" ]]; then
        # Perform cleanup
        cleanup

        # Git clone repo
        git clone "${GIT_REPO}" "${WORK_DIR}"

        # checkout release
        pushd "${WORK_DIR}"
          git checkout "${release}"
        popd

        # Build wheels
        OVERRIDE_WHEEL_OUTPUT_PATH="${OVERRIDE_WHEEL_OUTPUT_PATH:-${OUTPUT_WHEEL_PATH}/${release}}"
        mkdir -p "${OVERRIDE_WHEEL_OUTPUT_PATH}"
        /opt/openstack-wheel-builder.py --report-file "${REPORT_DIR}/${release}.json" \
                                        --link-pool "${LINK_PATH}" \
                                        --local-path "${WORK_DIR}" \
                                        --storage-pool ${STORAGE_POOL} \
                                        --release-directory "${OVERRIDE_WHEEL_OUTPUT_PATH}" \
                                        --add-on-repos ${ADDON_REPOS}

    fi
    echo "Complete [ ${release} ]"
done

# Perform cleanup
cleanup

# Remove lock file on job completion
lock_file_remove