Run reports + class to manage responses

- Stream reports per line

Author: mrm

Atomx/AtomxClient.php

@@ -2,6 +2,7 @@
 
 use Exception;
 use GuzzleHttp\Message\Response;
+use GuzzleHttp\Stream\Stream;
 
 class AtomxClient extends ApiClient {
     protected $apiBase = null;
@@ -74,6 +75,10 @@ public function login()
             throw new ApiException('Unable to login to API!');
         }
 
+        if ($response instanceof Stream) {
+            $response = json_decode($response->getContents(), true);
+        }
+
         $this->shouldSendToken = true;
 
         if ($response['success'] !== true)

Atomx/ReportStreamer.php

@@ -0,0 +1,57 @@
+<?php namespace Atomx;
+
+use Atomx\AccountStore;
+use GuzzleHttp\Stream\Stream;
+
+
+class ReportStreamer {
+    private $stream;
+
+    private $overflow = '';
+    private $columns = [];
+    private $count = 0;
+
+    public function __construct(Stream $stream, $report = null)
+    {
+        if (!is_null($report)) {
+            $this->columns = array_merge($report['report']['groups'], (isset($report['report']['sums']) ? $report['report']['sums'] : $report['report']['metrics']));
+            $this->count = $report['report']['count'];
+        }
+
+        $this->stream = $stream;
+    }
+
+    public function getOverflow()
+    {
+        return $this->overflow;
+    }
+
+    public function readLine()
+    {
+        $buffer = $this->overflow;
+
+        $this->overflow = '';
+
+        if (empty($buffer)) {
+            if ($this->stream->eof())
+                return false;
+        }
+
+        while (($break = strpos($buffer, "\n")) === false && !$this->stream->eof()) {
+            $buffer .= $this->stream->read(1024);
+        }
+
+        if ($break !== false) {
+            if (strlen($buffer) > $break+1)
+                $this->overflow = substr($buffer, $break+1);
+
+            $buffer = substr($buffer, 0, $break);
+        }
+        $line = str_getcsv($buffer, "\t");
+
+        if (count($line) == 0)
+            return false;
+
+        return $line;
+    }
+}

Atomx/Resources/Report.php

@@ -0,0 +1,63 @@
+<?php namespace Atomx\Resources;
+
+use Atomx\ApiException;
+use Atomx\AtomxClient;
+use Atomx\ReportStreamer;
+use GuzzleHttp\Message\Response;
+use GuzzleHttp\Stream\Stream;
+
+class Report extends AtomxClient {
+    private $returnStream = false;
+
+    // Run report
+    public function run($json)
+    {
+        return $this->postUrl('report', compact('json'));
+    }
+
+    public function status($reportId)
+    {
+        return $this->getUrl('report/' . $reportId, ['status' => true]);
+    }
+
+    public function isReady($report)
+    {
+        if (isset($report['report']['is_ready'])) {
+            return $report['report']['is_ready'];
+        }
+
+        return false;
+    }
+
+    public function download($report)
+    {
+        $reportId = $report['report']['id'];
+        $stream = Stream::factory();
+
+        $this->returnStream = true;
+
+        $stream = $this->getUrl('report/' . $reportId, [], [
+//            'save_to' => $stream,
+//            'timeout'         => 0,
+//            'connect_timeout' => 0
+        ]);
+
+        $this->returnStream = false;
+        return new ReportStreamer($stream);
+    }
+
+    protected function handleResponse(Response $response)
+    {
+        if ($response->getStatusCode() == 200) {
+            $stream = $response->getBody();
+
+            if ($this->returnStream == true)
+                return $stream;
+            else
+                return $stream->getContents();
+        }
+
+        throw new ApiException('Request failed, received the following status: ' .
+            $response->getStatusCode() . ' Body: ' . $response->getBody()->getContents());
+    }
+}

Atomx/Response.php

@@ -0,0 +1,33 @@
+<?php namespace Atomx;
+
+class Response {
+
+    private $namespace;
+    private $response;
+
+    public function __construct($response)
+    {
+        $this->response = $response;
+        $this->namespace = $response['resource'];
+    }
+
+    public function isValid()
+    {
+        return $this->response['success'] == true;
+    }
+
+    public function isMultiple()
+    {
+        return isset($this->response['count']);
+    }
+
+    public function count()
+    {
+       return count($this->response['count']);
+    }
+
+    public function __get($key)
+    {
+        return $this->response[$this->namespace][$key];
+    }
+}