#!/bin/bash

# Pull down all the lambda code for the named stack

set -o errexit -o nounset -o pipefail

: "${DIST_DIR:="./dist"}"

if [[ $# -ne 1 ]]

then

echo "Error: stack-name argument is missing" 1>&2

echo "Usage: $0 <stack-name>" 1>&2

exit 1

fi

stack_name="$1"

function pull_lambda_code(){

local api_name="$1"

local endpoint_name="$2"

lambda_name="nhsd-nrlf--${stack_name}--api--${api_name}--${endpoint_name}"

echo -n "- Downloading code for lambda ${lambda_name}.... "

code_url="$(aws lambda get-function --function-name "${lambda_name}" | jq -r .Code.Location)"

curl "${code_url}" 2>/dev/null > "${DIST_DIR}/${api_name}-${endpoint_name}.zip"

echo "✅"

return 0

}

function pull_layer_code(){

local name="$1"

layer_name="nhsd-nrlf--${stack_name}--${name}"

layer_pkg_name="$(echo "${name}" | tr '-' '_').zip"

layer_version="$(aws lambda list-layer-versions --layer-name "${layer_name}" | jq -r '.LayerVersions[0].Version')"

echo -n "- Downloading code for layer ${layer_name} version ${layer_version}.... "

code_url="$(aws lambda get-layer-version --layer-name "${layer_name}" --version-number "${layer_version}" | jq -r .Content.Location)"

curl "${code_url}" 2>/dev/null > "${DIST_DIR}/${layer_pkg_name}"

echo "✅"

return 0

}

mkdir -p "${DIST_DIR}"

echo

echo "Pulling code for consumer API lambdas...."

for endpoint_path in api/consumer/*

do

if [[ ! -d "${endpoint_path}" ]]

then

continue

fi

endpoint_name="$(basename "${endpoint_path}")"

pull_lambda_code "consumer" "${endpoint_name}"

done

echo

echo "Pulling code for producer API lambdas...."

for endpoint_path in api/producer/*

do

if [[ ! -d "${endpoint_path}" ]]

then

continue

fi

endpoint_name="$(basename "${endpoint_path}")"

pull_lambda_code "producer" "${endpoint_name}"

done

echo

echo "Pulling code for layers...."

for layer_name in nrlf dependency-layer nrlf-permissions

do

pull_layer_code "${layer_name}"

done

echo

echo "✅ Done. Code is in ${DIST_DIR}"

echo