Fix memory leak

App/Request/PathInfoProcessor.php

@@ -0,0 +1,48 @@
+<?php
+/**
+ * Copyright © OpenGento, All rights reserved.
+ */
+declare(strict_types=1);
+
+namespace Opengento\StorePathUrl\App\Request;
+
+use Magento\Framework\App\RequestInterface;
+use Magento\Framework\App\Router\Base;
+use Magento\Framework\Exception\LocalizedException;
+use Magento\Store\Api\StoreRepositoryInterface;
+use Magento\Store\App\Request\PathInfoProcessor as AppPathInfoProcessor;
+use Magento\Store\App\Request\StorePathInfoValidator;
+use Opengento\StorePathUrl\Service\PathResolver;
+
+use function str_starts_with;
+use function strlen;
+use function substr;
+
+class PathInfoProcessor extends AppPathInfoProcessor
+{
+    public function __construct(
+        private StorePathInfoValidator $storePathInfoValidator,
+        private PathResolver $pathResolver,
+        private StoreRepositoryInterface $storeRepository
+    ) {}
+
+    public function process(RequestInterface $request, $pathInfo): string
+    {
+        $storeCode = $this->storePathInfoValidator->getValidStoreCode($request, $pathInfo);
+        if (!empty($storeCode)) {
+            try {
+                $path = $this->pathResolver->resolve($this->storeRepository->getActiveStoreByCode($storeCode));
+            } catch (LocalizedException) {
+                return $pathInfo;
+            }
+            if (!$request->isDirectAccessFrontendName($path)) {
+                $pathInfo = substr($pathInfo, strlen($path) + (int)str_starts_with($pathInfo, '/')) ?: '/';
+            } else {
+                //no route in case we're trying to access a store that has the same code as a direct access
+                $request->setActionName(Base::NO_ROUTE);
+            }
+        }
+
+        return $pathInfo;
+    }
+}

Plugin/App/Request/StorePathInfoValidator.php

@@ -0,0 +1,36 @@
+<?php
+/**
+ * Copyright © OpenGento, All rights reserved.
+ */
+declare(strict_types=1);
+
+namespace Opengento\StorePathUrl\Plugin\App\Request;
+
+use Magento\Framework\App\Request\Http;
+use Magento\Store\Api\StoreRepositoryInterface;
+use Magento\Store\App\Request\StorePathInfoValidator as Subject;
+use Magento\Store\Model\Store;
+use Opengento\StorePathUrl\Model\Config;
+
+class StorePathInfoValidator
+{
+    public function __construct(
+        private Config $config,
+        private StoreRepositoryInterface $storeRepository
+    ) {}
+
+    public function beforeGetValidStoreCode(Subject $subject, Http $request, string $pathInfo = ''): array
+    {
+        if ($this->config->isEnabled()) {
+            $uri = $request->getUriString();
+            /** @var Store $store */
+            foreach ($this->storeRepository->getList() as $store) {
+                if ($store->getId() && str_starts_with($uri, $store->getBaseUrl())) {
+                    $pathInfo = $store->getCode();
+                }
+            }
+        }
+
+        return [$request, $pathInfo];
+    }
+}

etc/di.xml

@@ -15,10 +15,15 @@
             </argument>
         </arguments>
     </type>
+    <type name="Magento\Backend\App\Request\PathInfoProcessor">
+        <arguments>
+            <argument name="subject" xsi:type="object">Opengento\StorePathUrl\App\Request\PathInfoProcessor</argument>
+        </arguments>
+    </type>
     <type name="Magento\Framework\Url\ScopeInterface">
         <plugin name="Opengento_StorePathUrl::prepend_custom_path" type="Opengento\StorePathUrl\Plugin\Url\Scope"/>
     </type>
-    <type name="Magento\Framework\App\Request\PathInfo">
-        <plugin name="Opengento_StorePathUrl::replace_country_store_code" type="Opengento\StorePathUrl\Plugin\App\Request\PathInfo"/>
+    <type name="Magento\Store\App\Request\StorePathInfoValidator">
+        <plugin name="Opengento_StorePathUrl::resolve_store_code" type="Opengento\StorePathUrl\Plugin\App\Request\StorePathInfoValidator"/>
     </type>
 </config>