Less code duplication.HEADmaster

1 files changed, 4 insertions, 21 deletions

diff --git a/bocelli/module/base.rb b/bocelli/module/base.rb
index 45b09d7..f338c34 100644
--- a/bocelli/module/base.rb
+++ b/bocelli/module/base.rb
@@ -1,29 +1,12 @@
+require_relative '../core/router'
+
 module Bocelli
   module Module
     module Base
-      attr_reader :routes
+      include Bocelli::Core::Router
 
       def setup
-        @routes = {}
-      end
-
-      def on(route, &block)
-        raise 'no block given' if block.nil?
-
-        @routes[route] = block
-      end
-
-      def match?(str, route)
-        case route
-        when Regexp
-          str =~ route
-        when String
-          str == route
-        end
-      end
-
-      def match(str)
-        @routes.detect { |k, _| match?(str, k) }
+        setup_router
       end
 
       class << self