yml file changed

kiote · Nov 26, 2014 · 01835e7 · 01835e7
1 parent 454ff8e
commit 01835e7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/activeadmin-poro-decorator.yml.example b/activeadmin-poro-decorator.yml.example
@@ -0,0 +1,2 @@
+modelname: Decorator
+use_helpers: true
diff --git a/lib/activeadmin-poro-decorator.rb b/lib/activeadmin-poro-decorator.rb
@@ -41,7 +41,8 @@ def initialize(collection, class_name)
     end
 
     def klass
-      "#{@class_name}Decorator".constantize
+      config = Config::Reader.new
+      "#{@class_name}#{config['modelname']}".constantize
     end
 
     def wrapped_collection

diff --git a/lib/activeadmin-poro-decorator/config.rb b/lib/activeadmin-poro-decorator/config.rb
@@ -4,12 +4,8 @@ module ActiveadminPoroDecorator
   module Config
     class Reader
       def initialize
-        @config = YAML.load_file('activeadmin-poro-decorator.yml.example')
+        @config = YAML.load_file('activeadmin-poro-decorator.yml')
       end
     end
-
-    def read(option)
-      @config[option]
-    end
   end
 end