lib/sfn/command.rb in sfn-0.3.8 vs lib/sfn/command.rb in sfn-0.4.0

@@ -46,7 +46,17 @@
       end
       opts[:config] = cwd.join(File::SEPARATOR) unless cwd.empty?
       opts
     end
 
+    # @return [Class] attempt to return customized configuration class
+    def config_class
+      klass_name = self.class.name.split('::').last
+      if(Sfn::Config.const_defined?(klass_name))
+        Sfn::Config.const_get(klass_name)
+      else
+        super
+      end
+    end
+
   end
 end