#!/bin/bash
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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.

function usage() {
    echo ""
    echo "usage: ./package.sh [-h|--help] -d|--distribution <name> [-r|--release <version>] [-p|--pack oss|OSS|noredist|NOREDIST] [-s|--simulator default|DEFAULT|simulator|SIMULATOR]"
    echo ""
    echo "The supported arguments are:"
    echo "  To package with only redistributable libraries (default)"
    echo "    -p|--pack oss|OSS"
    echo "  To package with non-redistributable libraries"
    echo "    -p|--pack noredist|NOREDIST"
    echo "  To build a package for a distribution (mandatory)"
    echo "    -d|--distribution centos7|centos63|fedora20|fedora21"
    echo "  To set the package release version (optional)"
    echo "  (default is 1 for normal and prereleases, empty for SNAPSHOT)"
    echo "    -r|--release version(integer)"
    echo "  To build for Simulator (optional)"
    echo "    -s|--simulator default|DEFAULT|simulator|SIMULATOR"
    echo "  To display this information"
    echo "    -h|--help"
    echo ""
    echo "Examples: ./package.sh --pack oss"
    echo "          ./package.sh --pack noredist"
    echo "          ./package.sh --pack oss --distribution centos7 --release 42"
    echo "          ./package.sh --distribution centos7 --release 42"
    echo "          ./package.sh --distribution centos7"
}

# packaging
#   $1 redist flag
#   $2 simulator flag
#   $3 distribution name
#   $4 package release version
function packaging() {
    CWD=$(pwd)
    RPMDIR=$CWD/../dist/rpmbuild
    PACK_PROJECT=cloudstack
    if [ -n "$1" ] ; then
        DEFOSSNOSS="-D_ossnoss $1"
    fi
    if [ -n "$2" ] ; then
        DEFSIM="-D_sim $2"
    fi

    DISTRO=$3
    MVN=$(which mvn)
    if [ -z "$MVN" ] ; then
        MVN=$(locate bin/mvn | grep -e mvn$ | tail -1)
        if [ -z "$MVN" ] ; then
            echo -e "mvn not found\n cannot retrieve version to package\n RPM Build Failed"
            exit 2
        fi
    fi
    VERSION=$(cd ../; $MVN org.apache.maven.plugins:maven-help-plugin:2.1.1:evaluate -Dexpression=project.version | grep --color=none '^[0-9]\.')
    if echo "$VERSION" | grep -q SNAPSHOT ; then
        REALVER=$(echo "$VERSION" | cut -d '-' -f 1)
        if [ -n "$4" ] ; then
            DEFPRE="-D_prerelease $4"
            DEFREL="-D_rel SNAPSHOT$4"
        else
            DEFPRE="-D_prerelease 1"
            DEFREL="-D_rel SNAPSHOT"
        fi
    else
        REALVER="$VERSION"
        if [ -n "$4" ] ; then
            DEFREL="-D_rel $4"
        else
            DEFREL="-D_rel 1"
        fi
    fi
    DEFVER="-D_ver $REALVER"

    echo "Preparing to package Apache CloudStack $VERSION"

    mkdir -p "$RPMDIR/SPECS"
    mkdir -p "$RPMDIR/BUILD"
    mkdir -p "$RPMDIR/RPMS"
    mkdir -p "$RPMDIR/SRPMS"
    mkdir -p "$RPMDIR/SOURCES/$PACK_PROJECT-$VERSION"

    echo ". preparing source tarball"
    (cd ../; tar -c --exclude .git --exclude dist . | tar -C "$RPMDIR/SOURCES/$PACK_PROJECT-$VERSION" -x )
    (cd "$RPMDIR/SOURCES/"; tar -czf "$PACK_PROJECT-$VERSION.tgz" "$PACK_PROJECT-$VERSION")

    echo ". executing rpmbuild"
    cp "$DISTRO/cloud.spec" "$RPMDIR/SPECS"

    (cd "$RPMDIR"; rpmbuild --define "_topdir ${RPMDIR}" "${DEFVER}" "${DEFREL}" ${DEFPRE+"$DEFPRE"} ${DEFOSSNOSS+"$DEFOSSNOSS"} ${DEFSIM+"$DEFSIM"} -bb SPECS/cloud.spec)
    if [ $? -ne 0 ]; then
        echo "RPM Build Failed "
        exit 3
    else
        echo "RPM Build Done"
    fi
    exit
}

TARGETDISTRO=""
SIM=""
PACKAGEVAL=""
RELEASE=""

SHORTOPTS="hp:s:d:r:"
LONGOPTS="help,pack:simulator:distribution:release:"
ARGS=$(getopt -s bash -u -a --options "$SHORTOPTS"  --longoptions "$LONGOPTS" --name "$0" -- "$@")
eval set -- "$ARGS"
echo "$ARGS"
while [ $# -gt 0 ] ; do
    case "$1" in
        -h | --help)
            usage
            exit 0
            ;;
        -p | --pack)
            echo "Packaging CloudStack..."
            PACKAGEVAL=$2
            echo "$PACKAGEVAL"
            if [ "$PACKAGEVAL" == "oss" -o "$PACKAGEVAL" == "OSS" ] ; then
                PACKAGEVAL=""
            elif [ "$PACKAGEVAL" == "noredist" -o "$PACKAGEVAL" == "NOREDIST" ] ; then
                PACKAGEVAL="noredist"
            else
                echo "Error: Unsupported value for --pack"
                usage
                exit 1
            fi
            shift
            ;;
        -s | --simulator)
            SIM=$2
            echo "$SIM"
            if [ "$SIM" == "default" -o "$SIM" == "DEFAULT" ] ; then
                SIM="false"
            elif [ "$SIM" == "simulator" -o "$SIM" == "SIMULATOR" ] ; then
                SIM="simulator"
            else
                echo "Error: Unsupported value for --simulator"
                usage
                exit 1
            fi
            shift
            ;;
        -d | --distribution)
            TARGETDISTRO=$2
            if [ -z "$TARGETDISTRO" ] ; then
                echo "Error: Missing target distribution"
                usage
                exit 1
            fi
            shift
            ;;
        -r | --release)
            RELEASE=$2
            shift
            ;;
        -)
            echo "Error: Unrecognized option"
            usage
            exit 1
            ;;
        *)
            shift
            ;;
    esac
done

packaging "$PACKAGEVAL" "$SIM" "$TARGETDISTRO" "$RELEASE"