lib/draper/finders.rb in draper-1.0.0.beta1 vs lib/draper/finders.rb in draper-1.0.0.beta2

@@ -1,31 +1,26 @@
 module Draper
   module Finders
 
-    attr_reader :finder_class
-    def finder_class=(klass)
-      @finder_class = klass.to_s.camelize.constantize
-    end
-
     def find(id, options = {})
-      decorate(finder_class.find(id), options)
+      decorate(source_class.find(id), options)
     end
 
     def all(options = {})
-      decorate_collection(finder_class.all, options)
+      decorate_collection(source_class.all, options)
     end
 
     def first(options = {})
-      decorate(finder_class.first, options)
+      decorate(source_class.first, options)
     end
 
     def last(options = {})
-      decorate(finder_class.last, options)
+      decorate(source_class.last, options)
     end
 
     def method_missing(method, *args, &block)
-      result = finder_class.send(method, *args, &block)
+      result = super
       options = args.extract_options!
 
       case method.to_s
       when /^find_((last_)?by_|or_(initialize|create)_by_)/
         decorate(result, options)
@@ -33,12 +28,7 @@
         decorate_collection(result, options)
       else
         result
       end
     end
-
-    def respond_to?(method, include_private = false)
-      super || finder_class.respond_to?(method)
-    end
-
   end
 end