module Redwood
class ThreadViewMode < LineCursorMode
+ ## this holds all info we need to lay out a message
+ class MessageLayout
+ attr_accessor :top, :bot, :prev, :next, :depth, :width, :state, :color, :star_color, :orig_new
+ end
+
+ class ChunkLayout
+ attr_accessor :state
+ end
+
DATE_FORMAT = "%B %e %Y %l:%M%P"
+ INDENT_SPACES = 2 # how many spaces to indent child messages
register_keymap do |k|
- k.add :toggle_detailed_header, "Toggle detailed header", 'd'
+ k.add :toggle_detailed_header, "Toggle detailed header", 'h'
k.add :show_header, "Show full message header", 'H'
- k.add :toggle_expanded, "Expand/collapse item", :enter
+ k.add :activate_chunk, "Expand/collapse or activate item", :enter
k.add :expand_all_messages, "Expand/collapse all messages", 'E'
- k.add :edit_message, "Edit message (drafts only)", 'e'
+ k.add :edit_draft, "Edit draft", 'e'
+ k.add :edit_labels, "Edit or add labels for a thread", 'l'
k.add :expand_all_quotes, "Expand/collapse all quotes in a message", 'o'
k.add :jump_to_next_open, "Jump to next open message", 'n'
k.add :jump_to_prev_open, "Jump to previous open message", 'p'
k.add :toggle_starred, "Star or unstar message", '*'
- k.add :collapse_non_new_messages, "Collapse all but new messages", 'N'
+ k.add :toggle_new, "Toggle new/read status of message", 'N'
+# k.add :collapse_non_new_messages, "Collapse all but new messages", 'N'
k.add :reply, "Reply to a message", 'r'
k.add :forward, "Forward a message", 'f'
+ k.add :alias, "Edit alias/nickname for a person", 'i'
+ k.add :edit_as_new, "Edit message as new", 'D'
k.add :save_to_disk, "Save message/attachment to disk", 's'
+ k.add :search, "Search for messages from particular people", 'S'
+ k.add :compose, "Compose message to person", 'm'
+ k.add :archive_and_kill, "Archive thread and kill buffer", 'a'
+ k.add :delete_and_kill, "Delete thread and kill buffer", 'd'
end
+ ## there are a couple important instance variables we hold to format
+ ## the thread and to provide line-based functionality. @layout is a
+ ## map from Messages to MessageLayouts, and @chunk_layout from
+ ## Chunks to ChunkLayouts. @message_lines is a map from row #s to
+ ## Message objects. @chunk_lines is a map from row #s to Chunk
+ ## objects. @person_lines is a map from row #s to Person objects.
+
def initialize thread, hidden_labels=[]
super()
@thread = thread
- @state = {}
@hidden_labels = hidden_labels
+ @layout = SavingHash.new { MessageLayout.new }
+ @chunk_layout = SavingHash.new { ChunkLayout.new }
earliest, latest = nil, nil
latest_date = nil
+ altcolor = false
+
@thread.each do |m, d, p|
next unless m
earliest ||= m
- @state[m] = initial_state_for m
+ @layout[m].state = initial_state_for m
+ @layout[m].color = altcolor ? :alternate_patina_color : :message_patina_color
+ @layout[m].star_color = altcolor ? :alternate_starred_patina_color : :starred_patina_color
+ @layout[m].orig_new = m.has_label? :read
+ altcolor = !altcolor
if latest_date.nil? || m.date > latest_date
latest_date = m.date
latest = m
end
end
- @state[latest] = :open if @state[latest] == :closed
- @state[earliest] = :detailed if earliest.has_label?(:unread) || @thread.size == 1
+ @layout[latest].state = :open if @layout[latest].state == :closed
+ @layout[earliest].state = :detailed if earliest.has_label?(:unread) || @thread.size == 1
- BufferManager.say "Loading message bodies..." do
- regen_chunks
- regen_text
- end
+ @thread.remove_label :unread
+ regen_text
end
def draw_line ln, opts={}
def [] i; @text[i]; end
def show_header
- return unless(m = @message_lines[curpos])
+ m = @message_lines[curpos] or return
BufferManager.spawn_unless_exists("Full header") do
TextMode.new m.raw_header
end
end
def toggle_detailed_header
- return unless(m = @message_lines[curpos])
- @state[m] = (@state[m] == :detailed ? :open : :detailed)
+ m = @message_lines[curpos] or return
+ @layout[m].state = (@layout[m].state == :detailed ? :open : :detailed)
update
end
def reply
- return unless(m = @message_lines[curpos])
+ m = @message_lines[curpos] or return
mode = ReplyMode.new m
BufferManager.spawn "Reply to #{m.subj}", mode
end
def forward
- return unless(m = @message_lines[curpos])
+ m = @message_lines[curpos] or return
mode = ForwardMode.new m
BufferManager.spawn "Forward of #{m.subj}", mode
- mode.edit
+ mode.edit_message
+ end
+
+ include CanAliasContacts
+ def alias
+ p = @person_lines[curpos] or return
+ alias_contact p
+ update
+ end
+
+ def search
+ p = @person_lines[curpos] or return
+ mode = PersonSearchResultsMode.new [p]
+ BufferManager.spawn "Search for #{p.name}", mode
+ mode.load_threads :num => mode.buffer.content_height
+ end
+
+ def compose
+ p = @person_lines[curpos]
+ mode =
+ if p
+ ComposeMode.new :to => [p]
+ else
+ ComposeMode.new
+ end
+ BufferManager.spawn "Compose message", mode
+ mode.edit_message
+ end
+
+ def edit_labels
+ m = @message_lines[curpos] or return
+ new_labels = BufferManager.ask_for_labels :label, "Labels for message: ", m.labels
+
+ return unless new_labels
+ m.labels = new_labels
+ new_labels.each { |l| LabelManager << l }
+ ## TODO: don't recalculate EVERYTHING
+ update
+ UpdateManager.relay self, :label, m
end
def toggle_starred
- return unless(m = @message_lines[curpos])
- if m.has_label? :starred
- m.remove_label :starred
+ m = @message_lines[curpos] or return
+ toggle_label m, :starred
+ end
+
+ def toggle_new
+ m = @message_lines[curpos] or return
+ toggle_label m, :unread
+ end
+
+ def toggle_label m, label
+ if m.has_label? label
+ m.remove_label label
else
- m.add_label :starred
+ m.add_label label
end
## TODO: don't recalculate EVERYTHING just to add a stupid little
## star to the display
update
- UpdateManager.relay :starred, m
+ UpdateManager.relay self, :label, m
end
- def toggle_expanded
- return unless(chunk = @chunk_lines[curpos])
- case chunk
- when Message, Message::Quote, Message::Signature
- @state[chunk] = (@state[chunk] != :closed ? :closed : :open)
- cursor_down if @state[chunk] == :closed
- when Message::Attachment
- view_attachment chunk
+ ## called when someone presses enter when the cursor is highlighting
+ ## a chunk. for expandable chunks (including messages) we toggle
+ ## open/closed state; for viewable chunks (like attachments) we
+ ## view.
+ def activate_chunk
+ chunk = @chunk_lines[curpos] or return
+ layout =
+ if chunk.is_a?(Message)
+ @layout[chunk]
+ elsif chunk.expandable?
+ @chunk_layout[chunk]
+ end
+ if layout
+ layout.state = (layout.state != :closed ? :closed : :open)
+ #cursor_down if layout.state == :closed # too annoying
+ update
+ elsif chunk.viewable?
+ view chunk
end
- update
end
- def save fn
- if File.exists? fn
- return unless BufferManager.ask_yes_or_no "File exists. Overwrite?"
- end
- begin
- File.open(fn, "w") { |f| yield f }
- BufferManager.flash "Successfully wrote #{fn}."
- rescue SystemCallError => e
- BufferManager.flash "Error writing to file: #{e.message}"
- end
+ def edit_as_new
+ m = @message_lines[curpos] or return
+ mode = ComposeMode.new(:body => m.basic_body_lines, :to => m.to, :cc => m.cc, :subj => m.subj, :bcc => m.bcc)
+ BufferManager.spawn "edit as new", mode
+ mode.edit_message
end
- private :save
def save_to_disk
- return unless(chunk = @chunk_lines[curpos])
+ chunk = @chunk_lines[curpos] or return
case chunk
- when Message::Attachment
- fn = BufferManager.ask :filename, "save attachment to file: ", chunk.filename
- save(fn) { |f| f.print chunk } if fn
+ when Chunk::Attachment
+ fn = BufferManager.ask_for_filename :filename, "Save attachment to file: ", chunk.filename
+ save_to_file(fn) { |f| f.print chunk.raw_content } if fn
else
m = @message_lines[curpos]
- fn = BufferManager.ask :filename, "save message to file: "
- save(fn) { |f| f.print m.raw_full_message } if fn
+ fn = BufferManager.ask_for_filename :filename, "Save message to file: "
+ return unless fn
+ save_to_file(fn) do |f|
+ m.each_raw_message_line { |l| f.print l }
+ end
end
end
- def edit_message
- return unless(m = @message_lines[curpos])
+ def edit_draft
+ m = @message_lines[curpos] or return
if m.is_draft?
mode = ResumeMode.new m
BufferManager.spawn "Edit message", mode
- mode.edit
+ BufferManager.kill_buffer self.buffer
+ mode.edit_message
else
BufferManager.flash "Not a draft message!"
end
end
+ def jump_to_first_open
+ m = @message_lines[0] or return
+ if @layout[m].state != :closed
+ jump_to_message m
+ else
+ jump_to_next_open
+ end
+ end
+
def jump_to_next_open
- return unless(m = @message_lines[curpos])
- while nextm = @messages[m][3]
- break if @state[nextm] == :open
+ m = @message_lines[curpos] or return
+ while nextm = @layout[m].next
+ break if @layout[nextm].state != :closed
m = nextm
end
jump_to_message nextm if nextm
end
def jump_to_prev_open
- return unless(m = @message_lines[curpos])
+ m = @message_lines[curpos] or return
## jump to the top of the current message if we're in the body;
## otherwise, to the previous message
- top = @messages[m][0]
+
+ top = @layout[m].top
if curpos == top
- while(prevm = @messages[m][2])
- break if @state[prevm] != :closed
+ while(prevm = @layout[m].prev)
+ break if @layout[prevm].state != :closed
m = prevm
end
jump_to_message prevm if prevm
end
def jump_to_message m
- top, bot, prevm, nextm, depth = @messages[m]
- jump_to_line top unless top >= topline &&
- top <= botline && bot >= topline && bot <= botline
- jump_to_col depth * 2 # sorry!!!!
- set_cursor_pos top
+ l = @layout[m]
+ left = l.depth * INDENT_SPACES
+ right = left + l.width
+
+ ## jump to the top line unless both top and bottom fit in the current view
+ jump_to_line l.top unless l.top >= topline && l.top <= botline && l.bot >= topline && l.bot <= botline
+
+ ## jump to the left columns unless both left and right fit in the current view
+ jump_to_col left unless left >= leftcol && left <= rightcol && right >= leftcol && right <= rightcol
+
+ ## either way, move the cursor to the first line
+ set_cursor_pos l.top
end
def expand_all_messages
@global_message_state ||= :closed
@global_message_state = (@global_message_state == :closed ? :open : :closed)
- @state.each { |m, v| @state[m] = @global_message_state if m.is_a? Message }
+ @layout.each { |m, l| l.state = @global_message_state }
update
end
-
def collapse_non_new_messages
- @messages.each { |m, v| @state[m] = m.has_label?(:unread) ? :open : :closed }
+ @layout.each { |m, l| l.state = l.orig_new ? :open : :closed }
update
end
def expand_all_quotes
if(m = @message_lines[curpos])
- quotes = @chunks[m].select { |c| c.is_a?(Message::Quote) || c.is_a?(Message::Signature) }
- open, closed = quotes.partition { |c| @state[c] == :open }
- newstate = open.length > closed.length ? :closed : :open
- Redwood::log "#{open.length} opened, #{closed.length} closed, new state is thus #{newstate}"
- quotes.each { |c| @state[c] = newstate }
+ quotes = m.chunks.select { |c| (c.is_a?(Chunk::Quote) || c.is_a?(Chunk::Signature)) && c.lines.length > 1 }
+ numopen = quotes.inject(0) { |s, c| s + (@chunk_layout[c].state == :open ? 1 : 0) }
+ newstate = numopen > quotes.length / 2 ? :closed : :open
+ quotes.each { |c| @chunk_layout[c].state = newstate }
update
end
end
- ## kinda slow for large threads. TODO: make faster
def cleanup
- BufferManager.say "Marking messages as read..." do
- @thread.each do |m, d, p|
- if m && m.has_label?(:unread)
- m.remove_label :unread
- UpdateManager.relay :read, m
- end
- end
- end
- @messages = @chunks = @text = nil
+ @layout = @chunk_layout = @text = nil # for good luck
+ end
+
+ def archive_and_kill
+ @thread.remove_label :inbox
+ UpdateManager.relay self, :archived, @thread
+ BufferManager.kill_buffer_safely buffer
end
-private
+ def delete_and_kill
+ @thread.apply_label :deleted
+ UpdateManager.relay self, :deleted, @thread
+ BufferManager.kill_buffer_safely buffer
+ end
+
+private
def initial_state_for m
if m.has_label?(:starred) || m.has_label?(:unread)
buffer.mark_dirty if buffer
end
- def regen_chunks
- @chunks = {}
- @thread.each { |m, d, p| @chunks[m] = m.to_chunks if m.is_a?(Message) }
- end
-
+ ## here we generate the actual content lines. we accumulate
+ ## everything into @text, and we set @chunk_lines and
+ ## @message_lines, and we update @layout.
def regen_text
@text = []
@chunk_lines = []
@message_lines = []
- @messages = {}
+ @person_lines = []
- prev_m = nil
+ prevm = nil
@thread.each do |m, depth, parent|
- ## we're occasionally called on @threads that have had messages
- ## added to them since initialization. luckily we regen_text on
- ## the entire thread every time the user does anything besides
- ## scrolling (basically), so we can just slap this on here.
- ##
- ## to pick nits, the niceness that i do in the constructor with
- ## 'latest' might not be valid, but i don't see that as a huge
- ## issue.
- @state[m] ||= initial_state_for m if m
-
- text = chunk_to_lines m, @state[m], @text.length, depth, parent
+ unless m.is_a? Message # handle nil and :fake_root
+ @text += chunk_to_lines m, nil, @text.length, depth, parent
+ next
+ end
+ l = @layout[m]
+
+ ## is this still necessary?
+ next unless @layout[m].state # skip discarded drafts
+
+ ## build the patina
+ text = chunk_to_lines m, l.state, @text.length, depth, parent, l.color, l.star_color
+
+ l.top = @text.length
+ l.bot = @text.length + text.length # updated below
+ l.prev = prevm
+ l.next = nil
+ l.depth = depth
+ # l.state we preserve
+ l.width = 0 # updated below
+ @layout[l.prev].next = m if l.prev
+
(0 ... text.length).each do |i|
@chunk_lines[@text.length + i] = m
@message_lines[@text.length + i] = m
+ lw = text[i].flatten.select { |x| x.is_a? String }.map { |x| x.length }.sum
end
- ## sorry i store all this shit in an array. very, very sorry.
- ## also sorry about the * 2. very, very sorry.
- @messages[m] = [@text.length, @text.length + text.length, prev_m, nil, depth]
- @messages[prev_m][3] = m if prev_m
- prev_m = m
-
@text += text
- if @state[m] != :closed && @chunks.member?(m)
- @chunks[m].each do |c|
- @state[c] ||= :closed
- text = chunk_to_lines c, @state[c], @text.length, depth
+ prevm = m
+ if l.state != :closed
+ m.chunks.each do |c|
+ cl = @chunk_layout[c]
+
+ ## set the default state for chunks
+ cl.state ||=
+ if c.is_a?(Chunk::Attachment) && c.expandable?
+ :open
+ else
+ :closed
+ end
+
+ text = chunk_to_lines c, cl.state, @text.length, depth
(0 ... text.length).each do |i|
@chunk_lines[@text.length + i] = c
@message_lines[@text.length + i] = m
+ lw = text[i].flatten.select { |x| x.is_a? String }.map { |x| x.length }.sum - (depth * INDENT_SPACES)
+ l.width = lw if lw > l.width
end
@text += text
end
- @messages[m][1] = @text.length
+ @layout[m].bot = @text.length
end
end
end
- def message_patina_lines m, state, parent, prefix
- prefix_widget = [:message_patina_color, prefix]
- widget =
- case state
- when :closed
- [:message_patina_color, "+ "]
- when :open, :detailed
- [:message_patina_color, "- "]
- end
- imp_widget =
+ def message_patina_lines m, state, start, parent, prefix, color, star_color
+ prefix_widget = [color, prefix]
+
+ open_widget = [color, (state == :closed ? "+ " : "- ")]
+ new_widget = [color, (m.has_label?(:unread) ? "N" : " ")]
+ starred_widget =
if m.has_label?(:starred)
- [:starred_patina_color, "* "]
+ [star_color, "* "]
else
- [:message_patina_color, " "]
+ [color, " "]
end
case state
when :open
- [[prefix_widget, widget, imp_widget,
- [:message_patina_color,
+ @person_lines[start] = m.from
+ [[prefix_widget, open_widget, new_widget, starred_widget,
+ [color,
"#{m.from ? m.from.mediumname : '?'} to #{m.recipients.map { |l| l.shortname }.join(', ')} #{m.date.to_nice_s} (#{m.date.to_nice_distance_s})"]]]
-# (m.to.empty? ? [] : [[[:message_patina_color, prefix + " To: " + m.recipients.map { |x| x.mediumname }.join(", ")]]]) +
+
when :closed
- [[prefix_widget, widget, imp_widget,
- [:message_patina_color,
+ @person_lines[start] = m.from
+ [[prefix_widget, open_widget, new_widget, starred_widget,
+ [color,
"#{m.from ? m.from.mediumname : '?'}, #{m.date.to_nice_s} (#{m.date.to_nice_distance_s}) #{m.snippet}"]]]
+
when :detailed
- labels = m.labels# - @hidden_labels
- x = [[prefix_widget, widget, imp_widget, [:message_patina_color, "From: #{m.from ? m.from.longname : '?'}"]]] +
- ((m.to.empty? ? [] : break_into_lines(" To: ", m.to.map { |x| x.longname })) +
- (m.cc.empty? ? [] : break_into_lines(" Cc: ", m.cc.map { |x| x.longname })) +
- (m.bcc.empty? ? [] : break_into_lines(" Bcc: ", m.bcc.map { |x| x.longname })) +
- [" Date: #{m.date.strftime DATE_FORMAT} (#{m.date.to_nice_distance_s})"] +
- [" Subject: #{m.subj}"] +
- [(parent ? " In reply to: #{parent.from.mediumname}'s message of #{parent.date.strftime DATE_FORMAT}" : nil)] +
- [labels.empty? ? nil : " Labels: #{labels.join(', ')}"]
- ).flatten.compact.map { |l| [[:message_patina_color, prefix + " " + l]] }
- #raise x.inspect
- x
+ @person_lines[start] = m.from
+ from = [[prefix_widget, open_widget, new_widget, starred_widget,
+ [color, "From: #{m.from ? format_person(m.from) : '?'}"]]]
+
+ rest = []
+ unless m.to.empty?
+ m.to.each_with_index { |p, i| @person_lines[start + rest.length + from.length + i] = p }
+ rest += format_person_list " To: ", m.to
+ end
+ unless m.cc.empty?
+ m.cc.each_with_index { |p, i| @person_lines[start + rest.length + from.length + i] = p }
+ rest += format_person_list " Cc: ", m.cc
+ end
+ unless m.bcc.empty?
+ m.bcc.each_with_index { |p, i| @person_lines[start + rest.length + from.length + i] = p }
+ rest += format_person_list " Bcc: ", m.bcc
+ end
+
+ rest += [
+ " Date: #{m.date.strftime DATE_FORMAT} (#{m.date.to_nice_distance_s})",
+ " Subject: #{m.subj}",
+ (parent ? " In reply to: #{parent.from.mediumname}'s message of #{parent.date.strftime DATE_FORMAT}" : nil),
+ m.labels.empty? ? nil : " Labels: #{m.labels.join(', ')}",
+ ].compact
+
+ from + rest.map { |l| [[color, prefix + " " + l]] }
end
end
- def break_into_lines prefix, list
+ def format_person_list prefix, people
+ ptext = people.map { |p| format_person p }
pad = " " * prefix.length
- [prefix + list.first + (list.length > 1 ? "," : "")] +
- list[1 .. -1].map_with_index do |e, i|
- pad + e + (i == list.length - 1 ? "" : ",")
+ [prefix + ptext.first + (ptext.length > 1 ? "," : "")] +
+ ptext[1 .. -1].map_with_index do |e, i|
+ pad + e + (i == ptext.length - 1 ? "" : ",")
end
end
+ def format_person p
+ p.longname + (ContactManager.is_contact?(p) ? " (#{ContactManager.alias_for p})" : "")
+ end
- def chunk_to_lines chunk, state, start, depth, parent=nil
- prefix = " " * depth
+ ## todo: check arguments on this overly complex function
+ def chunk_to_lines chunk, state, start, depth, parent=nil, color=nil, star_color=nil
+ prefix = " " * INDENT_SPACES * depth
case chunk
when :fake_root
- [[[:message_patina_color, "#{prefix}<one or more unreceived messages>"]]]
+ [[[:missing_message_color, "#{prefix}<one or more unreceived messages>"]]]
when nil
- [[[:message_patina_color, "#{prefix}<an unreceived message>"]]]
+ [[[:missing_message_color, "#{prefix}<an unreceived message>"]]]
when Message
- message_patina_lines(chunk, state, parent, prefix) +
+ message_patina_lines(chunk, state, start, parent, prefix, color, star_color) +
(chunk.is_draft? ? [[[:draft_notification_color, prefix + " >>> This message is a draft. To edit, hit 'e'. <<<"]]] : [])
- when Message::Attachment
- [[[:mime_color, "#{prefix}+ MIME attachment #{chunk.content_type}#{chunk.desc ? ' (' + chunk.desc + ')': ''}"]]]
- when Message::Text
- t = chunk.lines
- if t.last =~ /^\s*$/
- t.pop while t[t.length - 2] =~ /^\s*$/
- end
- t.map { |line| [[:none, "#{prefix}#{line}"]] }
- when Message::Quote
- case state
- when :closed
- [[[:quote_patina_color, "#{prefix}+ #{chunk.lines.length} quoted lines"]]]
- when :open
- t = chunk.lines
- [[[:quote_patina_color, "#{prefix}- #{chunk.lines.length} quoted lines"]]] +
- t.map { |line| [[:quote_color, "#{prefix}#{line}"]] }
- end
- when Message::Signature
- case state
- when :closed
- [[[:sig_patina_color, "#{prefix}+ #{chunk.lines.length}-line signature"]]]
- when :open
- t = chunk.lines
- [[[:sig_patina_color, "#{prefix}- #{chunk.lines.length}-line signature"]]] +
- t.map { |line| [[:sig_color, "#{prefix}#{line}"]] }
- end
+
else
- raise "unknown chunk type #{chunk.class.name}"
+ raise "Bad chunk: #{chunk.inspect}" unless chunk.respond_to?(:inlineable?) ## debugging
+ if chunk.inlineable?
+ chunk.lines.map { |line| [[chunk.color, "#{prefix}#{line}"]] }
+ elsif chunk.expandable?
+ case state
+ when :closed
+ [[[chunk.patina_color, "#{prefix}+ #{chunk.patina_text}"]]]
+ when :open
+ [[[chunk.patina_color, "#{prefix}- #{chunk.patina_text}"]]] + chunk.lines.map { |line| [[chunk.color, "#{prefix}#{line}"]] }
+ end
+ else
+ [[[chunk.patina_color, "#{prefix}x #{chunk.patina_text}"]]]
+ end
end
end
- def view_attachment a
- BufferManager.flash "viewing #{a.content_type} attachment..."
- a.view!
+ def view chunk
+ BufferManager.flash "viewing #{chunk.content_type} attachment..."
+ success = chunk.view!
BufferManager.erase_flash
BufferManager.completely_redraw_screen
+ unless success
+ BufferManager.spawn "Attachment: #{chunk.filename}", TextMode.new(chunk.to_s)
+ BufferManager.flash "Couldn't execute view command, viewing as text."
+ end
end
-
end
end