From b8c9e836610cc7874d5c27d2a2276f9735220a36 Mon Sep 17 00:00:00 2001
From: romkatv <roman.perepelitsa@gmail.com>
Date: Sat, 11 Jan 2020 11:45:23 +0100
Subject: [PATCH] fix a few bugs in parsing

---
 internal/parse.zsh | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/internal/parse.zsh b/internal/parse.zsh
index ba18a2d..472f558 100644
--- a/internal/parse.zsh
+++ b/internal/parse.zsh
@@ -21,6 +21,9 @@ typeset -gA _p9k_skip_token=(
   'while' ''
   'until' ''
   'do' ''
+  'done' ''
+  'esac' ''
+  'end' ''
   'coproc' ''
   'nocorrect' ''
   'time' ''
@@ -45,6 +48,7 @@ typeset -gA _p9k_skip_token=(
   '>>&' '*'
   '&>>|' '*'
   '>>|' '*'
+  'foreach' '\(*\)'
 )
 
 typeset -gA _p9k_redirect=(
@@ -75,10 +79,25 @@ typeset -gA _p9k_term=(
   ';;' ''
   ';&' ''
   ';|' ''
+  '(' ''
   ')' ''
+  '{' ''
+  '}' ''
   '()' ''
 )
 
+# False positives:
+#
+#   for x (y) z
+#   {} always {}
+#
+# Completely broken:
+#
+#   Setup:
+#     setopt interactive_comments
+#     alias x='#'
+#   Punchline:
+#     x; y
 function _p9k_extract_commands() {
   local rcquotes
   [[ -o rcquotes ]] && rcquotes=(-o rcquotes)
@@ -183,6 +202,7 @@ function _p9k_extract_commands() {
     fi
 
     commands+=${:-${(Q)${~token}}}
+    [[ $commands[-1] == '(('*'))' ]] && commands[-1]=()
     skip='^'
   done