68 lines
1.7 KiB
Bash
68 lines
1.7 KiB
Bash
if command -v aws_completer &> /dev/null; then
|
|
complete -C "$(command -v aws_completer)" aws
|
|
fi
|
|
|
|
_CACHE_DIR="${XDG_CACHE_HOME:-${HOME}/.cache}/aws_profile_set"
|
|
mkdir -p "${_CACHE_DIR}"
|
|
|
|
_PREVIOUS_AWS_PROFILE_FILE="${_CACHE_DIR}/last_set_value"
|
|
|
|
aws_profile_set() {
|
|
local AWS_CONFIG_PATH="${HOME}/.aws/config"
|
|
if ! [ -r "${AWS_CONFIG_PATH}" ]; then
|
|
>&2 echo "${AWS_CONFIG_PATH} does not exist"
|
|
return 1
|
|
fi
|
|
|
|
local PROFILE=""
|
|
|
|
PROFILE="$(awk '/^\[profile /{print substr($2, 1, length($2)-1)}' \
|
|
"${AWS_CONFIG_PATH}" |
|
|
fzf --query "${1:-}" -1 -m 1 --height="$((LINES/4))"
|
|
)"
|
|
|
|
if [ -z "${PROFILE}" ]; then
|
|
>&2 echo "failed to select a profile"
|
|
return 2
|
|
fi
|
|
|
|
echo "setting AWS profile to ${PROFILE}"
|
|
|
|
echo "$PROFILE" > "$_PREVIOUS_AWS_PROFILE_FILE"
|
|
export AWS_PROFILE="$PROFILE"
|
|
|
|
export _ORIG_PS1="${_ORIG_PS1:-$PS1}"
|
|
|
|
export PS1="🌩 ${AWS_PROFILE}🌩 ${_ORIG_PS1}"
|
|
}
|
|
|
|
if [ -s "$_PREVIOUS_AWS_PROFILE_FILE" ]; then
|
|
aws_profile_set "$(<"$_PREVIOUS_AWS_PROFILE_FILE")" < /dev/null &> /dev/null
|
|
fi
|
|
|
|
aws_logs() {
|
|
(
|
|
set -euo pipefail
|
|
|
|
if [ -z "${LOG_GROUP:-}" ]; then
|
|
LOG_GROUP="$(aws logs describe-log-groups | jq -r '.logGroups | .[] | .logGroupName' | fzf -1)"
|
|
fi
|
|
if [ -z "${LOG_STREAM_PREFIX:-}" ]; then
|
|
LOG_STREAM_PREFIX="$(aws logs describe-log-streams --log-group-name "$LOG_GROUP" | jq -r '.logStreams | .[] | .logStreamName' | fzf -1 --print-query | head -n 1)"
|
|
fi
|
|
|
|
if [ -n "$LOG_STREAM_PREFIX" ]; then
|
|
aws logs tail "${LOG_GROUP}" \
|
|
--log-stream-name-prefix "$LOG_STREAM_PREFIX" \
|
|
"$@"
|
|
|
|
exit $?
|
|
fi
|
|
|
|
aws logs tail "${LOG_GROUP}" "$@"
|
|
)
|
|
}
|
|
|
|
aws_json_logs() {
|
|
aws_logs --format json "$@" | grep -Ev '^[0-9]{4}-[0-9]{1,2}-[0-9]{1,2}T'
|
|
}
|