From 6d42d803b229f49bce75a9c8a3b45e2bee371214 Mon Sep 17 00:00:00 2001
From: Joshua Cold <joshua.cold@securitymetrics.com>
Date: Sat, 7 May 2022 11:51:17 -0600
Subject: [PATCH] Fix zsh hanging when tab completing add/checkout

---
 completion/zsh/_yadm | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/completion/zsh/_yadm b/completion/zsh/_yadm
index 9423ac5..4fbfc9d 100644
--- a/completion/zsh/_yadm
+++ b/completion/zsh/_yadm
@@ -7,6 +7,19 @@ zstyle -T ':completion:*:yadm:argument-1:descriptions:' format && \
 zstyle -T ':completion:*:yadm:*:yadm' group-name && \
     zstyle ':completion:*:yadm:*:yadm' group-name ''
 
+function _yadm-add(){
+  yadm_path="$(yadm rev-parse --show-toplevel)"
+  yadm_options=$(yadm status --porcelain=v1 |
+    awk -v yadm_path=${yadm_path} '{printf "%s/\"%s\"\\:\"%s\" ",  yadm_path, $2, $1 }' )
+  _alternative \
+    "args:custom arg:(($yadm_options))" \
+    'files:filename:_files'
+}
+
+function _yadm-checkout(){ 
+    _yadm-add 
+}
+
 _yadm-alt() {
     return 0
 }