nvm.plugin.zsh

 1# Don't try to load nvm if command already available
 2# Note: nvm is a function so we need to use `which`
 3which nvm &>/dev/null && return
 4
 5# See https://github.com/nvm-sh/nvm#installation-and-update
 6if [[ -z "$NVM_DIR" ]]; then
 7  if [[ -d "$HOME/.nvm" ]]; then
 8    export NVM_DIR="$HOME/.nvm"
 9  elif [[ -d "${XDG_CONFIG_HOME:-$HOME/.config}/nvm" ]]; then
10    export NVM_DIR="${XDG_CONFIG_HOME:-$HOME/.config}/nvm"
11  elif (( $+commands[brew] )); then
12    NVM_HOMEBREW="${NVM_HOMEBREW:-${HOMEBREW_PREFIX:-$(brew --prefix)}/opt/nvm}"
13    if [[ -d "$NVM_HOMEBREW" ]]; then
14      export NVM_DIR="$NVM_HOMEBREW"
15    fi
16  fi
17fi
18
19if [[ -z "$NVM_DIR" ]] || [[ ! -f "$NVM_DIR/nvm.sh" ]]; then
20  return
21fi
22
23function _omz_nvm_setup_completion {
24  local _nvm_completion
25  # Load nvm bash completion
26  for _nvm_completion in "$NVM_DIR/bash_completion" "$NVM_HOMEBREW/etc/bash_completion.d/nvm"; do
27    if [[ -f "$_nvm_completion" ]]; then
28      # Load bashcompinit
29      autoload -U +X bashcompinit && bashcompinit
30      # Bypass compinit call in nvm bash completion script. See:
31      # https://github.com/nvm-sh/nvm/blob/4436638/bash_completion#L86-L93
32      ZSH_VERSION= source "$_nvm_completion"
33      break
34    fi
35  done
36  unfunction _omz_nvm_setup_completion
37}
38
39function _omz_nvm_setup_autoload {
40  if ! zstyle -t ':omz:plugins:nvm' autoload; then
41    unfunction _omz_nvm_setup_autoload
42    return
43  fi
44
45  # Autoload nvm when finding a .nvmrc file in the current directory
46  # Adapted from: https://github.com/nvm-sh/nvm#zsh
47  function load-nvmrc {
48    local nvmrc_path="$(nvm_find_nvmrc)"
49    local nvm_silent=""
50    zstyle -t ':omz:plugins:nvm' silent-autoload && nvm_silent="--silent"
51
52    if [[ -n "$nvmrc_path" ]]; then
53      local nvmrc_node_version=$(nvm version $(command cat "$nvmrc_path" | tr -dc '[:print:]'))
54
55      if [[ "$nvmrc_node_version" = "N/A" ]]; then
56        nvm install
57      elif [[ "$nvmrc_node_version" != "$(nvm version)" ]]; then
58        nvm use $nvm_silent
59      fi
60    elif [[ -n "$(PWD=$OLDPWD nvm_find_nvmrc)" ]] && [[ "$(nvm version)" != "$(nvm version default)" ]]; then
61      [[ -z $nvm_silent ]] && echo "Reverting to nvm default version"
62
63      nvm use default $nvm_silent
64    fi
65  }
66
67  autoload -U add-zsh-hook
68  add-zsh-hook chpwd load-nvmrc
69
70  load-nvmrc
71  unfunction _omz_nvm_setup_autoload
72}
73
74if zstyle -t ':omz:plugins:nvm' lazy; then
75  # Call nvm when first using nvm, node, npm, pnpm, yarn, corepack or other commands in lazy-cmd
76  zstyle -a ':omz:plugins:nvm' lazy-cmd nvm_lazy_cmd
77  nvm_lazy_cmd=(_omz_nvm_load nvm node npm npx pnpm pnpx yarn corepack $nvm_lazy_cmd) # default values
78  eval "
79    function $nvm_lazy_cmd {
80      for func in $nvm_lazy_cmd; do
81        if (( \$+functions[\$func] )); then
82          unfunction \$func
83        fi
84      done
85      # Load nvm if it exists in \$NVM_DIR
86      [[ -f \"\$NVM_DIR/nvm.sh\" ]] && source \"\$NVM_DIR/nvm.sh\"
87      _omz_nvm_setup_completion
88      _omz_nvm_setup_autoload
89      if [[ \"\$0\" != _omz_nvm_load ]]; then
90        \"\$0\" \"\$@\"
91      fi
92    }
93  "
94  unset nvm_lazy_cmd
95else
96  source "$NVM_DIR/nvm.sh"
97  _omz_nvm_setup_completion
98  _omz_nvm_setup_autoload
99fi