grokify / mime-builder-ruby

require 'mime/types'

module MIMEBuilder

  class Filepath

    attr_accessor :mime

    attr_accessor :filepath

    # {:base64_encode bool, :content_type string, :content_id_disable bool}

    def initialize(filepath, opts={})

      if opts.has_key?(:base64_encode) && opts[:base64_encode]

        @base64_encode = true

      else

        @base64_encode = false

      end

      @mime = create_mime(filepath)

      if opts.has_key?(:content_id_disable)

        @mime.headers.delete('Content-Id')

      end

      @mime.headers.set(

        'Content-Type',

        get_file_content_type(filepath, opts[:content_type])

      )

      if opts.has_key?(:is_attachment) && opts[:is_attachment]

        @mime.headers.set(

          'Content-Disposition',

          get_attachment_content_disposition(filepath)

        )

      end

    end

    def create_mime(filepath)

      bytes = read_file_bytes(filepath)

      if @base64_encode

        mime = MIME::Text.new(Base64.encode64(bytes))

        mime.headers.set('Content-Transfer-Encoding', 'base64')

        return mime

      end

      return MIME::Application.new(bytes)

    end

    def read_file_bytes(filepath=nil)

      unless File.file?(filepath.to_s)

        raise "File \"#{filepath.to_s}\" does not exist or cannot be read"

      end

      # Ruby 1.8.7 support

      file_bytes = RUBY_VERSION < '1.9' \

        ? File.open(filepath, 'rb')        { |f| f.read } \

        : File.open(filepath, 'rb:BINARY') { |f| f.read }

      return file_bytes

    end

    def get_file_content_type(filepath=nil, content_type=nil)

      if (content_type.is_a?(String) && content_type =~ /^[^\/\s]+\/[^\/\s]+/)

        return content_type

      end

      return MIME::Types.type_for(filepath).first.content_type \

        || 'application/octet-stream'

    end

  end

end