class Mail::AttachmentsList

def [](index_value)
  if index_value.is_a?(Integer)
    self.fetch(index_value)
  else
    self.select { |a| a.filename == index_value }.first
  end
end

def []=(name, value)
  encoded_name = Mail::Encodings.decode_encode name, :encode
  default_values = { :content_type => "#{set_mime_type(name)}; filename=\"#{encoded_name}\"",
                     :content_transfer_encoding => "#{guess_encoding}",
                     :content_disposition => "#{@content_disposition_type}; filename=\"#{encoded_name}\"" }
  if value.is_a?(Hash)
    if path = value.delete(:filename)
      value[:content] ||= File.open(path, 'rb') { |f| f.read }
    end
    default_values[:body] = value.delete(:content) if value[:content]
    default_values[:body] = value.delete(:data) if value[:data]
    encoding = value.delete(:transfer_encoding) || value.delete(:encoding)
    if encoding
      if Mail::Encodings.defined? encoding
        default_values[:content_transfer_encoding] = encoding
      else
        raise "Do not know how to handle Content Transfer Encoding #{encoding}, please choose either quoted-printable or base64"
      end
    end
    if value[:mime_type]
      default_values[:content_type] = value.delete(:mime_type)
      @mime_type = MiniMime.lookup_by_content_type(default_values[:content_type])
      default_values[:content_transfer_encoding] ||= guess_encoding
    end
    hash = default_values.merge(value)
  else
    default_values[:body] = value
    hash = default_values
  end
  if hash[:body].respond_to? :force_encoding and hash[:body].respond_to? :valid_encoding?
    if not hash[:body].valid_encoding? and default_values[:content_transfer_encoding].casecmp('binary').zero?
      hash[:body] = hash[:body].dup if hash[:body].frozen?
      hash[:body].force_encoding("BINARY")
    end
  end
  attachment = Part.new(hash)
  attachment.add_content_id(hash[:content_id])
  @parts_list << attachment
end

def guess_encoding
  if @mime_type && !@mime_type.binary?
    "7bit"
  else
    "binary"
  end
end

def initialize(parts_list)
  @parts_list = parts_list
  @content_disposition_type = 'attachment'
  parts_list.map { |p|
    if p.mime_type == 'message/rfc822'
      Mail.new(p.body.encoded).attachments
    elsif p.parts.empty?
      p if p.attachment?
    else
      p.attachments
    end
  }.flatten.compact.each { |a| self << a }
  self
end

def inline
  @content_disposition_type = 'inline'
  self
end

def set_mime_type(filename)
  filename = filename.encode(Encoding::UTF_8) if filename.respond_to?(:encode)
  @mime_type = MiniMime.lookup_by_filename(filename)
  @mime_type && @mime_type.content_type
end