Contents

require 'action_view'
if ActionView.version < Gem::Version.new("5.1")
require 'better_html/better_erb/erubis_implementation'
else
require 'better_html/better_erb/erubi_implementation'
end
require 'better_html/better_erb/validated_output_buffer'


class BetterHtml::BetterErb
  cattr_accessor :content_types
  if ActionView.version < Gem::Version.new("5.1")
    self.content_types = {
      'html.erb' => BetterHtml::BetterErb::ErubisImplementation
    }
  else
    self.content_types = {
      'html.erb' => BetterHtml::BetterErb::ErubiImplementation
    }
  end

  def self.prepend!
    ActionView::Template::Handlers::ERB.prepend(ConditionalImplementation)
  end

  private

  module ConditionalImplementation

    def call(template)
      generate(template)
    end

    private

    def generate(template)
      # First, convert to BINARY, so in case the encoding is
      # wrong, we can still find an encoding tag
      # (<%# encoding %>) inside the String using a regular
      # expression

      filename = template.identifier.split("/").last
      exts = filename.split(".")
      exts = exts[1..exts.length].join(".")
      template_source = template.source.dup.force_encoding(Encoding::ASCII_8BIT)

      erb = template_source.gsub(ActionView::Template::Handlers::ERB::ENCODING_TAG, '')
      encoding = $2

      erb.force_encoding valid_encoding(template.source.dup, encoding)

      # Always make sure we return a String in the default_internal
      erb.encode!

      excluded_template = !!BetterHtml.config.template_exclusion_filter&.call(template.identifier)
      klass = BetterHtml::BetterErb.content_types[exts] unless excluded_template
      klass ||= self.class.erb_implementation

      generator = klass.new(
        erb,
        :escape => (self.class.escape_whitelist.include? template.type),
        :trim => (self.class.erb_trim_mode == "-")
      )
      generator.validate! if generator.respond_to?(:validate!)
      generator.src
    end
  end
end

Version data entries

14 entries across 14 versions & 1 rubygems

Version	Path
better_html-1.0.11	lib/better_html/better_erb.rb
better_html-1.0.10	lib/better_html/better_erb.rb
better_html-1.0.9	lib/better_html/better_erb.rb
better_html-1.0.8	lib/better_html/better_erb.rb
better_html-1.0.7	lib/better_html/better_erb.rb
better_html-1.0.6	lib/better_html/better_erb.rb
better_html-1.0.5	lib/better_html/better_erb.rb
better_html-1.0.4	lib/better_html/better_erb.rb
better_html-1.0.3	lib/better_html/better_erb.rb
better_html-1.0.2	lib/better_html/better_erb.rb
better_html-1.0.1	lib/better_html/better_erb.rb
better_html-1.0.0	lib/better_html/better_erb.rb
better_html-0.0.12	lib/better_html/better_erb.rb
better_html-0.0.11	lib/better_html/better_erb.rb