Merge pull request #134 from jingu/save-donut-headers

Add ResourceDonut header to save headers
bearsunday · Mar 26, 2024 · de00720 · de00720
2 parents 6cbbef9 + 858370e
commit de00720
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/src/ResourceDonut.php b/src/ResourceDonut.php
@@ -22,8 +22,10 @@ final class ResourceDonut
 
     private const URI_REGEX = '/\[le:(.+)]/';
 
+    /** @param array<string, string> $headers */
     public function __construct(
         private string $template,
+        private array $headers,
         /** @readonly */
         public int|null $ttl,
         /** @readonly */
@@ -45,8 +47,9 @@ public function refresh(ResourceInterface $resource, ResourceObject $ro): Resour
             return (string) $ro->view;
         }, $this->template);
 
-        $etags->setSurrogateHeader($ro);
+        $ro->headers = $this->headers;
         $ro->view = $refreshView;
+        $etags->setSurrogateHeader($ro);
 
         return $ro;
     }
@@ -72,6 +75,6 @@ public static function create(ResourceObject $ro, DonutRendererInterface $storag
         unset($maybeRequest);
         $donutTemplate = (string) $ro;
 
-        return new self($donutTemplate, $ttl, $isCacheble);
+        return new self($donutTemplate, $ro->headers, $ttl, $isCacheble);
     }
 }
diff --git a/src/ResourceStorageSaver.php b/src/ResourceStorageSaver.php
@@ -19,7 +19,7 @@ public function __invoke(string $key, mixed $value, CacheItemPoolInterface $pool
         assert($cacheItem instanceof CacheItem);
         $cacheItem->tag($tags);
 
-        if ($ttl) {
+        if ($ttl !== null && $ttl > 0) {
             $cacheItem->expiresAfter($ttl);
         }
 

diff --git a/tests/DonutCacheTest.php b/tests/DonutCacheTest.php
@@ -30,10 +30,12 @@ protected function setUp(): void
 
     public function testGetState(): void
     {
-        $donut = new ResourceDonut('cmt=[le:page://self/html/comment]', null, true);
+        $headers = ['Content-Type' => 'application/xml;'];
+        $donut = new ResourceDonut('cmt=[le:page://self/html/comment]', $headers, null, true);
         $blog = $this->resource->get('page://self/html/blog-posting');
         $ro = $donut->refresh($this->resource, $blog);
         $this->assertInstanceOf(ResourceObject::class, $ro);
         $this->assertSame('cmt=comment01', $ro->view);
+        $this->assertSame($headers['Content-Type'], $ro->headers['Content-Type']);
     }
 }
diff --git a/tests/DonutRepositoryTest.php b/tests/DonutRepositoryTest.php
@@ -116,6 +116,7 @@ public function testRefresh(): void
         assert($queryRepository->purge(new Uri('page://self/html/comment')));
         $donutRo = $resource->get('page://self/html/blog-posting');
         $this->assertSame('r', $donutRo->headers[Header::ETAG][-1]);
+        $this->assertStringContainsString('blog-posting-page', $donutRo->headers[Header::SURROGATE_KEY]);
     }
 
     public function testInvalidateTags(): void