Update ZSH completions.

1 files changed, 54 insertions, 7 deletions

diff --git a/completions/zsh/_rest-run b/completions/zsh/_rest-run
index dfa6873..003caaf 100644
--- a/completions/zsh/_rest-run
+++ b/completions/zsh/_rest-run
@@ -6,17 +6,31 @@ function __debug {
   echo "$1" >> /tmp/_rest-run.log
 }
 
+function _rest-run_read_domain () {
+  local ARG="${words[2]}"
+
+  if [ ${#ARG} -lt 7 ]; then
+    ARG="${words[3]}"
+  fi
+
+  echo "$ARG" | sed 's/[a-z]*:\/\///g' | sed 's/\/.*//g' | sed 's/\?.*//g'
+}
+
 function _rest-run {
   local line
   local -a args
 
   args+=(
     '1:*:(POST GET PUT PATCH DELETE)'
-    '--no-edit[Skip request body edit]'
-    '--body[Body options]: :_rest-run_body_opts'
-    '--response[Response options]: :_rest-run_response_opts'
+    '-ne[Skip request body edit]'
+    '-b[Body options]: :_rest-run_body_opts'
+    '-r[Response options]: :_rest-run_response_opts'
+    '-e[Extract request data]: :_rest-run_extract_opts'
+    '-rr[Read response file]: :_rest-run_select_response_opts'
     '*-h[HTTP header]'
-    '--paginate[Paginate request response]'
+    '-p[Paginate request response]'
+    '--no-color[Disable ANSI color scape codes (default when output is being piped)]'
+    '--color[Enable ANSI color scape codes]'
   )
 
   _arguments -C $args
@@ -27,9 +41,9 @@ function _rest-run_body_opts {
 
   _descriptions=(
     'default -- create a request body using the default editor'
-    'last -- use the last request'
+    'last    -- use the last request'
     'history -- pick from the request history'
-    'stdin -- reads from piped input'
+    'stdin   -- reads from piped input'
   )
 
   _values=(
@@ -47,7 +61,7 @@ function _rest-run_response_opts {
 
   _descriptions=(
     'default -- store the request on cache'
-    'last -- read last request, skip request'
+    'last    -- read last request, skip request'
     'history -- pick from the response history, skip request'
   )
 
@@ -60,6 +74,39 @@ function _rest-run_response_opts {
   compadd -d _descriptions -a _values
 }
 
+function _rest-run_extract_opts {
+  local -a _descriptions _values
+
+  _descriptions=(
+    'default  -- whole request and response, headers and body'
+    'body     -- only response body (default when output is piped to another process)'
+    'headers  -- only response headers'
+    'rbody    -- only request body'
+    'rheaders -- only request headers'
+  )
+
+  _values=(
+    'default'
+    'body'
+    'headers'
+    'rbody'
+    'rheaders'
+  )
+
+  compadd -d _descriptions -a _values
+}
+
+function _rest-run_select_response_opts {
+  local domain="$(_rest-run_read_domain)"
+  local -a _values
+
+  for f in $(find $HOME/.cache/rest-run/$domain -type f -iname '*response*'); do
+    _values+=("$(echo "$f" | sed 's/^.*\/\.cache\/rest-run\///g')")
+  done
+
+  compadd -a _values
+}
+
 if [ "$funcstack[1]" = "_rest-run" ]; then
   _rest-run
 fi