]> git.cworth.org Git - sup/blobdiff - lib/sup/modes/thread-index-mode.rb
add #pipe-to-process to text-mode
[sup] / lib / sup / modes / thread-index-mode.rb
index f196c8f87daea08a697501ad2bc646a5a8e83775..542257e909f050aa1737e6702ab6bb886cec4b9b 100644 (file)
@@ -8,6 +8,12 @@ class ThreadIndexMode < LineCursorMode
   MIN_FROM_WIDTH = 15
   LOAD_MORE_THREAD_NUM = 20
 
+  HookManager.register "index-mode-size-widget", <<EOS
+Generates the per-thread size widget for each thread.
+Variables:
+  thread: The message thread to be formatted.
+EOS
+
   register_keymap do |k|
     k.add :load_threads, "Load #{LOAD_MORE_THREAD_NUM} more threads", 'M'
     k.add :reload, "Refresh view", '@'
@@ -23,23 +29,29 @@ class ThreadIndexMode < LineCursorMode
     k.add :jump_to_next_new, "Jump to next new thread", :tab
     k.add :reply, "Reply to latest message in a thread", 'r'
     k.add :forward, "Forward latest message in a thread", 'f'
-    k.add :toggle_tagged, "Tag/untag current line", 't'
+    k.add :toggle_tagged, "Tag/untag selected thread", 't'
+    k.add :toggle_tagged_all, "Tag/untag all threads", 'T'
+    k.add :tag_matching, "Tag matching threads", 'g'
     k.add :apply_to_tagged, "Apply next command to all tagged threads", ';'
   end
 
   def initialize hidden_labels=[], load_thread_opts={}
     super()
-    @mutex = Mutex.new
+    @mutex = Mutex.new # covers the following variables:
+    @threads = {}
+    @hidden_threads = {}
+    @size_widget_width = nil
+    @size_widgets = {}
+    @tags = Tagger.new self
+
+    ## these guys, and @text and @lines, are not covered
     @load_thread = nil
     @load_thread_opts = load_thread_opts
     @hidden_labels = hidden_labels + LabelManager::HIDDEN_RESERVED_LABELS
     @date_width = DATE_WIDTH
-    @size_width = nil
     
-    @tags = Tagger.new self
-    
-    initialize_threads
-    update
+    initialize_threads # defines @ts and @ts_mutex
+    update # defines @text and @lines
 
     UpdateManager.register self
 
@@ -54,7 +66,7 @@ class ThreadIndexMode < LineCursorMode
 
   def lines; @text.length; end
   def [] i; @text[i]; end
-  def contains_thread? t; !@lines[t].nil?; end
+  def contains_thread? t; @threads.include?(t) end
 
   def reload
     drop_all_threads
@@ -64,12 +76,17 @@ class ThreadIndexMode < LineCursorMode
 
   ## open up a thread view window
   def select t=nil
-    t ||= @threads[curpos] or return
-
-    ## TODO: don't regen text completely
-    Redwood::reporting_thread do
-      BufferManager.say("Loading message bodies...") do |sid|
-        t.each { |m, *o| m.load_from_source! if m }
+    t ||= cursor_thread or return
+
+    Redwood::reporting_thread("load messages for thread-view-mode") do
+      num = t.size
+      message = "Loading #{num.pluralize 'message body'}..."
+      BufferManager.say(message) do |sid|
+        t.each_with_index do |(m, *o), i|
+          next unless m
+          BufferManager.say "#{message} (#{i}/#{num})", sid if t.size > 1
+          m.load_from_source! 
+        end
       end
       mode = ThreadViewMode.new t, @hidden_labels
       BufferManager.spawn t.subj, mode
@@ -80,7 +97,7 @@ class ThreadIndexMode < LineCursorMode
       ## are set, and the second to show the cursor having moved
 
       update_text_for_line curpos
-      UpdateManager.relay self, :read, t
+      UpdateManager.relay self, :read, t.first
     end
   end
 
@@ -88,27 +105,31 @@ class ThreadIndexMode < LineCursorMode
     threads.each { |t| select t }
   end
   
-  def handle_label_update sender, m
-    t = @ts.thread_for(m) or return
-    handle_label_thread_update sender, t
+  def handle_single_message_labeled_update sender, m
+    ## no need to do anything different here; we don't differentiate 
+    ## messages from their containing threads
+    handle_labeled_update sender, m
+  end
+
+  def handle_labeled_update sender, m
+    if(t = thread_containing(m)) 
+      l = @lines[t] or return
+      update_text_for_line l
+    elsif is_relevant?(m)
+      add_or_unhide m
+    end
   end
 
-  def handle_label_thread_update sender, t
+  def handle_read_update sender, m
+    t = thread_containing(m) or return
     l = @lines[t] or return
     update_text_for_line l
-    BufferManager.draw_screen
-  end
-
-  def handle_read_update sender, t
-    l = @lines[t] or return
-    update_text_for_line @lines[t]
-    BufferManager.draw_screen
   end
 
   def handle_archived_update *a; handle_read_update(*a); end
 
-  def handle_deleted_update sender, t
-    handle_read_update sender, t
+  def handle_deleted_update sender, m
+    t = thread_containing(m) or return
     hide_thread t
     regen_text
   end
@@ -116,31 +137,36 @@ class ThreadIndexMode < LineCursorMode
   ## overwrite me!
   def is_relevant? m; false; end
 
-  def handle_add_update sender, m
-    if is_relevant?(m) || @ts.is_relevant?(m)
-      @ts.load_thread_for_message m
-      update
-      BufferManager.draw_screen
-    end
+  def handle_added_update sender, m
+    add_or_unhide m
+    BufferManager.draw_screen
   end
 
-  def handle_delete_update sender, mid
-    if @ts.contains_id? mid
-      @ts.remove mid
-      update
-      BufferManager.draw_screen
+  def handle_deleted_update sender, m
+    @ts_mutex.synchronize do
+      return unless @ts.contains? m
+      @ts.remove_id m.id
     end
+    update
+  end
+
+  def handle_undeleted_update sender, m
+    add_or_unhide m
   end
 
   def update
-    ## let's see you do THIS in python
-    @threads = @ts.threads.select { |t| !@hidden_threads[t] }.sort_by { |t| t.date }.reverse
-    @size_width = (@threads.max_of { |t| t.size } || 0).num_digits
+    @mutex.synchronize do
+      ## let's see you do THIS in python
+      @threads = @ts.threads.select { |t| !@hidden_threads[t] }.sort_by { |t| t.date }.reverse
+      @size_widgets = @threads.map { |t| size_widget_for_thread t }
+      @size_widget_width = @size_widgets.max_of { |w| w.length }
+    end
+
     regen_text
   end
 
   def edit_message
-    return unless(t = @threads[curpos])
+    return unless(t = cursor_thread)
     message, *crap = t.find { |m, *o| m.has_label? :draft }
     if message
       mode = ResumeMode.new message
@@ -153,15 +179,15 @@ class ThreadIndexMode < LineCursorMode
   def actually_toggle_starred t
     if t.has_label? :starred # if ANY message has a star
       t.remove_label :starred # remove from all
-      UpdateManager.relay self, :unstarred, t
+      UpdateManager.relay self, :unstarred, t.first
     else
       t.first.add_label :starred # add only to first
-      UpdateManager.relay self, :starred, t
+      UpdateManager.relay self, :starred, t.first
     end
   end  
 
   def toggle_starred 
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     actually_toggle_starred t
     update_text_for_line curpos
     cursor_down
@@ -175,35 +201,35 @@ class ThreadIndexMode < LineCursorMode
   def actually_toggle_archived t
     if t.has_label? :inbox
       t.remove_label :inbox
-      UpdateManager.relay self, :archived, t
+      UpdateManager.relay self, :archived, t.first
     else
       t.apply_label :inbox
-      UpdateManager.relay self, :unarchived, t
+      UpdateManager.relay self, :unarchived, t.first
     end
   end
 
   def actually_toggle_spammed t
     if t.has_label? :spam
       t.remove_label :spam
-      UpdateManager.relay self, :unspammed, t
+      UpdateManager.relay self, :unspammed, t.first
     else
       t.apply_label :spam
-      UpdateManager.relay self, :spammed, t
+      UpdateManager.relay self, :spammed, t.first
     end
   end
 
   def actually_toggle_deleted t
     if t.has_label? :deleted
       t.remove_label :deleted
-      UpdateManager.relay self, :undeleted, t
+      UpdateManager.relay self, :undeleted, t.first
     else
       t.apply_label :deleted
-      UpdateManager.relay self, :deleted, t
+      UpdateManager.relay self, :deleted, t.first
     end
   end
 
   def toggle_archived 
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     actually_toggle_archived t
     update_text_for_line curpos
   end
@@ -214,7 +240,7 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def toggle_new
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     t.toggle_label :unread
     update_text_for_line curpos
     cursor_down
@@ -226,12 +252,15 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def multi_toggle_tagged threads
-    @tags.drop_all_tags
+    @mutex.synchronize { @tags.drop_all_tags }
     regen_text
   end
 
   def jump_to_next_new
-    n = ((curpos + 1) ... lines).find { |i| @threads[i].has_label? :unread } || (0 ... curpos).find { |i| @threads[i].has_label? :unread }
+    n = @mutex.synchronize do
+      ((curpos + 1) ... lines).find { |i| @threads[i].has_label? :unread } ||
+        (0 ... curpos).find { |i| @threads[i].has_label? :unread }
+    end
     if n
       ## jump there if necessary
       jump_to_line n unless n >= topline && n < botline
@@ -242,7 +271,7 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def toggle_spam
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     multi_toggle_spam [t]
   end
 
@@ -262,7 +291,7 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def toggle_deleted
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     multi_toggle_deleted [t]
   end
 
@@ -276,7 +305,7 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def kill
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     multi_kill [t]
   end
 
@@ -286,11 +315,11 @@ class ThreadIndexMode < LineCursorMode
       hide_thread t
     end
     regen_text
-    BufferManager.flash "Thread#{threads.size == 1 ? '' : 's'} killed."
+    BufferManager.flash "#{threads.size.pluralize 'Thread'} killed."
   end
 
   def save
-    dirty_threads = (@threads + @hidden_threads.keys).select { |t| t.dirty? }
+    dirty_threads = @mutex.synchronize { (@threads + @hidden_threads.keys).select { |t| t.dirty? } }
     return if dirty_threads.empty?
 
     BufferManager.say("Saving threads...") do |say_id|
@@ -315,16 +344,29 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def toggle_tagged
-    t = @threads[curpos] or return
-    @tags.toggle_tag_for t
+    t = cursor_thread or return
+    @mutex.synchronize { @tags.toggle_tag_for t }
     update_text_for_line curpos
     cursor_down
   end
+  
+  def toggle_tagged_all
+    @mutex.synchronize { @threads.each { |t| @tags.toggle_tag_for t } }
+    regen_text
+  end
+
+  def tag_matching
+    query = BufferManager.ask :search, "tag threads matching: "
+    return if query.nil? || query.empty?
+    query = /#{query}/i
+    @mutex.synchronize { @threads.each { |t| @tags.tag t if thread_matches?(t, query) } }
+    regen_text
+  end
 
   def apply_to_tagged; @tags.apply_to_tagged; end
 
   def edit_labels
-    thread = @threads[curpos] or return
+    thread = cursor_thread or return
     speciall = (@hidden_labels + LabelManager::RESERVED_LABELS).uniq
     keepl, modifyl = thread.labels.partition { |t| speciall.member? t }
 
@@ -333,7 +375,7 @@ class ThreadIndexMode < LineCursorMode
     return unless user_labels
     thread.labels = keepl + user_labels
     user_labels.each { |l| LabelManager << l }
-    update_text_for_line curpos
+    UpdateManager.relay self, :labeled, thread.first
   end
 
   def multi_edit_labels threads
@@ -352,7 +394,7 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def reply
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     m = t.latest_message
     return if m.nil? # probably won't happen
     m.load_from_source!
@@ -361,31 +403,30 @@ class ThreadIndexMode < LineCursorMode
   end
 
   def forward
-    t = @threads[curpos] or return
+    t = cursor_thread or return
     m = t.latest_message
     return if m.nil? # probably won't happen
     m.load_from_source!
-    mode = ForwardMode.new m
-    BufferManager.spawn "Forward of #{m.subj}", mode
-    mode.edit_message
+    ForwardMode.spawn_nicely :message => m
   end
 
   def load_n_threads_background n=LOAD_MORE_THREAD_NUM, opts={}
     return if @load_thread # todo: wrap in mutex
-    @load_thread = Redwood::reporting_thread do
+    @load_thread = Redwood::reporting_thread("load threads for thread-index-mode") do
       num = load_n_threads n, opts
       opts[:when_done].call(num) if opts[:when_done]
       @load_thread = nil
     end
   end
 
+  ## TODO: figure out @ts_mutex in this method
   def load_n_threads n=LOAD_MORE_THREAD_NUM, opts={}
     @mbid = BufferManager.say "Searching for threads..."
     orig_size = @ts.size
-    last_update = Time.now - 9999 # oh yeah
+    last_update = Time.now
     @ts.load_n_threads(@ts.size + n, opts) do |i|
-      BufferManager.say "Loaded #{i} threads...", @mbid
       if (Time.now - last_update) >= 0.25
+        BufferManager.say "Loaded #{i.pluralize 'thread'}...", @mbid
         update
         BufferManager.draw_screen
         last_update = Time.now
@@ -399,7 +440,7 @@ class ThreadIndexMode < LineCursorMode
     BufferManager.draw_screen
     @ts.size - orig_size
   end
-  synchronized :load_n_threads
+  ignore_concurrent_calls :load_n_threads
 
   def status
     if (l = lines) == 0
@@ -415,7 +456,7 @@ class ThreadIndexMode < LineCursorMode
     myopts = @load_thread_opts.merge({ :when_done => (lambda do |num|
       opts[:when_done].call(num) if opts[:when_done]
       if num > 0
-        BufferManager.flash "Found #{num} threads."
+        BufferManager.flash "Found #{num.pluralize 'thread'}."
       else
         BufferManager.flash "No matches."
       end
@@ -427,6 +468,7 @@ class ThreadIndexMode < LineCursorMode
       load_n_threads n, myopts
     end
   end
+  ignore_concurrent_calls :load_threads
 
   def resize rows, cols
     regen_text
@@ -435,7 +477,33 @@ class ThreadIndexMode < LineCursorMode
 
 protected
 
-  def cursor_thread; @threads[curpos]; end
+  def add_or_unhide m
+    if @hidden_threads[m]
+      @hidden_threads.delete m
+      ## now it will re-appear when #update is called
+    else
+      @ts_mutex.synchronize do
+        return unless is_relevant?(m) || @ts.is_relevant?(m)
+        @ts.load_thread_for_message m
+      end
+    end
+
+    update
+  end
+
+  def thread_containing m; @ts_mutex.synchronize { @ts.thread_for m } end
+
+  ## used to tag threads by query. this can be made a lot more sophisticated,
+  ## but for right now we'll do the obvious this.
+  def thread_matches? t, query
+    t.subj =~ query || t.snippet =~ query || t.participants.any? { |x| x.longname =~ query }
+  end
+
+  def size_widget_for_thread t
+    HookManager.run("index-mode-size-widget", :thread => t) || default_size_widget_for(t)
+  end
+
+  def cursor_thread; @mutex.synchronize { @threads[curpos] }; end
 
   def drop_all_threads
     @tags.drop_all_tags
@@ -444,30 +512,40 @@ protected
   end
 
   def hide_thread t
-    raise "already hidden" if @hidden_threads[t]
-    @hidden_threads[t] = true
-    @threads.delete t
-    @tags.drop_tag_for t
-  end
-
-  def show_thread t
-    if @hidden_threads[t]
-      @hidden_threads.delete t
-    else
-      @ts.add_thread t
+    @mutex.synchronize do
+      i = @threads.index(t) or return
+      raise "already hidden" if @hidden_threads[t]
+      @hidden_threads[t] = true
+      @threads.delete_at i
+      @size_widgets.delete_at i
+      @tags.drop_tag_for t
     end
-    update
   end
 
   def update_text_for_line l
     return unless l # not sure why this happens, but it does, occasionally
-    @text[l] = text_for_thread @threads[l]
-    buffer.mark_dirty if buffer
+    
+    need_update = false
+
+    @mutex.synchronize do
+      @size_widgets[l] = size_widget_for_thread @threads[l]
+
+      ## if the widget size has increased, we need to redraw everyone
+      need_update = @size_widgets[l].size > @size_widget_width
+    end
+
+    if need_update
+      update
+    else
+      @text[l] = text_for_thread_at l
+      buffer.mark_dirty if buffer
+    end
   end
 
   def regen_text
-    @text = @threads.map_with_index { |t, i| text_for_thread t }
-    @lines = @threads.map_with_index { |t, i| [t, i] }.to_h
+    threads = @mutex.synchronize { @threads }
+    @text = threads.map_with_index { |t, i| text_for_thread_at i }
+    @lines = threads.map_with_index { |t, i| [t, i] }.to_h
     buffer.mark_dirty if buffer
   end
   
@@ -494,7 +572,9 @@ protected
     authors.compact.uniq.map { |a| [a, new[a]] }
   end
 
-  def text_for_thread t
+  def text_for_thread_at line
+    t, size_widget = @mutex.synchronize { [@threads[line], @size_widgets[line]] }
+
     date = t.date.to_nice_s
 
     new = t.has_label?(:unread)
@@ -544,6 +624,8 @@ protected
 
     snippet = t.snippet + (t.snippet.empty? ? "" : "...")
 
+    size_widget_text = sprintf "%#{ @size_widget_width}s", size_widget
+
     [ 
       [:tagged_color, @tags.tagged?(t) ? ">" : " "],
       [:none, sprintf("%#{@date_width}s", date)],
@@ -551,7 +633,7 @@ protected
     ] +
       from +
       [
-      [:none, t.size == 1 ? " " * (@size_width + 2) : sprintf("(%#{@size_width}d)", t.size)],
+      [subj_color, size_widget_text],
       [:to_me_color, dp ? " >" : (p ? ' +' : "  ")],
       [subj_color, t.subj + (t.subj.empty? ? "" : " ")],
     ] +
@@ -561,10 +643,19 @@ protected
 
   end
 
-  def dirty?; (@hidden_threads.keys + @threads).any? { |t| t.dirty? }; end
+  def dirty?; @mutex.synchronize { (@hidden_threads.keys + @threads).any? { |t| t.dirty? } } end
 
 private
 
+  def default_size_widget_for t
+    case t.size
+    when 1
+      ""
+    else
+      "(#{t.size})"
+    end
+  end
+
   def from_width
     [(buffer.content_width.to_f * 0.2).to_i, MIN_FROM_WIDTH].max
   end