Add visual mappping to ctr

Vigemus · Jan 24, 2017 · bb62179 · bb62179
1 parent 2b9dc3f
commit bb62179
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/plugin/iron.vim b/plugin/iron.vim
@@ -1,3 +1,4 @@
 nmap <silent> ctr :set opfunc=IronSendMotion<CR>g@
+vmap <silent> ctr :call IronSendMotion('visual')<CR>
 "Call previous command again
 nmap <silent> cp :call IronSend("\u001b\u005b\u0041")<CR>
diff --git a/rplugin/python3/iron/__init__.py b/rplugin/python3/iron/__init__.py
@@ -113,10 +113,13 @@ def mapping_send(self, args):
         if fn:
             fn(self)
 
-    @neovim.function("IronSendMotion")
-    def send_motion_to_repl(self, args):
+    @neovim.function("IronSendMotion", range=True)
+    def send_motion_to_repl(self, args, rng=None):
+        logger.debug("Supplied data: {}".format(", ".join(args)))
         if args[0] == 'line':
             self.call_cmd("""normal! '[V']"sy""")
+        if args[0] == 'visual':
+            self.call_cmd("""normal! `<v`>"sy""")
         else:
             self.call_cmd("""normal! `[v`]"sy""")
 
@@ -131,7 +134,6 @@ def send_to_repl(self, args):
             return None
 
         logger.debug("Supplied data: {}".format(args[0]))
-
         logger.info("Sending data to repl -> {}".format(repl))
 
         if 'multiline' in repl: