asteris-llc/converge

#!/bin/sh

# Copyright © 2016 Asteris, LLC
#
# 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.

# This script downloads converge to a system.
# If possible, the Operating System and processor
# type will be detected.

# install-converge.sh [-v <version>]
# -d            Directory for converge binary (default /usr/local/bin/)
# -v            Converge version to install

set -e

__run() {
    # Default to latest stable
    version="0.6.0"

    install_dir="/usr/local/bin"

    base_url="https://github.com/asteris-llc/converge/releases/download"

    machine=$(uname -m)
    os=$(uname -s)

    # Get a temporary directory
    if [ -z "$TMPDIR" ]; then
        tmp="/tmp"
    else
        tmp="$TMPDIR"
    fi
    tmp_dir="${tmp}/install-converge.sh.$$"
    (umask 077 && mkdir "${tmp_dir}") || exit 1


    check_dir () {
        if [ ! -d "$1" ]; then
            echo "Directory $1 does not exist, exiting"
            exit 1
        fi
    }

    usage () {
        echo "Usage: install-converge.sh [-v <version>]"
        echo " -d            Directory for converge binary (default ${install_dir})"
        echo " -v            Converge version to install   (default ${version})"
        if test "${#}" -gt 0; then
            echo
            echo "${@}"
        fi
        exit 1
    }

    not_supported () {
        echo "Unable to download converge binary for $os-$machine"
        echo "Most likely means that there is not an official release for that os+machine pair"
        exit 1
    }

    do_wget () {
        echo "Trying wget..."
        wget -O "$2" "$1" 2>"$tmp_dir/stderr"
        rc="$?"

        grep "ERROR404" "$tmp_dir/stderr" >/dev/null 2>&1 && not_supported

        if test "$rc" -ne 0 || test ! -s "$2"; then
            return 1
        fi

        return 0
    }

    do_curl () {
        echo "Trying curl..."
        echo "$1"
        curl --retry 5 -sL -D "$tmp_dir/stderr" "$1" > "$2"
        rc="$?"

        grep "404 Not Found" "$tmp_dir/stderr" >/dev/null 2>&1 && not_supported

        if test "$rc" -ne 0 || test ! -s "$2"; then
            return 1
        fi

        return 0
    }

    do_fetch() {
        echo "Trying fetch..."
        fetch -o "$2" "$1" 2>"$tmp_dir/stderr"
        test "$?" -ne 0 && return 1
        return 0
    }

    do_perl () {
        echo "Trying perl..."
        perl -r 'use LWP::Simple; getprint($ARGV[0]);' "$1" > "$2" 2>"$tmp_dir/stderr"
        rc="$?"

        grep "404 Not Found" "$tmp_dir/stderr" >/dev/null 2>&1 && not_supported

        if test "$rc" -ne 0 || test ! -s "$2"; then
            return 1
        fi

        return 0
    }

    do_python () {
        echo "Trying python..."
        python -c "import sys,urllib2 ; sys.stdout.write(urllib2.urlopen(sys.argv[1]).read())" "$1" > "$2" 2>"$tmp_dir/stderr"

        rc="$?"

        grep "HTTP Error 404" "$tmp_dir/stderr" >/dev/null 2>&1 && not_supported

        if test "$rc" -ne 0 || test ! -s "$2"; then
            return 1
        fi

        return 0
    }

    exists() {
        command -v "$1" >/dev/null 2>&1
    }

    do_download () {
        url="$1"
        dest="$2"

        if exists wget; then
            do_wget "$url" "$dest" && return 0
        fi

        if exists curl; then
            do_curl "$url" "$dest" && return 0
        fi

        if exists fetch; then
            do_fetch "$url" "$dest" && return 0
        fi

        if exists perl; then
            do_perl "$url" "$dest" && return 0
        fi

        if exists python; then
            do_python "$url" "$dest" && return 0
        fi

        echo "Unable to retrieve package"
        exit 1
    }

    extract() {
        file="$1"
        dest="$2"
        cd "$dest" && tar zxvf "$file" converge
    }

    while getopts :d:r:v: opt; do
        case "$opt" in
        d)
            install_dir="$OPTARG"
            ;;
        v)
            version="$OPTARG"
            ;;
        \?)
            usage "Invalid flag: $OPTARG"
            ;;
        esac
    done

    case "${machine}" in
        "x86_64" | "amd64" | "x64")
            machine="amd64"
            ;;
        "i386" | "i86pc" | "x86" | "i686")
            machine="386"
            ;;
        "armv6l" | "armv7l")
            machine="arm"
            ;;
        "arm" | "arm64" | "ppc64" | "ppc64le")
            # Nothing required
            ;;
        *)
            echo "Unsupported machine type: ${machine}"
            exit 1
            ;;
    esac

    case "${os}" in
        "Darwin")
            os="darwin"
            ;;
        "FreeBSD")
            os="freebsd"
            ;;
        "OpenBSD")
            os="openbsd"
            ;;
        "Linux")
            os="linux"
            ;;
        "solaris")
            # Nothing to do
            ;;
        *)
            echo "Unsupported OS type: $os"
            exit 1
            ;;
    esac

    check_dir "${install_dir}"
    do_download "${base_url}/${version}/converge_${version}_${os}_${machine}.tar.gz" "${tmp_dir}/converge_${version}_${os}_${machine}.tar.gz"
    extract "${tmp_dir}/converge_${version}_${os}_${machine}.tar.gz" "${install_dir}"
    chmod 0755 "${install_dir}/converge"

    if [ -n "$tmp_dir" ]; then
        rm -r "${tmp_dir}"
    fi
}

__run