add sf document classes as param

php-arsenal · Sep 16, 2021 · 7109539 · 7109539
1 parent 433591b
commit 7109539
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/DependencyInjection/SalesforceMapperExtension.php b/src/DependencyInjection/SalesforceMapperExtension.php
@@ -2,6 +2,7 @@
 
 namespace PhpArsenal\SalesforceMapperBundle\DependencyInjection;
 
+use PhpArsenal\SalesforceMapperBundle\Builder\SalesforceDocumentClassTreeBuilder;
 use Symfony\Component\Config\FileLocator;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Loader;
@@ -38,6 +39,10 @@ public function load(array $configs, ContainerBuilder $container)
 
         $yamlLoader->load('param_converter.yml');
         $container->setParameter('salesforce_mapper.param_converter', $config['param_converter']);
+
+        /** @var SalesforceDocumentClassTreeBuilder $sfDocumentClassTreeBuilder */
+        $sfDocumentClassTreeBuilder = $container->get(SalesforceDocumentClassTreeBuilder::class);
+        $container->setParameter('salesforce_mapper.document_classes', $sfDocumentClassTreeBuilder->build());
     }
 }