LocalAwareRedirectResponseTrait.php 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. <?php
  2. namespace Drupal\Core\Routing;
  3. use Drupal\Component\Utility\UrlHelper;
  4. /**
  5. * Provides a trait which ensures that a URL is safe to redirect to.
  6. */
  7. trait LocalAwareRedirectResponseTrait {
  8. /**
  9. * The request context.
  10. *
  11. * @var \Drupal\Core\Routing\RequestContext
  12. */
  13. protected $requestContext;
  14. /**
  15. * {@inheritdoc}
  16. */
  17. protected function isLocal($url) {
  18. return !UrlHelper::isExternal($url) || UrlHelper::externalIsLocal($url, $this->getRequestContext()->getCompleteBaseUrl());
  19. }
  20. /**
  21. * Returns the request context.
  22. *
  23. * @return \Drupal\Core\Routing\RequestContext
  24. */
  25. protected function getRequestContext() {
  26. if (!isset($this->requestContext)) {
  27. $this->requestContext = \Drupal::service('router.request_context');
  28. }
  29. return $this->requestContext;
  30. }
  31. /**
  32. * Sets the request context.
  33. *
  34. * @param \Drupal\Core\Routing\RequestContext $request_context
  35. * The request context.
  36. *
  37. * @return $this
  38. */
  39. public function setRequestContext(RequestContext $request_context) {
  40. $this->requestContext = $request_context;
  41. return $this;
  42. }
  43. }