run_docker.sh

#!/bin/bash
#-*- mode: shell-script; -*-
#
# Copyright (C) 2020 Simons Observatory
#
# Keywords: docker, pspipe
# Requirements: docker
# Status: not intended to be distributed yet

pspipe_workspace=${PSPIPE_WORKSPACE:-/tmp/workspace}
githubs=(
    "https://github.com/simonsobs/pixell"
    "https://github.com/simonsobs/pspy"
    "https://github.com/LSSTDESC/NaMaster,easier_libsharp"
    "https://github.com/simonsobs/LAT_MFLike"
    "https://github.com/simonsobs/PSpipe"
)

function msg_error()
{
    echo -en "\\033[0;31mERROR: $@\\033[0;39m\n"
    return 0
}

function msg_notice()
{
    echo -en "\\033[0;34mNOTICE: $@\\033[0;39m\n"
    return 0
}

function msg_warning()
{
    echo -en "\\033[0;35mWARNING: $@\\033[0;39m\n"
    return 0
}

function prepare_workspace() {
    echo -en "\\033[0;34mNOTICE: Location of PSpipe workspace [default ${pspipe_workspace}]"
    if [ ! -z ${PSPIPE_WORKSPACE} ]; then
        echo -e "$@\\033[0;39m"
        return 0
    else
        echo -en ": "
        read -e ws
    fi
    echo -e "$@\\033[0;39m"
    [[ ! -z ${ws} ]] && pspipe_workspace=${ws}
    if [ ! -d "${pspipe_workspace}" ]; then
        echo -e "\\033[0;35mWARNING: Directory '${pspipe_workspace}' does not exist. Create it ?"
        yesno=("yes" "no")
        select ans in "${yesno[@]}"; do
            case $ans in
                [Yy]*)
                    mkdir -p ${pspipe_workspace}
                    break
                    ;;
                [Nn]*)
                    msg_warning "No workspace has been set."
                    return 1
                    break
                    ;;
            esac
        done
        echo -e "\\033[0;39m"
    fi

    (
        cd ${pspipe_workspace}
        for gh in "${githubs[@]}"; do
            local http=$(echo ${gh} | awk -F, '{print $1}')
            local branch=$(echo ${gh} | awk -F, '{print $2}')
            local name=$(echo ${http} | awk -F/ '{print $NF}')
            if [ ! -d ${name} ]; then
                msg_notice "Cloning '${name}' into ${psipe_workspace}..."
                git clone ${http}
                [[ ! -z ${branch} ]] && git checkout ${branch} -b ${branch}
                continue
            fi
            (
                msg_notice "Updating '${name}'..."
                cd ${name}
                git pull
                [[ ! -z ${branch} ]] && git checkout ${branch}
            )
        done
    )
    cid_file="${pspipe_workspace}/.cid"
    return 0
}

function start_docker() {
    msg_notice "Starting PSpipe docker..."
    msg_notice "Type exit when you are done"
    [[ -f ${cid_file} ]] && rm -f ${cid_file}

    local docker_options="-it -p 8888:8888"
    if [ -z ${cid_file} ]; then
        docker_options+=" --rm"
    else
        docker_options+=" --cidfile ${cid_file} -v ${pspipe_workspace}:/home/pspipe/workspace"
    fi

    docker run $(echo ${docker_options}) simonsobs/pspipe:latest /bin/bash
    return 0
}

function stop_docker() {
    msg_notice "Stopping PSpipe docker."
    if [ ! -z ${cid_file} ]; then
        cid_value=$(cat ${cid_file})
        docker commit ${cid_value} simonsobs/pspipe:latest > /dev/null 2>&1
        docker rm ${cid_value} > /dev/null 2>&1
    fi
    return 0
}

function main() {
    prepare_workspace
    start_docker
    stop_docker
}

main