From a1cb94127e9710ffc6d46cb63a9e855831ebd465 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Rapha=C3=ABl=20Gertz?= <git@rapsys.eu>
Date: Thu, 31 Oct 2024 07:46:48 +0100
Subject: [PATCH] Add repository factory

---
 Factory.php | 66 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 Factory.php

diff --git a/Factory.php b/Factory.php
new file mode 100644
index 0000000..1856f3f
--- /dev/null
+++ b/Factory.php
@@ -0,0 +1,66 @@
+<?php declare(strict_types=1);
+
+/*
+ * This file is part of the Rapsys TreeBundle package.
+ *
+ * (c) Raphaël Gertz <symfony@rapsys.eu>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Rapsys\TreeBundle;
+
+use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\ORM\Repository\RepositoryFactory;
+use Doctrine\Persistence\ObjectRepository;
+
+use Psr\Container\ContainerInterface;
+
+use Rapsys\PackBundle\Util\SluggerUtil;
+use Rapsys\UserBundle\Factory as BaseFactory;
+
+use Symfony\Component\HttpFoundation\RequestStack;
+use Symfony\Component\Routing\RouterInterface;
+use Symfony\Contracts\Translation\TranslatorInterface;
+
+/**
+ * This factory is used to create default repository objects for entities at runtime.
+ */
+final class Factory extends BaseFactory {
+	/**
+	 * Initializes a new RepositoryFactory instance
+	 *
+	 * @param ContainerInterface $container The container instance
+	 * @param RequestStack $request The request stack
+	 * @param RouterInterface $router The router instance
+	 * @param SluggerUtil $slugger The SluggerUtil instance
+	 * @param TranslatorInterface $translator The TranslatorInterface instance
+	 * @param string $locale The current locale
+	 * @param array $languages The languages list
+	 */
+	public function __construct(private ContainerInterface $container, private RequestStack $request, private RouterInterface $router, private SluggerUtil $slugger, private TranslatorInterface $translator, private string $locale, private array $languages) {
+	}
+
+	/**
+	 * Create a new repository instance for an entity class
+	 *
+	 * @param EntityManagerInterface $entityManager The EntityManager instance.
+	 * @param string $entityName The name of the entity.
+	 */
+	protected function createRepository(EntityManagerInterface $entityManager, string $entityName): ObjectRepository {
+		//Get class metadata
+		$metadata = $entityManager->getClassMetadata($entityName);
+
+		//Get repository class
+		$repositoryClass = $metadata->customRepositoryClassName ?: $entityManager->getConfiguration()->getDefaultRepositoryClassName();
+
+		//Set to current locale
+		//XXX: current request is not yet populated in constructor
+		$this->locale = $this->request->getCurrentRequest()?->getLocale() ?? $this->locale;
+
+		//Return repository class instance
+		//XXX: router, slugger, translator, languages and locale arguments will be ignored by default
+		return new $repositoryClass($entityManager, $metadata, $this->container, $this->router, $this->slugger, $this->translator, $this->locale, $this->languages);
+	}
+}
-- 
2.41.1