From f0290b08ddd4719b5b0c9129d2566d317f597b9b Mon Sep 17 00:00:00 2001
From: Martin Grenfell <martin_grenfell@msn.com>
Date: Tue, 21 Jul 2009 19:23:44 +1200
Subject: [PATCH] add another api callback to determine whether menu items
 should appear

this way we can e.g. not show the git submenu unless a git repo is
actually present
---
 plugin/NERD_tree.vim | 72 +++++++++++++++++++++++++++++++++-----------
 1 file changed, 55 insertions(+), 17 deletions(-)

diff --git a/plugin/NERD_tree.vim b/plugin/NERD_tree.vim
index 399f566..f4679dd 100644
--- a/plugin/NERD_tree.vim
+++ b/plugin/NERD_tree.vim
@@ -429,7 +429,6 @@ endfunction
 "CLASS: MenuCallback {{{2
 "============================================================
 let s:MenuCallback = {}
-
 "FUNCTION: MenuCallback.All() {{{3
 function! s:MenuCallback.All()
     if !exists("s:menuCallbacks")
@@ -438,16 +437,58 @@ function! s:MenuCallback.All()
     return s:menuCallbacks
 endfunction
 
-"FUNCTION: MenuCallback.Create(text, shortcut, callback) {{{3
-function! s:MenuCallback.Create(text, shortcut, callback)
+"FUNCTION: MenuCallback.AllEnabledCallbacks() {{{3
+function! s:MenuCallback.AllEnabledCallbacks()
+    let toReturn = []
+    for i in s:MenuCallback.All()
+        if i.enabled()
+            call add(toReturn, i)
+        endif
+    endfor
+    return toReturn
+endfunction
+
+"FUNCTION: MenuCallback.FindByShortcut(shortcut) {{{3
+function! s:MenuCallback.FindByShortcut(shortcut)
+    for i in s:MenuCallback.All()
+        if i.shortcut ==# a:shortcut
+            return i
+        endif
+    endfor
+    return {}
+endfunction
+
+"FUNCTION: MenuCallback.Create(options) {{{3
+function! s:MenuCallback.Create(options)
     let newCallback = {}
     let newCallback = copy(self)
-    let newCallback.text = a:text
-    let newCallback.shortcut = a:shortcut
-    let newCallback.callback = a:callback
+
+    let shortcut = a:options['shortcut']
+    let callback = a:options['callback']
+
+
+    let newCallback.text = a:options['text']
+    let newCallback.shortcut = a:options['shortcut']
+    let newCallback.callback = a:options['callback']
+    if has_key(a:options, 'check_to_enable_callback')
+        let newCallback.check_to_enable_callback = a:options['check_to_enable_callback']
+    endif
     call add(s:MenuCallback.All(), newCallback)
 endfunction
 
+"FUNCTION: MenuCallback.enabled() {{{3
+function! s:MenuCallback.enabled()
+    if has_key(self, "check_to_enable_callback")
+        return {self.check_to_enable_callback}()
+    endif
+    return 1
+endfunction
+
+"FUNCTION: MenuCallback.execute() {{{3
+function! s:MenuCallback.execute()
+    call {self.callback}()
+endfunction
+
 "FUNCTION: MenuCallback.ShowMenu() {{{3
 function! s:MenuCallback.ShowMenu()
     let curNode = s:TreeFileNode.GetSelected()
@@ -459,23 +500,20 @@ function! s:MenuCallback.ShowMenu()
     let prompt = "NERDTree Menu\n" .
        \ "==========================================================\n"
 
-    for i in s:MenuCallback.All()
+    for i in s:MenuCallback.AllEnabledCallbacks()
         let prompt .= i.text . "\n"
     endfor
 
     echo prompt
 
-    let choice = nr2char(getchar())
+    let callback = s:MenuCallback.FindByShortcut(nr2char(getchar()))
+    if !empty(callback) && callback.enabled()
+        redraw
+        call callback.execute()
+    endif
 
-    for i in s:MenuCallback.All()
-        if choice ==# i.shortcut
-            exec "call " . i.callback . "()"
-            return
-        endif
-    endfor
 endfunction
 
-
 "CLASS: TreeFileNode {{{2
 "This class is the parent of the TreeDirNode class and constitures the
 "'Component' part of the composite design pattern between the treenode
@@ -2388,8 +2426,8 @@ function! NERDTreeGetCurrentPath()
     endif
 endfunction
 
-function! NERDTreeAddMenuItem(text, shortcut, callback)
-    call s:MenuCallback.Create(a:text, a:shortcut, a:callback)
+function! NERDTreeAddMenuItem(options)
+    call s:MenuCallback.Create(a:options)
 endfunction
 
 function! NERDTreeRender()