From 9a549a36884dfeb7f26f656cee064ff4e1ec5c1c Mon Sep 17 00:00:00 2001
From: SpaghettiOverload <imyours@mail.bg>
Date: Wed, 20 Sep 2023 11:44:34 +0300
Subject: [PATCH] update: edit nft msg handler

---
 database/types/nft.go     |  1 +
 modules/nft/handle_msg.go | 24 ++++++++++++++----------
 2 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/database/types/nft.go b/database/types/nft.go
index 9f04d2834..57c29def7 100644
--- a/database/types/nft.go
+++ b/database/types/nft.go
@@ -4,4 +4,5 @@ type EditNftQuery struct {
 	Name     string `json:"name"`
 	URI      string `json:"uri"`
 	DataText string `json:"data_text"`
+	DataJSON string `json:"data_json"`
 }
diff --git a/modules/nft/handle_msg.go b/modules/nft/handle_msg.go
index fed1ce41e..ff0dce514 100644
--- a/modules/nft/handle_msg.go
+++ b/modules/nft/handle_msg.go
@@ -87,36 +87,40 @@ func (m *Module) handleMsgMintNFT(index int, tx *juno.Tx, msg *nftTypes.MsgMintN
 }
 
 func (m *Module) handleMsgEditNFT(msg *nftTypes.MsgEditNFT) error {
-	log.Debug().Str("module", "nft").Str("denomId", msg.DenomId).Str("tokenId", msg.Id).Msg("handling message edit nft")
-
 	nftID := msg.Id
 	denomID := msg.DenomId
-	dataJSON, dataText := utils.GetData(msg.Data)
+	log.Debug().Str("module", "nft").Str("denomId", denomID).Str("tokenId", nftID).Msg("handling message edit nft")
+
+	// Parse newly proposed changes
+	newDataJSON, newDataText := utils.GetData(msg.Data)
 	newNftData := dbtypes.EditNftQuery{
 		Name:     msg.Name,
 		URI:      msg.URI,
-		DataText: msg.Data,
+		DataText: newDataText,
+		DataJSON: newDataJSON,
 	}
 
+	// Getting previous DB records
 	var currentNftData dbtypes.EditNftQuery
-	err := m.db.SQL.Select(&currentNftData, `SELECT name, uri, data_text FROM nft_nft WHERE id = $1 AND denom_id =$2`, nftID, denomID)
+	err := m.db.SQL.Select(&currentNftData, `SELECT name, uri, data_text, data_json FROM nft_nft WHERE id = $1 AND denom_id = $2`, nftID, denomID)
 	if err != nil {
 		return err
 	}
 
-	if newNftData.Name == nftTypes.DoNotModify || newNftData.Name == currentNftData.Name {
+	if newNftData.Name == nftTypes.DoNotModify {
 		newNftData.Name = currentNftData.Name
 	}
 
-	if newNftData.URI == nftTypes.DoNotModify || newNftData.URI == currentNftData.URI {
+	if newNftData.URI == nftTypes.DoNotModify {
 		newNftData.URI = currentNftData.URI
 	}
 
-	if newNftData.DataText == nftTypes.DoNotModify || newNftData.DataText == currentNftData.DataText {
-		dataJSON, dataText = utils.GetData(currentNftData.DataText)
+	if msg.Data == nftTypes.DoNotModify {
+		newNftData.DataText = currentNftData.DataText
+		newNftData.DataJSON = currentNftData.DataJSON
 	}
 
-	return m.db.UpdateNFT(nftID, denomID, newNftData.Name, newNftData.URI, utils.SanitizeUTF8(dataJSON), dataText)
+	return m.db.UpdateNFT(nftID, denomID, newNftData.Name, newNftData.URI, newNftData.DataJSON, newNftData.DataText)
 }
 
 func (m *Module) handleMsgTransferNFT(tx *juno.Tx, msg *nftTypes.MsgTransferNft) error {