From 3bd3adfff29d5f7996cc9667a639126a5797b33f Mon Sep 17 00:00:00 2001
From: Tom Ryder <tom@sanctum.geek.nz>
Date: Sun, 6 Jan 2019 13:16:13 +1300
Subject: Refactor Vim mail header field shortcuts

---
 vim/autoload/mail.vim | 34 ++++++++++++++++++++++++----------
 1 file changed, 24 insertions(+), 10 deletions(-)

diff --git a/vim/autoload/mail.vim b/vim/autoload/mail.vim
index baff4bbf..f37a7865 100644
--- a/vim/autoload/mail.vim
+++ b/vim/autoload/mail.vim
@@ -1,19 +1,33 @@
+" Add a header to a mail message
+function! mail#AddHeaderField(name, body) abort
+  let l:num = 0
+  while l:num < line('$') && getline(l:num + 1) !=# ''
+    let l:num += 1
+  endwhile
+  call append(l:num, a:name.': '.a:body)
+endfunction
+
+" Add a set of headers to a mail message
+function! mail#AddHeaderFields(fields) abort
+  for l:name in sort(keys(a:fields))
+    call mail#AddHeaderField(l:name, a:fields[l:name])
+  endfor
+endfunction
+
 " Flag a message as important
 function! mail#FlagImportant() abort
-  call cursor(1, 1)
-  call search('^$')
-  -
-  call append(line('.'), 'X-Priority: 1')
-  call append(line('.'), 'Importance: High')
+  call mail#AddHeaderFields({
+        \ 'Importance': 'High',
+        \ 'X-Priority': 1
+        \ })
 endfunction
 
 " Flag a message as unimportant
 function! mail#FlagUnimportant() abort
-  call cursor(1, 1)
-  call search('^$')
-  -
-  call append(line('.'), 'X-Priority: 5')
-  call append(line('.'), 'Importance: Low')
+  call mail#AddHeaderFields({
+        \ 'Importance': 'Low',
+        \ 'X-Priority': 5
+        \ })
 endfunction
 
 " Move through quoted paragraphs like normal-mode `{` and `}`
-- 
cgit v1.2.3