PATCH: Add localhistory state to ZLE_STATE

[Mikael Magnusson <mikachu@xxxxxxxxx>]

The sorting is possibly a bit of a hack.

This lets you make wrappers that use set-local-history without changing
the global setting, or show the current setting in the prompt, or
whatever.

function _set_local_history() {
  local oldstate=$ZLE_STATE
  zle .set-local-history
  if [[ $ZLE_STATE == $oldstate ]]; then
    return
  fi
  if [[ $ZLE_STATE == *globalhistory* ]]; then
    psvar[8]=(1)
  else
    psvar[8]=()
  fi
  zle reset-prompt
}
zle -N set-local-history _set_local_history

function _local_history_command {
  if [[ $ZLE_STATE == *localhistory* ]]; then
    zle ${WIDGET#local-}
    return
  fi
  zle .set-local-history -n 1
  zle ${WIDGET#local-}
  zle .set-local-history -n 0
}
zle -N local-down-history _local_history_command
zle -N local-up-history _local_history_command
bindkey '^X^P' local-up-history
bindkey '^X^N' local-down-history

---
 Doc/Zsh/zle.yo       |   19 ++++++++++++++-----
 Src/Zle/zle_params.c |   28 ++++++++++++++++++----------
 2 files changed, 32 insertions(+), 15 deletions(-)

diff --git a/Doc/Zsh/zle.yo b/Doc/Zsh/zle.yo
index f9c20de..86e139f 100644
--- a/Doc/Zsh/zle.yo
+++ b/Doc/Zsh/zle.yo
@@ -918,11 +918,20 @@ item(tt(ZLE_STATE) (scalar))(
 Contains a set of space-separated words that describe the current tt(zle)
 state.
 
-Currently, the only state shown is the insert mode as set by the
-tt(overwrite-mode) or tt(vi-replace) widgets.  The string contains
-`tt(insert)' if characters to be inserted on the command line move existing
-characters to the right, `tt(overwrite)' if characters to be inserted
-overwrite existing characters.
+Currently, the states shown are the insert mode as set by the
+tt(overwrite-mode) or tt(vi-replace) widgets and whether history commands
+will visit imported entries as controlled by the set-local-history widget.
+The string contains `tt(insert)' if characters to be inserted on the
+command line move existing characters to the right or `tt(overwrite)'
+if characters to be inserted overwrite existing characters. It contains
+`tt(localhistory)' if only local history commands will be visited or
+`tt(globalhistory)' if imported history commands will also be visited.
+
+The substrings are sorted in alphabetical order so that if you want to
+test for two specific substrings in a future-proof way, you can do match
+by doing:
+
+example(if [[ $ZLE_STATE == *insert*globalhistory* ]]; then ...; fi)
 )
 enditem()
 
diff --git a/Src/Zle/zle_params.c b/Src/Zle/zle_params.c
index 4c32d18..e050cd6 100644
--- a/Src/Zle/zle_params.c
+++ b/Src/Zle/zle_params.c
@@ -712,21 +712,17 @@ get_context(UNUSED(Param pm))
 static char *
 get_zle_state(UNUSED(Param pm))
 {
-    char *zle_state = NULL, *ptr = NULL;
+    char *zle_state = NULL, *ptr = NULL, **arr = NULL;
     int itp, istate, len = 0;
 
     /*
-     * When additional substrings are added, they should be kept in
-     * alphabetical order, so the user can easily match against this
-     * parameter: if [[ $ZLE_STATE == *bar*foo*zonk* ]]; then ...; fi
+     * Substrings are sorted at the end, so the user can
+     * easily match against this parameter:
+     * if [[ $ZLE_STATE == *bar*foo*zonk* ]]; then ...; fi
      */
     for (itp = 0; itp < 2; itp++) {
 	char *str;
-	/*
-	 * Currently there is only one state: insert or overwrite.
-	 * This loop is to make it easy to add others.
-	 */
-	for (istate = 0; istate < 1; istate++) {
+	for (istate = 0; istate < 2; istate++) {
 	    int slen;
 	    switch (istate) {
 	    case 0:
@@ -736,6 +732,13 @@ get_zle_state(UNUSED(Param pm))
 		    str = "overwrite";
 		}
 		break;
+	    case 1:
+		if (hist_skip_flags & HIST_FOREIGN) {
+		    str = "globalhistory";
+		} else {
+		    str = "localhistory";
+		}
+		break;
 
 	    default:
 		str = "";
@@ -749,7 +752,7 @@ get_zle_state(UNUSED(Param pm))
 	    } else {
 		/* Accumulating string */
 		if (istate)
-		    *ptr++ = ' ';
+		    *ptr++ = ':';
 		memcpy(ptr, str, slen);
 		ptr += slen;
 	    }
@@ -761,6 +764,11 @@ get_zle_state(UNUSED(Param pm))
 	    *ptr = '\0';
 	}
     }
+    
+    arr = colonsplit(zle_state, 0);
+    strmetasort(arr, SORTIT_ANYOLDHOW, NULL);
+    zle_state = zjoin(arr, ' ', 1);
+    freearray(arr);
 
     return zle_state;
 }
-- 
1.7.10.GIT