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