git.notmuchmail.org Git - sup/blob - bin/sup

   1 #!/usr/bin/env ruby
   2
   3 require 'rubygems'
   4 require 'ncurses'
   5 require 'curses'
   6 require 'fileutils'
   7 require 'trollop'
   8 require "sup"
   9
  10 $opts = Trollop::options do
  11   version "sup v#{Redwood::VERSION}"
  12   banner <<EOS
  13 Sup is a curses-based email client.
  14
  15 Usage:
  16   sup [options]
  17
  18 Options are:
  19 EOS
  20   opt :list_hooks, "List all hooks and descriptions thereof, and quit."
  21   opt :no_threads, "Turn of threading. Helps with debugging. (Necessarily disables background polling for new messages.)"
  22   opt :search, "Search for threads ", :type => String
  23 end
  24
  25 if $opts[:list_hooks]
  26   Redwood::HookManager.print_hooks
  27   exit
  28 end
  29
  30 Thread.abort_on_exception = true # make debugging possible
  31
  32 module Redwood
  33
  34 global_keymap = Keymap.new do |k|
  35   k.add :quit, "Quit Redwood", 'q'
  36   k.add :help, "Show help", 'H', '?'
  37   k.add :roll_buffers, "Switch to next buffer", 'b'
  38 #  k.add :roll_buffers_backwards, "Switch to previous buffer", 'B'
  39   k.add :kill_buffer, "Kill the current buffer", 'x'
  40   k.add :list_buffers, "List all buffers", 'B'
  41   k.add :list_contacts, "List contacts", 'C'
  42   k.add :redraw, "Redraw screen", :ctrl_l
  43   k.add :search, "Search all messages", '\\'
  44   k.add :list_labels, "List labels", 'L'
  45   k.add :poll, "Poll for new messages", 'P'
  46   k.add :compose, "Compose new message", 'm', 'c'
  47   k.add :nothing, "Do nothing", :ctrl_g
  48   k.add :recall_draft, "Edit most recent draft message", 'R'
  49 end
  50
  51 def start_cursing
  52   Ncurses.initscr
  53   Ncurses.noecho
  54   Ncurses.cbreak
  55   Ncurses.stdscr.keypad 1
  56   Ncurses.curs_set 0
  57   Ncurses.start_color
  58   $cursing = true
  59 end
  60
  61 def stop_cursing
  62   return unless $cursing
  63   Ncurses.curs_set 1
  64   Ncurses.echo
  65   Ncurses.endwin
  66 end
  67 module_function :start_cursing, :stop_cursing
  68
  69 Index.new
  70 begin
  71   Index.lock
  72 rescue Index::LockError => e
  73   require 'highline'
  74
  75   h = HighLine.new
  76   h.wrap_at = :auto
  77   h.say Index.fancy_lock_error_message_for(e)
  78
  79   case h.ask("Should I ask that process to kill itself? ")
  80   when /^\s*y\s*$/i
  81     h.say "Ok, suggesting sepuku..."
  82     FileUtils.touch Redwood::SUICIDE_FN
  83     sleep SuicideManager::DELAY * 2
  84     FileUtils.rm_f Redwood::SUICIDE_FN
  85     h.say "Let's try that again."
  86     retry
  87   else
  88     h.say <<EOS
  89 Ok, giving up. If the process crashed and left a stale lockfile, you
  90 can fix this by manually deleting #{Index.lockfile}.
  91 EOS
  92     exit
  93   end
  94 end
  95
  96 begin
  97   extend CanSpawnComposeMode
  98   Redwood::start
  99   Index.load
 100
 101   if(s = Index.source_for DraftManager.source_name)
 102     DraftManager.source = s
 103   else
 104     Redwood::log "no draft source, auto-adding..."
 105     Index.add_source DraftManager.new_source
 106   end
 107
 108   if(s = Index.source_for SentManager.source_name)
 109     SentManager.source = s
 110   else
 111     Redwood::log "no sent mail source, auto-adding..."
 112     Index.add_source SentManager.new_source
 113   end
 114
 115   log "starting curses"
 116   start_cursing
 117
 118   log "initializing colormap"
 119   Colormap.new do |c|
 120     c.add :status_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLUE, Ncurses::A_BOLD
 121     c.add :index_old_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK
 122     c.add :index_new_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK,
 123            Ncurses::A_BOLD
 124     c.add :index_starred_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK,
 125            Ncurses::A_BOLD
 126     c.add :labellist_old_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK
 127     c.add :labellist_new_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK,
 128            Ncurses::A_BOLD
 129     c.add :twiddle_color, Ncurses::COLOR_BLUE, Ncurses::COLOR_BLACK
 130     c.add :label_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
 131     c.add :message_patina_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_GREEN
 132     c.add :alternate_patina_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_BLUE
 133     c.add :missing_message_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_RED
 134     c.add :attachment_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
 135     c.add :cryptosig_valid_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK, Ncurses::A_BOLD
 136     c.add :cryptosig_unknown_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
 137     c.add :cryptosig_invalid_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_RED, Ncurses::A_BOLD
 138     c.add :generic_notice_patina_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
 139     c.add :quote_patina_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
 140     c.add :sig_patina_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
 141     c.add :quote_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
 142     c.add :sig_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
 143     c.add :to_me_color, Ncurses::COLOR_GREEN, Ncurses::COLOR_BLACK
 144     c.add :starred_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK,
 145           Ncurses::A_BOLD
 146     c.add :starred_patina_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_GREEN,
 147           Ncurses::A_BOLD
 148     c.add :alternate_starred_patina_color, Ncurses::COLOR_YELLOW,
 149           Ncurses::COLOR_BLUE, Ncurses::A_BOLD
 150     c.add :snippet_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
 151     c.add :option_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK
 152     c.add :tagged_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK,
 153           Ncurses::A_BOLD
 154     c.add :draft_notification_color, Ncurses::COLOR_RED, Ncurses::COLOR_BLACK,
 155           Ncurses::A_BOLD
 156     c.add :completion_character_color, Ncurses::COLOR_WHITE,
 157           Ncurses::COLOR_BLACK, Ncurses::A_BOLD
 158     c.add :reply_mode_selected_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK, Ncurses::A_BOLD
 159     c.add :reply_mode_unselected_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
 160     c.add :reply_mode_label_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
 161     c.add :search_highlight_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_YELLOW, Ncurses::A_BOLD, :highlight => :search_highlight_color
 162   end
 163
 164   log "initializing buffer manager"
 165   bm = BufferManager.new
 166
 167   log "initializing mail index buffer"
 168   imode = InboxMode.new
 169   ibuf = bm.spawn "Inbox", imode
 170
 171   log "ready for interaction!"
 172   Logger.make_buf
 173
 174   bm.draw_screen
 175
 176   begin
 177     Index.usual_sources.each { |s| s.check }
 178   rescue SourceError
 179     # do nothing! we'll report it at the next step
 180   end
 181   Redwood::report_broken_sources
 182
 183   Index.usual_sources.each do |s|
 184     reporting_thread do
 185       begin
 186         s.connect
 187       rescue SourceError => e
 188         Redwood::log "fatal error loading from #{s}: #{e.message}"
 189       end
 190     end if s.respond_to? :connect
 191   end
 192
 193   imode.load_threads :num => ibuf.content_height, :when_done => lambda { reporting_thread { sleep 1; PollManager.poll } unless $opts[:no_threads] }
 194
 195   unless $opts[:no_threads]
 196     PollManager.start
 197     SuicideManager.start
 198     Index.start_lock_update_thread
 199   end
 200
 201   if $opts[:search]
 202     SearchResultsMode.spawn_from_query $opts[:search]
 203   end
 204
 205   until $exception || SuicideManager.die?
 206     c = Ncurses.nonblocking_getch
 207     next unless c
 208     bm.erase_flash
 209
 210     unless bm.handle_input(c)
 211       x = global_keymap.action_for c
 212       case x
 213       when :quit
 214         break if bm.kill_all_buffers_safely
 215       when :help
 216         curmode = bm.focus_buf.mode
 217         bm.spawn_unless_exists("<help for #{curmode.name}>") { HelpMode.new curmode, global_keymap }
 218       when :roll_buffers
 219         bm.roll_buffers
 220       when :roll_buffers_backwards
 221         bm.roll_buffers_backwards
 222       when :kill_buffer
 223         bm.kill_buffer_safely bm.focus_buf
 224       when :list_buffers
 225         bm.spawn_unless_exists("Buffer List") { BufferListMode.new }
 226       when :list_contacts
 227         b = bm.spawn_unless_exists("Contact List") { ContactListMode.new }
 228         b.mode.load_in_background
 229       when :search
 230         query = BufferManager.ask :search, "query: "
 231         next unless query && query !~ /^\s*$/
 232         SearchResultsMode.spawn_from_query query
 233       when :list_labels
 234         labels = LabelManager.listable_labels.map { |l| LabelManager.string_for l }
 235         user_label = bm.ask_with_completions :label, "Show threads with label (enter for listing): ", labels
 236         user_label =
 237           case user_label
 238           when nil, /^\s*$/
 239             bm.spawn_modal("Label list", LabelListMode.new) if user_label && user_label.empty?
 240           else
 241             LabelManager.label_for user_label
 242           end
 243
 244         case user_label
 245         when nil
 246         when :inbox
 247           BufferManager.raise_to_front InboxMode.instance.buffer
 248         else
 249           b = BufferManager.spawn_unless_exists("All threads with label '#{user_label}'") do
 250             mode = LabelSearchResultsMode.new([user_label])
 251           end
 252           b.mode.load_threads :num => b.content_height
 253         end
 254
 255       when :compose
 256         spawn_compose_mode
 257       when :poll
 258         reporting_thread { PollManager.poll }
 259       when :recall_draft
 260         case Index.num_results_for :label => :draft
 261         when 0
 262           bm.flash "No draft messages."
 263         when 1
 264           m = nil
 265           Index.each_id_by_date(:label => :draft) { |mid, builder| m = builder.call }
 266           r = ResumeMode.new(m)
 267           BufferManager.spawn "Edit message", r
 268           r.edit_message
 269         else
 270           b = BufferManager.spawn_unless_exists("All drafts") do
 271             mode = LabelSearchResultsMode.new [:draft]
 272           end
 273           b.mode.load_threads :num => b.content_height
 274         end
 275       when :nothing
 276       when :redraw
 277         bm.completely_redraw_screen
 278       else
 279         bm.flash "Unknown keypress '#{c.to_character}' for #{bm.focus_buf.mode.name}."
 280       end
 281     end
 282
 283     bm.draw_screen
 284   end
 285 rescue Exception => e
 286   $exception ||= e
 287 ensure
 288   unless $opts[:no_threads]
 289     PollManager.stop if PollManager.instantiated?
 290     SuicideManager.stop if PollManager.instantiated?
 291     Index.stop_lock_update_thread
 292   end
 293
 294   Redwood::finish
 295   stop_cursing
 296   Redwood::log "stopped cursing"
 297
 298   if SuicideManager.instantiated? && SuicideManager.die?
 299     Redwood::log "I've been ordered to commit sepuku. I obey!"
 300   end
 301
 302   case $exception
 303   when nil
 304     Redwood::log "no fatal errors. good job, william."
 305     Index.save
 306   else
 307     Redwood::log "oh crap, an exception"
 308   end
 309
 310   Index.unlock
 311 end
 312
 313 if $exception
 314   $stderr.puts <<EOS
 315 ----------------------------------------------------------------
 316 I'm very sorry, but it seems that an error occurred in Sup.
 317 Please accept my sincere apologies. If you don't mind, please
 318 send the backtrace below and a brief report of the circumstances
 319 to sup-talk at rubyforge dot orgs so that I might address this
 320 problem. Thank you!
 321
 322 Sincerely,
 323 William
 324 ----------------------------------------------------------------
 325
 326 The problem was: '#{$exception.message}' (error type #{$exception.class.name})
 327 A backtrace follows:
 328 EOS
 329   raise $exception
 330 end
 331
 332 end