From 6d60849bd3fb0ac4b945560e71dc0fbdd0f436df Mon Sep 17 00:00:00 2001
From: Eslam-Nawara <eslamnawara44@gmail.com>
Date: Thu, 20 Feb 2025 11:17:51 +0200
Subject: [PATCH] update response from create node

---
 pkg/registrar_gateway/registrar_gateway.go | 19 ++++++++++++++++---
 pkg/types/types.go                         |  2 --
 2 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/pkg/registrar_gateway/registrar_gateway.go b/pkg/registrar_gateway/registrar_gateway.go
index 7bf9bef9..66961503 100644
--- a/pkg/registrar_gateway/registrar_gateway.go
+++ b/pkg/registrar_gateway/registrar_gateway.go
@@ -385,6 +385,7 @@ func (r *registrarGateway) getTwin(url string, twinID uint64) (result types.Acco
 	defer resp.Body.Close()
 
 	err = json.NewDecoder(resp.Body).Decode(&result)
+
 	return result, err
 }
 
@@ -422,6 +423,9 @@ func (r *registrarGateway) getTwinByPubKey(url string, pk []byte) (result uint64
 	var account types.Account
 	err = json.NewDecoder(resp.Body).Decode(&account)
 
+	if r.twinID == 0 {
+		r.twinID = account.TwinID
+	}
 	return account.TwinID, err
 }
 
@@ -486,10 +490,13 @@ func (r *registrarGateway) createNode(url string, node types.UpdateNodeRequest)
 
 	defer resp.Body.Close()
 
-	err = json.NewDecoder(resp.Body).Decode(&nodeID)
+	result := struct {
+		NodeID uint64 `json:"node_id"`
+	}{}
+	err = json.NewDecoder(resp.Body).Decode(&result)
 
-	r.nodeID = nodeID
-	return nodeID, err
+	r.nodeID = result.NodeID
+	return result.NodeID, err
 }
 
 func (r *registrarGateway) getFarm(url string) (farm types.Farm, err error) {
@@ -538,6 +545,9 @@ func (r *registrarGateway) getNode(url string) (node types.Node, err error) {
 		return
 	}
 
+	if r.nodeID == 0 {
+		r.nodeID = node.NodeID
+	}
 	return node, err
 }
 
@@ -580,6 +590,9 @@ func (r *registrarGateway) getNodeByTwinID(url string, twin uint64) (result uint
 		return result, ErrorRecordNotFound
 	}
 
+	if r.nodeID == 0 {
+		r.nodeID = nodes[0].NodeID
+	}
 	return nodes[0].NodeID, nil
 }
 
diff --git a/pkg/types/types.go b/pkg/types/types.go
index 810c1c6f..33c6d9d6 100644
--- a/pkg/types/types.go
+++ b/pkg/types/types.go
@@ -51,8 +51,6 @@ type Farm struct {
 	FarmName  string `json:"farm_name"`
 	TwinID    uint64 `json:"twin_id"` // Farmer account reference
 	Dedicated bool   `json:"dedicated"`
-	CreatedAt time.Time
-	UpdatedAt time.Time
 }
 
 type Node struct {