From afd585e84cce7833c6e9c7829ba1bedb900603a2 Mon Sep 17 00:00:00 2001
From: Tim Pope <code@tpope.net>
Date: Thu, 24 Jan 2008 03:15:36 +0000
Subject: [PATCH] Added dot.vim support to surround.vim

---
 plugin/surround.vim | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/plugin/surround.vim b/plugin/surround.vim
index a3bc81d..a7a0f4c 100644
--- a/plugin/surround.vim
+++ b/plugin/surround.vim
@@ -496,6 +496,11 @@ function! s:dosurround(...) " {{{1
     call setreg('"',removed,regtype)
     let s:lastdel = removed
     let &clipboard = cb_save
+    if newchar == ""
+        silent! call dot#set("\<Plug>Dsurround".char,scount)
+    else
+        silent! call dot#set("\<Plug>Csurround".char.newchar,scount)
+    endif
 endfunction " }}}1
 
 function! s:changesurround() " {{{1
@@ -561,6 +566,9 @@ function! s:opfunc(type,...) " {{{1
     call setreg(reg,reg_save,reg_type)
     let &selection = sel_save
     let &clipboard = cb_save
+    if a:type =~ '^\d\+$'
+        silent! call dot#set("\<Plug>Y".(a:0 ? "S" : "s")."surround".char,a:type)
+    endif
 endfunction
 
 function! s:opfunc2(arg)