Add repeat support

Commands should be able to be repeated. Makes it easier to mark several
to-dos complete in a row, for example.

1 files changed, 12 insertions, 0 deletions

diff --git a/autoload/todo.vim b/autoload/todo.vim
index 8cfa986..070d3f5 100644
--- a/autoload/todo.vim
+++ b/autoload/todo.vim
@@ -4,6 +4,8 @@ function! todo#MarkComplete()
 	keeppatterns s/\v^(\s*)[-_!xS] /\1v /
 
 	call winrestview(view)
+
+	silent! call repeat#set("\<Plug>TwodoMarkComplete", v:count)
 endfunction
 
 
@@ -13,6 +15,8 @@ function! todo#MarkDeleted()
 	keeppatterns s/\v^(\s*)[-_!vS] /\1x /
 
 	call winrestview(view)
+
+	silent! call repeat#set("\<Plug>TwodoMarkDeleted", v:count)
 endfunction
 
 
@@ -22,6 +26,8 @@ function! todo#MarkPartiallyCompleted()
 	keeppatterns s/\v^(\s*)[-_!xv] /\1S /
 
 	call winrestview(view)
+
+	silent! call repeat#set("\<Plug>TwodoMarkPartiallyCompleted", v:count)
 endfunction
 
 
@@ -31,6 +37,8 @@ function! todo#RemoveOldTodos()
 	keeppatterns g/^\s*[vx] /d
 
 	call winrestview(view)
+
+	silent! call repeat#set("\<Plug>TwodoRemoveOldTodos", v:count)
 endfunction
 
 
@@ -49,6 +57,8 @@ function! todo#Escalate()
 	endif
 
 	call setline(line('.'), todo)
+
+	silent! call repeat#set("\<Plug>TwodoEscalate", v:count)
 endfunction
 
 
@@ -67,4 +77,6 @@ function! todo#Descalate()
 	endif
 
 	call setline(line('.'), todo)
+
+	silent! call repeat#set("\<Plug>TwodoDescalate", v:count)
 endfunction