correct fold line counts
[notmuch] / vim / plugin / notmuch.vim
1 " notmuch.vim plugin --- run notmuch within vim
2 "
3 " Copyright © Carl Worth
4 "
5 " This file is part of Notmuch.
6 "
7 " Notmuch is free software: you can redistribute it and/or modify it
8 " under the terms of the GNU General Public License as published by
9 " the Free Software Foundation, either version 3 of the License, or
10 " (at your option) any later version.
11 "
12 " Notmuch is distributed in the hope that it will be useful, but
13 " WITHOUT ANY WARRANTY; without even the implied warranty of
14 " MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 " General Public License for more details.
16 "
17 " You should have received a copy of the GNU General Public License
18 " along with Notmuch.  If not, see <http://www.gnu.org/licenses/>.
19 "
20 " Authors: Bart Trojanowski <bart@jukie.net>
21
22 " --- configuration defaults {{{1
23
24 let s:notmuch_defaults = {
25         \ 'g:notmuch_cmd':                           'notmuch'                    ,
26         \
27         \ 'g:notmuch_search_newest_first':           1                            ,
28         \ 'g:notmuch_search_from_column_width':      20                           ,
29         \
30         \ 'g:notmuch_show_fold_signatures':          1                            ,
31         \ 'g:notmuch_show_fold_citations':           1                            ,
32         \ 'g:notmuch_show_fold_bodies':              0                            ,
33         \ 'g:notmuch_show_fold_headers':             1                            ,
34         \
35         \ 'g:notmuch_show_message_begin_regexp':     '^\fmessage{'                ,
36         \ 'g:notmuch_show_message_end_regexp':       '^\fmessage}'                ,
37         \ 'g:notmuch_show_header_begin_regexp':      '^\fheader{'                 ,
38         \ 'g:notmuch_show_header_end_regexp':        '^\fheader}'                 ,
39         \ 'g:notmuch_show_body_begin_regexp':        '^\fbody{'                   ,
40         \ 'g:notmuch_show_body_end_regexp':          '^\fbody}'                   ,
41         \ 'g:notmuch_show_attachment_begin_regexp':  '^\fattachment{'             ,
42         \ 'g:notmuch_show_attachment_end_regexp':    '^\fattachment}'             ,
43         \ 'g:notmuch_show_part_begin_regexp':        '^\fpart{'                   ,
44         \ 'g:notmuch_show_part_end_regexp':          '^\fpart}'                   ,
45         \ 'g:notmuch_show_marker_regexp':            '^\f\\(message\\|header\\|body\\|attachment\\|part\\)[{}].*$',
46         \
47         \ 'g:notmuch_show_message_parse_regexp':     '\(id:[^ ]*\) depth:\([0-9]*\) filename:\(.*\)$',
48         \ 'g:notmuch_show_tags_regexp':              '(\([^)]*\))$'               ,
49         \
50         \ 'g:notmuch_show_signature_regexp':         '^\(-- \?\|_\+\)$'           ,
51         \ 'g:notmuch_show_signature_lines_max':      12                           ,
52         \
53         \ 'g:notmuch_show_citation_regexp':          '^\s*>'                      ,
54         \ }
55
56 " defaults for g:notmuch_initial_search_words
57 " override with: let g:notmuch_initial_search_words = [ ... ]
58 let s:notmuch_initial_search_words_defaults = [
59         \ 'tag:inbox and tag:unread',
60         \ ]
61
62 " defaults for g:notmuch_show_headers
63 " override with: let g:notmuch_show_headers = [ ... ]
64 let s:notmuch_show_headers_defaults = [
65         \ 'Subject',
66         \ 'To',
67         \ 'Cc',
68         \ 'Bcc',
69         \ 'Date',
70         \ ]
71
72 " --- keyboard mapping definitions {{{1
73
74 " --- --- bindings for search screen {{{2
75 let g:notmuch_search_maps = {
76         \ '<Enter>':    ':call <SID>NM_search_show_thread()<CR>',
77         \ 'a':          ':call <SID>NM_search_archive_thread()<CR>',
78         \ 'f':          ':call <SID>NM_search_filter()<CR>',
79         \ 'm':          ':call <SID>NM_new_mail()<CR>',
80         \ 'o':          ':call <SID>NM_search_toggle_order()<CR>',
81         \ 'r':          ':call <SID>NM_search_reply_to_thread()<CR>',
82         \ 's':          ':call <SID>NM_search_prompt()<CR>',
83         \ 'S':          ':call <SID>NM_search_edit()<CR>',
84         \ 't':          ':call <SID>NM_search_filter_by_tag()<CR>',
85         \ 'q':          ':call <SID>NM_kill_this_buffer()<CR>',
86         \ '+':          ':call <SID>NM_search_add_tags([])<CR>',
87         \ '-':          ':call <SID>NM_search_remove_tags([])<CR>',
88         \ '=':          ':call <SID>NM_search_refresh_view()<CR>',
89         \ }
90
91 " --- --- bindings for show screen {{{2
92 let g:notmuch_show_maps = {
93         \ '<C-P>':      ':call <SID>NM_show_prev(1)<CR>',
94         \ '<C-N>':      ':call <SID>NM_show_next(1)<CR>',
95         \ 'q':          ':call <SID>NM_kill_this_buffer()<CR>',
96         \
97         \ 'b':          ':call <SID>NM_show_fold_toggle(''b'', ''bdy'', !g:notmuch_show_fold_bodies)<CR>',
98         \ 'c':          ':call <SID>NM_show_fold_toggle(''c'', ''cit'', !g:notmuch_show_fold_citations)<CR>',
99         \ 'h':          ':call <SID>NM_show_fold_toggle(''h'', ''hdr'', !g:notmuch_show_fold_headers)<CR>',
100         \ 's':          ':call <SID>NM_show_fold_toggle(''s'', ''sig'', !g:notmuch_show_fold_signatures)<CR>',
101         \
102         \ 'a':          ':call <SID>NM_show_archive_thread()<CR>',
103         \ 'A':          ':call <SID>NM_show_mark_read_then_archive_thread()<CR>',
104         \ 'N':          ':call <SID>NM_show_mark_read_then_next_open_message()<CR>',
105         \ 'v':          ':call <SID>NM_show_view_all_mime_parts()<CR>',
106         \ '+':          ':call <SID>NM_show_add_tag()<CR>',
107         \ '-':          ':call <SID>NM_show_remove_tag()<CR>',
108         \ '<Space>':    ':call <SID>NM_show_advance_marking_read_and_archiving()<CR>',
109         \ '\|':         ':call <SID>NM_show_pipe_message()<CR>',
110         \
111         \ 'r':          ':call <SID>NM_show_reply()<CR>',
112         \ 'm':          ':call <SID>NM_new_mail()<CR>',
113         \ }
114
115
116 " --- implement search screen {{{1
117
118 function! s:NM_cmd_search(words)
119         let cmd = ['search']
120         if g:notmuch_search_newest_first
121                 let cmd = cmd + ['--sort=newest-first']
122         else
123                 let cmd = cmd + ['--sort=oldest-first']
124         endif
125         let data = s:NM_run(cmd + a:words)
126         "let data = substitute(data, '27/27', '25/27', '')
127         "let data = substitute(data, '\[4/4\]', '[0/4]', '')
128         let lines = split(data, "\n")
129         let disp = copy(lines)
130         "call map(disp, 'substitute(v:val, "^thread:\\S* ", "", "")' )
131         call map(disp, 's:NM_cmd_search_fmtline(v:val)')
132
133         call <SID>NM_newBuffer('search', join(disp, "\n"))
134         let b:nm_raw_lines = lines
135         let b:nm_search_words = a:words
136
137         call <SID>NM_cmd_search_mksyntax()
138         call <SID>NM_set_map(g:notmuch_search_maps)
139         setlocal cursorline
140         setlocal nowrap
141 endfunction
142 function! s:NM_cmd_search_fmtline(line)
143         let m = matchlist(a:line, '^\(thread:\S\+\)\s\([^]]\+\]\) \([^;]\+\); \(.*\) (\([^(]*\))$')
144         if !len(m)
145                 return 'ERROR PARSING: ' . a:line
146         endif
147         let max = g:notmuch_search_from_column_width
148         let from = m[3]
149         if strlen(from) >= max
150                 let from = m[3][0:max-4] . '...'
151         endif
152         return printf('%s %-20s | %s (%s)', m[2], from, m[4], m[5])
153 endfunction
154 function! s:NM_cmd_search_mksyntax()
155         syntax clear nmSearchFrom
156         "syntax region nmSearchFrom start='\]\@<=' end='.'me=e+5,he=e+5,re=e+5 oneline contained
157         "syntax match nmSearchFrom /\]\@<=.\{10\}/ oneline contained
158         exec printf('syntax match nmSearchFrom /\(\] \)\@<=.\{%d\}/ oneline contained', g:notmuch_search_from_column_width)
159         "exec printf('syntax region nmSearchFrom start=''\%%%dv'' end=''\%%%dv'' oneline contained', 20, 30)
160 endfunction
161
162 " --- --- search screen action functions {{{2
163
164 function! s:NM_search_show_thread()
165         let id = <SID>NM_search_find_thread_id()
166         if id != ''
167                 call <SID>NM_cmd_show([id])
168         endif
169 endfunction
170
171 function! s:NM_search_prompt()
172         " TODO: input() can support completion
173         let text = input('NotMuch Search: ')
174         if strlen(text)
175                 let tags = split(text)
176         else
177                 let tags = s:notmuch_initial_search_words_defaults
178         endif
179         setlocal bufhidden=delete
180         call <SID>NM_cmd_search(tags)
181 endfunction
182
183 function! s:NM_search_edit()
184         " TODO: input() can support completion
185         let text = input('NotMuch Search: ', join(b:nm_search_words, ' '))
186         if strlen(text)
187                 call <SID>NM_cmd_search(split(text))
188         endif
189 endfunction
190
191 function! s:NM_search_archive_thread()
192         call <SID>NM_add_remove_tags_on_screen('-', ['inbox'])
193         call <SID>NM_add_remove_tags('-', ['inbox'])
194         norm j
195 endfunction
196
197 function! s:NM_search_filter()
198         call <SID>NM_search_filter_helper('Filter: ', '')
199 endfunction
200
201 function! s:NM_search_filter_by_tag()
202         call <SID>NM_search_filter_helper('Filter Tag(s): ', 'tag:')
203 endfunction
204
205 function! s:NM_search_filter_helper(prompt, prefix)
206         " TODO: input() can support completion
207         let text = input(a:prompt)
208         if !strlen(text)
209                 return
210         endif
211
212         let tags = split(text)
213         map(tags, 'and a:prefix . v:val')
214         let tags = b:nm_search_words + tags
215         echo tags
216
217         let prev_bufnr = bufnr('%')
218         setlocal bufhidden=hide
219         call <SID>NM_cmd_search(tags)
220         setlocal bufhidden=delete
221         let b:nm_prev_bufnr = prev_bufnr
222 endfunction
223
224 function! s:NM_search_toggle_order()
225         let g:notmuch_search_newest_first = !g:notmuch_search_newest_first
226         " FIXME: maybe this would be better done w/o reading re-reading the lines
227         "         reversing the b:nm_raw_lines and the buffer lines would be better
228         call <SID>NM_search_refresh_view()
229 endfunction
230
231 function! s:NM_search_reply_to_thread()
232         echo 'not implemented'
233 endfunction
234
235 function! s:NM_search_add_tags(tags)
236         call <SID>NM_search_add_remove_tags('Add Tag(s): ', '+', a:tags)
237 endfunction
238
239 function! s:NM_search_remove_tags(tags)
240         call <SID>NM_search_add_remove_tags('Remove Tag(s): ', '-', a:tags)
241 endfunction
242
243 function! s:NM_search_refresh_view()
244         let lno = line('.')
245         setlocal bufhidden=delete
246         call <SID>NM_cmd_search(b:nm_search_words)
247         " FIXME: should find the line of the thread we were on if possible
248         exec printf('norm %dG', lno)
249 endfunction
250
251 " --- --- search screen helper functions {{{2
252
253 function! s:NM_search_find_thread_id()
254         if !exists('b:nm_raw_lines')
255                 echoe 'no b:nm_raw_lines'
256                 return ''
257         else
258                 let line = line('.')
259                 let info = b:nm_raw_lines[line-1]
260                 let what = split(info, '\s\+')[0]
261                 return what
262         endif
263 endfunction
264
265 function! s:NM_search_add_remove_tags(prompt, prefix, intags)
266         if type(a:intags) != type([]) || len(a:intags) == 0
267                 " TODO: input() can support completion
268                 let text = input(a:prompt)
269                 if !strlen(text)
270                         return
271                 endif
272                 let tags = split(text, ' ')
273         else
274                 let tags = a:intags
275         endif
276         call <SID>NM_add_remove_tags(a:prefix, tags)
277         call <SID>NM_add_remove_tags_on_screen(a:prefix, tags)
278 endfunction
279
280 " --- implement show screen {{{1
281
282 function! s:NM_cmd_show(words)
283         let prev_bufnr = bufnr('%')
284         let data = s:NM_run(['show'] + a:words)
285         let lines = split(data, "\n")
286
287         let info = s:NM_cmd_show_parse(lines)
288
289         setlocal bufhidden=hide
290         call <SID>NM_newBuffer('show', join(info['disp'], "\n"))
291         setlocal bufhidden=delete
292         let b:nm_raw_info = info
293         let b:nm_prev_bufnr = prev_bufnr
294
295         call <SID>NM_cmd_show_mkfolds()
296         call <SID>NM_cmd_show_mksyntax()
297         call <SID>NM_set_map(g:notmuch_show_maps)
298         setlocal foldtext=NM_cmd_show_foldtext()
299         setlocal fillchars=
300         setlocal foldcolumn=6
301
302 endfunction
303
304 function! s:NM_show_prev(can_change_thread)
305         let info = b:nm_raw_info
306         let lnum = line('.')
307         for msg in reverse(copy(info['msgs']))
308                 if lnum <= msg['start']
309                         continue
310                 endif
311
312                 exec printf('norm %dG', msg['start'])
313                 " TODO: try to fit the message on screen
314                 norm zz
315                 return
316         endfor
317         if !a:can_change_thread
318                 return
319         endif
320         call <SID>NM_kill_this_buffer()
321         if line('.') != line('0')
322                 norm k
323                 call <SID>NM_search_show_thread()
324                 norm G
325                 call <SID>NM_show_prev(0)
326         else
327                 echo 'No more messages.'
328         endif
329 endfunction
330
331 function! s:NM_show_next(can_change_thread)
332         let info = b:nm_raw_info
333         let lnum = line('.')
334         for msg in info['msgs']
335                 if lnum >= msg['start']
336                         continue
337                 endif
338
339                 exec printf('norm %dG', msg['start'])
340                 " TODO: try to fit the message on screen
341                 norm zz
342                 return
343         endfor
344         if !a:can_change_thread
345                 return
346         endif
347         call <SID>NM_kill_this_buffer()
348         if line('.') != line('$')
349                 norm j
350                 call <SID>NM_search_show_thread()
351         else
352                 echo 'No more messages.'
353         endif
354 endfunction
355
356 function! s:NM_show_archive_thread()
357         echo 'not implemented'
358 endfunction
359
360 function! s:NM_show_mark_read_then_archive_thread()
361         echo 'not implemented'
362 endfunction
363
364 function! s:NM_show_next_message()
365         echo 'not implemented'
366 endfunction
367
368 function! s:NM_show_mark_read_then_next_open_message()
369         echo 'not implemented'
370 endfunction
371
372 function! s:NM_show_previous_message()
373         echo 'not implemented'
374 endfunction
375
376 function! s:NM_show_reply()
377         echo 'not implemented'
378 endfunction
379
380 function! s:NM_show_view_all_mime_parts()
381         echo 'not implemented'
382 endfunction
383
384 function! s:NM_show_view_raw_message()
385         echo 'not implemented'
386 endfunction
387
388 function! s:NM_show_add_tag()
389         echo 'not implemented'
390 endfunction
391
392 function! s:NM_show_remove_tag()
393         echo 'not implemented'
394 endfunction
395
396 function! s:NM_show_advance_marking_read_and_archiving()
397         echo 'not implemented'
398 endfunction
399
400 function! s:NM_show_pipe_message()
401         echo 'not implemented'
402 endfunction
403
404 " --- --- search screen helper functions {{{2
405
406 function! s:NM_show_fold_toggle(key, type, fold)
407         let info = b:nm_raw_info
408         let act = 'open'
409         if a:fold
410                 let act = 'close'
411         endif
412         for fld in info['folds']
413                 if fld[0] == a:type
414                         exec printf('%dfold%s', fld[1], act)
415                 endif
416         endfor
417         exec printf('nnoremap <buffer> %s :call <SID>NM_show_fold_toggle(''%s'', ''%s'', %d)<CR>', a:key, a:key, a:type, !a:fold)
418 endfunction
419
420
421 " s:NM_cmd_show_parse returns the following dictionary:
422 "    'disp':     lines to display
423 "    'msgs':     message info dicts { start, end, id, depth, filename, descr, header }
424 "    'folds':    fold info arrays [ type, start, end ]
425 "    'foldtext': fold text indexed by start line
426 function! s:NM_cmd_show_parse(inlines)
427         let info = { 'disp': [],       
428                    \ 'msgs': [],       
429                    \ 'folds': [],      
430                    \ 'foldtext': {} }  
431         let msg = {}
432         let hdr = {}
433
434         let in_message = 0
435         let in_header = 0
436         let in_body = 0
437         let in_part = ''
438
439         let body_start = -1
440         let part_start = -1
441
442         let mode_type = ''
443         let mode_start = -1
444
445         let inlnum = 0
446         for line in a:inlines
447                 let inlnum = inlnum + 1
448                 let foldinfo = []
449
450                 if strlen(in_part)
451                         let part_end = 0
452
453                         if match(line, g:notmuch_show_part_end_regexp) != -1
454                                 let part_end = len(info['disp'])
455                         else
456                                 call add(info['disp'], line)
457                         endif
458
459                         if in_part == 'text/plain'
460                                 if !part_end && mode_type == ''
461                                         if match(line, g:notmuch_show_signature_regexp) != -1
462                                                 let mode_type = 'sig'
463                                                 let mode_start = len(info['disp'])
464                                         elseif match(line, g:notmuch_show_citation_regexp) != -1
465                                                 let mode_type = 'cit'
466                                                 let mode_start = len(info['disp'])
467                                         endif
468                                 elseif mode_type == 'cit'
469                                         if part_end || match(line, g:notmuch_show_citation_regexp) == -1
470                                                 let outlnum = len(info['disp'])
471                                                 if mode_start != outlnum
472                                                         let foldinfo = [ mode_type, mode_start, outlnum-1,
473                                                                        \ printf('[ %d-line citation.  Press "c" to show. ]', outlnum - mode_start) ]
474                                                 endif
475                                                 let mode_type = ''
476                                         endif
477                                 elseif mode_type == 'sig'
478                                         let outlnum = len(info['disp'])
479                                         if (outlnum - mode_start) > g:notmuch_show_signature_lines_max
480                                                 let mode_type = ''
481                                         elseif part_end
482                                                 if mode_start != outlnum
483                                                         let foldinfo = [ mode_type, mode_start, outlnum-1,
484                                                                        \ printf('[ %d-line signature.  Press "s" to show. ]', outlnum - mode_start) ]
485                                                 endif
486                                                 let mode_type = ''
487                                         endif
488                                 endif
489                         endif
490
491                         if part_end
492                                 " FIXME: this is a hack for handling two folds being added for one line
493                                 "         we should handle addinga fold in a function
494                                 if len(foldinfo)
495                                         call add(info['folds'], foldinfo[0:2])
496                                         let info['foldtext'][foldinfo[1]] = foldinfo[3]
497                                 endif
498
499                                 let foldinfo = [ 'text', part_start, part_end,
500                                                \ printf('[ %d-line %s.  Press "p" to show. ]', part_end - part_start, in_part) ]
501                                 let in_part = ''
502                                 call add(info['disp'], '')
503                         endif
504
505                 elseif in_body
506                         if !has_key(msg,'body_start')
507                                 let msg['body_start'] = len(info['disp']) + 1
508                         endif
509                         if match(line, g:notmuch_show_body_end_regexp) != -1
510                                 let body_end = len(info['disp'])
511                                 let foldinfo = [ 'bdy', body_start, body_end,
512                                                \ printf('[ BODY %d - %d lines ]', len(info['msgs']), body_end - body_start) ]
513
514                                 let in_body = 0
515
516                         elseif match(line, g:notmuch_show_part_begin_regexp) != -1
517                                 let m = matchlist(line, 'ID: \(\d\+\), Content-type: \(\S\+\)')
518                                 let in_part = 'unknown'
519                                 if len(m)
520                                         let in_part = m[2]
521                                 endif
522                                 call add(info['disp'],
523                                          \ printf('--- %s ---', in_part))
524                                 let part_start = len(info['disp']) + 1
525                         endif
526
527                 elseif in_header
528                         if in_header == 1
529                                 let msg['descr'] = line
530                                 call add(info['disp'], line)
531                                 let in_header = 2
532                                 let msg['hdr_start'] = len(info['disp']) + 1
533
534                         else
535                                 if match(line, g:notmuch_show_header_end_regexp) != -1
536                                         let hdr_start = msg['hdr_start']+1
537                                         let hdr_end = len(info['disp'])
538                                         let foldinfo = [ 'hdr', hdr_start, hdr_end,
539                                                \ printf('[ %d-line headers.  Press "h" to show. ]', hdr_end + 1 - hdr_start) ]
540                                         let msg['header'] = hdr
541                                         let in_header = 0
542                                         let hdr = {}
543                                 else
544                                         let m = matchlist(line, '^\(\w\+\):\s*\(.*\)$')
545                                         if len(m)
546                                                 let hdr[m[1]] = m[2]
547                                                 if match(g:notmuch_show_headers, m[1]) != -1
548                                                         call add(info['disp'], line)
549                                                 endif
550                                         endif
551                                 endif
552                         endif
553
554                 elseif in_message
555                         if match(line, g:notmuch_show_message_end_regexp) != -1
556                                 let msg['end'] = len(info['disp'])
557                                 call add(info['disp'], '')
558
559                                 let foldinfo = [ 'msg', msg['start'], msg['end'],
560                                                \ printf('[ MSG %d - %s ]', len(info['msgs']), msg['descr']) ]
561
562                                 call add(info['msgs'], msg)
563                                 let msg = {}
564                                 let in_message = 0
565                                 let in_header = 0
566                                 let in_body = 0
567                                 let in_part = ''
568
569                         elseif match(line, g:notmuch_show_header_begin_regexp) != -1
570                                 let in_header = 1
571                                 continue
572
573                         elseif match(line, g:notmuch_show_body_begin_regexp) != -1
574                                 let body_start = len(info['disp']) + 1
575                                 let in_body = 1
576                                 continue
577                         endif
578
579                 else
580                         if match(line, g:notmuch_show_message_begin_regexp) != -1
581                                 let msg['start'] = len(info['disp']) + 1
582
583                                 let m = matchlist(line, g:notmuch_show_message_parse_regexp)
584                                 if len(m)
585                                         let msg['id'] = m[1]
586                                         let msg['depth'] = m[2]
587                                         let msg['filename'] = m[3]
588                                 endif
589
590                                 let in_message = 1
591                         endif
592                 endif
593
594                 if len(foldinfo)
595                         call add(info['folds'], foldinfo[0:2])
596                         let info['foldtext'][foldinfo[1]] = foldinfo[3]
597                 endif
598         endfor
599         return info
600 endfunction
601
602 function! s:NM_cmd_show_mkfolds()
603         let info = b:nm_raw_info
604
605         for afold in info['folds']
606                 exec printf('%d,%dfold', afold[1], afold[2])
607                 if (afold[0] == 'sig' && g:notmuch_show_fold_signatures)
608                  \ || (afold[0] == 'cit' && g:notmuch_show_fold_citations)
609                  \ || (afold[0] == 'bdy' && g:notmuch_show_fold_bodies)
610                  \ || (afold[0] == 'hdr' && g:notmuch_show_fold_headers)
611                         exec printf('%dfoldclose', afold[1])
612                 else
613                         exec printf('%dfoldopen', afold[1])
614                 endif
615         endfor
616 endfunction
617
618 function! s:NM_cmd_show_mksyntax()
619         let info = b:nm_raw_info
620         let cnt = 0
621         for msg in info['msgs']
622                 let cnt = cnt + 1
623                 let start = msg['start']
624                 let hdr_start = msg['hdr_start']
625                 let body_start = msg['body_start']
626                 let end = msg['end']
627                 exec printf('syntax region nmShowMsg%dDesc start=''\%%%dl'' end=''\%%%dl'' contains=@nmShowMsgDesc', cnt, start, start+1)
628                 exec printf('syntax region nmShowMsg%dHead start=''\%%%dl'' end=''\%%%dl'' contains=@nmShowMsgHead', cnt, hdr_start, body_start)
629                 exec printf('syntax region nmShowMsg%dBody start=''\%%%dl'' end=''\%%%dl'' contains=@nmShowMsgBody', cnt, body_start, end)
630         endfor
631 endfunction
632
633 function! NM_cmd_show_foldtext()
634         let foldtext = b:nm_raw_info['foldtext']
635         return foldtext[v:foldstart]
636 endfunction
637
638
639 " --- notmuch helper functions {{{1
640
641 function! s:NM_newBuffer(ft, content)
642         enew
643         setlocal buftype=nofile readonly modifiable
644         silent put=a:content
645         keepjumps 0d
646         setlocal nomodifiable
647         execute printf('set filetype=notmuch-%s', a:ft)
648         execute printf('set syntax=notmuch-%s', a:ft)
649 endfunction
650
651 function! s:NM_run(args)
652         let cmd = g:notmuch_cmd . ' ' . join(a:args) . '< /dev/null'
653
654         let start = reltime()
655         let out = system(cmd)
656         let err = v:shell_error
657         let delta = reltime(start)
658
659         echo printf('[%s] {%s} %s', reltimestr(delta), string(err), string(cmd))
660
661         if err
662                 echohl Error
663                 echo substitute(out, '\n*$', '', '')
664                 echohl None
665                 return ''
666         else
667                 return out
668         endif
669 endfunction
670
671 " --- external mail handling helpers {{{1
672
673 function! s:NM_new_mail()
674         echo 'not implemented'
675 endfunction
676
677 " --- other helpers {{{1
678
679 function! s:NM_kill_this_buffer()
680         if exists('b:nm_prev_bufnr')
681                 setlocal bufhidden=delete
682                 exec printf(":buffer %d", b:nm_prev_bufnr)
683         else
684                 echo "Nothing to kill."
685         endif
686 endfunction
687
688 function! s:NM_add_remove_tags(prefix, tags)
689         let id = <SID>NM_search_find_thread_id()
690         if id == ''
691                 echoe 'Eeek! I couldn''t find the thead id!'
692         endif
693         call map(a:tags, 'a:prefix . v:val')
694         " TODO: handle errors
695         call <SID>NM_run(['tag'] + a:tags + ['--', id])
696 endfunction
697
698 function! s:NM_add_remove_tags_on_screen(prefix, tags)
699         let online = ''
700         setlocal modifiable
701         if a:prefix == '-'
702                 for tagname in a:tags
703                         exec printf('silent %ss/(\([^)]*\)\<%s\>\([^)]*\))$/(\1\2)/', online, tagname)
704                 endfor
705         else
706                 for tagname in a:tags
707                         exec printf('silent %ss/(\([^)]*\)\([^)]*\))$/(\1 %s)/', online, tagname)
708                 endfor
709         endif
710         setlocal nomodifiable
711 endfunction
712
713 " --- process and set the defaults {{{1
714
715 function! NM_set_defaults(force)
716         for [key, dflt] in items(s:notmuch_defaults)
717                 let cmd = ''
718                 if !a:force && exists(key) && type(dflt) == type(eval(key))
719                         continue
720                 elseif type(dflt) == type(0)
721                         let cmd = printf('let %s = %d', key, dflt)
722                 elseif type(dflt) == type('')
723                         let cmd = printf('let %s = ''%s''', key, dflt)
724                 "elseif type(dflt) == type([])
725                 "        let cmd = printf('let %s = %s', key, string(dflt))
726                 else
727                         echoe printf('E: Unknown type in NM_set_defaults(%d) using [%s,%s]',
728                                                 \ a:force, key, string(dflt))
729                         continue
730                 endif
731                 exec cmd
732         endfor
733 endfunction
734 call NM_set_defaults(0)
735
736 " for some reason NM_set_defaults() didn't work for arrays...
737 if !exists('g:notmuch_show_headers')
738         let g:notmuch_show_headers = s:notmuch_show_headers_defaults
739 endif
740 if !exists('g:notmuch_initial_search_words')
741         let g:notmuch_initial_search_words = s:notmuch_initial_search_words_defaults
742 endif
743
744
745 " --- assign keymaps {{{1
746
747 function! s:NM_set_map(maps)
748         nmapclear
749         for [key, code] in items(a:maps)
750                 exec printf('nnoremap <buffer> %s %s', key, code)
751         endfor
752         " --- this is a hack for development :)
753         nnoremap ,nmr :source ~/.vim/plugin/notmuch.vim<CR>:call NotMuch('')<CR>
754 endfunction
755
756 " --- command handler {{{1
757
758 function! NotMuch(args)
759         if !strlen(a:args)
760                 if exists('b:nm_search_words')
761                         let words = b:nm_search_words
762                 else
763                         let words = g:notmuch_initial_search_words
764                 endif
765                 call <SID>NM_cmd_search(words)
766                 return
767         endif
768
769         echo "blarg!"
770
771         let words = split(a:args)
772         " TODO: handle commands passed as arguments
773 endfunction
774 function! CompleteNotMuch(arg_lead, cmd_line, cursor_pos)
775         return []
776 endfunction
777
778
779 " --- glue {{{1
780
781 command! -nargs=* -complete=customlist,CompleteNotMuch NotMuch call NotMuch(<q-args>)
782 cabbrev  notmuch <c-r>=(getcmdtype()==':' && getcmdpos()==1 ? 'NotMuch' : 'notmuch')<CR>
783
784 " vim: set ft=vim ts=8 sw=8 et foldmethod=marker :