lib/pretty_routes.rb in pretty_routes-0.0.1 vs lib/pretty_routes.rb in pretty_routes-0.1.0

@@ -11,13 +11,24 @@
     require 'action_dispatch/routing/inspector'
     ROUTE_INSPECTOR = ActionDispatch::Routing::RoutesInspector.new([])
   end
 
   def self.format_routes(routes = all_routes)
-    # ActionDispatch::Routing::RoutesInspector.new.collect_routes(_routes.routes)
-    ROUTE_INSPECTOR.send :collect_routes, routes
+    # Rails routes
+    rails_routes = ROUTE_INSPECTOR.send :collect_routes, routes
+    # Parse rails routes to find constraints
+    rails_routes.each do |route|
+      next unless route[:reqs].include?(' {')
+      reqs = route[:reqs].split(' ')
+      route[:reqs] = reqs.first
+      route[:constraints] = reqs.last
+    end
+    # Return parsed
+    rails_routes
   end
 
+  # Reload and return Rails routes.
+  #
   def self.all_routes
     Rails.application.reload_routes!
     Rails.application.routes.routes
   end
 end