]> git.cworth.org Git - sup/blobdiff - lib/sup/message.rb
removing some warnings
[sup] / lib / sup / message.rb
index 69f4f02ae7f24f2a33c2022239394fa0679a7a3e..facbc312518f67f7e18619a2ca5f5a73b1121d54 100644 (file)
@@ -27,31 +27,34 @@ class Message
   end
 
   class Attachment
-    attr_reader :content_type, :desc
+    attr_reader :content_type, :desc, :filename
     def initialize content_type, desc, part
       @content_type = content_type
       @desc = desc
       @part = part
       @file = nil
+      desc =~ /filename="(.*?)"/ && @filename = $1
     end
 
     def view!
       unless @file
         @file = Tempfile.new "redwood.attachment"
-        @file.print @part.decode
+        @file.print self
         @file.close
       end
 
       ## TODO: handle unknown mime-types
       system "/usr/bin/run-mailcap --action=view #{@content_type}:#{@file.path}"
     end
+
+    def to_s; @part.decode; end
   end
 
   class Text
     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
 
@@ -73,24 +76,30 @@ class Message
   BLOCK_QUOTE_PATTERN = /^-----\s*Original Message\s*----+$/
   QUOTE_START_PATTERN = /(^\s*Excerpts from)|(^\s*In message )|(^\s*In article )|(^\s*Quoting )|((wrote|writes|said|says)\s*:\s*$)/
   SIG_PATTERN = /(^-- ?$)|(^\s*----------+\s*$)|(^\s*_________+\s*$)/
-  SIG_DISTANCE = 15 # lines from the end
+  MAX_SIG_DISTANCE = 15 # lines from the end
   DEFAULT_SUBJECT = "(missing subject)"
   DEFAULT_SENDER = "(missing sender)"
 
   attr_reader :id, :date, :from, :subj, :refs, :replytos, :to, :source,
               :cc, :bcc, :labels, :list_address, :recipient_email, :replyto,
-              :source_info, :mbox_status
-
-  bool_reader :dirty
-
-  def initialize source, source_info, labels, snippet=nil
-    @source = source
-    @source_info = source_info
+              :source_info
+
+  bool_reader :dirty, :source_marked_read
+
+  ## if you specify a :header, will use values from that. otherwise, will try and
+  ## load the header from the source.
+  def initialize opts
+    @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
-    @snippet = snippet
-    @labels = labels
 
-    header = @source.load_header @source_info
+    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|
@@ -99,22 +108,19 @@ class Message
     end
 
     begin
-      @date = Time.parse header["date"]
+      date = 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 =
@@ -124,15 +130,13 @@ class Message
         nil
       end
 
-    @recipient_email = header["delivered-to"]
-    @mbox_status = header["status"]
-  end
-
-  def snippet
-    to_chunks unless @snippet
-    @snippet
+    @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.broken?; end
+  def snippet; @snippet || to_chunks && @snippet; end
   def is_list_message?; !@list_address.nil?; end
   def is_draft?; DraftLoader === @source; end
   def draft_filename
@@ -141,6 +145,7 @@ class Message
   end
 
   def save index
+    return if broken?
     index.update_message self if @dirty
     @dirty = false
   end
@@ -166,23 +171,65 @@ class Message
     @dirty = true
   end
 
+  ## this is called when the message body needs to actually be loaded.
   def to_chunks
-    m = @source.load_message @source_info
-    message_to_chunks m
+    @chunks ||=
+      if @source.broken?
+        [Text.new(error_message(@source.broken_msg.split("\n")))]
+      else
+        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 header_text
-    @source.load_header_text @source_info
+  def raw_header
+    begin
+      @source.raw_header @source_info
+    rescue SourceError => e
+      error_message e.message
+    end
+  end
+
+  def raw_full_message
+    begin
+      @source.raw_full_message @source_info
+    rescue SourceError => e
+      error_message(e.message)
+    end
   end
 
   def content
     [
-      from && from.longname,
-      to.map { |p| p.longname },
-      cc.map { |p| p.longname },
-      bcc.map { |p| p.longname },
+      from && "#{from.name} #{from.email}",
+      to.map { |p| "#{p.name} #{p.email}" },
+      cc.map { |p| "#{p.name} #{p.email}" },
+      bcc.map { |p| "#{p.name} #{p.email}" },
       to_chunks.select { |c| c.is_a? Text }.map { |c| c.lines },
-      subj,
+      Message.normalize_subj(subj),
     ].flatten.compact.join " "
   end
 
@@ -206,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
@@ -224,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 = []
@@ -232,16 +276,19 @@ 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) < SIG_DISTANCE
+        elsif line =~ SIG_PATTERN && (lines.length - i) < MAX_SIG_DISTANCE
           newstate = :sig
         elsif line =~ BLOCK_QUOTE_PATTERN
           newstate = :block_quote
         end
+
         if newstate
           chunks << Text.new(chunk_lines) unless chunk_lines.empty?
           chunk_lines = [line]
@@ -249,15 +296,18 @@ 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) < SIG_DISTANCE
+        elsif line =~ SIG_PATTERN && (lines.length - i) < MAX_SIG_DISTANCE
           newstate = :sig
         else
           newstate = :text
         end
+
         if newstate
           if chunk_lines.empty?
             # nothing
@@ -269,21 +319,19 @@ 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*$/
-        @snippet = (@snippet ? @snippet + " " : "") + line.gsub(/^\s+/, "").gsub(/[\r\n]/, "").gsub(/\s+/, " ")
-        @snippet = @snippet[0 ... SNIPPET_LEN]
+      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].chomp
       end
-#      if @snippet.nil? && state == :text && (line.length > 40 ||
-#                                             line =~ /\S+.*[^,!:]\s*$/)
-#        @snippet = line.gsub(/^\s+/, "").gsub(/[\r\n]/, "")[0 .. 80]
-#      end
     end
 
     ## final object