git.cworth.org Git - notmuch-old/blob - vim/notmuch.vim

   1 if exists("g:loaded_notmuch")
   2         finish
   3 endif
   4
   5 if !has("ruby") || version < 700
   6         finish
   7 endif
   8
   9 let g:loaded_notmuch = "yep"
  10
  11 let g:notmuch_folders_maps = {
  12         \ '<Enter>':    'folders_show_search()',
  13         \ 's':          'folders_search_prompt()',
  14         \ '=':          'folders_refresh()',
  15         \ 'c':          'compose()',
  16         \ }
  17
  18 let g:notmuch_search_maps = {
  19         \ 'q':          'kill_this_buffer()',
  20         \ '<Enter>':    'search_show_thread(1)',
  21         \ '<Space>':    'search_show_thread(2)',
  22         \ 'A':          'search_tag("-inbox -unread")',
  23         \ 'I':          'search_tag("-unread")',
  24         \ 't':          'search_tag("")',
  25         \ 's':          'search_search_prompt()',
  26         \ '=':          'search_refresh()',
  27         \ '?':          'search_info()',
  28         \ 'c':          'compose()',
  29         \ }
  30
  31 let g:notmuch_show_maps = {
  32         \ 'q':          'kill_this_buffer()',
  33         \ 'A':          'show_tag("-inbox -unread")',
  34         \ 'I':          'show_tag("-unread")',
  35         \ 't':          'show_tag("")',
  36         \ 'o':          'show_open_msg()',
  37         \ 'e':          'show_extract_msg()',
  38         \ 's':          'show_save_msg()',
  39         \ 'p':          'show_save_patches()',
  40         \ 'r':          'show_reply()',
  41         \ '?':          'show_info()',
  42         \ '<Tab>':      'show_next_msg()',
  43         \ 'c':          'compose()',
  44         \ }
  45
  46 let g:notmuch_compose_maps = {
  47         \ ',s':         'compose_send()',
  48         \ ',q':         'compose_quit()',
  49         \ }
  50
  51 let s:notmuch_folders_default = [
  52         \ [ 'new', 'tag:inbox and tag:unread' ],
  53         \ [ 'inbox', 'tag:inbox' ],
  54         \ [ 'unread', 'tag:unread' ],
  55         \ ]
  56
  57 let s:notmuch_date_format_default = '%d.%m.%y'
  58 let s:notmuch_datetime_format_default = '%d.%m.%y %H:%M:%S'
  59 let s:notmuch_reader_default = 'mutt -f %s'
  60 let s:notmuch_sendmail_default = 'sendmail'
  61 let s:notmuch_folders_count_threads_default = 0
  62 let s:notmuch_compose_start_insert_default = 1
  63
  64 function! s:new_file_buffer(type, fname)
  65         exec printf('edit %s', a:fname)
  66         execute printf('set filetype=notmuch-%s', a:type)
  67         execute printf('set syntax=notmuch-%s', a:type)
  68         ruby $curbuf.init(VIM::evaluate('a:type'))
  69 endfunction
  70
  71 function! s:on_compose_delete()
  72         if b:compose_done
  73                 return
  74         endif
  75         if input('[s]end/[q]uit? ') =~ '^s'
  76                 call s:compose_send()
  77         endif
  78 endfunction
  79
  80 "" actions
  81
  82 function! s:compose_quit()
  83         let b:compose_done = 1
  84         call s:kill_this_buffer()
  85 endfunction
  86
  87 function! s:compose_send()
  88         let b:compose_done = 1
  89         let fname = expand('%')
  90         let lines = getline(5, '$')
  91
  92 ruby << EOF
  93         # Generate proper mail to send
  94         text = VIM::evaluate('lines').join("\n")
  95         fname = VIM::evaluate('fname')
  96         transport = Mail.new(text)
  97         transport.message_id = generate_message_id
  98         transport.charset = 'utf-8'
  99         File.write(fname, transport.to_s)
 100 EOF
 101
 102         let cmdtxt = g:notmuch_sendmail . ' -t -f ' . s:reply_from . ' < ' . fname
 103         let out = system(cmdtxt)
 104         let err = v:shell_error
 105         if err
 106                 echohl Error
 107                 echo 'Eeek! unable to send mail'
 108                 echo out
 109                 echohl None
 110                 return
 111         endif
 112         call delete(fname)
 113         echo 'Mail sent successfully.'
 114         call s:kill_this_buffer()
 115 endfunction
 116
 117 function! s:show_next_msg()
 118 ruby << EOF
 119         r, c = $curwin.cursor
 120         n = $curbuf.line_number
 121         i = $messages.index { |m| n >= m.start && n <= m.end }
 122         m = $messages[i + 1]
 123         if m
 124                 r = m.body_start + 1
 125                 VIM::command("normal #{m.start}zt")
 126                 $curwin.cursor = r, c
 127         end
 128 EOF
 129 endfunction
 130
 131 function! s:show_reply()
 132         ruby open_reply get_message.mail
 133         let b:compose_done = 0
 134         call s:set_map(g:notmuch_compose_maps)
 135         autocmd BufDelete <buffer> call s:on_compose_delete()
 136         if g:notmuch_compose_start_insert
 137                 startinsert!
 138         end
 139 endfunction
 140
 141 function! s:compose()
 142         ruby open_compose
 143         let b:compose_done = 0
 144         call s:set_map(g:notmuch_compose_maps)
 145         autocmd BufDelete <buffer> call s:on_compose_delete()
 146         if g:notmuch_compose_start_insert
 147                 startinsert!
 148         end
 149 endfunction
 150
 151 function! s:show_info()
 152         ruby vim_puts get_message.inspect
 153 endfunction
 154
 155 function! s:show_extract_msg()
 156 ruby << EOF
 157         m = get_message
 158         m.mail.attachments.each do |a|
 159                 File.open(a.filename, 'w') do |f|
 160                         f.write a.body.decoded
 161                         print "Extracted '#{a.filename}'"
 162                 end
 163         end
 164 EOF
 165 endfunction
 166
 167 function! s:show_open_msg()
 168 ruby << EOF
 169         m = get_message
 170         mbox = File.expand_path('~/.notmuch/vim_mbox')
 171         cmd = VIM::evaluate('g:notmuch_reader') % mbox
 172         system "notmuch show --format=mbox id:#{m.message_id} > #{mbox} && #{cmd}"
 173 EOF
 174 endfunction
 175
 176 function! s:show_save_msg()
 177         let file = input('File name: ')
 178 ruby << EOF
 179         file = VIM::evaluate('file')
 180         m = get_message
 181         system "notmuch show --format=mbox id:#{m.message_id} > #{file}"
 182 EOF
 183 endfunction
 184
 185 function! s:show_save_patches()
 186 ruby << EOF
 187         q = $curbuf.query($cur_thread)
 188         t = q.search_threads.first
 189         n = 0
 190         t.toplevel_messages.first.replies.each do |m|
 191                 next if not m['subject'] =~ /^\[PATCH.*\]/
 192                 file = "%04d.patch" % [n += 1]
 193                 system "notmuch show --format=mbox id:#{m.message_id} > #{file}"
 194         end
 195         vim_puts "Saved #{n} patches"
 196 EOF
 197 endfunction
 198
 199 function! s:show_tag(intags)
 200         if empty(a:intags)
 201                 let tags = input('tags: ')
 202         else
 203                 let tags = a:intags
 204         endif
 205         ruby do_tag(get_cur_view, VIM::evaluate('l:tags'))
 206         call s:show_next_thread()
 207 endfunction
 208
 209 function! s:search_search_prompt()
 210         let text = input('Search: ')
 211         if text == ""
 212           return
 213         endif
 214         setlocal modifiable
 215 ruby << EOF
 216         $cur_search = VIM::evaluate('text')
 217         $curbuf.reopen
 218         search_render($cur_search)
 219 EOF
 220         setlocal nomodifiable
 221 endfunction
 222
 223 function! s:search_info()
 224         ruby vim_puts get_thread_id
 225 endfunction
 226
 227 function! s:search_refresh()
 228         setlocal modifiable
 229         ruby $curbuf.reopen
 230         ruby search_render($cur_search)
 231         setlocal nomodifiable
 232 endfunction
 233
 234 function! s:search_tag(intags)
 235         if empty(a:intags)
 236                 let tags = input('tags: ')
 237         else
 238                 let tags = a:intags
 239         endif
 240         ruby do_tag(get_thread_id, VIM::evaluate('l:tags'))
 241         norm j
 242 endfunction
 243
 244 function! s:folders_search_prompt()
 245         let text = input('Search: ')
 246         call s:search(text)
 247 endfunction
 248
 249 function! s:folders_refresh()
 250         setlocal modifiable
 251         ruby $curbuf.reopen
 252         ruby folders_render()
 253         setlocal nomodifiable
 254 endfunction
 255
 256 "" basic
 257
 258 function! s:show_cursor_moved()
 259 ruby << EOF
 260         if $render.is_ready?
 261                 VIM::command('setlocal modifiable')
 262                 $render.do_next
 263                 VIM::command('setlocal nomodifiable')
 264         end
 265 EOF
 266 endfunction
 267
 268 function! s:show_next_thread()
 269         call s:kill_this_buffer()
 270         if line('.') != line('$')
 271                 norm j
 272                 call s:search_show_thread(0)
 273         else
 274                 echo 'No more messages.'
 275         endif
 276 endfunction
 277
 278 function! s:kill_this_buffer()
 279 ruby << EOF
 280         $curbuf.close
 281         VIM::command("bdelete!")
 282 EOF
 283 endfunction
 284
 285 function! s:set_map(maps)
 286         nmapclear <buffer>
 287         for [key, code] in items(a:maps)
 288                 let cmd = printf(":call <SID>%s<CR>", code)
 289                 exec printf('nnoremap <buffer> %s %s', key, cmd)
 290         endfor
 291 endfunction
 292
 293 function! s:new_buffer(type)
 294         enew
 295         setlocal buftype=nofile bufhidden=hide
 296         keepjumps 0d
 297         execute printf('set filetype=notmuch-%s', a:type)
 298         execute printf('set syntax=notmuch-%s', a:type)
 299         ruby $curbuf.init(VIM::evaluate('a:type'))
 300 endfunction
 301
 302 function! s:set_menu_buffer()
 303         setlocal nomodifiable
 304         setlocal cursorline
 305         setlocal nowrap
 306 endfunction
 307
 308 "" main
 309
 310 function! s:show(thread_id)
 311         call s:new_buffer('show')
 312         setlocal modifiable
 313 ruby << EOF
 314         thread_id = VIM::evaluate('a:thread_id')
 315         $cur_thread = thread_id
 316         $messages.clear
 317         $curbuf.render do |b|
 318                 q = $curbuf.query(get_cur_view)
 319                 q.sort = Notmuch::SORT_OLDEST_FIRST
 320                 $exclude_tags.each { |t|
 321                         q.add_tag_exclude(t)
 322                 }
 323                 msgs = q.search_messages
 324                 msgs.each do |msg|
 325                         m = Mail.read(msg.filename)
 326                         part = m.find_first_text
 327                         nm_m = Message.new(msg, m)
 328                         $messages << nm_m
 329                         date_fmt = VIM::evaluate('g:notmuch_datetime_format')
 330                         date = Time.at(msg.date).strftime(date_fmt)
 331                         nm_m.start = b.count
 332                         b << "%s %s (%s)" % [msg['from'], date, msg.tags]
 333                         b << "Subject: %s" % [msg['subject']]
 334                         b << "To: %s" % msg['to']
 335                         b << "Cc: %s" % msg['cc']
 336                         b << "Date: %s" % msg['date']
 337                         nm_m.body_start = b.count
 338                         b << "--- %s ---" % part.mime_type
 339                         part.convert.each_line do |l|
 340                                 b << l.chomp
 341                         end
 342                         b << ""
 343                         nm_m.end = b.count
 344                 end
 345                 b.delete(b.count)
 346         end
 347         $messages.each_with_index do |msg, i|
 348                 VIM::command("syntax region nmShowMsg#{i}Desc start='\\%%%il' end='\\%%%il' contains=@nmShowMsgDesc" % [msg.start, msg.start + 1])
 349                 VIM::command("syntax region nmShowMsg#{i}Head start='\\%%%il' end='\\%%%il' contains=@nmShowMsgHead" % [msg.start + 1, msg.body_start])
 350                 VIM::command("syntax region nmShowMsg#{i}Body start='\\%%%il' end='\\%%%dl' contains=@nmShowMsgBody" % [msg.body_start, msg.end])
 351         end
 352 EOF
 353         setlocal nomodifiable
 354         call s:set_map(g:notmuch_show_maps)
 355 endfunction
 356
 357 function! s:search_show_thread(mode)
 358 ruby << EOF
 359         mode = VIM::evaluate('a:mode')
 360         id = get_thread_id
 361         case mode
 362         when 0;
 363         when 1; $cur_filter = nil
 364         when 2; $cur_filter = $cur_search
 365         end
 366         VIM::command("call s:show('#{id}')")
 367 EOF
 368 endfunction
 369
 370 function! s:search(search)
 371         call s:new_buffer('search')
 372 ruby << EOF
 373         $cur_search = VIM::evaluate('a:search')
 374         search_render($cur_search)
 375 EOF
 376         call s:set_menu_buffer()
 377         call s:set_map(g:notmuch_search_maps)
 378         autocmd CursorMoved <buffer> call s:show_cursor_moved()
 379 endfunction
 380
 381 function! s:folders_show_search()
 382 ruby << EOF
 383         n = $curbuf.line_number
 384         s = $searches[n - 1]
 385         VIM::command("call s:search('#{s}')")
 386 EOF
 387 endfunction
 388
 389 function! s:folders()
 390         call s:new_buffer('folders')
 391         ruby folders_render()
 392         call s:set_menu_buffer()
 393         call s:set_map(g:notmuch_folders_maps)
 394 endfunction
 395
 396 "" root
 397
 398 function! s:set_defaults()
 399         if !exists('g:notmuch_date_format')
 400                 if exists('g:notmuch_rb_date_format')
 401                         let g:notmuch_date_format = g:notmuch_rb_date_format
 402                 else
 403                         let g:notmuch_date_format = s:notmuch_date_format_default
 404                 endif
 405         endif
 406
 407         if !exists('g:notmuch_datetime_format')
 408                 if exists('g:notmuch_rb_datetime_format')
 409                         let g:notmuch_datetime_format = g:notmuch_rb_datetime_format
 410                 else
 411                         let g:notmuch_datetime_format = s:notmuch_datetime_format_default
 412                 endif
 413         endif
 414
 415         if !exists('g:notmuch_reader')
 416                 if exists('g:notmuch_rb_reader')
 417                         let g:notmuch_reader = g:notmuch_rb_reader
 418                 else
 419                         let g:notmuch_reader = s:notmuch_reader_default
 420                 endif
 421         endif
 422
 423         if !exists('g:notmuch_sendmail')
 424                 if exists('g:notmuch_rb_sendmail')
 425                         let g:notmuch_sendmail = g:notmuch_rb_sendmail
 426                 else
 427                         let g:notmuch_sendmail = s:notmuch_sendmail_default
 428                 endif
 429         endif
 430
 431         if !exists('g:notmuch_folders_count_threads')
 432                 if exists('g:notmuch_rb_count_threads')
 433                         let g:notmuch_count_threads = g:notmuch_rb_count_threads
 434                 else
 435                         let g:notmuch_folders_count_threads = s:notmuch_folders_count_threads_default
 436                 endif
 437         endif
 438
 439         if !exists('g:notmuch_compose_start_insert')
 440                 let g:notmuch_compose_start_insert = s:notmuch_compose_start_insert_default
 441         endif
 442
 443         if !exists('g:notmuch_custom_search_maps') && exists('g:notmuch_rb_custom_search_maps')
 444                 let g:notmuch_custom_search_maps = g:notmuch_rb_custom_search_maps
 445         endif
 446
 447         if !exists('g:notmuch_custom_show_maps') && exists('g:notmuch_rb_custom_show_maps')
 448                 let g:notmuch_custom_show_maps = g:notmuch_rb_custom_show_maps
 449         endif
 450
 451         if exists('g:notmuch_custom_search_maps')
 452                 call extend(g:notmuch_search_maps, g:notmuch_custom_search_maps)
 453         endif
 454
 455         if exists('g:notmuch_custom_show_maps')
 456                 call extend(g:notmuch_show_maps, g:notmuch_custom_show_maps)
 457         endif
 458
 459         if !exists('g:notmuch_folders')
 460                 if exists('g:notmuch_rb_folders')
 461                         let g:notmuch_folders = g:notmuch_rb_folders
 462                 else
 463                         let g:notmuch_folders = s:notmuch_folders_default
 464                 endif
 465         endif
 466 endfunction
 467
 468 function! s:NotMuch(...)
 469         call s:set_defaults()
 470
 471 ruby << EOF
 472         require 'notmuch'
 473         require 'rubygems'
 474         require 'tempfile'
 475         require 'socket'
 476         begin
 477                 require 'mail'
 478         rescue LoadError
 479         end
 480
 481         $db_name = nil
 482         $email = $email_name = $email_address = nil
 483         $exclude_tags = []
 484         $searches = []
 485         $threads = []
 486         $messages = []
 487         $mail_installed = defined?(Mail)
 488
 489         def get_config_item(item)
 490                 result = ''
 491                 IO.popen(['notmuch', 'config', 'get', item]) { |out|
 492                         result = out.read
 493                 }
 494                 return result.rstrip
 495         end
 496
 497         def get_config
 498                 $db_name = get_config_item('database.path')
 499                 $email_name = get_config_item('user.name')
 500                 $email_address = get_config_item('user.primary_email')
 501                 $email_name = get_config_item('user.name')
 502                 $email = "%s <%s>" % [$email_name, $email_address]
 503                 ignore_tags = get_config_item('search.exclude_tags')
 504                 $exclude_tags = ignore_tags.split("\n")
 505         end
 506
 507         def vim_puts(s)
 508                 VIM::command("echo '#{s.to_s}'")
 509         end
 510
 511         def vim_p(s)
 512                 VIM::command("echo '#{s.inspect}'")
 513         end
 514
 515         def author_filter(a)
 516                 # TODO email format, aliases
 517                 a.strip!
 518                 a.gsub!(/[\.@].*/, '')
 519                 a.gsub!(/^ext /, '')
 520                 a.gsub!(/ \(.*\)/, '')
 521                 a
 522         end
 523
 524         def get_thread_id
 525                 n = $curbuf.line_number - 1
 526                 return "thread:%s" % $threads[n]
 527         end
 528
 529         def get_message
 530                 n = $curbuf.line_number
 531                 return $messages.find { |m| n >= m.start && n <= m.end }
 532         end
 533
 534         def get_cur_view
 535                 if $cur_filter
 536                         return "#{$cur_thread} and (#{$cur_filter})"
 537                 else
 538                         return $cur_thread
 539                 end
 540         end
 541
 542         def generate_message_id
 543                 t = Time.now
 544                 random_tag = sprintf('%x%x_%x%x%x',
 545                         t.to_i, t.tv_usec,
 546                         $$, Thread.current.object_id.abs, rand(255))
 547                 return "<#{random_tag}@#{Socket.gethostname}.notmuch>"
 548         end
 549
 550         def open_compose_helper(lines, cur)
 551                 help_lines = [
 552                         'Notmuch-Help: Type in your message here; to help you use these bindings:',
 553                         'Notmuch-Help:   ,s    - send the message (Notmuch-Help lines will be removed)',
 554                         'Notmuch-Help:   ,q    - abort the message',
 555                         ]
 556
 557                 dir = File.expand_path('~/.notmuch/compose')
 558                 FileUtils.mkdir_p(dir)
 559                 Tempfile.open(['nm-', '.mail'], dir) do |f|
 560                         f.puts(help_lines)
 561                         f.puts
 562                         f.puts(lines)
 563
 564                         sig_file = File.expand_path('~/.signature')
 565                         if File.exists?(sig_file)
 566                                 f.puts("-- ")
 567                                 f.write(File.read(sig_file))
 568                         end
 569
 570                         f.flush
 571
 572                         cur += help_lines.size + 1
 573
 574                         VIM::command("let s:reply_from='%s'" % $email_address)
 575                         VIM::command("call s:new_file_buffer('compose', '#{f.path}')")
 576                         VIM::command("call cursor(#{cur}, 0)")
 577                 end
 578         end
 579
 580         def open_reply(orig)
 581                 reply = orig.reply do |m|
 582                         # fix headers
 583                         if not m[:reply_to]
 584                                 m.to = [orig[:from].to_s, orig[:to].to_s]
 585                         end
 586                         m.cc = orig[:cc]
 587                         m.from = $email
 588                         m.charset = 'utf-8'
 589                 end
 590
 591                 lines = []
 592
 593                 body_lines = []
 594                 if $mail_installed
 595                         addr = Mail::Address.new(orig[:from].value)
 596                         name = addr.name
 597                         name = addr.local + "@" if name.nil? && !addr.local.nil?
 598                 else
 599                         name = orig[:from]
 600                 end
 601                 name = "somebody" if name.nil?
 602
 603                 body_lines << "%s wrote:" % name
 604                 part = orig.find_first_text
 605                 part.convert.each_line do |l|
 606                         body_lines << "> %s" % l.chomp
 607                 end
 608                 body_lines << ""
 609                 body_lines << ""
 610                 body_lines << ""
 611
 612                 reply.body = body_lines.join("\n")
 613
 614                 lines += reply.present.lines.map { |e| e.chomp }
 615                 lines << ""
 616
 617                 cur = lines.count - 1
 618
 619                 open_compose_helper(lines, cur)
 620         end
 621
 622         def open_compose()
 623                 lines = []
 624
 625                 lines << "From: #{$email}"
 626                 lines << "To: "
 627                 cur = lines.count
 628
 629                 lines << "Cc: "
 630                 lines << "Bcc: "
 631                 lines << "Subject: "
 632                 lines << ""
 633                 lines << ""
 634                 lines << ""
 635
 636                 open_compose_helper(lines, cur)
 637         end
 638
 639         def folders_render()
 640                 $curbuf.render do |b|
 641                         folders = VIM::evaluate('g:notmuch_folders')
 642                         count_threads = VIM::evaluate('g:notmuch_folders_count_threads') == 1
 643                         $searches.clear
 644                         folders.each do |name, search|
 645                                 q = $curbuf.query(search)
 646                                 $exclude_tags.each { |t|
 647                                         q.add_tag_exclude(t)
 648                                 }
 649                                 $searches << search
 650                                 count = count_threads ? q.count_threads : q.count_messages
 651                                 b << "%9d %-20s (%s)" % [count, name, search]
 652                         end
 653                 end
 654         end
 655
 656         def search_render(search)
 657                 date_fmt = VIM::evaluate('g:notmuch_date_format')
 658                 q = $curbuf.query(search)
 659                 q.sort = Notmuch::SORT_NEWEST_FIRST
 660                 $exclude_tags.each { |t|
 661                         q.add_tag_exclude(t)
 662                 }
 663                 $threads.clear
 664                 t = q.search_threads
 665
 666                 $render = $curbuf.render_staged(t) do |b, items|
 667                         items.each do |e|
 668                                 authors = e.authors.to_utf8.split(/[,|]/).map { |a| author_filter(a) }.join(",")
 669                                 date = Time.at(e.newest_date).strftime(date_fmt)
 670                                 subject = e.messages.first['subject']
 671                                 if $mail_installed
 672                                         subject = Mail::Field.parse("Subject: " + subject).to_s
 673                                 else
 674                                         subject = subject.force_encoding('utf-8')
 675                                 end
 676                                 b << "%-12s %3s %-20.20s | %s (%s)" % [date, e.matched_messages, authors, subject, e.tags]
 677                                 $threads << e.thread_id
 678                         end
 679                 end
 680         end
 681
 682         def do_tag(filter, tags)
 683                 $curbuf.do_write do |db|
 684                         q = db.query(filter)
 685                         q.search_messages.each do |e|
 686                                 e.freeze
 687                                 tags.split.each do |t|
 688                                         case t
 689                                         when /^-(.*)/
 690                                                 e.remove_tag($1)
 691                                         when /^\+(.*)/
 692                                                 e.add_tag($1)
 693                                         when /^([^\+^-].*)/
 694                                                 e.add_tag($1)
 695                                         end
 696                                 end
 697                                 e.thaw
 698                                 e.tags_to_maildir_flags
 699                         end
 700                         q.destroy!
 701                 end
 702         end
 703
 704         module DbHelper
 705                 def init(name)
 706                         @name = name
 707                         @db = Notmuch::Database.new($db_name)
 708                         @queries = []
 709                 end
 710
 711                 def query(*args)
 712                         q = @db.query(*args)
 713                         @queries << q
 714                         q
 715                 end
 716
 717                 def close
 718                         @queries.delete_if { |q| ! q.destroy! }
 719                         @db.close
 720                 end
 721
 722                 def reopen
 723                         close if @db
 724                         @db = Notmuch::Database.new($db_name)
 725                 end
 726
 727                 def do_write
 728                         db = Notmuch::Database.new($db_name, :mode => Notmuch::MODE_READ_WRITE)
 729                         begin
 730                                 yield db
 731                         ensure
 732                                 db.close
 733                         end
 734                 end
 735         end
 736
 737         class Message
 738                 attr_accessor :start, :body_start, :end
 739                 attr_reader :message_id, :filename, :mail
 740
 741                 def initialize(msg, mail)
 742                         @message_id = msg.message_id
 743                         @filename = msg.filename
 744                         @mail = mail
 745                         @start = 0
 746                         @end = 0
 747                         mail.import_headers(msg) if not $mail_installed
 748                 end
 749
 750                 def to_s
 751                         "id:%s" % @message_id
 752                 end
 753
 754                 def inspect
 755                         "id:%s, file:%s" % [@message_id, @filename]
 756                 end
 757         end
 758
 759         class StagedRender
 760                 def initialize(buffer, enumerable, block)
 761                         @b = buffer
 762                         @enumerable = enumerable
 763                         @block = block
 764                         @last_render = 0
 765
 766                         @b.render { do_next }
 767                 end
 768
 769                 def is_ready?
 770                         @last_render - @b.line_number <= $curwin.height
 771                 end
 772
 773                 def do_next
 774                         items = @enumerable.take($curwin.height * 2)
 775                         return if items.empty?
 776                         @block.call @b, items
 777                         @last_render = @b.count
 778                 end
 779         end
 780
 781         class VIM::Buffer
 782                 include DbHelper
 783
 784                 def <<(a)
 785                         append(count(), a)
 786                 end
 787
 788                 def render_staged(enumerable, &block)
 789                         StagedRender.new(self, enumerable, block)
 790                 end
 791
 792                 def render
 793                         old_count = count
 794                         yield self
 795                         (1..old_count).each do
 796                                 delete(1)
 797                         end
 798                 end
 799         end
 800
 801         class Notmuch::Tags
 802                 def to_s
 803                         to_a.join(" ")
 804                 end
 805         end
 806
 807         class Notmuch::Message
 808                 def to_s
 809                         "id:%s" % message_id
 810                 end
 811         end
 812
 813         # workaround for bug in vim's ruby
 814         class Object
 815                 def flush
 816                 end
 817         end
 818
 819         module SimpleMessage
 820                 class Header < Array
 821                         def self.parse(string)
 822                                 return nil if string.empty?
 823                                 return Header.new(string.split(/,\s+/))
 824                         end
 825
 826                         def to_s
 827                                 self.join(', ')
 828                         end
 829                 end
 830
 831                 def initialize(string = nil)
 832                         @raw_source = string
 833                         @body = nil
 834                         @headers = {}
 835
 836                         return if not string
 837
 838                         if string =~ /(.*?(\r\n|\n))\2/m
 839                                 head, body = $1, $' || '', $2
 840                         else
 841                                 head, body = string, ''
 842                         end
 843                         @body = body
 844                 end
 845
 846                 def [](name)
 847                         @headers[name.to_sym]
 848                 end
 849
 850                 def []=(name, value)
 851                         @headers[name.to_sym] = value
 852                 end
 853
 854                 def format_header(value)
 855                         value.to_s.tr('_', '-').gsub(/(\w+)/) { $1.capitalize }
 856                 end
 857
 858                 def to_s
 859                         buffer = ''
 860                         @headers.each do |key, value|
 861                                 buffer << "%s: %s\r\n" %
 862                                         [format_header(key), value]
 863                         end
 864                         buffer << "\r\n"
 865                         buffer << @body
 866                         buffer
 867                 end
 868
 869                 def body=(value)
 870                         @body = value
 871                 end
 872
 873                 def from
 874                         @headers[:from]
 875                 end
 876
 877                 def decoded
 878                         @body
 879                 end
 880
 881                 def mime_type
 882                         'text/plain'
 883                 end
 884
 885                 def multipart?
 886                         false
 887                 end
 888
 889                 def reply
 890                         r = Mail::Message.new
 891                         r[:from] = self[:to]
 892                         r[:to] = self[:from]
 893                         r[:cc] = self[:cc]
 894                         r[:in_reply_to] = self[:message_id]
 895                         r[:references] = self[:references]
 896                         r
 897                 end
 898
 899                 HEADERS = [ :from, :to, :cc, :references, :in_reply_to, :reply_to, :message_id ]
 900
 901                 def import_headers(m)
 902                         HEADERS.each do |e|
 903                                 dashed = format_header(e)
 904                                 @headers[e] = Header.parse(m[dashed])
 905                         end
 906                 end
 907         end
 908
 909         module Mail
 910
 911                 if not $mail_installed
 912                         puts "WARNING: Install the 'mail' gem, without it support is limited"
 913
 914                         def self.read(filename)
 915                                 Message.new(File.open(filename, 'rb') { |f| f.read })
 916                         end
 917
 918                         class Message
 919                                 include SimpleMessage
 920                         end
 921                 end
 922
 923                 class Message
 924
 925                         def find_first_text
 926                                 return self if not multipart?
 927                                 return text_part || html_part
 928                         end
 929
 930                         def convert
 931                                 if mime_type != "text/html"
 932                                         text = decoded
 933                                 else
 934                                         IO.popen(VIM::evaluate('exists("g:notmuch_html_converter") ? ' +
 935                                                         'g:notmuch_html_converter : "elinks --dump"'), "w+") do |pipe|
 936                                                 pipe.write(decode_body)
 937                                                 pipe.close_write
 938                                                 text = pipe.read
 939                                         end
 940                                 end
 941                                 text
 942                         end
 943
 944                         def present
 945                                 buffer = ''
 946                                 header.fields.each do |f|
 947                                         buffer << "%s: %s\r\n" % [f.name, f.to_s]
 948                                 end
 949                                 buffer << "\r\n"
 950                                 buffer << body.to_s
 951                                 buffer
 952                         end
 953                 end
 954         end
 955
 956         class String
 957                 def to_utf8
 958                         RUBY_VERSION >= "1.9" ? force_encoding('utf-8') : self
 959                 end
 960         end
 961
 962         get_config
 963 EOF
 964         if a:0
 965           call s:search(join(a:000))
 966         else
 967           call s:folders()
 968         endif
 969 endfunction
 970
 971 command -nargs=* NotMuch call s:NotMuch(<f-args>)
 972
 973 " vim: set noexpandtab: