system-setup

#!/bin/bash

_APT_UPDATED=false
MEPH2_BRANCH="lp:maas-images"
SSTREAM_REPO="https://git.launchpad.net/simplestreams"
SSTREAM_BRANCH="master"
CROSS_DEPS=( qemu-user-static u-boot-tools )
MEPH2_D=""
# MEPH2_SIMPLESTREAMS can be:
# 'package': install the package
# 'system': use whatever 'import' provides
# 'branch': use the MEPH2_BRANCH
# 'auto': use system if available or distro (> trusty), or branch
MEPH2_SIMPLESTREAMS=${MEPH2_SIMPLESTREAMS:-"auto"}
MEPH2_UTIL_ONLY=${MEPH2_UTIL_ONLY:-"auto"}

error() { echo "$@" 1>&2; }
fail() { [ $# -eq 0 ] || error "$@"; exit 2; }

rel="$(lsb_release -sc)"
DEPS_UTIL_ONLY=(
    bzr
    python3
    python3-yaml
)

DEPS_FULL=(
    bzr
    cloud-image-utils      # for mount-image-callback
    dpkg                   # To compare bootloader package versions
    gpgv
    grub-common            # Used by the bootloaders
    python3-distro-info
    python3-requests
    python3-yaml
    squashfs-tools
    zerofree
)

apt_get() {
    local ret=""
    if [ "$1" != "update" ] && ! $_APT_UPDATED; then
        error "updating apt"
        apt_get update >/dev/null || {
            ret=$?;
            error "failed to update apt [$ret]";
            return $ret;
        }
        _APT_UPDATED=true
    fi
    sudo DEBIAN_FRONTEND=noninteractive apt-get --quiet \
        --assume-yes "$@" </dev/null
}

filter_installed_packages() {
    # write to stdout, a list of packages not installed locally
    local fmt='${Package} ${Version}\n'
    LC_ALL=C dpkg-query --show "--showformat=${fmt}" "$@" 2>&1 | awk '
        $0 ~ /[Nn]o packages/ {
            sub("[.]$","",$NF);
            pkgs[n]=$NF;
            n=n+1;
        }
        $2 == "" {
                pkgs[n]=$1;
                n=n+1;
        };
        END { for(p in pkgs) {printf("%s ",pkgs[p])}; printf("\n"); }' n=0
}

apt_install() {
    local needed
    needed=$(filter_installed_packages "$@")
    [ -z "$needed" ] && return 0
    error "installing: $needed"
    apt_get install "$@"
}


streams_package=true
def_meph2_only=false

case "$MEPH2_UTIL_ONLY" in
    false|0) meph2_only=false;;
    auto) meph2_only=$def_meph2_only;;
    *) meph2_only=true;;
esac

if $meph2_only; then
    DEPS=( "${DEPS_UTIL_ONLY[@]}" )
else    
    DEPS=( "${DEPS_FULL[@]}" )
fi

if [ "$MEPH2_SIMPLESTREAMS" = "auto" ]; then
    if python3 -c 'import simplestreams' >/dev/null 2>&1; then
       MEPH2_SIMPLESTREAMS="system"
    elif $streams_package; then
       MEPH2_SIMPLESTREAMS="package"
    else
       MEPH2_SIMPLESTREAMS="branch"
    fi
fi
[ "$MEPH2_SIMPLESTREAMS" = "package" ] &&
    DEPS=( "${DEPS[@]}" python3-simplestreams )

me=$(readlink -f "$0")
myd=$(dirname "$me")
if [ -f "$myd/bin/meph2-cloudimg-sync" ]; then
    MEPH2_D="$myd"
elif [ -f "$PWD/maas-images/bin/meph2-cloudimg-sync" ]; then
    MEPH2_D="$PWD/maas-images"
fi

error "MEPH2_SIMPLESTREAMS=$MEPH2_SIMPLESTREAMS MEPH2_UTIL_ONLY=$meph2_only"

arch=$(dpkg --print-architecture)
deps=( "${DEPS[@]}" )
if [ "$arch" = "amd64" ] && ! $meph2_only; then
    deps=( "${deps[@]}" "${CROSS_DEPS[@]}" )
fi

apt_install "${deps[@]}"

if [ -z "$MEPH2_D" ]; then
    error "getting $MEPH2_BRANCH to $PWD/maas-images"
    bzr branch "$MEPH2_BRANCH" maas-images ||
        fail "failed branch $MEPH2_BRANCH"
    MEPH2_D="$PWD/maas-images"
fi

if [ "$MEPH2_SIMPLESTREAMS" = "branch" ]; then
    sstream_d="$MEPH2_D/simplestreams.git"
    if [ ! -d "$sstream_d" ]; then
        git clone "--branch=${SSTREAM_BRANCH}" \
                "${SSTREAM_REPO}" "${sstream_d}" &&
            ln -snf "${sstream_d}/simplestreams" "$MEPH2_D/simplestreams" ||
            fail "failed to get $SSTREAM_BRANCH"
        out=$(cd "$MEPH2_D" && python3 -c 'import simplestreams' 2>&1) ||
            fail "error getting simplestreams: $out"
    fi
fi

# vi: ts=4 expandtab