louismullie/treat

# Unserialization of entities stored in a Mongo database.
class Treat::Workers::Formatters::Unserializers::Mongo
  
  require 'mongo'
  
  def self.unserialize(entity, options={})
    
    db = options.delete(:db)
    selector = options
    
    if !Treat.databases.mongo.db && !db
      raise Treat::Exception,
      'Must supply the database name in config. ' +
      '(Treat.databases.mongo.db = ...) or pass ' +
      'it as a parameter to #unserialize.'
    end
    
    @@database ||= Mongo::Connection.
    new(Treat.databases.mongo.host).
    db(db || Treat.databases.mongo.db)
    
    supertype =  Treat::Entities.const_get(
    entity.type.to_s.capitalize.intern).superclass.mn.downcase
    supertype = entity.type.to_s if supertype == 'entity'
    supertypes = supertype + 's'
    supertypes = 'documents' if entity.type == :collection
    coll = @@database.collection(supertypes)
    records = coll.find(selector).to_a
    
    if records.size == 0
      raise Treat::Exception,
      "Couldn't find any records using " +
      "selector #{selector.inspect}."
    end
    
    if entity.type == :document 
      if records.size == 1
        self.do_unserialize(
        records.first, options)
      else
        raise Treat::Exception,
        "More than one document matched" +
        "your selector #{selector.inspect}."
      end
    elsif entity.type == :collection
      collection = Treat::Entities::Collection.new
      records.each do |record|
        collection << self.
        do_unserialize(record, options)
      end
      collection
    end
    
  end

  def self.do_unserialize(record, options)
    
    entity = Treat::Entities.
    const_get(record['type'].
    capitalize.intern).new(
    record['value'], record['id'])
    features = record['features']
    new_feat = {}
    features.each do |feature, value|
      new_feat[feature.intern] = value
    end
    
    entity.features = new_feat
    
    record['children'].each do |c|
      entity << self.do_unserialize(c, options)
    end

    entity
    
  end
  
end