1284942Sbrd#!/bin/sh 2284942Sbrd#- 3284942Sbrd# Redistribution and use in source and binary forms, with or without 4284942Sbrd# modification, are permitted provided that the following conditions 5284942Sbrd# are met: 6284942Sbrd# 1. Redistributions of source code must retain the above copyright 7284942Sbrd# notice, this list of conditions and the following disclaimer. 8284942Sbrd# 2. Redistributions in binary form must reproduce the above copyright 9284942Sbrd# notice, this list of conditions and the following disclaimer in the 10284942Sbrd# documentation and/or other materials provided with the distribution. 11284942Sbrd# 12284942Sbrd# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 13284942Sbrd# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 14284942Sbrd# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 15284942Sbrd# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 16284942Sbrd# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 17284942Sbrd# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 18284942Sbrd# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 19284942Sbrd# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 20284942Sbrd# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 21284942Sbrd# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 22284942Sbrd# SUCH DAMAGE. 23284942Sbrd# 24284942Sbrd# Upload a Vagrant image to Hashicorp's Atlas service 25284942Sbrd# 26284942Sbrd# $FreeBSD: stable/10/release/scripts/atlas-upload.sh 320697 2017-07-06 00:43:43Z gjb $ 27284942Sbrd# 28284942Sbrd 29284942SbrdATLAS_API_URL='' 30320697SgjbATLAS_UPLOAD_URL='https://app.vagrantup.com' 31285814SgjbDESCRIPTION="FreeBSD Snapshot Build" 32284942Sbrd 33284942Sbrdusage() { 34284942Sbrd echo "${0} usage:" 35285814Sgjb echo "-b box-name -d 'box description' -f box-to-upload -k api-key -p provider -u user -v version" 36284942Sbrd return 1 37284942Sbrd} 38284942Sbrd 39284942Sbrdmain () { 40285814Sgjb while getopts "b:d:f:k:p:u:v:" arg; do 41284942Sbrd case "${arg}" in 42284942Sbrd b) 43284942Sbrd BOX="${OPTARG}" 44284942Sbrd ;; 45285814Sgjb d) 46285814Sgjb DESCRIPTION="${OPTARG}" 47285814Sgjb ;; 48284942Sbrd f) 49284942Sbrd FILE="${OPTARG}" 50284942Sbrd ;; 51284942Sbrd k) 52284942Sbrd KEY="${OPTARG}" 53284942Sbrd ;; 54284942Sbrd p) 55284942Sbrd PROVIDER="${OPTARG}" 56284942Sbrd ;; 57284942Sbrd u) 58284942Sbrd USERNAME="${OPTARG}" 59284942Sbrd ;; 60284942Sbrd v) 61284942Sbrd VERSION="${OPTARG}" 62284942Sbrd ;; 63284942Sbrd *) 64284942Sbrd ;; 65284942Sbrd esac 66284942Sbrd done 67284942Sbrd 68284942Sbrd if [ -z "${BOX}" -o \ 69284942Sbrd -z "${FILE}" -o \ 70284942Sbrd -z "${KEY}" -o \ 71284942Sbrd -z "${PROVIDER}" -o \ 72284942Sbrd -z "${USERNAME}" -o \ 73284942Sbrd -z "${VERSION}" ]; 74284942Sbrd then 75284942Sbrd usage || exit 0 76284942Sbrd fi 77284942Sbrd 78284942Sbrd # Check to see if the box exists or create it 79320697Sgjb BOXRESULT=$(/usr/local/bin/curl -s "${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}?access_token=${KEY}") 80284942Sbrd if [ $? != 0 ]; then 81284942Sbrd echo "Failed to connect to the API" 82284942Sbrd exit 2; 83284942Sbrd fi 84284942Sbrd echo $BOXRESULT | grep "\"name\":\"${BOX}\"" > /dev/null 85284942Sbrd if [ $? != 0 ]; then 86284942Sbrd echo "Creating box: ${BOX}" 87320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/boxes -X POST -d "box[name]=${BOX}" -d "access_token=${KEY}" > /dev/null 88320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX} -X PUT -d "box[is_private]=false" -d "access_token=${KEY}" > /dev/null 89320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX} -X PUT -d "box[description]='${DESCRIPTION}'" -d "access_token=${KEY}" > /dev/null 90284942Sbrd else 91284942Sbrd echo "Box already exists" 92284942Sbrd fi 93284942Sbrd 94284942Sbrd # Check to see if the version exists or create it 95320697Sgjb VERSIONRESULT=$(/usr/local/bin/curl -s "${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}?access_token=${KEY}") 96284942Sbrd if [ $? != 0 ]; then 97284942Sbrd echo "Failed to connect to the API" 98284942Sbrd exit 2; 99284942Sbrd fi 100320697Sgjb echo $VERSIONRESULT | grep "version/${VERSION}" > /dev/null 101284942Sbrd if [ $? != 0 ]; then 102284942Sbrd echo "Creating version: ${VERSION}" 103320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/versions -X POST -d "version[version]=${VERSION}" -d "access_token=${KEY}" > /dev/null 104320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION} -X PUT -d "version[description]=${DESCRIPTION}" -d "access_token=${KEY}" > /dev/null 105320697Sgjb VERSIONRESULT=$(/usr/local/bin/curl -s "${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}?access_token=${KEY}") 106320697Sgjb echo $VERSIONRESULT | grep "version/${VERSION}" > /dev/null 107284942Sbrd if [ $? != 0 ]; then 108284942Sbrd echo "Failed to create version" 109284942Sbrd exit 2 110284942Sbrd fi 111284942Sbrd else 112284942Sbrd echo "Version already exists" 113284942Sbrd fi 114284942Sbrd 115284942Sbrd # Check to see if the provider exists or create it 116320697Sgjb PROVIDERRESULT=$(/usr/local/bin/curl -s "${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}/provider/${PROVIDER}?access_token=${KEY}") 117284942Sbrd if [ $? != 0 ]; then 118284942Sbrd echo "Failed to connect to the API" 119284942Sbrd exit 2; 120284942Sbrd fi 121320697Sgjb echo $PROVIDERRESULT | grep "provider/${PROVIDER}" > /dev/null 122284942Sbrd if [ $? != 0 ]; then 123284942Sbrd echo "Creating provider: ${PROVIDER}" 124320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}/providers -X POST -d "provider[name]=${PROVIDER}" -d "access_token=${KEY}" > /dev/null 125284942Sbrd else 126284942Sbrd echo "Provider already exists" 127284942Sbrd fi 128284942Sbrd 129284942Sbrd # Request an upload token 130320697Sgjb TOKENRESULT=$(/usr/local/bin/curl -s "${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}/provider/${PROVIDER}/upload?access_token=${KEY}") 131284942Sbrd if [ $? != 0 ]; then 132284942Sbrd echo "Failed to get the token from the API" 133284942Sbrd exit 2; 134284942Sbrd fi 135320697Sgjb echo ${TOKENRESULT} | grep -E "upload_path" > /dev/null 136284942Sbrd if [ $? != 0 ]; then 137284942Sbrd echo "No token found from the API" 138284942Sbrd exit 2 139284942Sbrd else 140318264Sgjb TOKEN=$(echo $TOKENRESULT | sed -e 's/.*token":"//' -e 's/.*upload_path":"//' -e 's/}$//g' -e 's/"//g') 141284942Sbrd echo "Uploading to Atlas" 142320697Sgjb UPLOADRESULT=$(/usr/local/bin/curl -s -X PUT --upload-file ${FILE} "${TOKEN}") 143284942Sbrd 144284942Sbrd # Validate the Upload 145284942Sbrd echo "Validating" 146320697Sgjb VALIDRESULT=$(/usr/local/bin/curl -s "${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}/provider/${PROVIDER}?access_token=${KEY}") 147318264Sgjb HOSTED_TOKEN=$(echo $VALIDRESULT | sed -e 's/.*"hosted"://' -e 's/,.*$//') 148318264Sgjb if [ ! -z ${TOKEN} -a "${HOSTED_TOKEN}" != "true" ]; then 149284942Sbrd echo "Upload failed, try again." 150284942Sbrd exit 2 151284942Sbrd fi 152284942Sbrd 153284942Sbrd # Release the version 154284942Sbrd echo "Releasing ${VERSION} of ${BOX} in Atlas" 155320697Sgjb /usr/local/bin/curl -s ${ATLAS_UPLOAD_URL}/api/v1/box/${USERNAME}/${BOX}/version/${VERSION}/release -X PUT -d "access_token=${KEY}" > /dev/null 156284942Sbrd fi 157284942Sbrd} 158284942Sbrd 159284942Sbrdmain "$@" 160