Skip to content

Persist awsctx context across shells #74

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
lox wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Persist awsctx context across shells #74

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
88 changes: 88 additions & 0 deletions zsh/aws.zsh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,83 @@ _awsctx_regions() {
print -l ${(ou)regions}
}

_awsctx_state_file() {
print -r -- "${XDG_STATE_HOME:-$HOME/.local/state}/dotfiles/awsctx"
}

_awsctx_save_state() {
emulate -L zsh
setopt local_options no_aliases pipe_fail

local profile="$1"
local region="$2"
local state_file="$(_awsctx_state_file)"
local state_dir="${state_file:h}"
local tmp_file

mkdir -p -- "$state_dir" || return 1
tmp_file="$(mktemp "${state_file}.tmp.XXXXXX")" || return 1

(
umask 077
{
print -r -- "profile=${profile}"
print -r -- "region=${region}"
} > "$tmp_file"
) || {
rm -f -- "$tmp_file"
return 1
}

mv -f -- "$tmp_file" "$state_file" || {
rm -f -- "$tmp_file"
return 1
}
}

_awsctx_clear_state() {
local state_file="$(_awsctx_state_file)"
rm -f -- "$state_file"
}

_awsctx_restore_state() {
emulate -L zsh
setopt local_options no_aliases pipe_fail

local state_file="$(_awsctx_state_file)"
[[ -r "$state_file" ]] || return 1

local saved_profile=""
local saved_region=""
local key value

while IFS='=' read -r key value; do
case "$key" in
profile)
saved_profile="$value"
;;
region)
saved_region="$value"
;;
esac
done < "$state_file"

local restored=1

if [[ -z "${AWS_PROFILE:-}" ]] && [[ -n "$saved_profile" ]]; then
export AWS_PROFILE="$saved_profile"
restored=0
fi

if [[ -z "${AWS_REGION:-${AWS_DEFAULT_REGION:-}}" ]] && [[ -n "$saved_region" ]]; then
export AWS_REGION="$saved_region"
export AWS_DEFAULT_REGION="$saved_region"
restored=0
fi

return "$restored"
}

_awssh_instances() {
command -v aws >/dev/null 2>&1 || return

Expand Down Expand Up @@ -140,11 +217,14 @@ Examples:
awsctx buildkite-sandbox-pipelines-admin ap-southeast-2
awsctx buildkite-sandbox-pipelines-admin
awsctx --clear

Context is persisted across new shells. Use --clear to unset and forget it.
EOF
return 0
;;
off|clear|--clear)
unset AWS_PROFILE AWS_REGION AWS_DEFAULT_REGION
_awsctx_clear_state
echo "AWS context cleared"
return 0
;;
Expand All @@ -171,6 +251,10 @@ EOF
unset AWS_REGION AWS_DEFAULT_REGION
fi

if ! _awsctx_save_state "$AWS_PROFILE" "${AWS_REGION:-${AWS_DEFAULT_REGION:-}}"; then
echo "warning: unable to persist AWS context" >&2
fi

echo "AWS context: ${AWS_PROFILE}${AWS_REGION:+ (${AWS_REGION})}"
}

Expand Down Expand Up @@ -203,6 +287,8 @@ EOF
return 127
fi

_awsctx_restore_state >/dev/null 2>&1 || true

aws ssm start-session --target "$instance_id" "$@"
}

Expand Down Expand Up @@ -274,3 +360,5 @@ _awssh() {
;;
esac
}

_awsctx_restore_state >/dev/null 2>&1 || true
Loading