symfony · Mar 9, 2011 · Feb 18, 2011 · Feb 18, 2011 · Feb 18, 2011 · Feb 19, 2011
diff --git a/src/Symfony/Bundle/DoctrineMongoDBBundle/DependencyInjection/Configuration.php b/src/Symfony/Bundle/DoctrineMongoDBBundle/DependencyInjection/Configuration.php
@@ -0,0 +1,203 @@
+<?php
+
+namespace Symfony\Bundle\DoctrineMongoDBBundle\DependencyInjection;
+
+use Symfony\Component\Config\Definition\Builder\NodeBuilder;
+use Symfony\Component\Config\Definition\Builder\TreeBuilder;
+
+/**
+ * FrameworkExtension configuration structure.
+ *
+ * @author Ryan Weaver <[email protected]>
+ */
+class Configuration
+{
+    /**
+     * Generates the configuration tree.
+     *
+     * @param boolean $kernelDebug The kernel.debug DIC parameter
+     * @return \Symfony\Component\DependencyInjection\Configuration\NodeInterface
+     */
+    public function getConfigTree()
+    {
+        $treeBuilder = new TreeBuilder();
+        $rootNode = $treeBuilder->root('doctrine_mongo_db', 'array');
+
+        $this->addSingleDocumentManagerSection($rootNode);
+        $this->addDocumentManagersSection($rootNode);
+        $this->addSingleConnectionSection($rootNode);
+        $this->addConnectionsSection($rootNode);
+
+        $rootNode
+            ->scalarNode('proxy_namespace')->defaultValue('Proxies')->end()
+            ->scalarNode('auto_generate_proxy_classes')->defaultValue(false)->end()
+            ->scalarNode('hydrator_namespace')->defaultValue('Hydrators')->end()
+            ->scalarNode('auto_generate_hydrator_classes')->defaultValue(false)->end()
+        ;
+
+        return $treeBuilder->buildTree();
+    }
+
+    /**
+     * Builds the nodes responsible for the config that supports the single
+     * document manager.
+     */
+    private function addSingleDocumentManagerSection(NodeBuilder $rootNode)
+    {
+        $rootNode
+            ->scalarNode('default_document_manager')->defaultValue('default')->end()
+            ->scalarNode('default_database')->defaultValue('default')->end()
+            ->builder($this->getMetadataCacheDriverNode())
+            ->fixXmlConfig('mapping')
+            ->builder($this->getMappingsNode())
+        ;
+    }
+
+    /**
+     * Configures the "document_managers" section
+     */
+    private function addDocumentManagersSection(NodeBuilder $rootNode)
+    {
+        $rootNode
+            ->fixXmlConfig('document_manager')
+            ->arrayNode('document_managers')
+                ->useAttributeAsKey('id')
+                ->prototype('array')
+                    ->performNoDeepMerging()
+                    ->treatNullLike(array())
+                    ->builder($this->getMetadataCacheDriverNode())
+                    ->scalarNode('default_database')->end()
+                    ->scalarNode('connection')->end()
+                    ->scalarNode('database')->end()
+                    ->fixXmlConfig('mapping')
+                    ->builder($this->getMappingsNode())
+                ->end()
+            ->end()
+        ;
+    }
+
+    /**
+     * Configures the single-connection section:
+     *   * default_connection
+     *   * server
+     *   * options
+     */
+    private function addSingleConnectionSection(NodeBuilder $rootNode)
+    {
+        $rootNode
+            ->scalarNode('default_connection')->defaultValue('default')->end()
+            ->builder($this->addConnectionServerNode())
+            ->builder($this->addConnectionOptionsNode())
+        ;
+    }
+
+    /**
+     * Adds the configuration for the "connections" key
+     */
+    private function addConnectionsSection(NodeBuilder $rootNode)
+    {
+        $rootNode
+            ->fixXmlConfig('connection')
+            ->arrayNode('connections')
+                ->useAttributeAsKey('id')
+                ->prototype('array')
+                    ->performNoDeepMerging()
+                    ->builder($this->addConnectionServerNode())
+                    ->builder($this->addConnectionOptionsNode())
+                ->end()
+            ->end()
+        ;
+    }
+
+    /**
+     * Returns the array node used for "mappings".
+     *
+     * This is used in two different parts of the tree.
+     *
+     * @param NodeBuilder $rootNode The parent node
+     * @return NodeBuilder
+     */
+    protected function getMappingsNode()
+    {
+        $node = new Nodebuilder('mappings', 'array');
+        $node
+            ->useAttributeAsKey('name')
+            ->prototype('array')
+                ->beforeNormalization()
+                    // if it's not an array, then the scalar is the type key
+                    ->ifString()
+                    ->then(function($v) { return array ('type' => $v); })
+                ->end()
+                // I believe that "null" should *not* set the type
+                // it's guessed in AbstractDoctrineExtension::detectMetadataDriver
+                ->treatNullLike(array())
+                ->scalarNode('type')->end()
+                ->scalarNode('dir')->end()
+                ->scalarNode('prefix')->end()
+                ->scalarNode('alias')->end()
+                ->booleanNode('is_bundle')->end()
+                ->performNoDeepMerging()
+            ->end()
+        ;
+
+        return $node;
+    }
+
+    /**
+     * Adds the NodeBuilder for the "server" key of a connection.
+     */
+    private function addConnectionServerNode()
+    {
+        $node = new NodeBuilder('server', 'scalar');
+
+        $node
+            ->defaultValue(null)
+        ->end();
+
+        return $node;
+    }
+
+    /**
+     * Adds the NodeBuilder for the "options" key of a connection.
+     */
+    private function addConnectionOptionsNode()
+    {
+        $node = new NodeBuilder('options', 'array');
+
+        $node
+            ->performNoDeepMerging()
+            ->addDefaultsIfNotSet() // adds an empty array of omitted
+
+            // options go into the Mongo constructor
+            // http://www.php.net/manual/en/mongo.construct.php
+            ->booleanNode('connect')->end()
+            ->scalarNode('persist')->end()
+            ->scalarNode('timeout')->end()
+            ->booleanNode('replicaSet')->end()
+            ->scalarNode('username')->end()
+            ->scalarNode('password')->end()
+        ->end();
+
+        return $node;
+    }
+
+    private function getMetadataCacheDriverNode()
+    {
+        $node = new NodeBuilder('metadata_cache_driver', 'array');
+
+        $node
+            ->beforeNormalization()
+                // if scalar
+                ->ifTrue(function($v) { return !is_array($v); })
+                ->then(function($v) { return array('type' => $v); })
+            ->end()
+            ->scalarNode('type')->end()
+            ->scalarNode('class')->end()
+            ->scalarNode('host')->end()
+            ->scalarNode('port')->end()
+            ->scalarNode('instance_class')->end()
+        ->end();
+
+        return $node;
+    }
+}