File size: 2,673 Bytes
c981f27 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | #!/usr/bin/env zsh
# Editor and command suggestion action handlers
# Action handler: Open external editor for command composition
function _forge_action_editor() {
local initial_text="$1"
echo
# Determine editor in order of preference: FORGE_EDITOR > EDITOR > nano
local editor_cmd="${FORGE_EDITOR:-${EDITOR:-nano}}"
# Validate editor exists
if ! command -v "${editor_cmd%% *}" &>/dev/null; then
_forge_log error "Editor not found: $editor_cmd (set FORGE_EDITOR or EDITOR)"
return 1
fi
# Create .forge directory if it doesn't exist
local forge_dir=".forge"
if [[ ! -d "$forge_dir" ]]; then
mkdir -p "$forge_dir" || {
_forge_log error "Failed to create .forge directory"
return 1
}
fi
# Create temporary file with git-like naming: FORGE_EDITMSG.md
local temp_file="${forge_dir}/FORGE_EDITMSG.md"
touch "$temp_file" || {
_forge_log error "Failed to create temporary file"
return 1
}
# Ensure cleanup on exit
trap "rm -f '$temp_file'" EXIT INT TERM
# Pre-populate with initial text if provided
if [[ -n "$initial_text" ]]; then
echo "$initial_text" > "$temp_file"
fi
# Open editor in subshell with its own TTY session
(eval "$editor_cmd '$temp_file'" </dev/tty >/dev/tty 2>&1)
local editor_exit_code=$?
if [ $editor_exit_code -ne 0 ]; then
_forge_log error "Editor exited with error code $editor_exit_code"
_forge_reset
return 1
fi
# Read and process content
local content
content=$(cat "$temp_file" | tr -d '\r')
if [ -z "$content" ]; then
_forge_log info "Editor closed with no content"
BUFFER=""
CURSOR=0
zle reset-prompt
return 0
fi
# Insert into buffer with : prefix
BUFFER=": $content"
CURSOR=${#BUFFER}
zle reset-prompt
}
# Action handler: Generate shell command from natural language
# Usage: :? <description>
function _forge_action_suggest() {
local description="$1"
if [[ -z "$description" ]]; then
_forge_log error "Please provide a command description"
return 0
fi
echo
# Generate the command
local generated_command
generated_command=$(FORCE_COLOR=true CLICOLOR_FORCE=1 _forge_exec suggest "$description")
if [[ -n "$generated_command" ]]; then
# Replace the buffer with the generated command
BUFFER="$generated_command"
CURSOR=${#BUFFER}
zle reset-prompt
else
_forge_log error "Failed to generate command"
fi
}
|