Twitter streaming API reader: Cleanup input handling & split from HTTP headers to body

plugins/TwitterBridge/jsonstreamreader.php

@@ -93,18 +93,24 @@ abstract class JsonStreamReader
         $this->state = 'waiting';
     }
 
+    /**
+     * Send some fun data off to the server.
+     *
+     * @param string $buffer
+     */
     function send($buffer)
     {
-        echo "Writing...\n";
-        var_dump($buffer);
         fwrite($this->socket, $buffer);
     }
 
+    /**
+     * Read next packet of data from the socket.
+     *
+     * @return string
+     */
     function read()
     {
-        echo "Reading...\n";
         $buffer = fread($this->socket, 65536);
-        var_dump($buffer);
         return $buffer;
     }
 
@@ -195,6 +201,7 @@ abstract class JsonStreamReader
     {
         $lines = explode(self::CRLF, $buffer);
         foreach ($lines as $line) {
+            if ($this->state == 'headers') {
                 if ($line == '') {
                     $this->state = 'active';
                     common_log(LOG_DEBUG, "$this->id connection is active!");
@@ -202,6 +209,9 @@ abstract class JsonStreamReader
                     common_log(LOG_DEBUG, "$this->id read HTTP header: $line");
                     $this->responseHeaders[] = $line;
                 }
+            } else if ($this->state == 'active') {
+                $this->handleLineActive($line);
+            }
         }
     }
 
@@ -211,6 +221,16 @@ abstract class JsonStreamReader
         // Will we always deliver on packet boundaries?
         $lines = explode("\n", $buffer);
         foreach ($lines as $line) {
+            $this->handleLineActive($line);
+        }
+    }
+
+    function handleLineActive($line)
+    {
+        if ($line == '') {
+            // Server sends empty lines as keepalive.
+            return;
+        }
         $data = json_decode($line, true);
         if ($data) {
             $this->handleJson($data);
@@ -218,7 +238,6 @@ abstract class JsonStreamReader
             common_log(LOG_ERR, "$this->id received bogus JSON data: " . $line);
         }
     }
-    }
 
     abstract function handleJson(array $data);
 }

plugins/TwitterBridge/twitterstreamreader.php

@@ -94,11 +94,22 @@ abstract class TwitterStreamReader extends JsonStreamReader
 
     abstract function routeMessage($data);
 
-    function handleMessage($data, $forUserId=null)
+    /**
+     * Send the decoded JSON object out to any event listeners.
+     *
+     * @param array $data
+     * @param int $forUserId
+     */
+    function handleMessage(array $data, $forUserId=null)
     {
         $this->fireEvent('raw', $data, $forUserId);
-        $known = array('friends');
-        foreach ($known as $key) {
+
+        if (isset($data['id']) && isset($data['text']) && isset($data['user'])) {
+            $this->fireEvent('status', $data);
+        }
+
+        $knownMeta = array('friends', 'delete', 'scrubgeo', 'limit', 'event', 'direct_message');
+        foreach ($knownMeta as $key) {
             if (isset($data[$key])) {
                 $this->fireEvent($key, $data[$key], $forUserId);
             }