Updated twig extension

lg2 · Jan 23, 2023 · 06ea054 · 06ea054
1 parent acd3796
commit 06ea054
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/modules/site/web/TwigExtension.php b/modules/site/web/TwigExtension.php
@@ -4,6 +4,7 @@
 
 use modules\site\helpers\StringHelper;
 
+use Craft;
 use craft\base\Element;
 use craft\helpers\App;
 
@@ -25,11 +26,11 @@ public function getFunctions(): array
                 $url = null;
                 $site = $element->getSite();
 
-                if ($site->handle == 'en') {
+                if ($site->handle === 'en') {
                     $url = App::env('PREVIEW_EN_URL');
                 }
 
-                if ($site->handle == 'fr') {
+                if ($site->handle === 'fr') {
                     $url = App::env('PREVIEW_FR_URL');
                 }
 
@@ -43,6 +44,10 @@ public function getFunctions(): array
                     $url .= $element->uri;
                 }
 
+                if (Craft::$app->getConfig()->getGeneral()->addTrailingSlashesToUrls) {
+                    $url = StringHelper::ensureRight($url, '/');
+                }
+
                 return $url.'?preview=true';
             }),