ohmyzsh/plugins/zsh-vi-man/zsh-vi-man.zsh

108 lines
4.1 KiB
Bash

# zsh-vi-man.zsh -- Smart man page viewer for zsh vi mode
# https://github.com/TunaCuma/zsh-vi-man
#
# Press K in vi normal mode to open the man page for the current command.
# If your cursor is on an option (like -r or --recursive), it will jump
# directly to that option in the man page.
#
# MIT License - Copyright (c) 2025 Tuna Cuma
# Configuration variables (can be set before sourcing)
# ZVM_MAN_KEY: the key to trigger man page lookup (default: K)
# ZVM_MAN_PAGER: the pager to use (default: less)
: ${ZVM_MAN_KEY:=K}
: ${ZVM_MAN_PAGER:=less}
function zvm-man() {
# Get the word at cursor position
local left="${LBUFFER##*[[:space:]]}"
local right="${RBUFFER%%[[:space:]]*}"
local word="${left}${right}"
# Find the current command segment (handles pipes: tree | grep -A)
# Get everything after the last pipe before cursor
local current_segment="${LBUFFER##*|}"
# Trim leading whitespace
current_segment="${current_segment#"${current_segment%%[![:space:]]*}"}"
# Extract the command (first word of segment)
local cmd="${current_segment%%[[:space:]]*}"
if [[ -z "$cmd" ]]; then
zle -M "No command found"
return 1
fi
# Determine the man page to open
local man_page="$cmd"
local rest="${current_segment#*[[:space:]]}"
local potential_subcommand="${rest%%[[:space:]]*}"
# Check for subcommand man pages (e.g., git-commit, docker-run)
if [[ -n "$potential_subcommand" && ! "$potential_subcommand" =~ ^- ]]; then
if man -w "${cmd}-${potential_subcommand}" &>/dev/null; then
man_page="${cmd}-${potential_subcommand}"
fi
fi
# Build the search pattern for the current word
# Patterns match option definitions: lines starting with whitespace then dash
# Supports comma-separated (GNU style) and slash-separated (jq style) options
local pattern=""
if [[ -n "$word" ]]; then
# Long option with value: --color=always -> search for --color
if [[ "$word" =~ ^--[^=]+= ]]; then
local opt="${word%%=*}"
pattern="^[[:space:]]*${opt}([,/=:[[:space:]]|$)|^[[:space:]]*-.*[,/][[:space:]]+${opt}([,/=:[[:space:]]|$)"
# Combined short options: -rf -> search for -[rf] to find individual options
# Also includes fallback for single-dash long options like find's -name, -type
elif [[ "$word" =~ ^-[a-zA-Z]{2,}$ ]]; then
local chars="${word:1}"
# Pattern 1: individual chars (e.g., -r or -f from -rf)
# Pattern 2: the full word as-is (e.g., -name for find)
pattern="^[[:space:]]*-[${chars}][,/:[:space:]]|^[[:space:]]*-.*[,/][[:space:]]+-[${chars}][,/:[:space:]]|^[[:space:]]*${word}([,/:[:space:]]|$)|^[[:space:]]*-.*[,/][[:space:]]+${word}([,/:[:space:]]|$)"
# Single short option: -r -> match at start of option definition line
elif [[ "$word" =~ ^-[a-zA-Z]$ ]]; then
pattern="^[[:space:]]*${word}[,/:[:space:]]|^[[:space:]]*-.*[,/][[:space:]]+${word}([,/:[:space:]]|$)"
# Long option without value: --recursive
elif [[ "$word" =~ ^-- ]]; then
pattern="^[[:space:]]*${word}([,/=:[[:space:]]|$)|^[[:space:]]*-.*[,/][[:space:]]+${word}([,/=:[[:space:]]|$)"
fi
fi
# Clear screen and open man page
zle -I
if [[ -n "$pattern" ]]; then
man "$man_page" 2>/dev/null | ${ZVM_MAN_PAGER} -p "${pattern}" 2>/dev/null || \
man "$man_page" 2>/dev/null | ${ZVM_MAN_PAGER}
else
man "$man_page" 2>/dev/null || zle -M "No manual entry for ${man_page}"
fi
zle reset-prompt
}
# Register the widget and bind the key
zle -N zvm-man
function _zvm_man_bind_key() {
bindkey -M vicmd "${ZVM_MAN_KEY}" zvm-man
}
# Support both immediate binding and lazy loading with zsh-vi-mode
if (( ${+functions[zvm_after_lazy_keybindings]} )); then
# zsh-vi-mode is loaded with lazy keybindings, hook into it
if [[ -z "${ZVM_LAZY_KEYBINDINGS}" ]] || [[ "${ZVM_LAZY_KEYBINDINGS}" == true ]]; then
zvm_after_lazy_keybindings_commands+=(_zvm_man_bind_key)
else
_zvm_man_bind_key
fi
elif (( ${+functions[zvm_after_init]} )); then
# zsh-vi-mode without lazy keybindings
zvm_after_init_commands+=(_zvm_man_bind_key)
else
# Standalone or other vi-mode setups
_zvm_man_bind_key
fi