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