fix: DownloadResponse cache headers

codeigniter4 · Oct 25, 2024 · d3e1e51 · d3e1e51
1 parent 11d9721
commit d3e1e51
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 20 deletions.
diff --git a/system/Exceptions/DownloadException.php b/system/Exceptions/DownloadException.php
@@ -47,6 +47,8 @@ public static function forNotFoundDownloadSource()
     }
 
     /**
+     * @deprecated Since v4.5.6
+     *
      * @return static
      */
     public static function forCannotSetCache()

diff --git a/system/HTTP/DownloadResponse.php b/system/HTTP/DownloadResponse.php
@@ -242,16 +242,6 @@ public function noCache(): self
         return $this;
     }
 
-    /**
-     * Disables cache configuration.
-     *
-     * @throws DownloadException
-     */
-    public function setCache(array $options = [])
-    {
-        throw DownloadException::forCannotSetCache();
-    }
-
     /**
      * {@inheritDoc}
      *
@@ -290,10 +280,8 @@ public function buildHeaders()
             $this->setHeader('Content-Disposition', $this->getContentDisposition());
         }
 
-        $this->setHeader('Expires-Disposition', '0');
         $this->setHeader('Content-Transfer-Encoding', 'binary');
         $this->setHeader('Content-Length', (string) $this->getContentLength());
-        $this->noCache();
     }
 
     /**

diff --git a/tests/system/HTTP/DownloadResponseTest.php b/tests/system/HTTP/DownloadResponseTest.php
@@ -148,12 +148,25 @@ public function testNoCache(): void
         $this->assertSame('private, no-transform, no-store, must-revalidate', $response->getHeaderLine('Cache-control'));
     }
 
-    public function testCantSetCache(): void
+    public function testSetCache(): void
     {
         $response = new DownloadResponse('unit-test.txt', true);
 
-        $this->expectException(DownloadException::class);
-        $response->setCache();
+        $date = date('r');
+
+        $options = [
+            'etag'          => '12345678',
+            'last-modified' => $date,
+            'max-age'       => 300,
+            'must-revalidate',
+        ];
+
+        $response->setCache($options);
+        $response->buildHeaders();
+
+        $this->assertSame('12345678', $response->getHeaderLine('ETag'));
+        $this->assertSame($date, $response->getHeaderLine('Last-Modified'));
+        $this->assertSame('max-age=300, must-revalidate', $response->getHeaderLine('Cache-Control'));
     }
 
     public function testWhenFilepathIsSetBinaryCanNotBeSet(): void
@@ -200,30 +213,53 @@ public function testCanGetContentLength(): void
         $this->assertSame($size, $response->getContentLength());
     }
 
-    public function testIsSetDownloadableHeadlersFromBinary(): void
+    public function testIsSetDownloadableHeadersFromBinary(): void
     {
         $response = new DownloadResponse('unit test.txt', false);
 
         $response->setBinary('test');
         $response->buildHeaders();
 
+        $this->assertSame('private, no-transform, no-store, must-revalidate', $response->getHeaderLine('Cache-Control'));
         $this->assertSame('application/octet-stream', $response->getHeaderLine('Content-Type'));
         $this->assertSame('attachment; filename="unit test.txt"; filename*=UTF-8\'\'unit%20test.txt', $response->getHeaderLine('Content-Disposition'));
-        $this->assertSame('0', $response->getHeaderLine('Expires-Disposition'));
         $this->assertSame('binary', $response->getHeaderLine('Content-Transfer-Encoding'));
         $this->assertSame('4', $response->getHeaderLine('Content-Length'));
     }
 
-    public function testIsSetDownloadableHeadlersFromFile(): void
+    public function testIsSetDownloadableHeadersFromFile(): void
+    {
+        $response = new DownloadResponse('unit-test.php', false);
+
+        $response->setFilePath(__FILE__);
+        $response->buildHeaders();
+
+        $this->assertSame('private, no-transform, no-store, must-revalidate', $response->getHeaderLine('Cache-Control'));
+        $this->assertSame('application/octet-stream', $response->getHeaderLine('Content-Type'));
+        $this->assertSame('attachment; filename="unit-test.php"; filename*=UTF-8\'\'unit-test.php', $response->getHeaderLine('Content-Disposition'));
+        $this->assertSame('binary', $response->getHeaderLine('Content-Transfer-Encoding'));
+        $this->assertSame(filesize(__FILE__), (int) $response->getHeaderLine('Content-Length'));
+    }
+
+    public function testCustomHeaders(): void
     {
         $response = new DownloadResponse('unit-test.php', false);
 
         $response->setFilePath(__FILE__);
+
+        $response->setHeader('Last-Modified', 'Fri, 18 Oct 2024 13:17:37 GMT');
+        $response->setHeader('Expires', 'Sun, 17 Nov 2024 14:17:37 GMT');
+        $response->setHeader('Pragma', 'public');
+        $response->removeHeader('Cache-Control');
+        $response->setHeader('Cache-Control', 'public');
         $response->buildHeaders();
 
+        $this->assertSame('Fri, 18 Oct 2024 13:17:37 GMT', $response->getHeaderLine('Last-Modified'));
+        $this->assertSame('Sun, 17 Nov 2024 14:17:37 GMT', $response->getHeaderLine('Expires'));
+        $this->assertSame('public', $response->getHeaderLine('Pragma'));
+        $this->assertSame('public', $response->getHeaderLine('Cache-Control'));
         $this->assertSame('application/octet-stream', $response->getHeaderLine('Content-Type'));
         $this->assertSame('attachment; filename="unit-test.php"; filename*=UTF-8\'\'unit-test.php', $response->getHeaderLine('Content-Disposition'));
-        $this->assertSame('0', $response->getHeaderLine('Expires-Disposition'));
         $this->assertSame('binary', $response->getHeaderLine('Content-Transfer-Encoding'));
         $this->assertSame(filesize(__FILE__), (int) $response->getHeaderLine('Content-Length'));
     }

diff --git a/user_guide_src/source/changelogs/v4.5.6.rst b/user_guide_src/source/changelogs/v4.5.6.rst
@@ -30,8 +30,9 @@ Deprecations
 Bugs Fixed
 **********
 - **Session Library:** The session initialization debug message now uses the correct log type "debug" instead of "info".
-
 - **Validation:** Fixed the `getValidated()` method that did not return valid data when validation rules used multiple asterisks.
+- **DownloadResponse:** Fixed a bug that prevented setting custom cache headers. We can now also use the `setCache()` method.
+- **DownloadResponse:** Fixed a bug involving sending a custom "Expires-Disposition" header.
 
 See the repo's
 `CHANGELOG.md <https://github.com/codeigniter4/CodeIgniter4/blob/develop/CHANGELOG.md>`_