]> git.cworth.org Git - sup/blobdiff - bin/sup
hook system
[sup] / bin / sup
diff --git a/bin/sup b/bin/sup
index 36fd15263f57c75e06a4db6af0cbbaf3f631c9f5..918ec47636a368eb0daf8646123bab35dfbca789 100644 (file)
--- a/bin/sup
+++ b/bin/sup
@@ -2,8 +2,29 @@
 
 require 'rubygems'
 require 'ncurses'
+require 'fileutils'
+require 'trollop'
 require "sup"
 
+$opts = Trollop::options do
+  version "sup v#{Redwood::VERSION}"
+  banner <<EOS
+Sup is a curses-based email client.
+
+Usage:
+  sup [options]
+
+Options are:
+EOS
+  opt :list_hooks, "List all hooks and descriptions thereof, and quit."
+  opt :no_threads, "Turn of threading. Helps with debugging. (Necessarily disables background polling for new messages.)"
+end
+
+if $opts[:list_hooks]
+  Redwood::HookManager.print_hooks
+  exit
+end
+
 Thread.abort_on_exception = true # make debugging possible
 
 module Redwood
@@ -12,9 +33,9 @@ global_keymap = Keymap.new do |k|
   k.add :quit, "Quit Redwood", 'q'
   k.add :help, "Show help", 'H', '?'
   k.add :roll_buffers, "Switch to next buffer", 'b'
-  k.add :roll_buffers_backwards, "Switch to previous buffer", 'B'
+#  k.add :roll_buffers_backwards, "Switch to previous buffer", 'B'
   k.add :kill_buffer, "Kill the current buffer", 'x'
-  k.add :list_buffers, "List all buffers", 'A'
+  k.add :list_buffers, "List all buffers", 'B'
   k.add :list_contacts, "List contacts", 'C'
   k.add :redraw, "Redraw screen", :ctrl_l
   k.add :search, "Search messages", '/'
@@ -31,31 +52,60 @@ def start_cursing
   Ncurses.stdscr.keypad 1
   Ncurses.curs_set 0
   Ncurses.start_color
+  $cursing = true
 end
 
 def stop_cursing
+  return unless $cursing
   Ncurses.curs_set 1
   Ncurses.echo
   Ncurses.endwin
 end
 module_function :start_cursing, :stop_cursing
 
-Redwood::start
-Index.new.load
+Index.new
+begin
+  Index.lock
+rescue Index::LockError => e
+  require 'highline'
 
-if(s = Index.source_for DraftManager.source_name)
-  DraftManager.source = s
-else
-  Index.add_source DraftManager.new_source
-end
+  h = HighLine.new
+  h.wrap_at = :auto
+  h.say Index.fancy_lock_error_message_for(e)
 
-if(s = Index.source_for SentManager.source_name)
-  SentManager.source = s
-else
-  Index.add_source SentManager.new_source
+  case h.ask("Should I ask that process to kill itself? ")
+  when /^\s*y\s*$/i
+    h.say "Ok, suggesting sepuku..."
+    FileUtils.touch Redwood::SUICIDE_FN
+    sleep SuicideManager::DELAY * 2
+    FileUtils.rm_f Redwood::SUICIDE_FN
+    h.say "Let's try that again."
+    retry
+  else
+    h.say <<EOS
+Ok, giving up. If the process crashed and left a stale lockfile, you
+can fix this by manually deleting #{Index.lockfile}.
+EOS
+    exit
+  end
 end
 
 begin
+  Redwood::start
+  Index.load
+
+  if(s = Index.source_for DraftManager.source_name)
+    DraftManager.source = s
+  else
+    Index.add_source DraftManager.new_source
+  end
+
+  if(s = Index.source_for SentManager.source_name)
+    SentManager.source = s
+  else
+    Index.add_source SentManager.new_source
+  end
+
   log "starting curses"
   start_cursing
 
@@ -65,13 +115,17 @@ begin
     c.add :index_old_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK
     c.add :index_new_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK, 
            Ncurses::A_BOLD
+    c.add :index_starred_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK, 
+           Ncurses::A_BOLD
     c.add :labellist_old_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK
     c.add :labellist_new_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK, 
            Ncurses::A_BOLD
     c.add :twiddle_color, Ncurses::COLOR_BLUE, Ncurses::COLOR_BLACK
     c.add :label_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
     c.add :message_patina_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_GREEN
-    c.add :mime_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
+    c.add :alternate_patina_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_BLUE
+    c.add :missing_message_color, Ncurses::COLOR_BLACK, Ncurses::COLOR_RED
+    c.add :attachment_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
     c.add :quote_patina_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
     c.add :sig_patina_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
     c.add :quote_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK
@@ -81,12 +135,16 @@ begin
           Ncurses::A_BOLD
     c.add :starred_patina_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_GREEN,
           Ncurses::A_BOLD
+    c.add :alternate_starred_patina_color, Ncurses::COLOR_YELLOW,
+          Ncurses::COLOR_BLUE, Ncurses::A_BOLD
     c.add :snippet_color, Ncurses::COLOR_CYAN, Ncurses::COLOR_BLACK
     c.add :option_color, Ncurses::COLOR_WHITE, Ncurses::COLOR_BLACK
     c.add :tagged_color, Ncurses::COLOR_YELLOW, Ncurses::COLOR_BLACK,
           Ncurses::A_BOLD
     c.add :draft_notification_color, Ncurses::COLOR_RED, Ncurses::COLOR_BLACK,
           Ncurses::A_BOLD
+    c.add :completion_character_color, Ncurses::COLOR_WHITE,
+          Ncurses::COLOR_BLACK, Ncurses::A_BOLD
   end
 
   log "initializing buffer manager"
@@ -94,100 +152,155 @@ begin
 
   log "initializing mail index buffer"
   imode = InboxMode.new
-  ibuf = bm.spawn "inbox", imode
+  ibuf = bm.spawn "Inbox", imode
 
-  log "ready for (inter)action!"
+  log "ready for interaction!"
   Logger.make_buf
 
   bm.draw_screen
-  imode.load_more_threads ibuf.content_height
 
-  reporting_thread { sleep 5; PollManager.poll }
-  PollManager.start_thread
+  begin
+    Index.usual_sources.each { |s| s.check }
+  rescue SourceError
+    # do nothing! we'll report it at the next step
+  end
+  Redwood::report_broken_sources
+  
+  Index.usual_sources.each do |s|
+    reporting_thread do
+      begin
+        s.connect
+      rescue SourceError => e
+        Redwood::log "fatal error loading from #{s}: #{e.message}"
+      end
+    end if s.respond_to? :connect
+  end
 
-  until $exception
-    bm.draw_screen
+  imode.load_threads :num => ibuf.content_height, :when_done => lambda { reporting_thread { sleep 1; PollManager.poll } unless $opts[:no_threads] }
+
+  unless $opts[:no_threads]
+    PollManager.start
+    SuicideManager.start
+    Index.start_lock_update_thread
+  end
+
+  until $exception || SuicideManager.die?
     c = Ncurses.nonblocking_getch
+    next unless c
     bm.erase_flash
 
-    if c == Ncurses::KEY_RESIZE
-      bm.handle_resize
-    elsif c
-      unless bm.handle_input(c)
-        x = global_keymap.action_for c
-        case x
-        when :quit
-          break
-        when :help
-          curmode = bm.focus_buf.mode
-          bm.spawn_unless_exists("<help for #{curmode.name}>") { HelpMode.new curmode, global_keymap }
-        when :roll_buffers
-          bm.roll_buffers
-        when :roll_buffers_backwards
-          bm.roll_buffers_backwards
-        when :kill_buffer
-          bm.kill_buffer bm.focus_buf if bm.focus_buf.mode.killable?
-        when :list_buffers
-          bm.spawn_unless_exists("Buffer List") { BufferListMode.new }
-        when :list_contacts
-          bm.spawn_unless_exists("Contact List") { ContactListMode.new }
-        when :search
-          text = bm.ask :search, "query: "
-          next unless text && text !~ /^\s*$/
-
-          begin
-            qobj = Index.parse_user_query_string text
-            short_text = text.length < 20 ? text : text[0 ... 20] + "..."
-            log "built query from #{text.inspect}: #{qobj}"
-            mode = SearchResultsMode.new qobj
-            bm.spawn "search: \"#{short_text}\"", mode
-            mode.load_more_threads mode.buffer.content_height
-          rescue Ferret::QueryParser::QueryParseException => e
-            bm.flash "Couldn't parse query."
-          end
+    unless bm.handle_input(c)
+      x = global_keymap.action_for c
+      case x
+      when :quit
+        break if bm.kill_all_buffers_safely
+      when :help
+        curmode = bm.focus_buf.mode
+        bm.spawn_unless_exists("<help for #{curmode.name}>") { HelpMode.new curmode, global_keymap }
+      when :roll_buffers
+        bm.roll_buffers
+      when :roll_buffers_backwards
+        bm.roll_buffers_backwards
+      when :kill_buffer
+        bm.kill_buffer_safely bm.focus_buf
+      when :list_buffers
+        bm.spawn_unless_exists("Buffer List") { BufferListMode.new }
+      when :list_contacts
+        b = bm.spawn_unless_exists("Contact List") { ContactListMode.new }
+        b.mode.load_in_background
+      when :search
+        text = bm.ask :search, "query: "
+        next unless text && text !~ /^\s*$/
 
-        when :list_labels
-          b = bm.spawn_unless_exists("Label List") { LabelListMode.new }
-          b.mode.load_in_background
-        when :compose
-          mode = ComposeMode.new
-          bm.spawn "New Message", mode
-          mode.edit
-        when :poll
-          bm.raise_to_front PollManager.buffer
-          reporting_thread { PollManager.poll }
-        when :recall_draft
-          case Index.num_results_for :label => :draft
-          when 0
-            bm.flash "No draft messages."
-          when 1
-            m = nil
-            Index.each_id_by_date(:label => :draft) { |mid, builder| m = builder.call }
-            BufferManager.spawn "Edit message", ResumeMode.new(m)
-          else
-            b = BufferManager.spawn_unless_exists(:draft) do
-              mode = LabelSearchResultsMode.new [:draft]
-            end
-            b.mode.load_more_threads b.content_height
+        begin
+          qobj = Index.parse_user_query_string text
+          short_text = text.length < 20 ? text : text[0 ... 20] + "..."
+          log "built query from #{text.inspect}: #{qobj}"
+          mode = SearchResultsMode.new qobj
+          bm.spawn "search: \"#{short_text}\"", mode
+          mode.load_threads :num => mode.buffer.content_height
+        rescue Ferret::QueryParser::QueryParseException => e
+          bm.flash "Couldn't parse query."
+        end
+      when :list_labels
+        labels = LabelManager.listable_labels.map { |l| LabelManager.string_for l }
+        user_label = bm.ask_with_completions :label, "Show threads with label (enter for listing): ", labels
+        user_label = bm.spawn_modal("Label list", LabelListMode.new) if user_label && user_label.empty?
+        
+        case user_label
+        when nil
+        when :inbox
+          BufferManager.raise_to_front InboxMode.instance.buffer
+        else
+          b = BufferManager.spawn_unless_exists("All threads with label '#{user_label}'") do
+            mode = LabelSearchResultsMode.new([user_label])
           end
-        when :nothing
-        when :redraw
-          bm.completely_redraw_screen
+          b.mode.load_threads :num => b.content_height
+        end
+
+      when :compose
+        to = BufferManager.ask_for_contacts(:people, "To: ") or next
+        cc = BufferManager.ask_for_contacts(:people, "Cc: ") or next if $config[:ask_for_cc]
+        bcc = BufferManager.ask_for_contacts(:people, "Bcc: ") or next if $config[:ask_for_bcc]
+
+        mode = ComposeMode.new :to => to, :cc => cc, :bcc => bcc
+        bm.spawn "New Message", mode
+        mode.edit_message
+      when :poll
+        reporting_thread { PollManager.poll }
+      when :recall_draft
+        case Index.num_results_for :label => :draft
+        when 0
+          bm.flash "No draft messages."
+        when 1
+          m = nil
+          Index.each_id_by_date(:label => :draft) { |mid, builder| m = builder.call }
+          r = ResumeMode.new(m)
+          BufferManager.spawn "Edit message", r
+          r.edit_message
         else
-          bm.flash "Unknown key press '#{c.to_character}' for #{bm.focus_buf.mode.name}."
+          b = BufferManager.spawn_unless_exists("All drafts") do
+            mode = LabelSearchResultsMode.new [:draft]
+          end
+          b.mode.load_threads :num => b.content_height
         end
+      when :nothing
+      when :redraw
+        bm.completely_redraw_screen
+      else
+        bm.flash "Unknown key press '#{c.to_character}' for #{bm.focus_buf.mode.name}."
       end
     end
+
+    bm.draw_screen
   end
-  bm.kill_all_buffers
 rescue Exception => e
   $exception ||= e
 ensure
+  unless $opts[:no_threads]
+    PollManager.stop if PollManager.instantiated?
+    SuicideManager.stop if PollManager.instantiated?
+    Index.stop_lock_update_thread
+  end
+
   Redwood::finish
   stop_cursing
-end
+  Redwood::log "stopped cursing"
 
-Index.save unless $exception # TODO: think about this
+  if SuicideManager.instantiated? && SuicideManager.die?
+    Redwood::log "I've been ordered to commit sepuku. I obey!"
+  end
+
+  case $exception
+  when nil
+    Redwood::log "no fatal errors. good job, william."
+    Index.save
+  else
+    Redwood::log "oh crap, an exception"
+  end
+
+  Index.unlock
+end
 
 if $exception 
   $stderr.puts <<EOS
@@ -195,14 +308,14 @@ if $exception
 I'm very sorry, but it seems that an error occurred in Sup. 
 Please accept my sincere apologies. If you don't mind, please
 send the backtrace below and a brief report of the circumstances
-to wmorgan-sup at masanjin dot nets so that I might address this
+to sup-talk at rubyforge dot orgs so that I might address this
 problem. Thank you!
 
 Sincerely,
 William
 ----------------------------------------------------------------
 
-The problem was: #{$exception.message} (error type #{$exception.class.name})
+The problem was: '#{$exception.message}' (error type #{$exception.class.name})
 A backtrace follows:
 EOS
   raise $exception