libBuild.sh

#!/usr/bin/env bash

set -Eeuo pipefail

# Regions that support arm64 architecture
REGIONS_ARM=(
  af-south-1
	ap-northeast-1
	ap-northeast-2
	ap-northeast-3
	ap-south-1
	ap-southeast-1
	ap-southeast-2
	ap-southeast-3
	ca-central-1
	eu-central-1
	eu-north-1
	eu-south-1
	eu-west-1
	eu-west-2
	eu-west-3
	me-south-1
	sa-east-1
	us-east-1
	us-east-2
	us-west-1
	us-west-2
)

REGIONS_X86=(
  af-south-1
  ap-northeast-1
  ap-northeast-2
  ap-northeast-3
  ap-south-1
  ap-south-2
  ap-southeast-1
  ap-southeast-2
  ap-southeast-3
  ap-southeast-4
  ca-central-1
  eu-central-1
  eu-central-2
  eu-north-1
  eu-south-1
  eu-south-2
  eu-west-1
  eu-west-2
  eu-west-3
  me-central-1
  me-south-1
  sa-east-1
  us-east-1
  us-east-2
  us-west-1
  us-west-2
)

EXTENSION_DIST_DIR=extensions
EXTENSION_DIST_ZIP=extension.zip
EXTENSION_DIST_PREVIEW_FILE=preview-extensions-ggqizro707

EXTENSION_VERSION=2.3.6

function list_all_regions {
    aws ec2 describe-regions \
      --all-regions \
      --query "Regions[].{Name:RegionName}" \
      --output text | sort
}

function fetch_extension {
    arch=$1

    url="https://github.com/newrelic/newrelic-lambda-extension/releases/download/v${EXTENSION_VERSION}/newrelic-lambda-extension.${arch}.zip"
    rm -rf $EXTENSION_DIST_DIR $EXTENSION_DIST_ZIP
    curl -L $url -o $EXTENSION_DIST_ZIP
}

function download_extension {
    fetch_extension $@

    unzip $EXTENSION_DIST_ZIP -d .
    rm -f $EXTENSION_DIST_ZIP
}

function layer_name_str() {
    rt_part="LambdaExtension"
    arch_part=""

    case $1 in
    "java8.al2")
      rt_part="Java8"
      ;;
    "java11")
      rt_part="Java11"
      ;;
    "python3.7")
      rt_part="Python37"
      ;;
    "python3.8")
      rt_part="Python38"
      ;;
    "python3.9")
      rt_part="Python39"
      ;;
    "python3.10")
      rt_part="Python310"
      ;;
    "nodejs14.x")
      rt_part="NodeJS14X"
      ;;
    "nodejs16.x")
      rt_part="NodeJS16X"
      ;;
    "nodejs18.x")
      rt_part="NodeJS18X"
      ;;
    esac

    case $2 in
    "arm64")
      arch_part="ARM64"
      ;;
    "x86_64")
      arch_part=""
      ;;
    esac

    echo "NewRelic${rt_part}${arch_part}"
}

function s3_prefix() {
    name="nr-extension"

    case $1 in
    "java8.al2")
      name="java-8"
      ;;
    "java11")
      name="java-11"
      ;;
    "python3.7")
      name="nr-python3.7"
      ;;
    "python3.8")
      name="nr-python3.8"
      ;;
    "python3.9")
      name="nr-python3.9"
      ;;
    "python3.10")
      name="nr-python3.10"
      ;;
    "nodejs14.x")
      name="nr-nodejs14.x"
      ;;
    "nodejs16.x")
      name="nr-nodejs16.x"
      ;;
    "nodejs18.x")
      name="nr-nodejs18.x"
      ;;
    esac

    echo $name
}

function hash_file() {
    if command -v md5sum &> /dev/null ; then
        md5sum $1 | awk '{ print $1 }'
    else
        md5 -q $1
    fi
}

function publish_layer {
    layer_archive=$1
    region=$2
    runtime_name=$3
    arch=$4

    layer_name=$( layer_name_str $runtime_name $arch )

    hash=$( hash_file $layer_archive | awk '{ print $1 }' )

    bucket_name="nr-layers-${region}"
    s3_key="$( s3_prefix $runtime_name )/${hash}.${arch}.zip"

    compat_list=( $runtime_name )
    if [[ $runtime_name == "provided" ]]
    then compat_list=("provided" "provided.al2" "dotnetcore3.1" "dotnet6")
    fi

    echo "Uploading ${layer_archive} to s3://${bucket_name}/${s3_key}"
    aws --region "$region" s3 cp $layer_archive "s3://${bucket_name}/${s3_key}"

   if [[ ${REGIONS_ARM[*]} =~ $region ]];
   then arch_flag="--compatible-architectures $arch"
   else arch_flag=""
   fi

    echo "Publishing ${runtime_name} layer to ${region}"
    layer_version=$(aws lambda publish-layer-version \
      --layer-name ${layer_name} \
      --content "S3Bucket=${bucket_name},S3Key=${s3_key}" \
      --description "New Relic Layer for ${runtime_name} (${arch})" \
      --license-info "Apache-2.0" $arch_flag \
      --compatible-runtimes ${compat_list[*]} \
      --region "$region" \
      --output text \
      --query Version)
    echo "Published ${runtime_name} layer version ${layer_version} to ${region}"

    echo "Setting public permissions for ${runtime_name} layer version ${layer_version} in ${region}"
    aws lambda add-layer-version-permission \
      --layer-name ${layer_name} \
      --version-number "$layer_version" \
      --statement-id public \
      --action lambda:GetLayerVersion \
      --principal "*" \
      --region "$region"
    echo "Public permissions set for ${runtime_name} layer version ${layer_version} in region ${region}"
}