3 ## subclasses should implement:
6 class ThreadIndexMode < LineCursorMode
7 include CanSpawnForwardMode
9 DATE_WIDTH = Time::TO_NICE_S_MAX_LEN
11 LOAD_MORE_THREAD_NUM = 20
13 HookManager.register "index-mode-size-widget", <<EOS
14 Generates the per-thread size widget for each thread.
16 thread: The message thread to be formatted.
19 register_keymap do |k|
20 k.add :load_threads, "Load #{LOAD_MORE_THREAD_NUM} more threads", 'M'
21 k.add :reload, "Refresh view", '@'
22 k.add :toggle_archived, "Toggle archived status", 'a'
23 k.add :toggle_starred, "Star or unstar all messages in thread", '*'
24 k.add :toggle_new, "Toggle new/read status of all messages in thread", 'N'
25 k.add :edit_labels, "Edit or add labels for a thread", 'l'
26 k.add :edit_message, "Edit message (drafts only)", 'e'
27 k.add :toggle_spam, "Mark/unmark thread as spam", 'S'
28 k.add :toggle_deleted, "Delete/undelete thread", 'd'
29 k.add :kill, "Kill thread (never to be seen in inbox again)", '&'
30 k.add :save, "Save changes now", '$'
31 k.add :jump_to_next_new, "Jump to next new thread", :tab
32 k.add :reply, "Reply to latest message in a thread", 'r'
33 k.add :forward, "Forward latest message in a thread", 'f'
34 k.add :toggle_tagged, "Tag/untag selected thread", 't'
35 k.add :toggle_tagged_all, "Tag/untag all threads", 'T'
36 k.add :apply_to_tagged, "Apply next command to all tagged threads", ';'
39 def initialize hidden_labels=[], load_thread_opts={}
43 @load_thread_opts = load_thread_opts
44 @hidden_labels = hidden_labels + LabelManager::HIDDEN_RESERVED_LABELS
45 @date_width = DATE_WIDTH
46 @size_widget_width = nil
48 @tags = Tagger.new self
53 UpdateManager.register self
55 @last_load_more_size = nil
56 to_load_more do |size|
57 next if @last_load_more_size == 0
58 load_threads :num => 1, :background => false
59 load_threads :num => (size - 1),
60 :when_done => lambda { |num| @last_load_more_size = num }
64 def lines; @text.length; end
65 def [] i; @text[i]; end
66 def contains_thread? t; !@lines[t].nil?; end
70 BufferManager.draw_screen
71 load_threads :num => buffer.content_height
74 ## open up a thread view window
76 t ||= @threads[curpos] or return
78 ## TODO: don't regen text completely
79 Redwood::reporting_thread do
80 BufferManager.say("Loading message bodies...") do |sid|
81 t.each { |m, *o| m.load_from_source! if m }
83 mode = ThreadViewMode.new t, @hidden_labels
84 BufferManager.spawn t.subj, mode
85 BufferManager.draw_screen
86 mode.jump_to_first_open
87 BufferManager.draw_screen # lame TODO: make this unnecessary
88 ## the first draw_screen is needed before topline and botline
89 ## are set, and the second to show the cursor having moved
91 update_text_for_line curpos
92 UpdateManager.relay self, :read, t
96 def multi_select threads
97 threads.each { |t| select t }
100 def handle_label_update sender, m
101 t = @ts.thread_for(m) or return
102 handle_label_thread_update sender, t
105 def handle_label_thread_update sender, t
106 l = @lines[t] or return
107 update_text_for_line l
108 BufferManager.draw_screen
111 def handle_read_update sender, t
112 l = @lines[t] or return
113 update_text_for_line @lines[t]
114 BufferManager.draw_screen
117 def handle_archived_update *a; handle_read_update(*a); end
119 def handle_deleted_update sender, t
120 handle_read_update sender, t
126 def is_relevant? m; false; end
128 def handle_add_update sender, m
129 if is_relevant?(m) || @ts.is_relevant?(m)
130 @ts.load_thread_for_message m
132 BufferManager.draw_screen
136 def handle_delete_update sender, mid
137 if @ts.contains_id? mid
140 BufferManager.draw_screen
145 ## let's see you do THIS in python
146 @threads = @ts.threads.select { |t| !@hidden_threads[t] }.sort_by { |t| t.date }.reverse
147 @size_widgets = @threads.map { |t| size_widget_for_thread t }
148 @size_widget_width = @size_widgets.max_of { |w| w.length }
154 return unless(t = @threads[curpos])
155 message, *crap = t.find { |m, *o| m.has_label? :draft }
157 mode = ResumeMode.new message
158 BufferManager.spawn "Edit message", mode
160 BufferManager.flash "Not a draft message!"
164 def actually_toggle_starred t
165 if t.has_label? :starred # if ANY message has a star
166 t.remove_label :starred # remove from all
167 UpdateManager.relay self, :unstarred, t
169 t.first.add_label :starred # add only to first
170 UpdateManager.relay self, :starred, t
175 t = @threads[curpos] or return
176 actually_toggle_starred t
177 update_text_for_line curpos
181 def multi_toggle_starred threads
182 threads.each { |t| actually_toggle_starred t }
186 def actually_toggle_archived t
187 if t.has_label? :inbox
188 t.remove_label :inbox
189 UpdateManager.relay self, :archived, t
192 UpdateManager.relay self, :unarchived, t
196 def actually_toggle_spammed t
197 if t.has_label? :spam
199 UpdateManager.relay self, :unspammed, t
202 UpdateManager.relay self, :spammed, t
206 def actually_toggle_deleted t
207 if t.has_label? :deleted
208 t.remove_label :deleted
209 UpdateManager.relay self, :undeleted, t
211 t.apply_label :deleted
212 UpdateManager.relay self, :deleted, t
217 t = @threads[curpos] or return
218 actually_toggle_archived t
219 update_text_for_line curpos
222 def multi_toggle_archived threads
223 threads.each { |t| actually_toggle_archived t }
228 t = @threads[curpos] or return
229 t.toggle_label :unread
230 update_text_for_line curpos
234 def multi_toggle_new threads
235 threads.each { |t| t.toggle_label :unread }
239 def multi_toggle_tagged threads
245 n = ((curpos + 1) ... lines).find { |i| @threads[i].has_label? :unread } || (0 ... curpos).find { |i| @threads[i].has_label? :unread }
247 ## jump there if necessary
248 jump_to_line n unless n >= topline && n < botline
251 BufferManager.flash "No new messages"
256 t = @threads[curpos] or return
257 multi_toggle_spam [t]
260 ## both spam and deleted have the curious characteristic that you
261 ## always want to hide the thread after either applying or removing
262 ## that label. in all thread-index-views except for
263 ## label-search-results-mode, when you mark a message as spam or
264 ## deleted, you want it to disappear immediately; in LSRM, you only
265 ## see deleted or spam emails, and when you undelete or unspam them
266 ## you also want them to disappear immediately.
267 def multi_toggle_spam threads
269 actually_toggle_spammed t
276 t = @threads[curpos] or return
277 multi_toggle_deleted [t]
280 ## see comment for multi_toggle_spam
281 def multi_toggle_deleted threads
283 actually_toggle_deleted t
290 t = @threads[curpos] or return
294 def multi_kill threads
296 t.apply_label :killed
300 BufferManager.flash "Thread#{threads.size == 1 ? '' : 's'} killed."
304 dirty_threads = (@threads + @hidden_threads.keys).select { |t| t.dirty? }
305 return if dirty_threads.empty?
307 BufferManager.say("Saving threads...") do |say_id|
308 dirty_threads.each_with_index do |t, i|
309 BufferManager.say "Saving modified thread #{i + 1} of #{dirty_threads.length}...", say_id
316 UpdateManager.unregister self
320 BufferManager.clear @mbid if @mbid
321 sleep 0.1 # TODO: necessary?
322 BufferManager.erase_flash
329 t = @threads[curpos] or return
330 @tags.toggle_tag_for t
331 update_text_for_line curpos
335 def toggle_tagged_all
336 @threads.each { |t| @tags.toggle_tag_for t }
340 def apply_to_tagged; @tags.apply_to_tagged; end
343 thread = @threads[curpos] or return
344 speciall = (@hidden_labels + LabelManager::RESERVED_LABELS).uniq
345 keepl, modifyl = thread.labels.partition { |t| speciall.member? t }
347 user_labels = BufferManager.ask_for_labels :label, "Labels for thread: ", modifyl, @hidden_labels
349 return unless user_labels
350 thread.labels = keepl + user_labels
351 user_labels.each { |l| LabelManager << l }
352 update_text_for_line curpos
355 def multi_edit_labels threads
356 answer = BufferManager.ask :add_labels, "add labels: "
358 user_labels = answer.split(/\s+/).map { |l| l.intern }
360 hl = user_labels.select { |l| @hidden_labels.member? l }
362 threads.each { |t| user_labels.each { |l| t.apply_label l } }
363 user_labels.each { |l| LabelManager << l }
365 BufferManager.flash "'#{hl}' is a reserved label!"
371 t = @threads[curpos] or return
373 return if m.nil? # probably won't happen
375 mode = ReplyMode.new m
376 BufferManager.spawn "Reply to #{m.subj}", mode
380 t = @threads[curpos] or return
382 return if m.nil? # probably won't happen
387 def load_n_threads_background n=LOAD_MORE_THREAD_NUM, opts={}
388 return if @load_thread # todo: wrap in mutex
389 @load_thread = Redwood::reporting_thread do
390 num = load_n_threads n, opts
391 opts[:when_done].call(num) if opts[:when_done]
396 def load_n_threads n=LOAD_MORE_THREAD_NUM, opts={}
397 @mbid = BufferManager.say "Searching for threads..."
399 last_update = Time.now - 9999 # oh yeah
400 @ts.load_n_threads(@ts.size + n, opts) do |i|
401 BufferManager.say "Loaded #{i} threads...", @mbid
402 if (Time.now - last_update) >= 0.25
404 BufferManager.draw_screen
405 last_update = Time.now
408 @ts.threads.each { |th| th.labels.each { |l| LabelManager << l } }
411 BufferManager.clear @mbid
413 BufferManager.draw_screen
416 synchronized :load_n_threads
422 "line #{curpos + 1} of #{l} #{dirty? ? '*modified*' : ''}"
426 def load_threads opts={}
427 n = opts[:num] || ThreadIndexMode::LOAD_MORE_THREAD_NUM
429 myopts = @load_thread_opts.merge({ :when_done => (lambda do |num|
430 opts[:when_done].call(num) if opts[:when_done]
432 BufferManager.flash "Found #{num} threads."
434 BufferManager.flash "No matches."
438 if opts[:background] || opts[:background].nil?
439 load_n_threads_background n, myopts
441 load_n_threads n, myopts
445 def resize rows, cols
452 def size_widget_for_thread t
453 HookManager.run("index-mode-size-widget", :thread => t) || default_size_widget_for(t)
456 def cursor_thread; @threads[curpos]; end
465 raise "already hidden" if @hidden_threads[t]
466 @hidden_threads[t] = true
472 if @hidden_threads[t]
473 @hidden_threads.delete t
480 def update_text_for_line l
481 return unless l # not sure why this happens, but it does, occasionally
482 @size_widgets[l] = size_widget_for_thread @threads[l]
484 ## if the widget size has increased, we need to redraw everyone
485 if @size_widgets[l].size > @size_widget_width
488 @text[l] = text_for_thread_at l
489 buffer.mark_dirty if buffer
494 @text = @threads.map_with_index { |t, i| text_for_thread_at i }
495 @lines = @threads.map_with_index { |t, i| [t, i] }.to_h
496 buffer.mark_dirty if buffer
499 def authors; map { |m, *o| m.from if m }.compact.uniq; end
501 def author_names_and_newness_for_thread t
503 authors = t.map do |m, *o|
507 if AccountManager.is_account?(m.from)
509 elsif t.authors.size == 1
515 new[name] ||= m.has_label?(:unread)
519 authors.compact.uniq.map { |a| [a, new[a]] }
522 def text_for_thread_at line
524 size_widget = @size_widgets[line]
526 date = t.date.to_nice_s
528 new = t.has_label?(:unread)
529 starred = t.has_label?(:starred)
531 ## format the from column
533 ann = author_names_and_newness_for_thread t
535 ann.each_with_index do |(name, newness), i|
536 break if cur_width >= from_width
537 last = i == ann.length - 1
540 if cur_width + name.length > from_width
541 name[0 ... (from_width - cur_width - 1)] + "."
542 elsif cur_width + name.length == from_width
543 name[0 ... (from_width - cur_width)]
546 name[0 ... (from_width - cur_width)]
548 name[0 ... (from_width - cur_width - 1)] + ","
552 cur_width += abbrev.length
554 if last && from_width > cur_width
555 abbrev += " " * (from_width - cur_width)
558 from << [(newness ? :index_new_color : (starred ? :index_starred_color : :index_old_color)), abbrev]
561 dp = t.direct_participants.any? { |p| AccountManager.is_account? p }
562 p = dp || t.participants.any? { |p| AccountManager.is_account? p }
573 snippet = t.snippet + (t.snippet.empty? ? "" : "...")
575 size_widget_text = sprintf "%#{ @size_widget_width}s", size_widget
578 [:tagged_color, @tags.tagged?(t) ? ">" : " "],
579 [:none, sprintf("%#{@date_width}s", date)],
580 (starred ? [:starred_color, "*"] : [:none, " "]),
584 [subj_color, size_widget_text],
585 [:to_me_color, dp ? " >" : (p ? ' +' : " ")],
586 [subj_color, t.subj + (t.subj.empty? ? "" : " ")],
588 (t.labels - @hidden_labels).map { |label| [:label_color, "+#{label} "] } +
589 [[:snippet_color, snippet]
594 def dirty?; (@hidden_threads.keys + @threads).any? { |t| t.dirty? }; end
598 def default_size_widget_for t
608 [(buffer.content_width.to_f * 0.2).to_i, MIN_FROM_WIDTH].max
611 def initialize_threads
612 @ts = ThreadSet.new Index.instance, $config[:thread_by_subject]
613 @ts_mutex = Mutex.new