172 lines
4 KiB
Bash
172 lines
4 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
###############################################################################
|
||
|
|
# __ _
|
||
|
|
# \ \ _ __ | |__
|
||
|
|
# \ \ | '_ \| '_ \
|
||
|
|
# / / | | | | |_) |
|
||
|
|
# /_/ |_| |_|_.__/
|
||
|
|
#
|
||
|
|
# [nb] Command line and local web note-taking, bookmarking, and archiving with
|
||
|
|
# plain text data storage, encryption, filtering and search, pinning, #tagging,
|
||
|
|
# Git-backed versioning and syncing, Pandoc-backed conversion, global and local
|
||
|
|
# notebooks, customizable color themes, [[wiki-style linking]], plugins, and
|
||
|
|
# more in a single portable, user-friendly script.
|
||
|
|
#
|
||
|
|
# https://github.com/xwmx/nb
|
||
|
|
###############################################################################
|
||
|
|
|
||
|
|
_nb_subcommands() {
|
||
|
|
# _nb_cache_completions()
|
||
|
|
#
|
||
|
|
# Usage:
|
||
|
|
# _nb_cache_completions <path>
|
||
|
|
#
|
||
|
|
# Description:
|
||
|
|
# Cache completions for `nb`. Generating completions can be slow and
|
||
|
|
# native shell caching doesn't appear to help.
|
||
|
|
_nb_cache_completions() {
|
||
|
|
local _cache_path="${1:-}"
|
||
|
|
|
||
|
|
[[ -z "${_cache_path:-}" ]] && return 0
|
||
|
|
|
||
|
|
# Remove outdated cache files.
|
||
|
|
|
||
|
|
local _base_cache_path="${_cache_path%-*}"
|
||
|
|
|
||
|
|
local __suffix=
|
||
|
|
for __suffix in "zsh" "v1"
|
||
|
|
do
|
||
|
|
if [[ -e "${_base_cache_path:?}-${__suffix:?}" ]]
|
||
|
|
then
|
||
|
|
rm -f "${_base_cache_path:?}-${__suffix:?}"
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
|
||
|
|
# Rebuild completion cache.
|
||
|
|
|
||
|
|
local _commands=()
|
||
|
|
IFS=$'\n' _commands=($(nb subcommands))
|
||
|
|
|
||
|
|
local _notebooks=()
|
||
|
|
IFS=$'\n' _notebooks=($(nb notebooks --names --no-color --unarchived))
|
||
|
|
|
||
|
|
local _completions=()
|
||
|
|
IFS=$'\n' _completions=(${_commands[@]})
|
||
|
|
|
||
|
|
local _commands_cached=
|
||
|
|
local _notebooks_cached=
|
||
|
|
|
||
|
|
if [[ -e "${_cache_path}" ]]
|
||
|
|
then
|
||
|
|
local _counter=0
|
||
|
|
|
||
|
|
while IFS= read -r __line
|
||
|
|
do
|
||
|
|
_counter=$((_counter+1))
|
||
|
|
|
||
|
|
if [[ "${_counter}" == 1 ]]
|
||
|
|
then
|
||
|
|
_commands_cached="${__line}"
|
||
|
|
elif [[ "${_counter}" == 2 ]]
|
||
|
|
then
|
||
|
|
_notebooks_cached="${__line}"
|
||
|
|
else
|
||
|
|
break
|
||
|
|
fi
|
||
|
|
done < "${_cache_path}"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ "${_commands_cached}" != "${_commands[*]:-}" ]] ||
|
||
|
|
[[ "${_notebooks_cached}" != "${_notebooks[*]:-}" ]]
|
||
|
|
then
|
||
|
|
# Construct <notebook>:<subcommand> completions.
|
||
|
|
local __notebook=
|
||
|
|
for __notebook in "${_notebooks[@]}"
|
||
|
|
do
|
||
|
|
local __command=
|
||
|
|
for __command in "${_commands[@]}"
|
||
|
|
do
|
||
|
|
if [[ -n "${__notebook:-}" ]] &&
|
||
|
|
[[ -n "${__command:-}" ]]
|
||
|
|
then
|
||
|
|
_completions+=("${__notebook}:${__command}")
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
done
|
||
|
|
|
||
|
|
local _directory_path=
|
||
|
|
_directory_path="$(dirname "${_cache_path}")"
|
||
|
|
|
||
|
|
mkdir -p "${_directory_path}"
|
||
|
|
|
||
|
|
if [[ -f "${_cache_path:?}" ]]
|
||
|
|
then
|
||
|
|
rm -f "${_cache_path:?}"
|
||
|
|
fi
|
||
|
|
|
||
|
|
touch "${_cache_path:?}"
|
||
|
|
|
||
|
|
{
|
||
|
|
(IFS=$' '; printf "%s\\n" "${_commands[*]}")
|
||
|
|
(IFS=$' '; printf "%s\\n" "${_notebooks[*]}")
|
||
|
|
printf "%s\\n" "${_completions[@]}"
|
||
|
|
} >> "${_cache_path}"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
local _nb_dir=
|
||
|
|
_nb_dir="$(nb env | grep 'NB_DIR' | cut -d = -f 2)"
|
||
|
|
|
||
|
|
if [[ -z "${_nb_dir:?}" ]] ||
|
||
|
|
[[ ! -e "${_nb_dir}" ]]
|
||
|
|
then
|
||
|
|
return 0
|
||
|
|
elif [[ -L "${_nb_dir}" ]]
|
||
|
|
then
|
||
|
|
if hash "realpath" 2>/dev/null
|
||
|
|
then
|
||
|
|
_nb_dir="$(realpath "${_nb_dir}")"
|
||
|
|
else
|
||
|
|
_nb_dir="$(readlink "${_nb_dir}")"
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ ! -d "${_nb_dir}" ]]
|
||
|
|
then
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
local _cache_path="${_nb_dir:?}/.cache/nb-completion-cache-v2"
|
||
|
|
local _completions_cached=()
|
||
|
|
|
||
|
|
if [[ ! -e "${_cache_path}" ]]
|
||
|
|
then
|
||
|
|
_nb_cache_completions "${_cache_path}"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ -e "${_cache_path}" ]]
|
||
|
|
then
|
||
|
|
local _counter=0
|
||
|
|
|
||
|
|
local __line=
|
||
|
|
while IFS= read -r __line
|
||
|
|
do
|
||
|
|
_counter=$((_counter+1))
|
||
|
|
|
||
|
|
if [[ "${_counter}" -gt 2 ]]
|
||
|
|
then
|
||
|
|
_completions_cached+=("${__line}")
|
||
|
|
fi
|
||
|
|
done < "${_cache_path}"
|
||
|
|
|
||
|
|
(_nb_cache_completions "${_cache_path}" &)
|
||
|
|
fi
|
||
|
|
|
||
|
|
local _current="${COMP_WORDS[COMP_CWORD]}"
|
||
|
|
COMPREPLY=()
|
||
|
|
|
||
|
|
COMPREPLY=($(compgen -W "${_completions_cached[*]}" -- "${_current}"))
|
||
|
|
}
|
||
|
|
|
||
|
|
complete -F _nb_subcommands nb
|