grokify / mime-builder-ruby

module MIMEBuilder

  class Text

    attr_accessor :mime

    attr_accessor :text

    def initialize(text, opts = {})

      @text = text

      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