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", '?'
  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 :search_unread, "Show all unread messages", 'U'
  69   k.add :list_labels, "List labels", 'L'
  70   k.add :poll, "Poll for new messages", 'P'
  71   k.add :compose, "Compose new message", 'm', 'c'
  72   k.add :nothing, "Do nothing", :ctrl_g
  73   k.add :recall_draft, "Edit most recent draft message", 'R'
  74 end
  75
  76 def start_cursing
  77   Ncurses.initscr
  78   Ncurses.noecho
  79   Ncurses.cbreak
  80   Ncurses.stdscr.keypad 1
  81   Ncurses.curs_set 0
  82   Ncurses.start_color
  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(es)?\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.populate_colormap
 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     Redwood::reporting_thread("be friendly") do
 178       id = BufferManager.say "Welcome to Sup! Press '?' at any point for help."
 179       sleep 10
 180       BufferManager.clear id
 181     end
 182   end
 183
 184   if $opts[:search]
 185     SearchResultsMode.spawn_from_query $opts[:search]
 186   end
 187
 188   until Redwood::exceptions.nonempty? || SuicideManager.die?
 189     c = Ncurses.nonblocking_getch
 190     next unless c
 191     bm.erase_flash
 192
 193     action =
 194       begin
 195         if bm.handle_input c
 196           :nothing
 197         else
 198           bm.resolve_input_with_keymap c, global_keymap
 199         end
 200       rescue InputSequenceAborted
 201         :nothing
 202       end
 203
 204     case action
 205     when :quit_now
 206       break if bm.kill_all_buffers_safely
 207     when :quit_ask
 208       if bm.ask_yes_or_no "Really quit?"
 209         break if bm.kill_all_buffers_safely
 210       end
 211     when :help
 212       curmode = bm.focus_buf.mode
 213       bm.spawn_unless_exists("<help for #{curmode.name}>") { HelpMode.new curmode, global_keymap }
 214     when :roll_buffers
 215       bm.roll_buffers
 216     when :roll_buffers_backwards
 217       bm.roll_buffers_backwards
 218     when :kill_buffer
 219       bm.kill_buffer_safely bm.focus_buf
 220     when :list_buffers
 221       bm.spawn_unless_exists("Buffer List") { BufferListMode.new }
 222     when :list_contacts
 223       b, new = bm.spawn_unless_exists("Contact List") { ContactListMode.new }
 224       b.mode.load_in_background if new
 225     when :search
 226       query = BufferManager.ask :search, "search all messages: "
 227       next unless query && query !~ /^\s*$/
 228       SearchResultsMode.spawn_from_query query
 229     when :search_unread
 230       SearchResultsMode.spawn_from_query "is:unread"
 231     when :list_labels
 232       labels = LabelManager.listable_labels.map { |l| LabelManager.string_for l }
 233       user_label = bm.ask_with_completions :label, "Show threads with label (enter for listing): ", labels
 234       unless user_label.nil?
 235         if user_label.empty?
 236           bm.spawn_unless_exists("Label list") { LabelListMode.new } if user_label && user_label.empty?
 237         else
 238           LabelSearchResultsMode.spawn_nicely user_label
 239         end
 240       end
 241     when :compose
 242       ComposeMode.spawn_nicely
 243     when :poll
 244       reporting_thread("user-invoked poll") { PollManager.poll }
 245     when :recall_draft
 246       case Index.num_results_for :label => :draft
 247       when 0
 248         bm.flash "No draft messages."
 249       when 1
 250         m = nil
 251         Index.each_id_by_date(:label => :draft) { |mid, builder| m = builder.call }
 252         r = ResumeMode.new(m)
 253         BufferManager.spawn "Edit message", r
 254         r.edit_message
 255       else
 256         b, new = BufferManager.spawn_unless_exists("All drafts") { LabelSearchResultsMode.new [:draft] }
 257         b.mode.load_threads :num => b.content_height if new
 258       end
 259     when :nothing, InputSequenceAborted
 260     when :redraw
 261       bm.completely_redraw_screen
 262     else
 263       bm.flash "Unknown keypress '#{c.to_character}' for #{bm.focus_buf.mode.name}."
 264     end
 265
 266     bm.draw_screen
 267   end
 268
 269   bm.kill_all_buffers if SuicideManager.die?
 270 rescue Exception => e
 271   Redwood::record_exception e, "main"
 272 ensure
 273   unless $opts[:no_threads]
 274     PollManager.stop if PollManager.instantiated?
 275     SuicideManager.stop if PollManager.instantiated?
 276     Index.stop_lock_update_thread
 277   end
 278
 279   Redwood::finish
 280   stop_cursing
 281   Redwood::log "stopped cursing"
 282
 283   if SuicideManager.instantiated? && SuicideManager.die?
 284     Redwood::log "I've been ordered to commit seppuku. I obey!"
 285   end
 286
 287   if Redwood::exceptions.empty?
 288     Redwood::log "no fatal errors. good job, william."
 289     Index.save
 290   else
 291     Redwood::log "oh crap, an exception"
 292   end
 293
 294   Index.unlock
 295 end
 296
 297 unless Redwood::exceptions.empty?
 298   File.open(File.join(BASE_DIR, "exception-log.txt"), "w") do |f|
 299     Redwood::exceptions.each do |e, name|
 300       f.puts "--- #{e.class.name} from thread: #{name}"
 301       f.puts e.message, e.backtrace
 302     end
 303   end
 304   $stderr.puts <<EOS
 305 ----------------------------------------------------------------
 306 I'm very sorry. It seems that an error occurred in Sup. Please
 307 accept my sincere apologies. If you don't mind, please send the
 308 contents of ~/.sup/exception-log.txt and a brief report of the
 309 circumstances to sup-talk at rubyforge dot orgs so that I might
 310 address this problem. Thank you!
 311
 312 Sincerely,
 313 William
 314 ----------------------------------------------------------------
 315 EOS
 316   Redwood::exceptions.each do |e, name|
 317     puts "--- #{e.class.name} from thread: #{name}"
 318     puts e.message, e.backtrace
 319   end
 320 end
 321
 322 end