]> git.cworth.org Git - sup/blobdiff - lib/sup/message.rb
removing some warnings
[sup] / lib / sup / message.rb
index f3ad31d7ea05bf2b4a420aeb82acbda6d2a0c81d..facbc312518f67f7e18619a2ca5f5a73b1121d54 100644 (file)
@@ -54,7 +54,7 @@ class Message
     attr_reader :lines
     def initialize lines
       ## do some wrapping
-      @lines = lines.map { |l| l.wrap 80 }.flatten
+      @lines = lines.map { |l| l.chomp.wrap 80 }.flatten
     end
   end
 
@@ -82,32 +82,25 @@ class Message
 
   attr_reader :id, :date, :from, :subj, :refs, :replytos, :to, :source,
               :cc, :bcc, :labels, :list_address, :recipient_email, :replyto,
-              :source_info, :status
+              :source_info
 
-  bool_reader :dirty
+  bool_reader :dirty, :source_marked_read
 
-  ## if index_entry is specified, will fill in values from that,
+  ## if you specify a :header, will use values from that. otherwise, will try and
+  ## load the header from the source.
   def initialize opts
-    if opts[:source]
-      @source = opts[:source]
-      @source_info = opts[:source_info] or raise ArgumentError, ":source but no :source_info"
-      @body = nil
-    else
-      @source = @source_info = nil
-      @body = opts[:body] or raise ArgumentError, "one of :body or :source must be specified"
-    end
+    @source = opts[:source] or raise ArgumentError, "source can't be nil"
+    @source_info = opts[:source_info] or raise ArgumentError, "source_info can't be nil"
     @snippet = opts[:snippet] || ""
+    @have_snippet = !opts[:snippet].nil?
     @labels = opts[:labels] || []
     @dirty = false
 
-    header = 
-      if opts[:header]
-        opts[:header]
-      else
-        header = @source.load_header @source_info
-        header.each { |k, v| header[k.downcase] = v }
-        header
-      end
+    read_header(opts[:header] || @source.load_header(@source_info))
+  end
+
+  def read_header header
+    header.each { |k, v| header[k.downcase] = v }
 
     %w(message-id date).each do |f|
       raise MessageFormatError, "no #{f} field in header #{header.inspect} (source #@source offset #@source_info)" unless header.include? f
@@ -116,22 +109,18 @@ class Message
 
     begin
       date = header["date"]
-      @date = (Time === date ? date : Time.parse(header["date"]))
+      @date = Time === date ? date : Time.parse(header["date"])
     rescue ArgumentError => e
       raise MessageFormatError, "unparsable date #{header['date']}: #{e.message}"
     end
 
-    if(@subj = header["subject"])
-      @subj = @subj.gsub(/\s+/, " ").gsub(/\s+$/, "")
-    else
-      @subj = DEFAULT_SUBJECT
-    end
+    @subj = header.member?("subject") ? header["subject"].gsub(/\s+/, " ").gsub(/\s+$/, "") : DEFAULT_SUBJECT
     @from = Person.for header["from"]
     @to = Person.for_several header["to"]
     @cc = Person.for_several header["cc"]
     @bcc = Person.for_several header["bcc"]
     @id = header["message-id"]
-    @refs = (header["references"] || "").scan(/<(.*?)>/).flatten
+    @refs = (header["references"] || "").gsub(/[<>]/, "").split(/\s+/).flatten
     @replytos = (header["in-reply-to"] || "").scan(/<(.*?)>/).flatten
     @replyto = Person.for header["reply-to"]
     @list_address =
@@ -141,11 +130,12 @@ class Message
         nil
       end
 
-    @recipient_email = header["delivered-to"]
-    @status = header["status"]
+    @recipient_email = header["envelope-to"] || header["x-original-to"] || header["delivered-to"]
+    @source_marked_read = header["status"] == "RO"
   end
+  private :read_header
 
-  def broken?; @source.nil?; end
+  def broken?; @source.broken?; end
   def snippet; @snippet || to_chunks && @snippet; end
   def is_list_message?; !@list_address.nil?; end
   def is_draft?; DraftLoader === @source; end
@@ -181,21 +171,55 @@ class Message
     @dirty = true
   end
 
+  ## this is called when the message body needs to actually be loaded.
   def to_chunks
     @chunks ||=
-      if @body
-        [Text.new(@body.split("\n"))]
+      if @source.broken?
+        [Text.new(error_message(@source.broken_msg.split("\n")))]
       else
-        message_to_chunks @source.load_message(@source_info)
+        begin
+          ## we need to re-read the header because it contains information
+          ## that we don't store in the index. actually i think it's just
+          ## the mailing list address (if any), so this is kinda overkill.
+          ## i could just store that in the index, but i think there might
+          ## be other things like that in the future, and i'd rather not
+          ## bloat the index.
+          read_header @source.load_header(@source_info)
+          message_to_chunks @source.load_message(@source_info)
+        rescue SourceError, SocketError, MessageFormatError => e
+          [Text.new(error_message(e.message))]
+        end
       end
   end
 
+  def error_message msg
+    <<EOS
+#@snippet...
+
+***********************************************************************
+* An error occurred while loading this message. It is possible that   *
+* the source has changed, or (in the case of remote sources) is down. *
+***********************************************************************
+
+The error message was:
+  #{msg}
+EOS
+  end
+
   def raw_header
-    @source.raw_header @source_info
+    begin
+      @source.raw_header @source_info
+    rescue SourceError => e
+      error_message e.message
+    end
   end
 
   def raw_full_message
-    @source.raw_full_message @source_info
+    begin
+      @source.raw_full_message @source_info
+    rescue SourceError => e
+      error_message(e.message)
+    end
   end
 
   def content
@@ -229,10 +253,8 @@ private
     ret = [] <<
       case m.header.content_type
       when "text/plain", nil
-        raise MessageFormatError, "no message body before decode" unless
-          m.body
-        body = m.decode or raise MessageFormatError, "no message body"
-        text_to_chunks body.gsub(/\t/, "    ").gsub(/\r/, "").split("\n")
+        m.body && body = m.decode or raise MessageFormatError, "for some bizarre reason, RubyMail was unable to parse this message."
+        text_to_chunks body.normalize_whitespace.split("\n")
       when /^multipart\//
         nil
       else
@@ -247,7 +269,6 @@ private
   ## parse the lines of text into chunk objects.  the heuristics here
   ## need tweaking in some nice manner. TODO: move these heuristics
   ## into the classes themselves.
-
   def text_to_chunks lines
     state = :text # one of :text, :quote, or :sig
     chunks = []
@@ -255,9 +276,11 @@ private
 
     lines.each_with_index do |line, i|
       nextline = lines[(i + 1) ... lines.length].find { |l| l !~ /^\s*$/ } # skip blank lines
+
       case state
       when :text
         newstate = nil
+
         if line =~ QUOTE_PATTERN || (line =~ QUOTE_START_PATTERN && (nextline =~ QUOTE_PATTERN || nextline =~ QUOTE_START_PATTERN))
           newstate = :quote
         elsif line =~ SIG_PATTERN && (lines.length - i) < MAX_SIG_DISTANCE
@@ -265,6 +288,7 @@ private
         elsif line =~ BLOCK_QUOTE_PATTERN
           newstate = :block_quote
         end
+
         if newstate
           chunks << Text.new(chunk_lines) unless chunk_lines.empty?
           chunk_lines = [line]
@@ -272,8 +296,10 @@ private
         else
           chunk_lines << line
         end
+
       when :quote
         newstate = nil
+
         if line =~ QUOTE_PATTERN || line =~ QUOTE_START_PATTERN || line =~ /^\s*$/
           chunk_lines << line
         elsif line =~ SIG_PATTERN && (lines.length - i) < MAX_SIG_DISTANCE
@@ -281,6 +307,7 @@ private
         else
           newstate = :text
         end
+
         if newstate
           if chunk_lines.empty?
             # nothing
@@ -292,17 +319,18 @@ private
           chunk_lines = [line]
           state = newstate
         end
+
       when :block_quote
         chunk_lines << line
+
       when :sig
         chunk_lines << line
       end
  
-      if state == :text && (@snippet.nil? || @snippet.length < SNIPPET_LEN) &&
-          line !~ /[=\*#_-]{3,}/ && line !~ /^\s*$/
+      if !@have_snippet && state == :text && (@snippet.nil? || @snippet.length < SNIPPET_LEN) && line !~ /[=\*#_-]{3,}/ && line !~ /^\s*$/
         @snippet += " " unless @snippet.empty?
         @snippet += line.gsub(/^\s+/, "").gsub(/[\r\n]/, "").gsub(/\s+/, " ")
-        @snippet = @snippet[0 ... SNIPPET_LEN]
+        @snippet = @snippet[0 ... SNIPPET_LEN].chomp
       end
     end