From 841a433ded2bad77236afbe355e06586c4868af6 Mon Sep 17 00:00:00 2001
From: Cerys Lewis <cal66@aber.ac.uk>
Date: Mon, 17 Jun 2024 16:55:46 +0100
Subject: [PATCH] better handing for different responses

---
 src/DataStructures/QueryResponseWrapper.php | 26 +++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/src/DataStructures/QueryResponseWrapper.php b/src/DataStructures/QueryResponseWrapper.php
index c9797c4..b868b3a 100644
--- a/src/DataStructures/QueryResponseWrapper.php
+++ b/src/DataStructures/QueryResponseWrapper.php
@@ -4,22 +4,40 @@ namespace Darksparrow\DeegraphInteractions\DataStructures;
 
 class QueryResponseWrapper
 {
+    public array $Nodes;
+
     public array $Rows;
     public string $RowFormat;
 
     public static function FromAPIResponse(string $response): QueryResponseWrapper
     {
-        $temp = json_decode($response, true);
+        $response = json_decode($response, true);
+
+        echo "\n\n\n";
+        var_dump($response);
+        echo "\n\n\n";
+
 
         $builder = new QueryResponseWrapper();
 
-        foreach($temp["@rows"] as $row) $builder->Rows[] = QueryResponseRow::FromArray(array: $row);
-        $builder->RowFormat = $temp["@row_format"];
+        if(array_key_exists(key: "@rows", array: $response))
+        {
+            $builder->Rows = [];
+            foreach($response["@rows"] as $row)
+                $builder->Rows[] = QueryResponseRow::FromArray(array: $row);
+            $builder->RowFormat = $response["@row_format"];
+        }
+        if(array_key_exists(key: "@nodes", array: $response))
+        {
+            $builder->Nodes = [];
+            foreach($response["@nodes"] as $row)
+                $builder->Nodes = $response["@nodes"];
+        }
 
         return $builder;
     }
 
-    public function Flatten(): array
+    public function FlattenRows(): array
     {
         $builder = [];