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 $opts = Trollop::options do
  25   version "sup v#{Redwood::VERSION}"
  26   banner <<EOS
  27 Sup is a curses-based email client.
  28
  29 Usage:
  30   sup [options]
  31
  32 Options are:
  33 EOS
  34   opt :list_hooks, "List all hooks and descriptions, and quit."
  35   opt :no_threads, "Turn off threading. Helps with debugging. (Necessarily disables background polling for new messages.)"
  36   opt :no_initial_poll, "Don't poll for new messages when starting."
  37   opt :search, "Search for this query upon startup", :type => String
  38   opt :compose, "Compose message to this recipient upon startup", :type => String
  39 end
  40
  41 Redwood::HookManager.register "startup", <<EOS
  42 Executes at startup
  43 No variables.
  44 No return value.
  45 EOS
  46
  47 Redwood::HookManager.register "shutdown", <<EOS
  48 Executes when sup is shutting down. May be run when sup is crashing,
  49 so don\'t do anything too important. Run before the label, contacts,
  50 and people are saved.
  51 No variables.
  52 No return value.
  53 EOS
  54
  55 if $opts[:list_hooks]
  56   Redwood::HookManager.print_hooks
  57   exit
  58 end
  59
  60 Thread.abort_on_exception = true # make debugging possible
  61
  62 module Redwood
  63
  64 global_keymap = Keymap.new do |k|
  65   k.add :quit_ask, "Quit Sup, but ask first", 'q'
  66   k.add :quit_now, "Quit Sup immediately", 'Q'
  67   k.add :help, "Show help", '?'
  68   k.add :roll_buffers, "Switch to next buffer", 'b'
  69   k.add :roll_buffers_backwards, "Switch to previous buffer", 'B'
  70   k.add :kill_buffer, "Kill the current buffer", 'x'
  71   k.add :list_buffers, "List all buffers", ';'
  72   k.add :list_contacts, "List contacts", 'C'
  73   k.add :redraw, "Redraw screen", :ctrl_l
  74   k.add :search, "Search all messages", '\\', 'F'
  75   k.add :search_unread, "Show all unread messages", 'U'
  76   k.add :list_labels, "List labels", 'L'
  77   k.add :poll, "Poll for new messages", 'P'
  78   k.add :compose, "Compose new message", 'm', 'c'
  79   k.add :nothing, "Do nothing", :ctrl_g
  80   k.add :recall_draft, "Edit most recent draft message", 'R'
  81 end
  82
  83 ## the following magic enables wide characters when used with a ruby
  84 ## ncurses.so that's been compiled against libncursesw. (note the w.) why
  85 ## this works, i have no idea. much like pretty much every aspect of
  86 ## dealing with curses.  cargo cult programming at its best.
  87 ##
  88 ## BSD users: if libc.so.6 is not found, try installing compat6x.
  89 require 'dl/import'
  90 module LibC
  91   extend DL.const_defined?(:Importer) ? DL::Importer : DL::Importable
  92   setlocale_lib = case Config::CONFIG['arch']
  93     when /darwin/; "libc.dylib"
  94     when /cygwin/; "cygwin1.dll"
  95     else; "libc.so.6"
  96   end
  97
  98   Redwood::log "dynamically loading setlocale() from #{setlocale_lib}"
  99   begin
 100     dlload setlocale_lib
 101     extern "void setlocale(int, const char *)"
 102     Redwood::log "setting locale..."
 103     LibC.setlocale(6, "")  # LC_ALL == 6
 104   rescue RuntimeError => e
 105     Redwood::log "cannot dlload setlocale(); ncurses wide character support probably broken."
 106     Redwood::log "dlload error was #{e.class}: #{e.message}"
 107     if Config::CONFIG['arch'] =~ /bsd/
 108       Redwood::log "BSD variant detected. You may have to install a compat6x package to acquire libc."
 109     end
 110   end
 111 end
 112
 113 def start_cursing
 114   Ncurses.initscr
 115   Ncurses.noecho
 116   Ncurses.cbreak
 117   Ncurses.stdscr.keypad 1
 118   Ncurses.use_default_colors
 119   Ncurses.curs_set 0
 120   Ncurses.start_color
 121   $cursing = true
 122 end
 123
 124 def stop_cursing
 125   return unless $cursing
 126   Ncurses.curs_set 1
 127   Ncurses.echo
 128   Ncurses.endwin
 129 end
 130 module_function :start_cursing, :stop_cursing
 131
 132 Index.new
 133 begin
 134   Index.lock
 135 rescue Index::LockError => e
 136   require 'highline'
 137
 138   h = HighLine.new
 139   h.wrap_at = :auto
 140   h.say Index.fancy_lock_error_message_for(e)
 141
 142   case h.ask("Should I ask that process to kill itself? ")
 143   when /^\s*y(es)?\s*$/i
 144     h.say "Ok, suggesting seppuku..."
 145     FileUtils.touch Redwood::SUICIDE_FN
 146     sleep SuicideManager::DELAY * 2
 147     FileUtils.rm_f Redwood::SUICIDE_FN
 148     h.say "Let's try that again."
 149     retry
 150   else
 151     h.say <<EOS
 152 Ok, giving up. If the process crashed and left a stale lockfile, you
 153 can fix this by manually deleting #{Index.lockfile}.
 154 EOS
 155     exit
 156   end
 157 end
 158
 159 begin
 160   Redwood::start
 161   Index.load
 162
 163   trap("TERM") { |x| SuicideManager.please_die! }
 164   trap("WINCH") { |x| BufferManager.sigwinch_happened! }
 165
 166   if(s = Redwood::SourceManager.source_for DraftManager.source_name)
 167     DraftManager.source = s
 168   else
 169     Redwood::log "no draft source, auto-adding..."
 170     Redwood::SourceManager.add_source DraftManager.new_source
 171   end
 172
 173   if(s = Redwood::SourceManager.source_for SentManager.source_uri)
 174     SentManager.source = s
 175   else
 176     Redwood::SourceManager.add_source SentManager.default_source
 177   end
 178
 179   HookManager.run "startup"
 180
 181   log "starting curses"
 182   start_cursing
 183
 184   bm = BufferManager.new
 185   Colormap.new.populate_colormap
 186
 187   log "initializing mail index buffer"
 188   imode = InboxMode.new
 189   ibuf = bm.spawn "Inbox", imode
 190
 191   log "ready for interaction!"
 192   Logger.make_buf
 193
 194   bm.draw_screen
 195
 196   Redwood::SourceManager.usual_sources.each do |s|
 197     next unless s.respond_to? :connect
 198     reporting_thread("call #connect on #{s}") do
 199       begin
 200         s.connect
 201       rescue SourceError => e
 202         Redwood::log "fatal error loading from #{s}: #{e.message}"
 203       end
 204     end
 205   end unless $opts[:no_initial_poll]
 206
 207   imode.load_threads :num => ibuf.content_height, :when_done => lambda { |num| reporting_thread("poll after loading inbox") { sleep 1; PollManager.poll } unless $opts[:no_threads] || $opts[:no_initial_poll] }
 208
 209   if $opts[:compose]
 210     ComposeMode.spawn_nicely :to_default => $opts[:compose]
 211   end
 212
 213   unless $opts[:no_threads]
 214     PollManager.start
 215     SuicideManager.start
 216     Index.start_lock_update_thread
 217   end
 218
 219   if $opts[:search]
 220     SearchResultsMode.spawn_from_query $opts[:search]
 221   end
 222
 223   until Redwood::exceptions.nonempty? || SuicideManager.die?
 224     c = begin
 225       Ncurses.nonblocking_getch
 226     rescue Interrupt => e
 227       raise if BufferManager.ask_yes_or_no "Die ungracefully now?"
 228       BufferManager.draw_screen
 229       nil
 230     end
 231
 232     if c.nil?
 233       if BufferManager.sigwinch_happened?
 234         Redwood::log "redrawing screen on sigwinch"
 235         BufferManager.completely_redraw_screen
 236       end
 237       next
 238     end
 239
 240     if c == 410
 241       ## this is ncurses's way of telling us it's detected a refresh.
 242       ## since we have our own sigwinch handler, we don't do anything.
 243       next
 244     end
 245
 246     bm.erase_flash
 247
 248     action = begin
 249       if bm.handle_input c
 250         :nothing
 251       else
 252         bm.resolve_input_with_keymap c, global_keymap
 253       end
 254     rescue InputSequenceAborted
 255       :nothing
 256     end
 257     case action
 258     when :quit_now
 259       break if bm.kill_all_buffers_safely
 260     when :quit_ask
 261       if bm.ask_yes_or_no "Really quit?"
 262         break if bm.kill_all_buffers_safely
 263       end
 264     when :help
 265       curmode = bm.focus_buf.mode
 266       bm.spawn_unless_exists("<help for #{curmode.name}>") { HelpMode.new curmode, global_keymap }
 267     when :roll_buffers
 268       bm.roll_buffers
 269     when :roll_buffers_backwards
 270       bm.roll_buffers_backwards
 271     when :kill_buffer
 272       bm.kill_buffer_safely bm.focus_buf
 273     when :list_buffers
 274       bm.spawn_unless_exists("buffer list", :system => true) { BufferListMode.new }
 275     when :list_contacts
 276       b, new = bm.spawn_unless_exists("Contact List") { ContactListMode.new }
 277       b.mode.load_in_background if new
 278     when :search
 279       query = BufferManager.ask :search, "search all messages: "
 280       next unless query && query !~ /^\s*$/
 281       SearchResultsMode.spawn_from_query query
 282     when :search_unread
 283       SearchResultsMode.spawn_from_query "is:unread"
 284     when :list_labels
 285       labels = LabelManager.all_labels.map { |l| LabelManager.string_for l }
 286       user_label = bm.ask_with_completions :label, "Show threads with label (enter for listing): ", labels
 287       unless user_label.nil?
 288         if user_label.empty?
 289           bm.spawn_unless_exists("Label list") { LabelListMode.new } if user_label && user_label.empty?
 290         else
 291           LabelSearchResultsMode.spawn_nicely user_label
 292         end
 293       end
 294     when :compose
 295       ComposeMode.spawn_nicely
 296     when :poll
 297       reporting_thread("user-invoked poll") { PollManager.poll }
 298     when :recall_draft
 299       case Index.num_results_for :label => :draft
 300       when 0
 301         bm.flash "No draft messages."
 302       when 1
 303         m = nil
 304         Index.each_id_by_date(:label => :draft) { |mid, builder| m = builder.call }
 305         r = ResumeMode.new(m)
 306         BufferManager.spawn "Edit message", r
 307         r.edit_message
 308       else
 309         b, new = BufferManager.spawn_unless_exists("All drafts") { LabelSearchResultsMode.new [:draft] }
 310         b.mode.load_threads :num => b.content_height if new
 311       end
 312     when :nothing, InputSequenceAborted
 313     when :redraw
 314       bm.completely_redraw_screen
 315     else
 316       bm.flash "Unknown keypress '#{c.to_character}' for #{bm.focus_buf.mode.name}."
 317     end
 318
 319     bm.draw_screen
 320   end
 321
 322   bm.kill_all_buffers if SuicideManager.die?
 323 rescue Exception => e
 324   Redwood::record_exception e, "main"
 325 ensure
 326   unless $opts[:no_threads]
 327     PollManager.stop if PollManager.instantiated?
 328     SuicideManager.stop if PollManager.instantiated?
 329     Index.stop_lock_update_thread
 330   end
 331
 332   HookManager.run "shutdown"
 333
 334   Redwood::finish
 335   stop_cursing
 336   Redwood::log "stopped cursing"
 337
 338   if SuicideManager.instantiated? && SuicideManager.die?
 339     Redwood::log "I've been ordered to commit seppuku. I obey!"
 340   end
 341
 342   if Redwood::exceptions.empty?
 343     Redwood::log "no fatal errors. good job, william."
 344     Index.save
 345   else
 346     Redwood::log "oh crap, an exception"
 347   end
 348
 349   Index.unlock
 350 end
 351
 352 unless Redwood::exceptions.empty?
 353   File.open(File.join(BASE_DIR, "exception-log.txt"), "w") do |f|
 354     Redwood::exceptions.each do |e, name|
 355       f.puts "--- #{e.class.name} from thread: #{name}"
 356       f.puts e.message, e.backtrace
 357     end
 358   end
 359   $stderr.puts <<EOS
 360 ----------------------------------------------------------------
 361 I'm very sorry. It seems that an error occurred in Sup. Please
 362 accept my sincere apologies. If you don't mind, please send the
 363 contents of ~/.sup/exception-log.txt and a brief report of the
 364 circumstances to sup-talk at rubyforge dot orgs so that I might
 365 address this problem. Thank you!
 366
 367 Sincerely,
 368 William
 369 ----------------------------------------------------------------
 370 EOS
 371   Redwood::exceptions.each do |e, name|
 372     puts "--- #{e.class.name} from thread: #{name}"
 373     puts e.message, e.backtrace
 374   end
 375 end
 376
 377 end