From 0fd1321ddb62643e757a08e34d056e2073f6a7da Mon Sep 17 00:00:00 2001
From: Laurentiu Ciobanu <laurentiu@ciobanu.dev>
Date: Mon, 20 May 2024 09:58:17 +0000
Subject: [PATCH] add missing esdt roles for cyrpto opcodes v2

---
 vmhost/vmhooks/baseOps.go     |  3 ++-
 vmhost/vmhooks/eei_helpers.go | 50 +++++++++++++++++++++++++++++++++--
 2 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/vmhost/vmhooks/baseOps.go b/vmhost/vmhooks/baseOps.go
index c8138b44a..170ab8c40 100644
--- a/vmhost/vmhooks/baseOps.go
+++ b/vmhost/vmhooks/baseOps.go
@@ -510,7 +510,8 @@ func (context *VMHooksImpl) GetESDTLocalRoles(tokenIdHandle int32) int64 {
 		return -1
 	}
 
-	return getESDTRoles(data)
+	enableEpochsHandler := context.host.EnableEpochsHandler()
+	return getESDTRoles(data, enableEpochsHandler.IsFlagEnabled(vmhost.CryptoOpcodesV2Flag))
 }
 
 // ValidateTokenIdentifier VMHooks implementation.
diff --git a/vmhost/vmhooks/eei_helpers.go b/vmhost/vmhooks/eei_helpers.go
index 681b61cb7..3f5fb2cf8 100644
--- a/vmhost/vmhooks/eei_helpers.go
+++ b/vmhost/vmhooks/eei_helpers.go
@@ -5,6 +5,12 @@ const esdtRoleLocalBurn = "ESDTRoleLocalBurn"
 const esdtRoleNFTCreate = "ESDTRoleNFTCreate"
 const esdtRoleNFTAddQuantity = "ESDTRoleNFTAddQuantity"
 const esdtRoleNFTBurn = "ESDTRoleNFTBurn"
+const esdtRoleNFTUpdateAttributes = "ESDTRoleNFTUpdateAttributes"
+const esdtRoleNFTAddURI = "ESDTRoleNFTAddURI"
+const esdtRoleNFTRecreate = "ESDTRoleNFTRecreate"
+const esdtRoleModifyCreator = "ESDTRoleModifyCreator"
+const esdtRoleModifyRoyalties = "ESDTRoleModifyRoyalties"
+const esdtRoleSetNewURI = "ESDTRoleSetNewURI"
 
 const tickerMinLength = 3
 const tickerMaxLength = 10
@@ -19,6 +25,12 @@ const (
 	RoleNFTCreate
 	RoleNFTAddQuantity
 	RoleNFTBurn
+	RoleNFTUpdateAttributes
+	RoleNFTAddURI
+	RoleNFTRecreate
+	RoleModifyCreator
+	RoleModifyRoyalties
+	RoleSetNewURI
 )
 
 func roleFromByteArray(bytes []byte) int64 {
@@ -39,7 +51,37 @@ func roleFromByteArray(bytes []byte) int64 {
 	}
 }
 
-func getESDTRoles(dataBuffer []byte) int64 {
+func roleFromByteArrayV2(bytes []byte) int64 {
+	stringValue := string(bytes)
+	switch stringValue {
+	case esdtRoleLocalMint:
+		return RoleMint
+	case esdtRoleLocalBurn:
+		return RoleBurn
+	case esdtRoleNFTCreate:
+		return RoleNFTCreate
+	case esdtRoleNFTAddQuantity:
+		return RoleNFTAddQuantity
+	case esdtRoleNFTBurn:
+		return RoleNFTBurn
+	case esdtRoleNFTUpdateAttributes:
+		return RoleNFTUpdateAttributes
+	case esdtRoleNFTAddURI:
+		return RoleNFTAddURI
+	case esdtRoleNFTRecreate:
+		return RoleNFTRecreate
+	case esdtRoleModifyCreator:
+		return RoleModifyCreator
+	case esdtRoleModifyRoyalties:
+		return RoleModifyRoyalties
+	case esdtRoleSetNewURI:
+		return RoleSetNewURI
+	default:
+		return 0
+	}
+}
+
+func getESDTRoles(dataBuffer []byte, cryptoOpcodesV2Enabled bool) int64 {
 	result := int64(0)
 	currentIndex := 0
 	valueLen := len(dataBuffer)
@@ -57,7 +99,11 @@ func getESDTRoles(dataBuffer []byte) int64 {
 		roleName := dataBuffer[currentIndex:endIndex]
 		currentIndex = endIndex
 
-		result |= roleFromByteArray(roleName)
+		if cryptoOpcodesV2Enabled {
+			result |= roleFromByteArrayV2(roleName)
+		} else {
+			result |= roleFromByteArray(roleName)
+		}
 	}
 	return result
 }