Add impersonation_path twig function to generate impersonation path

Author: PhilETaylor

src/Symfony/Bridge/Twig/Extension/SecurityExtension.php

@@ -69,12 +69,22 @@ public function getImpersonateExitPath(string $exitTo = null): string
         return $this->impersonateUrlGenerator->generateExitPath($exitTo);
     }
 
+    public function getImpersonatePath(string $identifier = null): string
+    {
+        if (null === $this->impersonateUrlGenerator) {
+            return '';
+        }
+
+        return $this->impersonateUrlGenerator->generateImpersonationPath($identifier);
+    }
+
     public function getFunctions(): array
     {
         return [
             new TwigFunction('is_granted', $this->isGranted(...)),
             new TwigFunction('impersonation_exit_url', $this->getImpersonateExitUrl(...)),
             new TwigFunction('impersonation_exit_path', $this->getImpersonateExitPath(...)),
+            new TwigFunction('impersonation_path', $this->getImpersonatePath(...)),
         ];
     }
 }

src/Symfony/Component/Security/Http/Impersonate/ImpersonateUrlGenerator.php

@@ -18,7 +18,7 @@
 use Symfony\Component\Security\Http\Firewall\SwitchUserListener;
 
 /**
- * Provides generator functions for the impersonate url exit.
+ * Provides generator functions for the impersonation urls.
  *
  * @author Amrouche Hamza <[email protected]>
  * @author Damien Fayet <[email protected]>
@@ -36,9 +36,14 @@ public function __construct(RequestStack $requestStack, FirewallMap $firewallMap
         $this->firewallMap = $firewallMap;
     }
 
+    public function generateImpersonationPath(string $identifier = null): string
+    {
+        return $this->buildPath(null, $identifier);
+    }
+
     public function generateExitPath(string $targetUri = null): string
     {
-        return $this->buildExitPath($targetUri);
+        return $this->buildPath($targetUri);
     }
 
     public function generateExitUrl(string $targetUri = null): string
@@ -47,27 +52,31 @@ public function generateExitUrl(string $targetUri = null): string
             return '';
         }
 
-        return $request->getUriForPath($this->buildExitPath($targetUri));
+        return $request->getUriForPath($this->buildPath($targetUri));
     }
 
     private function isImpersonatedUser(): bool
     {
         return $this->tokenStorage->getToken() instanceof SwitchUserToken;
     }
 
-    private function buildExitPath(string $targetUri = null): string
+    private function buildPath(string $targetUri = null, string $identifier = SwitchUserListener::EXIT_VALUE): string
     {
-        if (null === ($request = $this->requestStack->getCurrentRequest()) || !$this->isImpersonatedUser()) {
+        if (null === ($request = $this->requestStack->getCurrentRequest())) {
+            return '';
+        }
+
+        if (!$this->isImpersonatedUser() && $identifier == SwitchUserListener::EXIT_VALUE){
             return '';
         }
 
         if (null === $switchUserConfig = $this->firewallMap->getFirewallConfig($request)->getSwitchUser()) {
-            throw new \LogicException('Unable to generate the impersonate exit URL without a firewall configured for the user switch.');
+            throw new \LogicException('Unable to generate the impersonate URLs without a firewall configured for the user switch.');
         }
 
         $targetUri ??= $request->getRequestUri();
 
-        $targetUri .= (parse_url($targetUri, \PHP_URL_QUERY) ? '&' : '?').http_build_query([$switchUserConfig['parameter'] => SwitchUserListener::EXIT_VALUE], '', '&');
+        $targetUri .= (parse_url($targetUri, \PHP_URL_QUERY) ? '&' : '?').http_build_query([$switchUserConfig['parameter'] => $identifier], '', '&');
 
         return $targetUri;
     }