From 31f7a32e362f2e1fcbacc4cbba857352aa483645 Mon Sep 17 00:00:00 2001
From: Vladimir Zhelezov <zhelezov@gmail.com>
Date: Tue, 8 Dec 2020 07:23:30 +0100
Subject: [PATCH] Query filter completion for amt, real and status; clean
 whitespace

---
 shell-completion/hledger-completion.bash    | 39 +++++++++------------
 shell-completion/hledger-completion.bash.m4 | 39 +++++++++------------
 2 files changed, 34 insertions(+), 44 deletions(-)

diff --git a/shell-completion/hledger-completion.bash b/shell-completion/hledger-completion.bash
index f82f30ef4..ba1c95591 100644
--- a/shell-completion/hledger-completion.bash
+++ b/shell-completion/hledger-completion.bash
@@ -228,29 +228,24 @@ _hledger_compreply_query() {
 
     local hledgerArgs=()
     case $query in
-        acct:)
-            hledgerArgs=(accounts --flat)
-            ;;
-        code:)
-            hledgerArgs=(codes)
-            ;;
-        cur:)
-            hledgerArgs=(commodities)
-            ;;
-        desc:)
-            hledgerArgs=(descriptions)
-            ;;
-        note:)
-            hledgerArgs=(notes)
-            ;;
-        payee:)
-            hledgerArgs=(payees)
-            ;;
-        tag:)
-            hledgerArgs=(tags)
-            ;;
+        acct:)  hledgerArgs=(accounts --flat) ;;
+        code:)  hledgerArgs=(codes) ;;
+        cur:)   hledgerArgs=(commodities) ;;
+        desc:)  hledgerArgs=(descriptions) ;;
+        note:)  hledgerArgs=(notes) ;;
+        payee:) hledgerArgs=(payees) ;;
+        tag:)   hledgerArgs=(tags) ;;
         *)
-            return 1
+            local wordlist
+            case $query in
+                amt:)    wordlist="< <= > >=" ;;
+                real:)   wordlist="\  0" ;;
+                status:) wordlist="\  * !" ;;
+                *)       return 1 ;;
+            esac
+            _get_comp_words_by_ref -n '<=>' -c wordToComplete
+            _hledger_compreply "$(compgen -P "$query" -W "$wordlist" -- "${wordToComplete#*:}")"
+            return 0
             ;;
     esac
 
diff --git a/shell-completion/hledger-completion.bash.m4 b/shell-completion/hledger-completion.bash.m4
index fe3020b9f..bfc48b15b 100644
--- a/shell-completion/hledger-completion.bash.m4
+++ b/shell-completion/hledger-completion.bash.m4
@@ -228,29 +228,24 @@ _hledger_compreply_query() {
 
     local hledgerArgs=()
     case $query in
-        acct:)
-            hledgerArgs=(accounts --flat)
-            ;;
-        code:)
-            hledgerArgs=(codes)
-            ;;
-        cur:)
-            hledgerArgs=(commodities)
-            ;;
-        desc:)
-            hledgerArgs=(descriptions)
-            ;;
-        note:)
-            hledgerArgs=(notes)
-            ;;
-        payee:)
-            hledgerArgs=(payees)
-            ;;
-        tag:)
-            hledgerArgs=(tags)
-            ;;
+        acct:)  hledgerArgs=(accounts --flat) ;;
+        code:)  hledgerArgs=(codes) ;;
+        cur:)   hledgerArgs=(commodities) ;;
+        desc:)  hledgerArgs=(descriptions) ;;
+        note:)  hledgerArgs=(notes) ;;
+        payee:) hledgerArgs=(payees) ;;
+        tag:)   hledgerArgs=(tags) ;;
         *)
-            return 1
+            local wordlist
+            case $query in
+                amt:)    wordlist="< <= > >=" ;;
+                real:)   wordlist="\  0" ;;
+                status:) wordlist="\  * !" ;;
+                *)       return 1 ;;
+            esac
+            _get_comp_words_by_ref -n '<=>' -c wordToComplete
+            _hledger_compreply "$(compgen -P "$query" -W "$wordlist" -- "${wordToComplete#*:}")"
+            return 0
             ;;
     esac