From 1f25e16c0e17a6b9d5149a6ca8a23f872289cc76 Mon Sep 17 00:00:00 2001 From: Nalesso Moreira Date: Mon, 6 May 2013 05:27:10 +0200 Subject: [PATCH] Update engine.rb I've done a bit more of refactoring. --- lib/impressionist/engine.rb | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/lib/impressionist/engine.rb b/lib/impressionist/engine.rb index e16c8ee..f748f97 100644 --- a/lib/impressionist/engine.rb +++ b/lib/impressionist/engine.rb @@ -35,11 +35,14 @@ module Impressionist end def define_orm_type(str) - str = str.to_s - @orm = str.match(/active_record|mongo_mapper|mongoid|/) - :matched - + @orm = matcher(str.to_s) end + def matcher(str) + matched = str.match(/active_record|mongo_mapper|mongoid|/) + matched[0] + end + + end end