rails--rails/actionpack/lib/action_controller/metal/hide_actions.rb

49 lines
1.4 KiB
Ruby
Raw Normal View History

2010-02-01 02:32:28 +00:00
require 'active_support/core_ext/class/attribute'
module ActionController
2009-06-10 21:43:43 +00:00
# ActionController::HideActions adds the ability to prevent public methods on a controller
# to be called as actions.
module HideActions
extend ActiveSupport::Concern
included do
2010-02-01 02:32:28 +00:00
class_attribute :hidden_actions
self.hidden_actions = Set.new.freeze
end
2009-06-10 21:43:43 +00:00
private
# Overrides AbstractController::Base#action_method? to return false if the
# action name is in the list of hidden actions.
def method_for_action(action_name)
self.class.visible_action?(action_name) && super
2009-06-10 21:43:43 +00:00
end
2009-06-10 21:43:43 +00:00
module ClassMethods
# Sets all of the actions passed in as hidden actions.
#
# ==== Parameters
# *args<#to_s>:: A list of actions
2009-06-10 21:43:43 +00:00
def hide_action(*args)
self.hidden_actions = hidden_actions.dup.merge(args.map(&:to_s)).freeze
2009-06-10 21:43:43 +00:00
end
def inherited(klass)
klass.class_eval { @visible_actions = {} }
super
end
def visible_action?(action_name)
return @visible_actions[action_name] if @visible_actions.key?(action_name)
@visible_actions[action_name] = !hidden_actions.include?(action_name)
end
# Overrides AbstractController::Base#action_methods to remove any methods
# that are listed as hidden methods.
2009-06-10 21:43:43 +00:00
def action_methods
@action_methods ||= Set.new(super.reject { |name| hidden_actions.include?(name) })
end
2009-06-10 21:43:43 +00:00
end
end
end