enriclluelles/route_translator

# frozen_string_literal: true

module RouteTranslator
  module Translator
    module RouteHelpers
      TEST_CASE_HOOKS = %i[
        action_controller_test_case action_mailer_test_case action_view_test_case
      ].freeze

      module_function

      # Add standard route helpers for default locale e.g.
      #   I18n.locale = :de
      #   people_path -> people_de_path
      #   I18n.locale = :fr
      #   people_path -> people_fr_path
      def add(old_name, named_route_collection)
        helper_list = named_route_collection.helper_names

        %w[path url].each do |suffix|
          helper_container = named_route_collection.send(:"#{suffix}_helpers_module")
          new_helper_name = :"#{old_name}_#{suffix}"

          helper_list.push(new_helper_name) unless helper_list.include?(new_helper_name)

          helper_container.__send__(:define_method, new_helper_name) do |*args|
            __send__(Translator.route_name_for(args, old_name, suffix, self), *args)
          end

          next unless ENV.fetch('RAILS_ENV', nil) == 'test'

          TEST_CASE_HOOKS.each do |test_case_hook|
            ActiveSupport.on_load(test_case_hook) do
              include helper_container
            end
          end
        end
      end
    end
  end
end