require 'action_view'

# register an addition detail for the lookup context to understand,
# this will allow us to have the versions available upon lookup in
# the resolver.
ActionView::LookupContext.register_detail(:versions) { [] }

ActionView::PathResolver.class_eval do
  ActionView::PathResolver::EXTENSIONS.replace({
    locale: ".",
    formats: ".",
    versions: ".",
    variants: "+",
    handlers: "."
  })

  def initialize(pattern = nil)
    @pattern = pattern || ":prefix/:action{.:locale,}{.:formats,}{+:variants,}{.:versions,}{.:handlers,}"
    super()
  end

  # The default extract handler expects that the handler is the last extension and
  # the format is the next one. Since we are replacing the DEFAULT_PATTERN, we need to
  # make sure that we extract the format from the correct position.
  #
  # The version may be stuck inbetween the format and the handler. This is actually pretty tricky
  # because the version is optional and the locale is optional-which means when there are 3 'pieces'
  # it may be the locale, format and handler or the format, version and handler. To check this, we will
  # try one additional time if there are more pieces, which should cover all the cases:
  #
  # Cases:
  #  1: assume version is in the extension, pieces = ['html','erb']
  #  2: assume version is in the extension, pieces = ['html','v1','erb']
  #  3: assume version is in the extension, pieces = ['en','html','erb']
  #  4: assume version is in the extension, pieces = ['en','html','v1','erb']
  #
  def extract_handler_and_format_and_variant(path, default_formats=nil)
    pieces = File.basename(path).split('.'.freeze)
    pieces.shift

    extension = pieces.pop

    handler = ActionView::Template.handler_for_extension(extension)
    format, variant = get_format_and_variant_from_pieces(pieces, Mime)
    format &&= ActionView::Template::Types[format]

    [handler, format, variant]
  end

  # If there are still pieces and we didn't find a valid format, we may
  # have a version in the extension, so try one more time to pop the format.
  def get_format_and_variant_from_pieces(pieces, format_list)
    variant, format = nil
    pieces.reverse.each do |piece|
      piece, variant = piece.split(ActionView::PathResolver::EXTENSIONS[:variants], 2)

      format = format_list[piece]
      break unless format.nil?
    end
    [format, variant]
  end
end

ActionView::Template.class_eval do
  # the identifier method name filters out numbers,
  # but we want to preserve them for v1 etc.
  def identifier_method_name #:nodoc:
    inspect.gsub(/[^a-z0-9_]/, '_')
  end
end