add bash-it

2022-05-05 22:20:02 +02:00
parent 6a62a35d0d
commit 43701f0590
490 changed files with 33047 additions and 0 deletions
--- a/dot_bash_it/plugins/available/aws.plugin.bash
+++ b/dot_bash_it/plugins/available/aws.plugin.bash
@@ -0,0 +1,113 @@
+cite about-plugin
+about-plugin 'AWS helper functions'
+
+AWS_CONFIG_FILE="${AWS_CONFIG_FILE:-$HOME/.aws/config}"
+AWS_SHARED_CREDENTIALS_FILE="${AWS_SHARED_CREDENTIALS_FILE:-$HOME/.aws/credentials}"
+
+function awskeys {
+    about 'helper function for AWS credentials file'
+    group 'aws'
+
+    if [[ ! -f "${AWS_SHARED_CREDENTIALS_FILE}" ]]; then
+        echo "AWS credentials file not found"
+        return 1
+    fi
+
+    if [[ $# -eq 1 ]] && [[ "$1" = "list" ]]; then
+        __awskeys_list "$2"
+    elif [[ $# -eq 1 ]] && [[ "$1" = "unset" ]]; then
+        __awskeys_unset "$2"
+    elif [[ $# -eq 2 ]] && [[ "$1" = "show" ]]; then
+        __awskeys_show "$2"
+    elif [[ $# -eq 2 ]] && [[ "$1" = "export" ]]; then
+        __awskeys_export "$2"
+    else
+        __awskeys_help
+    fi
+}
+
+function __awskeys_help {
+    echo -e "Usage: awskeys [COMMAND] [profile]\n"
+    echo -e "Helper to AWS credentials file.\n"
+    echo -e "Commands:\n"
+    echo "   help    Show this help message"
+    echo "   list    List available AWS credentials profiles"
+    echo "   show    Show the AWS keys associated to a credentials profile"
+    echo "   export  Export an AWS credentials profile keys as environment variables"
+    echo "   unset   Unset the AWS keys variables from the environment"
+}
+
+function __awskeys_get {
+    local ln=$(grep -n "\[ *$1 *\]" "${AWS_SHARED_CREDENTIALS_FILE}" | cut -d ":" -f 1)
+    if [[ -n "${ln}" ]]; then
+        tail -n +${ln} "${AWS_SHARED_CREDENTIALS_FILE}" | egrep -m 2 "aws_access_key_id|aws_secret_access_key"
+        tail -n +${ln} "${AWS_SHARED_CREDENTIALS_FILE}" | egrep -m 1 "aws_session_token"
+    fi
+}
+
+function __awskeys_list {
+    local credentials_list="$((egrep '^\[ *[a-zA-Z0-9_-]+ *\]$' "${AWS_SHARED_CREDENTIALS_FILE}"; grep "\[profile" "${AWS_CONFIG_FILE}" | sed "s|\[profile |\[|g") | sort | uniq)"
+    if [[ -n $"{credentials_list}" ]]; then
+        echo -e "Available credentials profiles:\n"
+        for profile in ${credentials_list}; do
+            echo "    $(echo ${profile} | tr -d "[]")"
+        done
+        echo
+    else
+        echo "No profiles found in credentials file"
+    fi
+}
+
+function __awskeys_show {
+    local p_keys="$(__awskeys_get $1)"
+    if [[ -n "${p_keys}" ]]; then
+        echo "${p_keys}"
+    else
+        echo "Profile $1 not found in credentials file"
+    fi
+}
+
+function __awskeys_export {
+    if [[ $(__awskeys_list) == *"$1"* ]]; then
+        local p_keys=( $(__awskeys_get $1 | tr -d " ") )
+        if [[ -n "${p_keys}" ]]; then
+            for p_key in ${p_keys[@]}; do
+                local key="${p_key%=*}"
+                export "$(echo ${key} | tr [:lower:] [:upper:])=${p_key#*=}"
+            done
+        fi
+        export AWS_PROFILE="$1"
+    else
+        echo "Profile $1 not found in credentials file"
+    fi
+}
+
+function __awskeys_unset {
+    unset AWS_PROFILE AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN
+}
+
+function __awskeys_comp {
+    local cur prev opts prevprev
+    COMPREPLY=()
+    cur="${COMP_WORDS[COMP_CWORD]}"
+    prev="${COMP_WORDS[COMP_CWORD-1]}"
+
+    opts="help list show export unset"
+
+    case "${prev}" in
+        help|list|unset)
+            return 0
+            ;;
+        show|export)
+            local profile_list="$(__awskeys_list | grep "    ")"
+            COMPREPLY=( $(compgen -W "${profile_list}" -- ${cur}) )
+            return 0
+            ;;
+    esac
+
+    COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+
+    return 0
+}
+
+complete -F __awskeys_comp awskeys