Add layout specific injections

CHANGELOG.md

@@ -5,6 +5,7 @@
 - Enh #79: Add debug collector for yiisoft/yii-debug (@xepozz)
 - Bug #82: Fixed find for layout file due to compatibility with `yiisoft/view` (@rustamwin)
 - Enh #99: Make `viewPath` in `ViewRenderer` constructor optional (@vjik)
+- New #102: Add layout specific injections (@vjik) 
 
 ## 6.0.0 February 16, 2023
 

src/LayoutSpecificInjections.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Yii\View;
+
+final class LayoutSpecificInjections
+{
+    /**
+     * @var object[]
+     */
+    private array $injections;
+
+    public function __construct(
+        private string $layout,
+        object ...$injections
+    ) {
+        $this->injections = $injections;
+    }
+
+    /**
+     * @return object[]
+     */
+    public function getInjections(): array
+    {
+        return $this->injections;
+    }
+
+    public function getLayout(): string
+    {
+        return $this->layout;
+    }
+}

src/ViewRenderer.php

@@ -147,11 +147,11 @@ public function renderPartial(string $view, array $parameters = []): DataRespons
      *
      * @psalm-param array<string, mixed> $parameters
      *
-     * @throws RuntimeException If the view cannot be resolved.
+     * @return string The rendering result.
      * @throws Throwable If an error occurred during rendering.
      * @throws ViewNotFoundException If the view file does not exist.
      *
-     * @return string The rendering result.
+     * @throws RuntimeException If the view cannot be resolved.
      */
     public function renderAsString(string $view, array $parameters = []): string
     {
@@ -174,11 +174,11 @@ public function renderAsString(string $view, array $parameters = []): string
      *
      * @psalm-param array<string, mixed> $parameters
      *
-     * @throws RuntimeException If the view cannot be resolved.
+     * @return string The rendering result.
      * @throws Throwable If an error occurred during rendering.
      * @throws ViewNotFoundException If the view file does not exist.
      *
-     * @return string The rendering result.
+     * @throws RuntimeException If the view cannot be resolved.
      */
     public function renderPartialAsString(string $view, array $parameters = []): string
     {
@@ -290,11 +290,11 @@ public function withLocale(string $locale): self
      * @psalm-param array<string, mixed> $injectCommonParameters
      * @psalm-param array<string, mixed> $injectLayoutParameters
      *
+     * @return string The rendering result.
      * @throws RuntimeException If the view cannot be resolved.
      * @throws Throwable If an error occurred during rendering.
      * @throws ViewNotFoundException If the view file does not exist.
      *
-     * @return string The rendering result.
      */
     private function renderProxy(
         string $view,
@@ -348,10 +348,8 @@ private function renderProxy(
     private function getCommonParameters(): array
     {
         $parameters = [];
-        foreach ($this->injections as $injection) {
-            if ($injection instanceof CommonParametersInjectionInterface) {
-                $parameters = array_merge($parameters, $injection->getCommonParameters());
-            }
+        foreach ($this->getInjections($this->layout, CommonParametersInjectionInterface::class) as $injection) {
+            $parameters = array_merge($parameters, $injection->getCommonParameters());
         }
         return $parameters;
     }
@@ -366,10 +364,8 @@ private function getCommonParameters(): array
     private function getLayoutParameters(): array
     {
         $parameters = [];
-        foreach ($this->injections as $injection) {
-            if ($injection instanceof LayoutParametersInjectionInterface) {
-                $parameters = array_merge($parameters, $injection->getLayoutParameters());
-            }
+        foreach ($this->getInjections($this->layout, LayoutParametersInjectionInterface::class) as $injection) {
+            $parameters = array_merge($parameters, $injection->getLayoutParameters());
         }
         return $parameters;
     }
@@ -382,10 +378,8 @@ private function getLayoutParameters(): array
     private function getMetaTags(): array
     {
         $tags = [];
-        foreach ($this->injections as $injection) {
-            if ($injection instanceof MetaTagsInjectionInterface) {
-                $tags = array_merge($tags, $injection->getMetaTags());
-            }
+        foreach ($this->getInjections($this->layout, MetaTagsInjectionInterface::class) as $injection) {
+            $tags = array_merge($tags, $injection->getMetaTags());
         }
         return $tags;
     }
@@ -398,12 +392,34 @@ private function getMetaTags(): array
     private function getLinkTags(): array
     {
         $tags = [];
+        foreach ($this->getInjections($this->layout, LinkTagsInjectionInterface::class) as $injection) {
+            $tags = array_merge($tags, $injection->getLinkTags());
+        }
+        return $tags;
+    }
+
+    /**
+     * @psalm-template T
+     * @psalm-param class-string<T> $injectionInterface
+     * @psalm-return list<T>
+     */
+    private function getInjections(?string $layout, string $injectionInterface): array
+    {
+        $result = [];
         foreach ($this->injections as $injection) {
-            if ($injection instanceof LinkTagsInjectionInterface) {
-                $tags = array_merge($tags, $injection->getLinkTags());
+            if ($injection instanceof $injectionInterface) {
+                $result[] = $injection;
+                continue;
+            }
+            if ($injection instanceof LayoutSpecificInjections && $injection->getLayout() === $layout) {
+                foreach ($injection->getInjections() as $layoutInjection) {
+                    if ($layoutInjection instanceof $injectionInterface) {
+                        $result[] = $layoutInjection;
+                    }
+                }
             }
         }
-        return $tags;
+        return $result;
     }
 
     /**

tests/ViewRendererTest.php

@@ -17,6 +17,8 @@
 use Yiisoft\View\WebView;
 use Yiisoft\Yii\View\Exception\InvalidLinkTagException;
 use Yiisoft\Yii\View\Exception\InvalidMetaTagException;
+use Yiisoft\Yii\View\LayoutSpecificInjections;
+use Yiisoft\Yii\View\MetaTagsInjectionInterface;
 use Yiisoft\Yii\View\Tests\Support\FakeCntrl;
 use Yiisoft\Yii\View\Tests\Support\FakeController;
 use Yiisoft\Yii\View\Tests\Support\InvalidLinkTagInjection;
@@ -480,6 +482,38 @@ public function testWithoutViewPath(): void
         $viewRenderer->getViewPath();
     }
 
+    public function testLayoutSpecificInjections(): void
+    {
+        $renderer = $this
+            ->getRenderer()
+            ->withLayout('@views/nested-layout/layout')
+            ->withInjections(
+                new LayoutSpecificInjections(
+                    '@views/nested-layout/layout',
+                    new TitleInjection(),
+                ),
+                new LayoutSpecificInjections(
+                    '@views/layout',
+                    new TestInjection(),
+                ),
+                new class() implements MetaTagsInjectionInterface {
+                    public function getMetaTags(): array
+                    {
+                        return [
+                            ['charset' => 'windows-1251']
+                        ];
+                    }
+                }
+            );
+
+        $response = $renderer->render('empty');
+
+        $this->assertSame(
+            '<html><head><title>Hello</title><meta charset="windows-1251"></head><body><h1>Hello</h1></body></html>',
+            (string) $response->getBody(),
+        );
+    }
+
     public function testImmutability(): void
     {
         $original = $this->getRenderer();

tests/views/nested-layout/head.php

@@ -9,4 +9,5 @@
 
 echo '<head>';
 echo '<title>' . $title . '</title>';
+$this->head();
 echo '</head>';

tests/views/nested-layout/layout.php

@@ -8,10 +8,12 @@
  * @var string $title
  */
 
+$this->beginPage();
 echo '<html>';
 echo $this->render('head');
 echo '<body>';
 echo '<h1>' . $title . '</h1>';
 echo $content;
 echo '</body>';
 echo '</html>';
+$this->endPage();