<?php
 
namespace DrSlump\Protobuf;
 
use DrSlump\Protobuf;
 
/**
 * Keeps instances of the different message descriptors used.
 *
 */
class Registry
{
    /** @var array */
    protected $descriptors = array();
 
    /**
     * @param string|\DrSlump\Protobuf\Message $message
     * @param \DrSlump\Protobuf\Descriptor $descriptor
     */
    public function setDescriptor($message, Descriptor $descriptor)
    {
        $message = is_object($message) ? get_class($message) : $message;
        $message = ltrim($message, '\\');
 
        $this->descriptors[$message] = $descriptor;
    }
 
    /**
     * Obtains the descriptor for the given message class, obtaining
     * it if not yet loaded.
     *
     * @param string|\DrSlump\Protobuf\Message $message
     * @return \DrSlump\Protobuf\Descriptor
     */
    public function getDescriptor($message)
    {
        $message = is_object($message) ? get_class($message) : $message;
        $message = ltrim($message, '\\');
 
        // Build a descriptor for the message
        if (!isset($this->descriptors[$message])) {
            $class = '\\' . $message;
            if (!class_exists($class)) {
                throw Protobuf\Exception('Message class "' . $class . '" not available');
            }
 
            $this->descriptors[$message] = $class::descriptor();
        }
 
        return $this->descriptors[$message];
    }
 
    /**
     * @param string|\DrSlump\Protobuf\Message $message
     * @return bool
     */
    public function hasDescriptor($message)
    {
        $message = is_object($message) ? get_class($message) : $message;
        $message = ltrim($message, '\\');
 
        return isset($this->descriptors[$message]);
    }
 
    /**
     * @param string|\DrSlump\Protobuf\Message $message
     */
    public function unsetDescriptor($message)
    {
        $message = is_object($message) ? get_class($message) : $message;
        $message = ltrim($message, '\\');
 
        unset($this->descriptors[$message]);
    }
}