从带有参数的bash脚本中导出环境变量

Question

我正在尝试使用带有一个参数的 bash 脚本导出环境变量。我尝试使用 source 命令运行脚本，但它不起作用。

source ./script.sh dev

下面是我的示例脚本

#!/bin/bash
# Check 1 argument is passed with the script
if [ $# -ne 1 ]
then
        echo "Usage : $0 AWS account: e.g. $0 dev"
        exit 0
fi

# convert the input to uppercase
aws_env=$( tr '[:lower:]' '[:upper:]' <<<"$1" )


if ! [[ "$aws_env" =~ ^(DEV|UAT|TRN|PROD)$ ]]; then 
    # check that correct account is provided
    echo "Enter correct AWS account: dev or uat or trn or prod"
    exit 0

else
    # export environment variables
    file="/home/xyz/.aws/key_${aws_env}"
    IFS=$'\n' read -d '' -r -a lines < $file
    export AWS_ACCESS_KEY_ID=${lines[0]}
    export AWS_SECRET_ACCESS_KEY=${lines[1]}
    export AWS_DEFAULT_REGION=ap-southeast-2
    echo "AWS access keys and secret has been exported as environment variables for $aws_env account"
fi

文件 /home/xyz/.aws/key_DEV 的内容。它只是一个示例，不是真正的键。

$ cat key_DEV 
123
xyz

当我说它不起作用时，终端没有任何反应，当我运行脚本时它会关闭终端。

进一步更新：

当我从没有源 (./script.sh dev) 的终端按原样运行脚本时，它似乎工作正常，通过调试 (set -x)，我可以看到所有输出都是正确的。

但是，问题是当我使用源 (source ./script.sh dev) 运行时，它失败了（关闭终端，现在我知道原因了，因为退出 0），并且从调试命令捕获的输出中我可以看到它没有正确捕获 $1 参数。错误消息"Enter correct AWS account: dev or uat or trn or prod"。并且，$aws_env 变量的值为空。

我不知道为什么这两种行为不同以及如何解决它。

最终更新：

脚本看起来不错。这个问题是我电脑的本地问题。 tr 被定义为导致问题的 .bashrc 文件中的别名。我只是使用typeset -u aws_env="$1" 而不是aws_env=$( tr '[:lower:]' '[:upper:]' <<<"$1" )。感谢大家帮助我解决这个问题，特别是@markp-fuso。

Answer 1

尝试像这样使用mapfile 而不是read...

#!/usr/bin/env bash

# Check 1 argument is passed with the script
if [ $# -ne 1 ]
then
        echo "Usage : $0 AWS account: e.g. $0 dev"
        exit 0
fi

# convert the input to uppercase
aws_env=$( tr '[:lower:]' '[:upper:]' <<<"$1" )


if ! [[ "$aws_env" =~ ^(DEV|UAT|TRN|PROD)$ ]]; then 
    # check that correct account is provided
    echo "Enter correct AWS account: dev or uat or trn or prod"
    exit 0

else
    # export environment variables
    file="/home/xyz/.aws/key_${aws_env}"

    # IFS=$'\n' read -d '' -r -a lines < $file
    mapfile -t lines < "$file"

    export AWS_ACCESS_KEY_ID=${lines[0]}
    export AWS_SECRET_ACCESS_KEY=${lines[1]}
    export AWS_DEFAULT_REGION=ap-southeast-2
    echo "AWS access keys and secret has been exported as environment variables for $aws_env account"
fi

确保您还在"$file" 周围加上引号

另一个小技巧，Bash 支持直接将变量设为大写，如下所示：

var="upper"
echo "${var^^}"