nu_scripts/modules/aws/select-aws-profile.nu

# This alias lets you choose your aws environment variables with ease.
#
# Dependencies
#    * fzf
#
# Installation
#     1. store in ~/.config/nushell/select-aws-profile.nu
#     2. add to your config.nu: `use ~/.config/nushell/select-aws-profile.nu *`
#
# Usage
#     select-aws-profile
export def select-aws-profile [] {
    hide AWS_REGION;

    (do {
        let creds = (open ($env.HOME + "/.aws/credentials") | from toml)
            let selectedProfile = (for it in ($creds | transpose name creds) {
                echo $it.name
            })
            
            selectedProfile = selectedProfile  | str join "\n" | fzf | str trim

            if $selectedProfile != "" {
                let out = {
                    AWS_PROFILE: $selectedProfile,
                    AWS_ACCESS_KEY_ID: ($creds | get $selectedProfile | get "aws_access_key_id"),
                    AWS_SECRET_ACCESS_KEY: ($creds | get $selectedProfile | get "aws_secret_access_key"),
                }
                
                let region = ($creds | get $selectedProfile | get -i "region")
                if $region != "" {
                    $out | insert "AWS_REGION" $region
                } else {
                    $out
                }
            }
    } | load-env);

    {
        AWS_PROFILE: $env.AWS_PROFILE,
        AWS_ACCESS_KEY_ID: $env.AWS_ACCESS_KEY_ID,
        AWS_SECRET_ACCESS_KEY: $env.AWS_SECRET_ACCESS_KEY,
        AWS_REGION: $env.AWS_REGION
    }
}