zman

#!/usr/bin/env zsh
# Can be run as a script and as a function

emulate -LR zsh
setopt extendedglob warncreateglobal typesetsilent noshortloops

if [[ "$0" != declzsh || -n "$ZSH_SCRIPT" ]]; then
    # Handle $0 according to the Zsh Plugin Standard:
    # https://zdharma-continuum.github.io/Zsh-100-Commits-Club/Zsh-Plugin-Standard.html
    0="${${ZERO:-${0:#$ZSH_ARGZERO}}:-${(%):-%N}}"
    0="${${(M)0##/*}:-$PWD/$0}"
    fpath+=( "$0:h" )
fi

autoload -Uz za-man-handler

local -a match mbegin mend reply
local MATCH REPLY
integer MBEGIN MEND

local -A opthash
zparseopts -E -D -A opthash c h -help || \
    { echo "Improper options given, see help (-h/--help)"; return 1; }

# Converts plugin spec like "user/plugin" into two parts
# returned in reply, namely the user and plugin
-zinit-any-to-user-plugin() {
    # Two components given?
    # That's a pretty fast track to call this function this way
    if [[ -n "$2" ]];then
        2=${~2}
        reply=( "${1:-${${(M)2#/}:+%}}" "${${${(M)1#%}:+$2}:-${2//---//}}" )
        return 0
    fi

    # Is it absolute path?
    if [[ "$1" = "/"* ]]; then
        reply=( "%" "$1" )
        return 0
    fi

    # Is it absolute path in zinit format?
    if [[ "$1" = "%"* ]]; then
        reply=( "%" "${${${1/\%HOME/$HOME}/\%SNIPPETS/${ZINIT[SNIPPETS_DIR]}}#%}" )
        reply[2]=${~reply[2]}
        return 0
    fi

    # Rest is for single component given
    # It doesn't touch $2

    if [[ "$1" = */* ]]; then
        reply=( "${1%%/*}" "${1#*/}" )
        return 0
    fi

    reply=( "${${(M)1#*---}%---}" "${${${1#*---}//---//}:-_unknown}" )

    return 0
}

local ZMAN_REPO_DIR="$1" plgdir="$2" snipdir="$3" pors
shift 3

# TODO: full snippet-path handling
local -a parray sarray
parray=( "$plgdir"/"${^${@//\//---}}"(N) "$plgdir"/*---"$^@"(N) )
sarray=( "$snipdir"/"$^@"(N) )

for pors (${parray[@]}) {
    [[ "${pors%/}" = "$plgdir" ]] && continue

    if [[ -f ''(#i)$pors/README.md(#qN[1]) && \
        ( ! -f ''(#i)$pors/README(#qN[1]) || \
            ''(#i)$pors/README.md(#qN[1]) -nt ''(#i)$pors/README(#qN[1]) )
    ]]; then
        -zinit-any-to-user-plugin "${pors:t}"
        # local type="$1" user="$2" plugin="$3" id_as="$4" dir="$5"
        za-man-handler plugin "${reply[-2]}" "${reply[-1]}" \
                    id_as="${${pors:t}//---//}" "$pors"
    elif [[ ! -f ''(#i)$pors/README.md(#qN[1]) ]]; then
        -zinit-any-to-user-plugin "${pors:t}"
        print -P -- "%F{38}zman z-plugin: %F{154}No README.md in the \
%F{220}${reply[-1]}%F{154} plugin%f"
    fi
}

for pors (${sarray[@]}) {
    if [[ -f ''(#i)$pors/README.md(#qN[1]) && \
        ( ! -f ''(#i)$pors/README(#qN[1]) || \
            ''(#i)$pors/README.md(#qN[1]) -nt ''(#i)$pors/README(#qN[1]) )
    ]]; then
        local snippath="${pors#$snipdir/}"
        # local type="$1" url="$2" id_as="$3" dir="$4"
        za-man-handler snippet "$snippath" \
                    id_as="${${snippath:t}//--//}" "$pors"
    elif [[ ! -f ''(#i)$pors/README.md(#qN[1]) ]]; then
        local snippath="${pors#$snipdir/}"
        print -P -- "%F{38}zman z-plugin: %F{154}No README.md in the \
%F{220}${snippath}%F{154} snippet%f"
    fi
}

if (( !${+opthash[-c]} )); then
    local -a farray
    farray=( ${^parray[@]}/(#i)README(#qN)
            ${^sarray[@]}/(#i)README(#qN)
    )
    if (( ${#farray} )); then
        man ${farray[@]}
    else
        if (( ${#} && !${#parray} && !${#sarray} )); then
            print -P -- "%F{38}zman z-plugin: %F{154}No such plugin(s) or snippet(s): \
%F{220}${(j:, :)@}%F{154}%f"
        elif (( !${#} )); then
            print -P -- "%F{38}zman z-plugin: %F{154}Argument required – a plugin or snippet name%f"
        fi
    fi
else
    local -a farray
    farray=( ${^parray[@]}/zsdoc/*.1(N) ${^sarray[@]}/zsdoc/*.1(N) )
    if (( ${#farray} )); then
        local fle
        for fle (${farray[@]}) {
            man "$fle"
            sleep 2
        }
    else
        if (( ${#} && !${#parray} && !${#sarray} )); then
            print -P -- "%F{38}zman z-plugin: %F{154}No such plugin(s) or snippet(s) \
or no code documentation available for: %F{220}${(j:, :)@}%F{154}%f"
        elif (( !${#} )); then
            print -P -- "%F{38}zman z-plugin: %F{154}Argument required – a plugin or snippet name%f"
        fi
    fi
fi
# vim:ft=zsh:sw=4:sts=4:et