cloud_install.sh

#!/bin/bash

#
# Installs a specific SolrCloud
#

###############################################################################
# CONFIG
###############################################################################


if [[ -s "cloud.conf" ]]; then
    source "cloud.conf"     # Local overrides
fi
pushd ${BASH_SOURCE%/*} > /dev/null
if [[ -s "cloud.conf" ]]; then
    source "cloud.conf"     # Project overrides
fi
source general.conf
SHOME=`pwd`
: ${CACHE:=`pwd`/cache}
: ${CLOUD:=`pwd`/cloud}
: ${VERSION:="$1"}
popd > /dev/null

function usage() {
    echo "Usage: ./cloud_install.sh <`echo \"$VERSIONS\" | sed 's/ / | /g'`>"
    exit $1
}

check_parameters() {
    if [[ -z "$VERSION" ]]; then
        echo "No Solr version specified."$'\n'
        usage
    fi
}

################################################################################
# FUNCTIONS
################################################################################

# Input: Package
function check_package() {
    if [[ ! -s "${CACHE}/$1" ]]; then
        echo "Activating get_solr.sh as package $1 is not available at ${CACHE}/$1"
        SPECIFIC_SOLR="$VERSION"
        . $SHOME/get_solr.sh
        if [ ! -s ${CACHE}/$1 ]; then
            >&2 "Error: Package $P not available after call to get_solr.sh"
            exit 2
        fi
    fi
}

function zoo() {
    local ZPACK="$1"
    local FOLDER=`echo $ZPACK | sed -e 's/.gz$//' -e 's/.tar$//' -e 's/.tgz//'`
    echo "  - Installing ZooKeeper ensemble of size $ZOOS"
    local ZPORT=$ZOO_BASE_PORT
    
    for Z in `seq 1 $ZOOS`; do
        if [ ! -d zoo$Z ]; then
            echo "     - Copying ZooKeeper files for instance $Z"
            tar -xzovf ${CACHE}/$ZPACK > /dev/null
            mv $FOLDER zoo$Z
        else
            echo "   - ZooKeeper files for instance $Z already exists"
        fi
        local ZCONF="zoo$Z/conf/zoo.cfg"
        if [ ! -s "$ZCONF" ]; then
            echo "     - Creating new setup for ZooKeeper $Z"
            echo "dataDir=`pwd`/zoo$Z/data" >> "$ZCONF"
            echo "clientPort=$ZPORT" >> "$ZCONF"
        else
            echo "     - ZooKeeper $Z already configured"
        fi
        local ZPORT=$(( ZPORT + 1 ))
    done
}

function solr() {
    local SPACK="$1"
    local FOLDER=`echo $SPACK | sed -e 's/[.]gz$//' -e 's/[.]tar$//' -e 's/[.]tgz//'`
    echo "  - Installing $SOLRS Solrs"
    tar -xzovf  ${CACHE}/$SPACK > /dev/null
    if [ solr-* != $FOLDER ]; then
        mv solr-* $FOLDER
    fi
    
    if [ "." == ".`echo \" $LAYOUT2_VERSIONS \" | grep \" $DEST \"`" ]; then
        # We inject the JARs into the WAR to avoid fiddling with solrconfig.xml
        pushd $FOLDER/example/webapps/ > /dev/null
        if [ ! "." == ".$SPARSE_WAR" ]; then
            echo "     - Using sparse Solr WAR as Solr $VERSION"
            cp  ${CACHE}/$SPARSE_WAR solr.war
        fi
        mkdir -p WEB-INF/lib/
        cp ../../dist/solr-analysis-extras-*.jar WEB-INF/lib/
        cp ../../contrib/analysis-extras/lucene-libs/lucene-analyzers-icu-*.jar WEB-INF/lib/
        cp ../../contrib/analysis-extras/lib/icu4j*.jar WEB-INF/lib/
        echo "     - Patching Solr $VERSION WAR with needed JARs"
        zip solr.war WEB-INF > /dev/null
#        rm -r WEB_INF/
        popd > /dev/null
    fi
    
    for S in `seq 1 $SOLRS`; do
        if [ ! -d solr$S ]; then
            echo "     - Copying Solr files for instance $S"
            cp -r $FOLDER solr$S

            # The three libraries below are needed for collator sorting
            if [ ! "." == ".`echo \" $LAYOUT2_VERSIONS \" | grep \" $DEST \"`" ]; then
                local SOLR_LIB=solr$S/server/solr/lib/
                mkdir -p $SOLR_LIB
                cp $FOLDER/dist/solr-analysis-extras-*.jar $SOLR_LIB
                cp $FOLDER/contrib/analysis-extras/lucene-libs/lucene-analyzers-icu-*.jar $SOLR_LIB
                cp $FOLDER/contrib/analysis-extras/lib/icu4j*.jar $SOLR_LIB
            fi
        else
            echo "     - Solr $S already exists"
        fi
    done
}

function install() {
    local VERSION="$1"

    # Shared setup
    mkdir -p ${CLOUD}
    DEST=$VERSION
    echo "- Installing SolrCloud $DEST at ${CLOUD}/$DEST"
    if [ $VERSION == 4.10.4-sparse ]; then
        SPARSE_WAR=sparse-4.10.war
        VERSION=4.10.4
    fi
    SPACK=solr-${VERSION}.tgz
    ZPACK=$(basename "$ZOO_URL")
    check_package $SPACK
    check_package $ZPACK
    echo "  - Source packages: $SPACK and $ZPACK"
    mkdir -p ${CLOUD}/$DEST
    pushd ${CLOUD}/$DEST > /dev/null
    zoo $ZPACK

    # Version specific parts
    
    if [ ! "." == ".`echo \" $VERSIONS \" | grep \" $DEST \"`" ]; then
        solr $SPACK
        popd > /dev/null # ${CLOUD}/$DEST
        return
    fi

    >&2 echo "Error: Support for Solr version '$DEST' has not been implemented yet"
    usage 1
}

###############################################################################
# CODE
###############################################################################

check_parameters "$@"

pushd ${BASH_SOURCE%/*} > /dev/null
if [[ "." == ".$@" ]]; then
    install "$VERSION"
else
    for V in $@; do
        install $V
    done
fi
popd > /dev/null