fix(init): bound the LocalHostName cache and don't cache lookup failures

Keying the cache on $HOST alone can't notice a LocalHostName rename
while HostName stays fixed, which would pin a stale identity forever.
Expire the entry after a day, like lib/grep.zsh does, so a rename is
picked up without giving up the fork on almost every start.

Only write the cache when scutil actually answered, so a transient
failure falls back to ${HOST/.*/} for that shell instead of pinning the
fallback name for the next day.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>

.github/workflows/main.yml

@@ -36,8 +36,6 @@ jobs:
         run: |
           for file in ./oh-my-zsh.sh \
                       ./lib/*.zsh \
-                      ./functions/* \
-                      ./completions/_* \
                       ./plugins/*/*.plugin.zsh \
                       ./plugins/*/_* \
                       ./themes/*.zsh-theme; do

completions/_omz

@@ -1,90 +0,0 @@
-#compdef omz
-
-local -a cmds subcmds
-cmds=(
-  'changelog:Print the changelog'
-  'help:Usage information'
-  'plugin:Manage plugins'
-  'pr:Manage Oh My Zsh Pull Requests'
-  'reload:Reload the current zsh session'
-  'shop:Open the Oh My Zsh shop'
-  'theme:Manage themes'
-  'update:Update Oh My Zsh'
-  'version:Show the version'
-)
-
-if (( CURRENT == 2 )); then
-  _describe 'command' cmds
-elif (( CURRENT == 3 )); then
-  case "$words[2]" in
-    changelog) local -a refs
-      refs=("${(@f)$(builtin cd -q "$ZSH"; command git for-each-ref --format="%(refname:short):%(subject)" refs/heads refs/tags)}")
-      _describe 'command' refs ;;
-    plugin) subcmds=(
-      'disable:Disable plugin(s)'
-      'enable:Enable plugin(s)'
-      'info:Get plugin information'
-      'list:List plugins'
-      'load:Load plugin(s)'
-    )
-      _describe 'command' subcmds ;;
-    pr) subcmds=('clean:Delete all Pull Request branches' 'test:Test a Pull Request')
-      _describe 'command' subcmds ;;
-    theme) subcmds=('list:List themes' 'set:Set a theme in your .zshrc file' 'use:Load a theme')
-      _describe 'command' subcmds ;;
-  esac
-elif (( CURRENT == 4 )); then
-  case "${words[2]}::${words[3]}" in
-    plugin::(disable|enable|load))
-      local -aU valid_plugins
-
-      if [[ "${words[3]}" = disable ]]; then
-        # if command is "disable", only offer already enabled plugins
-        valid_plugins=($plugins)
-      else
-        valid_plugins=("$ZSH"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t) "$ZSH_CUSTOM"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t))
-        # if command is "enable", remove already enabled plugins
-        [[ "${words[3]}" = enable ]] && valid_plugins=(${valid_plugins:|plugins})
-      fi
-
-      _describe 'plugin' valid_plugins ;;
-    plugin::info)
-      local -aU plugins
-      plugins=("$ZSH"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t) "$ZSH_CUSTOM"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t))
-      _describe 'plugin' plugins ;;
-    plugin::list)
-      local -a opts
-      opts=('--enabled:List enabled plugins only')
-      _describe -o 'options' opts ;;
-    theme::(set|use))
-      local -aU themes
-      themes=("$ZSH"/themes/*.zsh-theme(-.N:t:r) "$ZSH_CUSTOM"/**/*.zsh-theme(-.N:r:gs:"$ZSH_CUSTOM"/themes/:::gs:"$ZSH_CUSTOM"/:::))
-      _describe 'theme' themes ;;
-  esac
-elif (( CURRENT > 4 )); then
-  case "${words[2]}::${words[3]}" in
-    plugin::(enable|disable|load))
-      local -aU valid_plugins
-
-      if [[ "${words[3]}" = disable ]]; then
-        # if command is "disable", only offer already enabled plugins
-        valid_plugins=($plugins)
-      else
-        valid_plugins=("$ZSH"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t) "$ZSH_CUSTOM"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t))
-        # if command is "enable", remove already enabled plugins
-        [[ "${words[3]}" = enable ]] && valid_plugins=(${valid_plugins:|plugins})
-      fi
-
-      # Remove plugins already passed as arguments
-      # NOTE: $(( CURRENT - 1 )) is the last plugin argument completely passed, i.e. that which
-      # has a space after them. This is to avoid removing plugins partially passed, which makes
-      # the completion not add a space after the completed plugin.
-      local -a args
-      args=(${words[4,$(( CURRENT - 1))]})
-      valid_plugins=(${valid_plugins:|args})
-
-      _describe 'plugin' valid_plugins ;;
-  esac
-fi
-
-return 0

lib/cli.zsh

@@ -1,3 +1,5 @@
+#!/usr/bin/env zsh
+
 function omz {
   setopt localoptions noksharrays
   [[ $# -gt 0 ]] || {
@@ -18,6 +20,102 @@ function omz {
   _omz::$command "$@"
 }
 
+function _omz {
+  local -a cmds subcmds
+  cmds=(
+    'changelog:Print the changelog'
+    'help:Usage information'
+    'plugin:Manage plugins'
+    'pr:Manage Oh My Zsh Pull Requests'
+    'reload:Reload the current zsh session'
+    'shop:Open the Oh My Zsh shop'
+    'theme:Manage themes'
+    'update:Update Oh My Zsh'
+    'version:Show the version'
+  )
+
+  if (( CURRENT == 2 )); then
+    _describe 'command' cmds
+  elif (( CURRENT == 3 )); then
+    case "$words[2]" in
+      changelog) local -a refs
+        refs=("${(@f)$(builtin cd -q "$ZSH"; command git for-each-ref --format="%(refname:short):%(subject)" refs/heads refs/tags)}")
+        _describe 'command' refs ;;
+      plugin) subcmds=(
+        'disable:Disable plugin(s)'
+        'enable:Enable plugin(s)'
+        'info:Get plugin information'
+        'list:List plugins'
+        'load:Load plugin(s)'
+      )
+        _describe 'command' subcmds ;;
+      pr) subcmds=('clean:Delete all Pull Request branches' 'test:Test a Pull Request')
+        _describe 'command' subcmds ;;
+      theme) subcmds=('list:List themes' 'set:Set a theme in your .zshrc file' 'use:Load a theme')
+        _describe 'command' subcmds ;;
+    esac
+  elif (( CURRENT == 4 )); then
+    case "${words[2]}::${words[3]}" in
+      plugin::(disable|enable|load))
+        local -aU valid_plugins
+
+        if [[ "${words[3]}" = disable ]]; then
+          # if command is "disable", only offer already enabled plugins
+          valid_plugins=($plugins)
+        else
+          valid_plugins=("$ZSH"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t) "$ZSH_CUSTOM"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t))
+          # if command is "enable", remove already enabled plugins
+          [[ "${words[3]}" = enable ]] && valid_plugins=(${valid_plugins:|plugins})
+        fi
+
+        _describe 'plugin' valid_plugins ;;
+      plugin::info)
+        local -aU plugins
+        plugins=("$ZSH"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t) "$ZSH_CUSTOM"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t))
+        _describe 'plugin' plugins ;;
+      plugin::list)
+        local -a opts
+        opts=('--enabled:List enabled plugins only')
+        _describe -o 'options' opts ;;
+      theme::(set|use))
+        local -aU themes
+        themes=("$ZSH"/themes/*.zsh-theme(-.N:t:r) "$ZSH_CUSTOM"/**/*.zsh-theme(-.N:r:gs:"$ZSH_CUSTOM"/themes/:::gs:"$ZSH_CUSTOM"/:::))
+        _describe 'theme' themes ;;
+    esac
+  elif (( CURRENT > 4 )); then
+    case "${words[2]}::${words[3]}" in
+      plugin::(enable|disable|load))
+        local -aU valid_plugins
+
+        if [[ "${words[3]}" = disable ]]; then
+          # if command is "disable", only offer already enabled plugins
+          valid_plugins=($plugins)
+        else
+          valid_plugins=("$ZSH"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t) "$ZSH_CUSTOM"/plugins/*/{_*,*.plugin.zsh}(-.N:h:t))
+          # if command is "enable", remove already enabled plugins
+          [[ "${words[3]}" = enable ]] && valid_plugins=(${valid_plugins:|plugins})
+        fi
+
+        # Remove plugins already passed as arguments
+        # NOTE: $(( CURRENT - 1 )) is the last plugin argument completely passed, i.e. that which
+        # has a space after them. This is to avoid removing plugins partially passed, which makes
+        # the completion not add a space after the completed plugin.
+        local -a args
+        args=(${words[4,$(( CURRENT - 1))]})
+        valid_plugins=(${valid_plugins:|args})
+
+        _describe 'plugin' valid_plugins ;;
+    esac
+  fi
+
+  return 0
+}
+
+# If run from a script, do not set the completion function
+if (( ${+functions[compdef]} )); then
+  compdef _omz omz
+fi
+
 ## Utility functions
 
 function _omz::confirm {
@@ -844,5 +942,3 @@ function _omz::version {
     printf "%s (%s)\n" "$version" "$commit"
   )
 }
-
-omz "$@"

lib/diagnostics.zsh

@@ -1,4 +1,4 @@
-# omz_diagnostic_dump
+# diagnostics.zsh
 #
 # Diagnostic and debugging support for oh-my-zsh
 
@@ -351,5 +351,3 @@ function _omz_diag_dump_os_specific_version() {
   done
 }
 
-
-omz_diagnostic_dump "$@"

oh-my-zsh.sh

@@ -78,9 +78,6 @@ fpath=($ZSH/{functions,completions} $ZSH_CUSTOM/{functions,completions} $fpath)
 # Load all stock functions (from $fpath files) called below.
 autoload -U compaudit compinit zrecompile
 
-# The omz CLI and omz_diagnostic_dump are loaded on first use
-autoload -Uz omz omz_diagnostic_dump
-
 is_plugin() {
   local base_dir=$1
   local name=$2
@@ -103,7 +100,23 @@ done
 # Figure out the SHORT hostname
 if [[ "$OSTYPE" = darwin* ]]; then
   # macOS's $HOST changes with dhcp, etc. Use LocalHostName if possible.
-  SHORT_HOST=$(scutil --get LocalHostName 2>/dev/null) || SHORT_HOST="${HOST/.*/}"
+  # scutil costs a fork on every start, so remember its answer for a day
+  # (like lib/grep.zsh) and re-check sooner if $HOST changes.
+  __omz_host_cache="$ZSH_CACHE_DIR/localhostname"
+  __omz_host_cached=("$__omz_host_cache"(Nm-1))
+  if [[ -n "$__omz_host_cached" ]]; then
+    { read -r __omz_host_key && read -r SHORT_HOST } < "$__omz_host_cache"
+  fi
+  if [[ "$__omz_host_key" != "$HOST" || -z "$SHORT_HOST" ]]; then
+    # only cache what scutil actually answered, so a transient failure
+    # doesn't pin the fallback name for a day
+    if SHORT_HOST=$(scutil --get LocalHostName 2>/dev/null) && [[ -n "$SHORT_HOST" ]]; then
+      [[ ! -w "$ZSH_CACHE_DIR" ]] || print -rl -- "$HOST" "$SHORT_HOST" >| "$__omz_host_cache"
+    else
+      SHORT_HOST="${HOST/.*/}"
+    fi
+  fi
+  unset __omz_host_cache __omz_host_cached __omz_host_key
 else
   SHORT_HOST="${HOST/.*/}"
 fi