class Middleman::CoreExtensions::InlineURLRewriter::Rack

Public Class Methods

new(app, options={}) click to toggle source
# File lib/middleman-core/core_extensions/inline_url_rewriter.rb, line 56
def initialize(app, options={})
  @rack_app = app
  @middleman_app = options.fetch(:middleman_app)
  @rewriters = options.fetch(:rewriters)

  all_source_exts = @rewriters
                    .reduce([]) { |sum, rewriter| sum + rewriter[:source_extensions] }
                    .flatten
                    .uniq
  @source_exts_regex_text = Regexp.union(all_source_exts).to_s

  @all_asset_exts = @rewriters
                    .reduce([]) { |sum, rewriter| sum + rewriter[:url_extensions] }
                    .flatten
                    .uniq
end

Public Instance Methods

call(env) click to toggle source
# File lib/middleman-core/core_extensions/inline_url_rewriter.rb, line 73
def call(env)
  status, headers, response = @rack_app.call(env)

  # Allow configuration or upstream request to skip all rewriting
  return [status, headers, response] if env['bypass_inline_url_rewriter'] == 'true'

  path = ::Middleman::Util.full_path(env['PATH_INFO'], @middleman_app)

  return [status, headers, response] unless path =~ /(^\/$)|(#{@source_exts_regex_text}$)/
  return [status, headers, response] unless body = ::Middleman::Util.extract_response_text(response)

  dirpath = ::Pathname.new(File.dirname(path))

  rewritten = ::Middleman::Util.instrument 'inline_url_rewriter', path: path do
    ::Middleman::Util.rewrite_paths(body, path, @all_asset_exts, @middleman_app) do |asset_path|
      uri = ::Middleman::Util.parse_uri(asset_path)

      relative_path = uri.host.nil?

      full_asset_path = if relative_path
        dirpath.join(asset_path).to_s
      else
        asset_path
      end

      @rewriters.each do |rewriter|
        uid = rewriter.fetch(:id)

        # Allow upstream request to skip this specific rewriting
        next if env["bypass_inline_url_rewriter_#{uid}"] == 'true'

        exts = rewriter.fetch(:url_extensions)
        next unless exts.include?(::File.extname(asset_path))

        source_exts = rewriter.fetch(:source_extensions)
        next unless source_exts.include?(::File.extname(path))

        ignore = rewriter.fetch(:ignore)
        next if ignore.any? { |r| ::Middleman::Util.should_ignore?(r, full_asset_path) }

        rewrite_ignore = Array(rewriter[:rewrite_ignore] || [])
        next if rewrite_ignore.any? { |i| ::Middleman::Util.path_match(i, path) }

        proc = rewriter.fetch(:proc)

        result = proc.call(asset_path, dirpath, path)
        asset_path = result if result
      end

      asset_path
    end
  end

  ::Rack::Response.new(
    rewritten,
    status,
    headers
  ).finish
end