# encoding: UTF-8

PryRails::Commands.create_command "show-routes" do
  group "Rails"
  description "Show all routes in match order."

  def options(opt)
    opt.banner unindent <<-USAGE
      Usage: show-routes [-G]

      show-routes displays the current Rails app's routes.
    USAGE

    opt.on :G, "grep", "Filter output by regular expression", :argument => true
  end

  def process
    Rails.application.reload_routes!
    all_routes = Rails.application.routes.routes

    formatted = case Rails.version.to_s
      when /^4/
        process_rails_4(all_routes)
      when /^3\.2/
        process_rails_3_2(all_routes)
      else
        process_rails_3_0_and_3_1(all_routes)
      end

    output.puts formatted.grep(Regexp.new(opts[:G] || ".")).join("\n")
  end

  # Cribbed from https://github.com/rails/rails/blob/3-1-stable/railties/lib/rails/tasks/routes.rake
  def process_rails_3_0_and_3_1(all_routes)
    routes = all_routes.collect do |route|
      reqs = route.requirements.dup
      reqs[:to] = route.app unless route.app.class.name.to_s =~ /^ActionDispatch::Routing/
        reqs = reqs.empty? ? "" : reqs.inspect

      {:name => route.name.to_s, :verb => route.verb.to_s, :path => route.path, :reqs => reqs}
    end

    # Skip the route if it's internal info route
    routes.reject! { |r| r[:path] =~ %r{/rails/info/properties|^/assets} }

    name_width = routes.map{ |r| r[:name].length }.max
    verb_width = routes.map{ |r| r[:verb].length }.max
    path_width = routes.map{ |r| r[:path].length }.max

    routes.map do |r|
      "#{r[:name].rjust(name_width)} #{r[:verb].ljust(verb_width)} #{r[:path].ljust(path_width)} #{r[:reqs]}"
    end
  end

  def process_rails_3_2(all_routes)
    require 'rails/application/route_inspector'
    Rails::Application::RouteInspector.new.format(all_routes)
  end

  def process_rails_4(all_routes)
    require 'action_dispatch/routing/inspector'
    ActionDispatch::Routing::RoutesInspector.new(all_routes).format(ActionDispatch::Routing::ConsoleFormatter.new).split(/\n/)
  end
end