obfusk-data w/ mongoid

obfusk-data-mongoid.rb

require 'obfusk/data/hash'
require 'mongoid'

Mongoid.load! "#{ Dir.home }/tmp/_/mongoid.yml", :development

module ValidMongoid
  module ClassMethods
    def data (*a, &b)
      self::ValidHash.data(*a, &b)
    end

    def union (*a, &b)
      self::ValidHash.union(*a, &b)
    end

    def instantiate (*a, &b)
      doc   = super
      attrs = doc.instance_variable_get :@attributes
      valid = self::ValidHash.new attrs
      doc.instance_variable_set :@attributes, valid
    end
  end

  def self.included (base)
    base.extend ClassMethods

    base.class_eval do
      const_set :ValidHash, Class.new(Obfusk::Data::ValidHash)

      def initialize (*a, &b)
        super
        @attributes = self.class::ValidHash.new @attributes
      end
    end
  end
end

class Foo
  include Mongoid::Document
  include ValidMongoid

  # field :foo
  # field :bar

  data do
    field %w{ _id foo bar }, []
  end
end