lib/cantango/rails/helpers/rest_helper.rb in cantango-0.8.9 vs lib/cantango/rails/helpers/rest_helper.rb in cantango-0.8.9.1

@@ -1,6 +1,6 @@
-class RestHelper
+module CanTango::Rails::Helpers::RestHelper
   CanTango.config.models.available_models.each do |model|
     class_eval %{
       def delete_#{model}_path obj, options = {}
         #{model}_path obj, {:method => 'delete'}.merge(options)
       end
@@ -14,28 +14,32 @@
   end
 
   def link_to_delete obj, user_type, options = {}
     return unless send(:"#{user_type}_can?", :delete, obj)
     # use i18n translation on label
-    link_to t(".delete"), send action_method(obj, :delete), obj, options
+    link_to t(".delete"), rest_action(obj, :delete, options)
   end
 
   def link_to_edit obj, user_type, options = {}
     return unless send(:"#{user_type}_can?", :edit, obj)
     # use i18n translation on label
-    link_to t(".edit"), send action_method(obj, :edit), obj, options
+    link_to t(".edit"), rest_action(obj, :edit, options)
   end
 
   def link_to_view obj, user_type, options = {}
     return unless send(:"#{user_type}_can?", :edit, obj)
     # use i18n translation on label
-    link_to t(".view"), send view_method(obj), obj, options)
+    link_to t(".view"), send(view_method(obj), obj, options)
   end
 
   protected
 
   def view_method obj
     "#{obj.class.to_s.underscore}_path"
+  end
+
+  def rest_action obj, action, options
+    send action_method(obj, :edit), obj, options
   end
 
   def action_method obj, action
     "#{action}_#{obj.class.to_s.underscore}_path"
   end