module Omelettes
  module ModelAdditions
    module ClassMethods
      def column_config(column_name)
        @column_config ||= {}
        @column_config[column_name.to_s]
      end
      
      def treat(column_name, style=nil, &block)
        @column_config ||= {}
        column = Column.new(column_name, style, &block)
        @column_config[column_name.to_s] = column
        column
      end
      alias :scramble :treat
      
      def ignore(column_name)
        @column_config ||= {}
        column = Column.new(column_name, :hardened)
        @column_config[column_name.to_s] = column
        column
      end
      alias :harden :ignore
    end

    def obfuscate(column_name)
      column = self.class.column_config(column_name)
      original_value = self.send(column_name)
      if column
        value = column.process(original_value)
      else
        value = Column.default(column_name, original_value)
      end
      self.update_attribute(column_name, value) if original_value != value
    end

    def self.included(base)
      base.extend ClassMethods
    end
  end
end

ActiveRecord::Base.class_eval do
  include Omelettes::ModelAdditions
end