upstream/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2025-04-08 18:43:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge 85f01240fb into fa396ad771

Browse Source
This commit is contained in:
Bean 2025-03-26 10:11:44 +09:00 committed by GitHub
commit 8190b60588
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
plugins/aws/aws.plugin.zsh
View File

@ -44,6 +44,8 @@ function asp() {
export AWS_PROFILE_REGION=$(aws configure get region)
aws-mfa $AWS_PROFILE
_aws_update_state
if [[ "$2" == "login" ]]; then
@ -57,6 +59,38 @@ function asp() {
fi
}
function aws-mfa() {
local -a available_profiles
available_profiles=($(aws_profiles))
if [[ -z "${available_profiles[(r)$1]}" ]]; then
echo "${fg[red]}Profile '$1' not found in '${AWS_CONFIG_FILE:-$HOME/.aws/config}'" >&2
echo "Available profiles: ${(j:, :)available_profiles:-no profiles found}${reset_color}" >&2
return 1
fi
local profile="$1"
local mfa_token
now=$(gdate +%s)
if [[ ! -z $(aws configure get --profile $profile mfa_serial) ]]; then
get_expire=`aws configure get --profile $profile expiration`
if [[ -z "${get_expire}" || "$now" -ge "$(gdate -d $get_expire +%s)" ]]; then
echo -n "Please input MFA token: "
read -r mfa_token
local output=$(aws sts get-session-token --serial-number `aws configure get mfa_serial` --token-code $mfa_token --profile $profile-long-term)
local aws_access_key_id=$(echo $output | jq -r '.Credentials | .AccessKeyId')
local aws_secret_access_key=$(echo $output | jq -r '.Credentials | .SecretAccessKey')
local aws_session_token=$(echo $output | jq -r '.Credentials | .SessionToken')
local expiration=$(echo $output | jq -r '.Credentials | .Expiration')
aws configure set --profile $profile aws_access_key_id $aws_access_key_id
aws configure set --profile $profile aws_secret_access_key $aws_secret_access_key
aws configure set --profile $profile aws_session_token $aws_session_token
aws configure set --profile $profile expiration $expiration
else
echo "SESSION TOKEN STILL VALIDED!!"
fi
fi
}
# AWS region selection
function asr() {
if [[ -z "$1" ]]; then