Add event to allow inspecting and changing multipart responses

sabre-io · Apr 29, 2024 · a9fe8fc · a9fe8fc
1 parent da8c1f2
commit a9fe8fc
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/lib/DAV/Server.php b/lib/DAV/Server.php
@@ -1635,6 +1635,8 @@ public function getResourceTypeForNode(INode $node)
      */
     public function generateMultiStatus($fileProperties, $strip404s = false)
     {
+        $this->emit('beforeMultiStatus', [&$fileProperties]);
+
         $w = $this->xml->getWriter();
         if (self::$streamMultiStatus) {
             return function () use ($fileProperties, $strip404s, $w) {

diff --git a/tests/Sabre/DAV/ServerEventsTest.php b/tests/Sabre/DAV/ServerEventsTest.php
@@ -12,6 +12,8 @@ class ServerEventsTest extends AbstractServerTestCase
 
     private $exception;
 
+    private $fileProperties;
+
     public function testAfterBindOfFile()
     {
         $this->server->on('afterBind', [$this, 'afterHandler']);
@@ -191,4 +193,39 @@ public function testMethod()
         // Fun fact, PHP 7.1 changes the order when sorting-by-callback.
         self::assertTrue($k >= 2 && $k <= 3);
     }
+
+    public function multiStatusHandler(&$fileProperties)
+    {
+        $this->fileProperties = $fileProperties;
+        $fileProperties = array_slice($fileProperties, 0, 1);
+    }
+
+    public function testBeforeMultiStatus()
+    {
+        $this->server->on('beforeMultiStatus', [$this, 'multiStatusHandler']);
+
+        $response = $this->server->generateMultiStatus([
+            [
+                'href' => 'foo',
+                '200' => [
+                    'd:getcontentlength' => 10,
+                ],
+            ],
+            [
+                'href' => 'bar',
+                '200' => [
+                    'd:getcontentlength' => 11,
+                ],
+            ],
+        ]);
+
+        $body = preg_replace("/xmlns(:[A-Za-z0-9_])?=(\"|\')DAV:(\"|\')/", 'xmlns\\1="urn:DAV"', $response);
+        $xml = simplexml_load_string($body);
+        $xml->registerXPathNamespace('d', 'urn:DAV');
+
+        $data = $xml->xpath('/d:multistatus/d:response/d:propstat/d:prop/d:getcontentlength');
+        $this->assertCount(1, $data);
+
+        $this->assertCount(2, $this->fileProperties);
+    }
 }