Merge branch 'main' into issue-3911issue-3911

1 files changed, 55 insertions, 0 deletions

diff --git a/lib/classes/StudipResponse.php b/lib/classes/StudipResponse.php
new file mode 100644
index 0000000..a9f1a4c
--- /dev/null
+++ b/lib/classes/StudipResponse.php
@@ -0,0 +1,55 @@
+<?php
+class StudipResponse extends Trails\Response
+{
+    /**
+     * Outputs this response to the client using "echo" and "header".
+     *
+     * This extension allows the body to be a callable and handles generators
+     * by outputting the chunks yielded by the generator.
+     */
+    public function output()
+    {
+        if (isset($this->status)) {
+            $this->send_header(
+                "{$_SERVER['SERVER_PROTOCOL']} {$this->status} {$this->reason}",
+                true,
+                $this->status
+            );
+        }
+
+        // Send headers
+        foreach ($this->headers as $k => $v) {
+            $this->send_header("{$k}: {$v}");
+        }
+
+        // Determine output
+        if (is_callable($this->body)) {
+            $output = call_user_func($this->body);
+        } else {
+            $output = $this->body;
+        }
+
+        if ($output instanceof Generator) {
+            // Clear output buffer
+            while (ob_get_level()) {
+                ob_end_clean();
+            }
+
+            // Ensure generator will run to the end
+            $abort = ignore_user_abort(true);
+
+            // Output chunks yielded by generator
+            foreach ($output as $chunk) {
+                if (!connection_aborted()) {
+                    echo $chunk;
+                    flush();
+                }
+            }
+
+            // Reset user abort to previous state
+            ignore_user_abort($abort);
+        } else {
+            echo $output;
+        }
+    }
+}