From a29a63f7ff3db981ae7fc38eace081ea4f2ad880 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Tue, 10 Mar 2020 01:12:58 +0100
Subject: todo: Add count support to TODO movement commands

---
 bundle/todo/autoload/todo.vim | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

(limited to 'bundle/todo/autoload/todo.vim')

diff --git a/bundle/todo/autoload/todo.vim b/bundle/todo/autoload/todo.vim
index d5579c1..6885b2b 100644
--- a/bundle/todo/autoload/todo.vim
+++ b/bundle/todo/autoload/todo.vim
@@ -1,9 +1,19 @@
-function! todo#NextTODO()
+function! s:MoveToTODO(search_flags)
+	let cnt = v:count1
 	normal! m'
-	call search('TODO')
+
+	let i = 0
+	while i < cnt
+		call search('TODO', a:search_flags)
+
+		let i += 1
+	endwhile
+endfunction
+
+function! todo#NextTODO()
+	call s:MoveToTODO('')
 endfunction
 
 function! todo#PreviousTODO()
-	normal! m'
-	call search('TODO', 'b')
+	call s:MoveToTODO('b')
 endfunction
-- 
cgit v1.2.3