Contents

# Breaks down email
module Mail2FrontMatter
  class Parser
    require 'nokogiri'
    require 'fileutils'

    attr_accessor :message, :metadata, :body

    ALLOWED_TYPES = {
      "audio" => "audio",
      "video" => "videos", 
      "image" => "images"
    }

    def initialize(message)
      @message = message
      raw_parsed_html = Nokogiri::HTML.parse(@message.html_part.body.raw_source.strip)
      @body = raw_parsed_html.at("body")

      # remove extraneous nesting
      while(@body.children.count == 1 && @body.children.first.name == "div") do
        @body = @body.children.first
      end

      attachments = {}

      @message.attachments.each do |attachment|
        if Parser::ALLOWED_TYPES.keys.include?(attachment.main_type)

          # save attachments
          media_directory = File.join(Dir.pwd, 'source', Parser::ALLOWED_TYPES[attachment.main_type])
          FileUtils.mkdir_p(media_directory)

          filepath = File.join(media_directory, attachment.filename) 

          # save attachment
          File.open(filepath, "w+b", 0644) {|f| f.write attachment.body.decoded}

          # retain metadata
          attachments[attachment.cid] = {
            maintype: attachment.main_type,
            mimetype: attachment.mime_type,
            filename: attachment.filename,
            filepath: filepath
          }

        # file type not allowed
        else
          # remove cooresponding node from html
          @body.xpath("//*[@src='cid:#{attachment.content_id}']").remove

        end
      end

      # convert body immediately to a string, why?
      # Plugins / pre-processors may wish to manipulate the body
      # however Nokogiri is strict and won't allow template tags
      # for obvious good reasons
      @body = @body.inner_html

      @metadata = {
        from:        message[:from].value,
        to:          message[:to].value,
        received:    message.date,
        subject:     message.subject,
        attachments: attachments
      }

    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
mail2frontmatter-0.0.5	lib/mail2frontmatter/parser.rb
mail2frontmatter-0.0.4	lib/mail2frontmatter/parser.rb
mail2frontmatter-0.0.3	lib/mail2frontmatter/parser.rb
mail2frontmatter-0.0.2	lib/mail2frontmatter/parser.rb