From 9a8b7a3f830a578e955c93ad8c0de811f21f68a4 Mon Sep 17 00:00:00 2001
From: Joseph Tibbertsma <josephtibbertsma@gmail.com>
Date: Fri, 18 Dec 2015 10:47:19 -0800
Subject: [PATCH] Wrote visualize method

---
 lib/router-visualizer/mapper.rb | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/lib/router-visualizer/mapper.rb b/lib/router-visualizer/mapper.rb
index d089704..9ae048f 100644
--- a/lib/router-visualizer/mapper.rb
+++ b/lib/router-visualizer/mapper.rb
@@ -4,8 +4,22 @@ module ActionDispatch
   module Routing
     class Mapper
       def visualize(options = {})
+        groups = options.delete(:groups) { ['development'] }
+        at = options.delete(:at) { '/routes' }
 
+        if matching_group? groups
+          mount RouterVisualizer::Engine, at: at
+        end
+        self
       end
+
+      private
+        def matching_group?(groups)
+          groups.each do |group|
+            return true if Rails.env.send("#{group}?")
+          end
+          false
+        end
     end
   end
 end
\ No newline at end of file