0
@@ -16,17 +16,20 @@ module Ardes
0
       def include_actions(controller, options = {})
0
         options.assert_valid_keys(:only, :except)
0
         raise ArgumentError, "you can only specify either :except or :only, not both" if options[:only] && options[:except]
0
-        
0
         mixin = self.dup
0
-        if only = options[:only]
0
-          only = Array(options[:only]).collect(&:to_s)
0
-          mixin.instance_methods.each {|m| mixin.send(:undef_method, m) unless only.include?(m)}
0
-        elsif except = options[:except]
0
-          except = Array(options[:except]).collect(&:to_s)
0
-          mixin.instance_methods.each {|m| mixin.send(:undef_method, m) if except.include?(m)}
0
-        end
0
+        methods_to_remove(options).each {|m| mixin.send(:undef_method, m)}
0
         controller.send :include, mixin
0
       end
0
+      
0
+      def methods_to_remove(options = {})
0
+        if options[:only]
0
+          instance_methods - options[:only].map(&:to_s)
0
+        elsif options[:except]
0
+          options[:except].map(&:to_s) & instance_methods
0
+        else
0
+          []
0
+        end
0
+      end
0
     end
0
   end
0
 end
0
\ No newline at end of file