lib/mime_builder/text.rb in mime_builder-0.0.2 vs lib/mime_builder/text.rb in mime_builder-0.0.3

@@ -5,11 +5,37 @@
     attr_accessor :mime
     attr_accessor :text
 
     def initialize(text, opts = {})
       @text = text
-      @mime = MIME::Text.new(text, 'plain')
+
+      if opts.key?(:content_type) && opts[:content_type].to_s.length>0
+        content_type = opts[:content_type]
+        if content_type =~ /^text\/([^\/]+)$/i
+          @mime = MIME::Text.new(text, $1.downcase)
+        else
+          raise "Unknown Content Type: " + opts[:content_type].to_s
+        end
+      else
+        @mime = MIME::Text.new(text, 'plain')
+      end
+
       @mime.headers.delete('Content-Id') \
         if opts.key?(:content_id_disable) && opts[:content_id_disable]
+
+      set_attachment_content_disposition(opts[:filename], opts[:is_attachment])
+    end
+
+    def set_attachment_content_disposition(filename, is_attachment)
+      @mime.headers.set(
+        'Content-Disposition',
+        get_attachment_content_disposition(filename)
+      ) if is_attachment
+    end
+
+    def get_attachment_content_disposition(filename = nil)
+      cd = filename.to_s.length > 0              \
+        ? "attachment; filename=\"#{filename}\"" \
+        : 'attachment'
     end
   end
 end