mhtml 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e6c8f08f08cddf731523a803d8ff1fd6237248cc
-  data.tar.gz: 5d529f065c8c1c1adaa0656c16ba95cf785b8e45
+  metadata.gz: 60fd056e7a5ded9db6a7e18e49c982100672e084
+  data.tar.gz: 1236377e35f0c3176bcaa25aaa6db13757d64d63
 SHA512:
-  metadata.gz: 8fbfe9a3c36c375f8493ce8df95134123edf4353447fda375be7a95ce367905c6631cbf10f8d599a5405421bf6b3040af1eea66affb227f45543d35f618adccb
-  data.tar.gz: bd81a7aa328a18d3437e3feeaedfdccea463ae0f9e683986c92fc637cc962877efca8d4a7e4fa8c9d31480e1b27d34760f5ab39de3a13793f3c46f999c1e1d3b
+  metadata.gz: 9cb37bbcc90d6e0c7658cde28c89bc69e2d79315709f6d86c8ac941c3284471e7ef0d7d7c25d04f0a8cb8a33ffe79b56216d2b09a5b4bc8a75eb16b7e4caace7
+  data.tar.gz: '095474af649049223490a686331be4368b6956c6f0281985436c4815f91fb9ddcbd92905c1c0c86af731801bfefd5de988efccbfa60427775a1bb2a9524a02f7'

data/lib/mhtml/document.rb

@@ -3,7 +3,13 @@ require 'http-parser'
 module Mhtml
   class Document
     attr_reader :chunked, :parser
-    attr_accessor :headers, :body, :is_quoted_printable, :encoding
+
+    attr_accessor :headers, 
+      :body, 
+      :is_quoted_printable, 
+      :encoding, 
+      :file_path, 
+      :root_doc
 
     def initialize(str = nil)
       @chunked = !str.is_a?(String)
@@ -61,6 +67,23 @@ module Mhtml
       header
     end
 
+    def relative_file_path
+      return nil if @file_path.nil?
+      return @file_path if @root_doc.nil? || @root_doc.file_path.nil?
+      return '.' if @file_path == @root_doc.file_path
+
+      str = nil
+      if @file_path.start_with?(@root_doc.file_path)
+        start = @root_doc.file_path.length
+        str = @file_path[start..@file_path.length - 1]
+      else
+        str = @file_path
+      end
+
+      str = str[1..(str.length - 1)] if str[0] == '/'
+      str
+    end
+
     # for testing only = no spec implemented
     def to_s
       @headers.join(LINE_BREAK) + Mhtml::DOUBLE_LINE_BREAK + @body
@@ -117,6 +140,10 @@ module Mhtml
           if !value.nil? && value.value == 'quoted-printable'
             @is_quoted_printable = true
           end
+        
+        elsif header.key == 'Content-Location'
+          value = header.values.first
+          @file_path = value.value unless value.nil?
         end
 
         @headers_proc.call(header) unless @headers_proc.nil?

data/lib/mhtml/root_document.rb

@@ -71,7 +71,9 @@ module Mhtml
           is_last_part = i + 1 == parts.length
           handle_chunked_body(part, is_last_part, is_last_subdoc)
         else
-          @sub_docs << Document.new(part)
+          sub_doc = Document.new(part)
+          sub_doc.root_doc = self
+          @sub_docs << sub_doc
         end
       end
     end
@@ -110,6 +112,7 @@ module Mhtml
 
     def create_chunked_subdoc
       @chunked_sub_doc = Document.new
+      @chunked_sub_doc.root_doc = self
 
       @chunked_sub_doc.on_header do |header| 
         @subdoc_header_proc.call(header) unless @subdoc_header_proc.nil?

data/lib/mhtml/version.rb

@@ -1,3 +1,3 @@
 module Mhtml
-  VERSION = '0.1.2'
+  VERSION = '0.1.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mhtml
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Ben Williams