vim - 改进 TwiddleCase 功能以在可视 block 模式下工作

Question

以下代码段可以在 vim wiki: 上找到

function! TwiddleCase(str)
  if a:str ==# toupper(a:str)
    let result = tolower(a:str)
  elseif a:str ==# tolower(a:str)
    let result = substitute(a:str,'\(\<\w\+\>\)', '\u\1', 'g')
  else
    let result = toupper(a:str)
  endif
  return result
endfunction
vnoremap ~ ygv"=TwiddleCase(@")<CR>Pgv

然而，这在视觉块模式下失败。鉴于以下
输入:

foo alpha
bar beta

将光标定位在 f 上的 foo并按
ctrl-v je~ 产生以下结果。

Foo
Bar alpha
 beta

而不是预期的

Foo alpha
Bar beta

如何改进此功能以在可视块模式下正常工作？

Answer 1

问题在于表达式寄存器( "= )的使用强制进行字符操作。

要解决这个问题，请操作默认寄存器(无论如何都包含原始提取的文本)，并通过 getregtype() 保持原始寄存器类型。 :

vnoremap ~ y:call setreg('', TwiddleCase(@"), getregtype(''))<CR>gv""Pgv