nushell support for shell-helpers

[b0lle]

Is your feature request related to a problem? Please describe.
It would be great to add nushell support for shell-helpers

Additional context
Nushell is a modern shell. aws-sso-cli could be a nice replacement for the oh-my-zsh aws plugin, which obviously does not work in nushell.

[synfinatic]

@b0lle: Lots of shells out there and while I'm sure nushell is awesome, I've honestly never heard of it which makes this a low priority for me. If someone provides a pull request to add the shell helper code, that would help a lot and dramatically improve the chances this happens.

[radahh-rest]

Here's how I hacked it into nushell:

#!/usr/bin/env nu

def --env aws-sso-profile [
    profile: string
] {
    let _args = (if ($env.AWS_SSO_HELPER_ARGS? | is-empty) { '-L error' } else { $env.AWS_SSO_HELPER_ARGS })
    if ($env.AWS_PROFILE? | is-not-empty) {
        print "Unable to assume a role while AWS_PROFILE is set"
        return 1
    }

    XONSH_VERSION=1 aws-sso eval -p $profile | lines | parse "${name} = '{value}'" | reduce -f {} {|it, acc| $acc | upsert $it.name $it.value} | load-env

    if ($env.AWS_SSO_PROFILE? != $profile) {
        return 1
    }
}

def --env aws-sso-clear [] {
    let _args = (if ($env.AWS_SSO_HELPER_ARGS? | is-empty) { '-L error' } else { $env.AWS_SSO_HELPER_ARGS })
    if ($env.AWS_SSO_PROFILE? | is-empty) {
        print "AWS_SSO_PROFILE is not set"
        return 1
    }

    for row in (XONSH_VERSION=1 aws-sso eval -c | lines | parse "del ${name}") { hide-env $row.name }
}

[Biktop]

I created version with autocompletion

# aws-sso.nu — Nushell port of aws-sso shell integration

# ── internal helper ──────────────────────────────────────────────────────────

def _sso_args [] {
    let val = $env.AWS_SSO_HELPER_ARGS?
    if ($val | is-empty) { ["-L" "error"] } else { $val | split words }
}

# ── completions ──────────────────────────────────────────────────────────────

# Completion driver for the aws-sso command (uses COMP_LINE protocol)
def _aws-sso-nu-complete [context: string, offset: int] {
    let line = if ($context | str ends-with " ") { $context } else { $"($context) " }
    with-env { COMP_LINE: $line, __NO_ESCAPE_COLONS: "1" } {
        ^aws-sso | lines | where { |it| ($it | str trim) != "" }
    }
}

# SSO instance (session) completions for --sso flag
def _aws-sso-session-nu-complete [context: string, offset: int] {
    ^aws-sso list ...(_sso_args) --csv SSO
        | lines
        | where { |it| ($it | str trim) != "" }
}

# Profile completions for aws-sso-profile
def _aws-sso-profile-nu-complete [context: string, offset: int] {
    let cur = $context | split row " " | last
    ^aws-sso list ...(_sso_args) --csv -P $"Profile=($cur)" Profile
        | lines
        | where { |it| ($it | str trim) != "" }
}

# Attach completions to the aws-sso external command
export extern aws-sso [...args: string@_aws-sso-nu-complete]

# ── commands ─────────────────────────────────────────────────────────────────

# Assume an AWS SSO profile, loading credentials into the environment
export def --env aws-sso-profile [
    --sso (-S): string@_aws-sso-session-nu-complete  # SSO instance name
    profile: string@_aws-sso-profile-nu-complete     # Profile to assume
] {
    if ($env.AWS_PROFILE? | default "" | is-not-empty) {
        error make { msg: "Unable to assume a role while AWS_PROFILE is set" }
    }

    let creds = if ($sso | is-not-empty) {
        ^aws-sso ...(_sso_args) -S $sso process -p $profile | from json
    } else {
        ^aws-sso ...(_sso_args) process -p $profile | from json
    }

    load-env {
        AWS_ACCESS_KEY_ID:     $creds.AccessKeyId
        AWS_SECRET_ACCESS_KEY: $creds.SecretAccessKey
        AWS_SESSION_TOKEN:     $creds.SessionToken
        AWS_SSO_PROFILE:       $profile
    }
}

# Clear current AWS SSO credentials from the environment
export def --env aws-sso-clear [] {
    if ($env.AWS_SSO_PROFILE? | default "" | is-empty) {
        error make { msg: "AWS_SSO_PROFILE is not set" }
    }

    hide-env AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN AWS_SSO_PROFILE
}
hide _sso_args
hide _aws-sso-session-nu-complete
hide _aws-sso-profile-nu-complete
hide _aws-sso-nu-complete