Merge branch 'commands-should-be-repeatable'

2 files changed, 13 insertions, 1 deletions

diff --git a/TODO b/TODO
index fb4cc91..0be7b16 100644
--- a/TODO
+++ b/TODO
@@ -17,6 +17,6 @@ Escalation and "descalation" should be an operator-pending mapping
 2020.03.21:
 - Add changelog
 v Don't save edits in search history
-- Add repeat support
+v Add repeat support
 - Add license headers
 v Add no_plugin_maps support
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