aws-context-switcher 0.1.5

pub fn shell_hook(shell: &str) {
    let bin = std::env::current_exe()
        .unwrap_or_else(|_| std::path::PathBuf::from("awsx"));
    let bin = bin.display();

    let hook = match shell {
        "zsh" | "bash" => format!(
            r#"
awsx() {{
    local output
    output=$({bin} "$@")
    local exit_code=$?
    while IFS= read -r line; do
        case "$line" in
            export\ *) eval "$line" ;;
            unset\ *) eval "$line" ;;
            *) echo "$line" ;;
        esac
    done <<< "$output"
    return $exit_code
}}
"#
        ),
        "fish" => format!(
            r#"
function awsx
    set -l output ({bin} $argv)
    set -l exit_code $status
    for line in $output
        if string match -q 'export *' -- $line
            set -l var (string replace 'export ' '' -- $line)
            set -l key (string split '=' -- $var)[1]
            set -l val (string split '=' -- $var)[2]
            set -gx $key $val
        else if string match -q 'unset *' -- $line
            set -l var (string replace 'unset ' '' -- $line)
            set -e $var
        else
            echo $line
        end
    end
    return $exit_code
end
"#
        ),
        _ => {
            eprintln!("Unsupported shell: {shell}. Use zsh, bash, or fish.");
            std::process::exit(1);
        }
    };
    print!("{hook}");
}

pub fn prompt_hook(shell: &str) {
    let hook = match shell {
        "zsh" => r#"
_awsx_prompt() {
    [[ -n "$AWSX_CONTEXT" ]] && echo "☁️ $AWSX_CONTEXT"
}
RPROMPT='$(_awsx_prompt) '"$RPROMPT"
"#,
        "bash" => r#"
_awsx_prompt() {
    [[ -n "$AWSX_CONTEXT" ]] && printf "☁️ %s " "$AWSX_CONTEXT"
}
PS1='$(_awsx_prompt)'"$PS1"
"#,
        "fish" => r#"
function fish_right_prompt
    if set -q AWSX_CONTEXT
        echo "☁️ $AWSX_CONTEXT"
    end
end
"#,
        _ => "",
    };
    print!("{hook}");
}