Add cross OM references support

DependencyInjection/Configuration.php

@@ -49,6 +49,7 @@ private function getVendorNode($name)
             ->prototype('array')
                 ->performNoDeepMerging()
                 ->children()
+
                     ->scalarNode('translatable')->defaultFalse()->end()
                     ->scalarNode('timestampable')->defaultFalse()->end()
                     ->scalarNode('blameable')->defaultFalse()->end()
@@ -59,6 +60,7 @@ private function getVendorNode($name)
                     ->scalarNode('softdeleteable')->defaultFalse()->end()
                     ->scalarNode('uploadable')->defaultFalse()->end()
                     ->scalarNode('reference_integrity')->defaultFalse()->end()
+                    ->scalarNode('references')->cannotBeEmpty()->end()
                 ->end()
             ->end()
         ;
@@ -114,6 +116,10 @@ private function getClassNode()
                     ->cannotBeEmpty()
                     ->defaultValue('Gedmo\\ReferenceIntegrity\\ReferenceIntegrityListener')
                 ->end()
+                ->scalarNode('references')
+                    ->cannotBeEmpty()
+                    ->defaultValue('Stof\\DoctrineExtensionsBundle\\Listener\\ReferencesListener')
+                ->end()
             ->end()
         ;
 

DependencyInjection/StofDoctrineExtensionsExtension.php

@@ -37,6 +37,8 @@ public function load(array $configs, ContainerBuilder $container)
         $useLoggable = false;
         $useBlameable = false;
 
+        $managers = array();
+
         foreach ($config['orm'] as $name => $listeners) {
             foreach ($listeners as $ext => $enabled) {
                 $listener = sprintf('stof_doctrine_extensions.listener.%s', $ext);
@@ -53,6 +55,10 @@ public function load(array $configs, ContainerBuilder $container)
                     }
                     $definition = $container->getDefinition($listener);
                     $definition->addTag('doctrine.event_subscriber', $attributes);
+
+                    if ('references' === $ext) {
+                        $managers[$enabled] = sprintf('doctrine.orm.%s_entity_manager', $name);
+                    }
                 }
             }
 
@@ -75,6 +81,10 @@ public function load(array $configs, ContainerBuilder $container)
                     }
                     $definition = $container->getDefinition($listener);
                     $definition->addTag('doctrine_mongodb.odm.event_subscriber', $attributes);
+
+                    if ('references' === $ext) {
+                        $managers[$enabled] = sprintf('doctrine_mongodb.odm.%s_document_manager', $name);
+                    }
                 }
             }
             $this->documentManagers[$name] = $listeners;
@@ -117,6 +127,10 @@ public function load(array $configs, ContainerBuilder $container)
             );
         }
 
+        $container
+            ->getDefinition('stof_doctrine_extensions.listener.references')
+            ->setArguments(array(new Reference('service_container'), $managers));
+
         foreach ($config['class'] as $listener => $class) {
             $container->setParameter(sprintf('stof_doctrine_extensions.listener.%s.class', $listener), $class);
         }

Listener/ReferencesListener.php

@@ -0,0 +1,28 @@
+<?php
+
+namespace Stof\DoctrineExtensionsBundle\Listener;
+
+use Gedmo\References\ReferencesListener as BaseReferencesListener;
+use Symfony\Component\DependencyInjection\ContainerAwareInterface;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * ReferencesListener
+ *
+ * @author Bulat Shakirzyanov <[email protected]>
+ */
+class ReferencesListener extends BaseReferencesListener
+{
+    private $container;
+
+    public function __construct(ContainerInterface $container, array $managers = array())
+    {
+        $this->container = $container;
+        parent::__construct($managers);
+    }
+
+    public function getManager($type)
+    {
+        return $this->container->get(parent::getManager($type));
+    }
+}

Resources/config/listeners.xml

@@ -14,6 +14,11 @@
     </parameters>
     <services>
         <!-- Doctrine listeners -->
+        <service id="stof_doctrine_extensions.listener.references" class="%stof_doctrine_extensions.listener.references.class%" public="false">
+            <call method="setAnnotationReader">
+                <argument type="service" id="annotation_reader" />
+            </call>
+        </service>
         <service id="stof_doctrine_extensions.listener.timestampable" class="%stof_doctrine_extensions.listener.timestampable.class%" public="false">
             <call method="setAnnotationReader">
                 <argument type="service" id="annotation_reader" />