Add support for transfer-encoding chunked (Stream Client)

xeoncross · Nov 13, 2013 · 25875ed · 25875ed
1 parent 4b77b0f
commit 25875ed
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/lib/PicoFeed/Clients/Stream.php b/lib/PicoFeed/Clients/Stream.php
@@ -57,10 +57,31 @@ public function doRequest()
 
         fclose($stream);
 
+        if (isset($headers['Transfer-Encoding']) && $headers['Transfer-Encoding'] === 'chunked') {
+            $body = $this->decodeChunked($body);
+        }
+
         return array(
             'status' => $status,
             'body' => $body,
             'headers' => $headers
         );
     }
+
+
+    public function decodeChunked($str)
+    {
+        for ($result = ''; ! empty($str); $str = trim($str)) {
+
+            // Get the chunk length
+            $pos = strpos($str, "\r\n");
+            $len = hexdec(substr($str, 0, $pos));
+
+            // Append the chunk to the result
+            $result .= substr($str, $pos + 2, $len);
+            $str = substr($str, $pos + 2 + $len);
+        }
+
+        return $result;
+    }
 }
diff --git a/tests/StreamTest.php b/tests/StreamTest.php
@@ -7,6 +7,16 @@
 
 class StreamTest extends PHPUnit_Framework_TestCase
 {
+    // public function testChunkedResponse()
+    // {
+    //     $client = new Stream;
+    //     $client->url = 'http://www.reddit.com/r/dwarffortress/.rss';
+    //     $result = $client->doRequest();
+
+    //     var_dump($result);
+    // }
+
+
     public function testDownload()
     {
         $client = new Stream;