Support zstyle to customize zsh-defer repo

mattmc3 · Jul 17, 2024 · fd90f64 · fd90f64
1 parent 620ded1
commit fd90f64
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/functions/__antidote_bulk_clone b/functions/__antidote_bulk_clone
@@ -4,8 +4,13 @@
 #function __antidote_bulk_clone {
   emulate -L zsh; setopt local_options $_adote_funcopts
 
+  # Allow the user to define zsh-defer repo in case they want to fork it.
+  local zsh_defer_bundle
+  zstyle -s ':antidote:defer' bundle 'zsh_defer_bundle' \
+    || zsh_defer_bundle='romkatv/zsh-defer'
+
   # get a list of clonable repos from a bundle file
-  $__adote_awkcmd '
+  $__adote_awkcmd -v ZSH_DEFER_BUNDLE=$zsh_defer_bundle '
     BEGIN { RS="[\r\n]" }
 
     # initialize vars
@@ -15,7 +20,7 @@
     /^ *(#.+)?$/ { next }
 
     # clone zsh-defer
-    /kind:defer/ { print "antidote-script --kind clone romkatv/zsh-defer &" }
+    /kind:defer/ { print "antidote-script --kind clone " ZSH_DEFER_BUNDLE " &" }
 
     # handle user/repo and URL forms
     $1~/^[^\/]+\/[^\/]+$/      { bundle=$1 }

diff --git a/functions/antidote-script b/functions/antidote-script
@@ -107,11 +107,14 @@
 
   # handle defers
   local source_cmd="source"
+  local zsh_defer_bundle
+  zstyle -s ':antidote:defer' bundle 'zsh_defer_bundle' \
+    || zsh_defer_bundle='romkatv/zsh-defer'
   if [[ "$o_kind[-1]" == "defer" ]]; then
     source_cmd="${zsh_defer} source"
     script+=(
       'if ! (( $+functions[zsh-defer] )); then'
-      "$(antidote-script 'romkatv/zsh-defer' | __antidote_indent)"
+      "$(antidote-script $zsh_defer_bundle | __antidote_indent)"
       'fi'
     )
   fi
@@ -123,7 +126,7 @@
     else
       script+=("fpath+=( ${bundle_path}/${o_autoload[-1]} )")
     fi
-    script+=( "builtin autoload -Uz \$fpath[-1]/*(N.:t)" )
+    script+=("builtin autoload -Uz \$fpath[-1]/*(N.:t)")
   fi
 
   # generate load script