From 477650ee441656c86b1d5f890158b043c2490343 Mon Sep 17 00:00:00 2001
From: Shohei YOSHIDA <syohex@gmail.com>
Date: Sun, 12 Apr 2020 11:03:08 +0900
Subject: [PATCH] Improve zsh completion

---
 scripts/cheat.zsh | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/scripts/cheat.zsh b/scripts/cheat.zsh
index 3e6d734..c65987d 100755
--- a/scripts/cheat.zsh
+++ b/scripts/cheat.zsh
@@ -2,12 +2,18 @@
 
 local cheats taglist pathlist
 
-_cheat_complete_cheatsheets()
+_cheat_complete_personal_cheatsheets()
 {
   cheats=("${(f)$(cheat -l -t personal | tail -n +2 | cut -d' ' -f1)}")
   _describe -t cheats 'cheats' cheats
 }
 
+_cheat_complete_full_cheatsheets()
+{
+  cheats=("${(f)$(cheat -l | tail -n +2 | cut -d' ' -f1)}")
+  _describe -t cheats 'cheats' cheats
+}
+
 _cheat_complete_tags()
 {
   taglist=("${(f)$(cheat -T)}")
@@ -26,7 +32,7 @@ _cheat() {
     '(--init)--init[Write a default config file to stdout]: :->none' \
     '(-c --colorize)'{-c,--colorize}'[Colorize output]: :->none' \
     '(-d --directories)'{-d,--directories}'[List cheatsheet directories]: :->none' \
-    '(-e --edit)'{-e,--edit}'[Edit <sheet>]: :->full' \
+    '(-e --edit)'{-e,--edit}'[Edit <sheet>]: :->personal' \
     '(-l --list)'{-l,--list}'[List cheatsheets]: :->full' \
     '(-p --path)'{-p,--path}'[Return only sheets found on path <name>]: :->pathlist' \
     '(-r --regex)'{-r,--regex}'[Treat search <phrase> as a regex]: :->none' \
@@ -34,13 +40,17 @@ _cheat() {
     '(-t --tag)'{-t,--tag}'[Return only sheets matching <tag>]: :->taglist' \
     '(-T --tags)'{-T,--tags}'[List all tags in use]: :->none' \
     '(-v --version)'{-v,--version}'[Print the version number]: :->none' \
-    '(--rm)--rm[Remove (delete) <sheet>]: :->full' \
+    '(--rm)--rm[Remove (delete) <sheet>]: :->personal' \
+    '(-)*: :->full'
 
   case $state in
     (none)
       ;;
     (full)
-      _cheat_complete_cheatsheets
+      _cheat_complete_full_cheatsheets
+      ;;
+    (personal)
+      _cheat_complete_personal_cheatsheets
       ;;
     (taglist)
       _cheat_complete_tags
@@ -49,7 +59,6 @@ _cheat() {
       _cheat_complete_paths
       ;;
     (*)
-      _cheat_complete_cheatsheets
       ;;
   esac
 }