sunitparekh/data-anonymization

module DataAnon
  module Strategy
    module Field

      class DefaultAnon

        DEFAULT_STRATEGIES = {:string => FieldStrategy::RandomString.new,
                              :integer => FieldStrategy::RandomIntegerDelta.new(5),
                              :fixnum => FieldStrategy::RandomIntegerDelta.new(5),
                              :bignum => FieldStrategy::RandomIntegerDelta.new(5000),
                              :float => FieldStrategy::RandomFloatDelta.new(5.0),
                              :bigdecimal => FieldStrategy::RandomBigDecimalDelta.new(500.0),
                              :datetime => FieldStrategy::DateTimeDelta.new,
                              :time => FieldStrategy::TimeDelta.new,
                              :date => FieldStrategy::DateDelta.new,
                              :array => FieldStrategy::AnonymizeArray.new(nil),
                              :trueclass => FieldStrategy::RandomBoolean.new,
                              :"bson::objectid" => FieldStrategy::Whitelist.new,
                              :falseclass => FieldStrategy::RandomBoolean.new
        }

        def initialize user_defaults = {}
          @user_defaults = DEFAULT_STRATEGIES.merge user_defaults
          FieldStrategy::AnonymizeArray.user_defaults @user_defaults
        end

        def anonymize field
          strategy = @user_defaults[field.value.class.to_s.downcase.to_sym]
          raise "No strategy defined for datatype #{field.value.class}. Use 'default_field_strategies' option in your script. Refer to  http://sunitparekh.github.com/data-anonymization/#default-field-strategies for more details. #{field.inspect}" unless strategy
          strategy.anonymize field
        end

      end


    end
  end
end