Merge pull request #75 from Chronial/fix-63

Rewrite keymap restoring to use :mkexrc (fix #63)

Author: joonty

plugin/python/vdebug/util.py

@@ -2,6 +2,7 @@
 import vdebug.log
 import vim
 import re
+import os
 import urllib
 
 class Keymapper:
@@ -25,24 +26,37 @@ def close_key(self):
     def map(self):
         if self.is_mapped:
             return
+        self._store_old_map()
         for func in self.keymaps:
-            key = self.keymaps[func]
             if func not in self.exclude:
-                vim.command("redir @z | silent noremap %s%s | redir END" %(self.leader,key))
-                self.__save_map_output( vim.eval("@z").strip() )
+                key = self.keymaps[func]
                 map_cmd = "noremap %s%s :python debugger.%s()<cr>" %\
                     (self.leader,key,func)
                 vim.command(map_cmd)
         self.is_mapped = True
 
-    def __save_map_output(self,output):
-        if output.startswith('No mapping'):
-            return False
-        else:
-            vdebug.log.Log("Storing existing key mapping, '%s' " % output,\
-                    vdebug.log.Logger.DEBUG)
-            self.existing.append(output)
-            return True
+    def _store_old_map(self):
+        vim.command('let tempfile=tempname()')
+        tempfile = vim.eval("tempfile")
+        vim.command('mkexrc! %s' % (tempfile))
+        regex = re.compile(r'^([nvxsoilc]|)(nore)?map!?')
+        split_regex = re.compile(r'\s+')
+        keys = set(v for (k,v) in self.keymaps.items() if k not in self.exclude)
+        special = set(["<buffer>", "<silent>", "<special>", "<script>", "<expr>", "<unique>"])
+        for line in open(tempfile, 'r'):
+            if not regex.match(line):
+                continue
+            parts = split_regex.split(line)[1:]
+            for p in parts:
+                if p in special:
+                    continue
+                elif p in keys:
+                    vdebug.log.Log("Storing existing key mapping, '%s' " % line,
+                                   vdebug.log.Logger.DEBUG)
+                    self.existing.append(line)
+                else:
+                    break
+        os.remove(tempfile)
 
     def unmap(self):
         if self.is_mapped:
@@ -55,18 +69,7 @@ def unmap(self):
             for mapping in self.existing:
                 vdebug.log.Log("Remapping key with '%s' " % mapping,\
                         vdebug.log.Logger.DEBUG)
-                regex = re.compile(r'\s+')
-                parts = regex.split(mapping)
-                mapcmd = 'noremap'
-                if len(parts)>2:
-                    modeRegex = re.compile(r'^[nvsxoilc!]$')
-                    if modeRegex.match(parts[0]):
-                        mapping = ' '.join(parts[1:])
-                        if parts[0]=='!':
-                            mapcmd = 'noremap!'
-                        else:
-                            mapcmd = '%snoremap' % parts[0]
-                vim.command("%s %s" % (mapcmd,mapping))
+                vim.command(mapping)
 
 class FilePath:
     is_win = False