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 BIN_VERSION = "git"
  11
  12 unless Redwood::VERSION == BIN_VERSION
  13   $stderr.puts <<EOS
  14
  15 Error: version mismatch!
  16 The sup executable is at version #{BIN_VERSION.inspect}.
  17 The sup libraries are at version #{Redwood::VERSION.inspect}.
  18
  19 Is your development environment conflicting with rubygems?
  20 EOS
  21   exit(-1)
  22 end
  23
  24 $exceptions = []
  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   Ncurses.use_default_colors
  83   $cursing = true
  84 end
  85
  86 def stop_cursing
  87   return unless $cursing
  88   Ncurses.curs_set 1
  89   Ncurses.echo
  90   Ncurses.endwin
  91 end
  92 module_function :start_cursing, :stop_cursing
  93
  94 Index.new
  95 begin
  96   Index.lock
  97 rescue Index::LockError => e
  98   require 'highline'
  99
 100   h = HighLine.new
 101   h.wrap_at = :auto
 102   h.say Index.fancy_lock_error_message_for(e)
 103
 104   case h.ask("Should I ask that process to kill itself? ")
 105   when /^\s*y\s*$/i
 106     h.say "Ok, suggesting seppuku..."
 107     FileUtils.touch Redwood::SUICIDE_FN
 108     sleep SuicideManager::DELAY * 2
 109     FileUtils.rm_f Redwood::SUICIDE_FN
 110     h.say "Let's try that again."
 111     retry
 112   else
 113     h.say <<EOS
 114 Ok, giving up. If the process crashed and left a stale lockfile, you
 115 can fix this by manually deleting #{Index.lockfile}.
 116 EOS
 117     exit
 118   end
 119 end
 120
 121 begin
 122   Redwood::start
 123   Index.load
 124
 125   if(s = Index.source_for DraftManager.source_name)
 126     DraftManager.source = s
 127   else
 128     Redwood::log "no draft source, auto-adding..."
 129     Index.add_source DraftManager.new_source
 130   end
 131
 132   if(s = Index.source_for SentManager.source_name)
 133     SentManager.source = s
 134   else
 135     Redwood::log "no sent mail source, auto-adding..."
 136     Index.add_source SentManager.new_source
 137   end
 138
 139   HookManager.run "startup"
 140
 141   log "starting curses"
 142   start_cursing
 143
 144   bm = BufferManager.new
 145   Colormap.new
 146
 147   log "initializing mail index buffer"
 148   imode = InboxMode.new
 149   ibuf = bm.spawn "Inbox", imode
 150
 151   log "ready for interaction!"
 152   Logger.make_buf
 153
 154   bm.draw_screen
 155
 156   Index.usual_sources.each do |s|
 157     next unless s.respond_to? :connect
 158     reporting_thread("call #connect on #{s}") do
 159       begin
 160         s.connect
 161       rescue SourceError => e
 162         Redwood::log "fatal error loading from #{s}: #{e.message}"
 163       end
 164     end
 165   end unless $opts[:no_initial_poll]
 166
 167   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] }
 168
 169   if $opts[:compose]
 170     ComposeMode.spawn_nicely :to_default => $opts[:compose]
 171   end
 172
 173   unless $opts[:no_threads]
 174     PollManager.start
 175     SuicideManager.start
 176     Index.start_lock_update_thread
 177   end
 178
 179   if $opts[:search]
 180     SearchResultsMode.spawn_from_query $opts[:search]
 181   end
 182
 183   until $exceptions.nonempty? || SuicideManager.die?
 184     c = Ncurses.nonblocking_getch
 185     next unless c
 186     bm.erase_flash
 187
 188     action =
 189       begin
 190         if bm.handle_input c
 191           :nothing
 192         else
 193           bm.resolve_input_with_keymap c, global_keymap
 194         end
 195       rescue InputSequenceAborted
 196         :nothing
 197       end
 198
 199     case action
 200     when :quit_now
 201       break if bm.kill_all_buffers_safely
 202     when :quit_ask
 203       if bm.ask_yes_or_no "Really quit?"
 204         break if bm.kill_all_buffers_safely
 205       end
 206     when :help
 207       curmode = bm.focus_buf.mode
 208       bm.spawn_unless_exists("<help for #{curmode.name}>") { HelpMode.new curmode, global_keymap }
 209     when :roll_buffers
 210       bm.roll_buffers
 211     when :roll_buffers_backwards
 212       bm.roll_buffers_backwards
 213     when :kill_buffer
 214       bm.kill_buffer_safely bm.focus_buf
 215     when :list_buffers
 216       bm.spawn_unless_exists("Buffer List") { BufferListMode.new }
 217     when :list_contacts
 218       b, new = bm.spawn_unless_exists("Contact List") { ContactListMode.new }
 219       b.mode.load_in_background if new
 220     when :search
 221       query = BufferManager.ask :search, "search all messages: "
 222       next unless query && query !~ /^\s*$/
 223       SearchResultsMode.spawn_from_query query
 224     when :list_labels
 225       labels = LabelManager.listable_labels.map { |l| LabelManager.string_for l }
 226       user_label = bm.ask_with_completions :label, "Show threads with label (enter for listing): ", labels
 227       unless user_label.nil?
 228         if user_label.empty?
 229           bm.spawn_unless_exists("Label list") { LabelListMode.new } if user_label && user_label.empty?
 230         else
 231           LabelSearchResultsMode.spawn_nicely user_label
 232         end
 233       end
 234     when :compose
 235       ComposeMode.spawn_nicely
 236     when :poll
 237       reporting_thread("user-invoked poll") { PollManager.poll }
 238     when :recall_draft
 239       case Index.num_results_for :label => :draft
 240       when 0
 241         bm.flash "No draft messages."
 242       when 1
 243         m = nil
 244         Index.each_id_by_date(:label => :draft) { |mid, builder| m = builder.call }
 245         r = ResumeMode.new(m)
 246         BufferManager.spawn "Edit message", r
 247         r.edit_message
 248       else
 249         b, new = BufferManager.spawn_unless_exists("All drafts") { LabelSearchResultsMode.new [:draft] }
 250         b.mode.load_threads :num => b.content_height if new
 251       end
 252     when :nothing, InputSequenceAborted
 253     when :redraw
 254       bm.completely_redraw_screen
 255     else
 256       bm.flash "Unknown keypress '#{c.to_character}' for #{bm.focus_buf.mode.name}."
 257     end
 258
 259     bm.draw_screen
 260   end
 261
 262   bm.kill_all_buffers if SuicideManager.die?
 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 seppuku. 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(File.join(BASE_DIR, "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