Feature request: Support "shell integration"

[deltazefiro]

Hey thanks for the awesome project!

I'm wondering if it would be possible we can add the "shell integration" feature similar to shell-gpt. It allows us to use a hotkey to feed the current input line (in natural language) to the AI and convert it into a command. It would be incredibly convenient to have something like this.

A possible implementation reference: https://github.com/TheR1D/shell_gpt/blob/main/sgpt/integration.py

[ricklamers]

I'd be open to a well tested PR!

[Deltik]

@deltazefiro: The hurdle appears to be that Shell-AI doesn't have a non-interactive mode, so it doesn't have a built-in way to print an output without prompting the user.

I can offer a hacky wrapper that orchestrates Shell-AI interactively with expect to make it look non-interactive, but I don't recommend it because Shell-AI's InquirerPy may change and no longer match in the future:

Bash

_shai_transform() {
    if [[ -n "$READLINE_LINE" ]]; then
        local original="$READLINE_LINE"
        # Dim + underline: subtle, accessible, theme-agnostic
        printf '\r\033[K\033[2;4m%s\033[0m 💭' "$original"
        READLINE_LINE=$(expect -c '
            log_user 0
            set timeout 30
            spawn env SHAI_SKIP_CONFIRM=false SHAI_SKIP_HISTORY=true uv tool run --from shell-ai shai '"$original"'
            expect {
                "Select a command:" { send "\r" }
                timeout { exit 1 }
            }
            expect {
                -re {108m([a-zA-Z0-9 _./:=-]+)} {
                    puts $expect_out(1,string)
                    send "\x03"
                }
                timeout { exit 1 }
            }
            expect eof
        ' 2>/dev/null | tail -1)
        READLINE_POINT=${#READLINE_LINE}
        printf '\r\033[K'
    fi
}
bind -x '"\C-g": _shai_transform'

Z shell

_shai_transform() {
    if [[ -n "$BUFFER" ]]; then
        local original="$BUFFER"
        zle -R "$original" "💭"
        BUFFER=$(expect -c '
            log_user 0
            set timeout 30
            spawn env SHAI_SKIP_CONFIRM=false SHAI_SKIP_HISTORY=true uv tool run --from shell-ai shai '"$original"'
            expect {
                "Select a command:" { send "\r" }
                timeout { exit 1 }
            }
            expect {
                -re {108m([a-zA-Z0-9 _./:=-]+)} {
                    puts $expect_out(1,string)
                    send "\x03"
                }
                timeout { exit 1 }
            }
            expect eof
        ' 2>/dev/null | tail -1)
        zle reset-prompt
        zle end-of-line
    fi
}
zle -N _shai_transform
bindkey '^g' _shai_transform

---

Alternative: I maintain a fork that started from @ricklamers's original Shell-AI and has since diverged significantly (rewritten in Rust). It includes built-in shell integration with the keybinding feature you're looking for:

# Install
cargo install shell-ai

# Generate shell integration with Ctrl+G keybinding (choose your shell)
shell-ai integration generate bash --preset full
shell-ai integration generate zsh --preset full
shell-ai integration generate fish --preset full
shell-ai integration generate powershell --preset full
# Then add the source line to your shell rc file as instructed

That said, the expect wrapper above should work if you'd prefer to stay with this project.

[deltazefiro]

@Deltik Your fork looks really awesome! I'll try it out later.