Contents

module SafeAttributes
  extend ActiveSupport::Concern

  module ClassMethods
    # 
    # Within your model call this method once with a list of 
    # methods matching either attribute() or attribute=() for 
    # attributes (column names) you do not want to create the
    # the normal method for.  You should not need to do this
    # but the option is there in case the default list is 
    # inadequate.
    #
    def bad_attribute_names(*attrs)
      @bad_attribute_names ||= lambda {
        methods = Array.new(attrs.collect { |x| x.to_sym })
        methods += ActiveRecord::Base.public_instance_methods.collect { |x| x.to_sym }
        methods += ActiveRecord::Base.protected_instance_methods.collect { |x| x.to_sym }
        methods += ActiveRecord::Base.private_instance_methods.collect { |x| x.to_sym }
        methods -= [:id]
        return methods
      }.call
    end

    # 
    # Override the default implementation to not create the
    # attribute() method if that method name is in the list of
    # bad names
    #
    def define_method_attribute(attr_name)
      return if (bad_attribute_names.include?(attr_name.to_sym))
      super(attr_name)
    end

    #
    # Override the default implementation to not create the
    # attribute= method if that method name is in the list of
    # bad names
    #
    def define_method_attribute=(attr_name)
      method = attr_name + '='
      return if (bad_attribute_names.include?(method.to_sym))
      super(attr_name)
    end

    def instance_method_already_implemented?(method_name)
      begin 
        return super(method_name)
      rescue ActiveRecord::DangerousAttributeError
        return true
      end
      return false
    end
  end

  def read_attribute_for_validation(attr)
    if (self.attributes.include?(attr.to_s))
      self[attr.to_sym]
    else
      self.send(attr.to_s) if (self.respond_to?(attr.to_sym))
    end
  end

end

require 'safe_attributes/railtie.rb' if defined?(Rails)

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
safe_attributes-1.0.8	lib/safe_attributes.rb