From ee1c1f41e65572ef9f91e12023d26bc3bd596eee Mon Sep 17 00:00:00 2001
From: Sotatek-HuyLe3a <huy.le3a@sotatek.com>
Date: Wed, 16 Oct 2024 17:27:59 +0700
Subject: [PATCH] fix: handle exception when getting script hash

---
 .../ScriptRedeemerDatumProcessor.java         | 43 ++++++++++++++-----
 .../script/processor/ScriptRefProcessor.java  | 36 ++++++++++------
 2 files changed, 55 insertions(+), 24 deletions(-)

diff --git a/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRedeemerDatumProcessor.java b/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRedeemerDatumProcessor.java
index 895286528..3f36280e5 100644
--- a/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRedeemerDatumProcessor.java
+++ b/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRedeemerDatumProcessor.java
@@ -177,22 +177,45 @@ private void handleScriptTransaction(EventMetadata metadata, Transaction transac
 
         //Create TxScript entities to save
         List<Script> plutusScripts = scriptsMap.values().stream()
-                        .map(plutusScript -> Script.builder()
-                                .scriptHash(getPlutusScriptHash(plutusScript))
-                                .scriptType(ScriptUtil.toPlutusScriptType(plutusScript.getType()))
-                                .content(JsonUtil.getJson(plutusScript))
-                                .build()).collect(Collectors.toList());
+                        .map(plutusScript -> {
+                            String scriptHash;
+                            try {
+                                scriptHash = ScriptUtil.getPlutusScriptHash(plutusScript);
+                            } catch (Exception e) {
+                                log.error("Error getting native script hash. Block hash: " + metadata.getBlockHash(), e);
+                                return null;
+                            }
+                            return Script.builder()
+                                    .scriptHash(scriptHash)
+                                    .scriptType(ScriptUtil.toPlutusScriptType(plutusScript.getType()))
+                                    .content(JsonUtil.getJson(plutusScript))
+                                    .build();
+                            }
+                        )
+                .filter(Objects::nonNull)
+                .collect(Collectors.toList());
         if (plutusScripts != null && plutusScripts.size() > 0)
             scriptStorage.saveScripts(plutusScripts);
 
         //Get all native scripts  and save
         if (transaction.getWitnesses().getNativeScripts() != null) {
             List<Script> nativeScripts = transaction.getWitnesses().getNativeScripts().stream()
-                    .map(nativeScript -> Script.builder()
-                            .scriptHash(ScriptUtil.getNativeScriptHash(nativeScript))
-                            .scriptType(ScriptType.NATIVE_SCRIPT)
-                            .content(JsonUtil.getJson(nativeScript))
-                            .build()).collect(Collectors.toList());
+                    .map(nativeScript -> {
+                        String scriptHash;
+                        try {
+                            scriptHash = ScriptUtil.getNativeScriptHash(nativeScript);
+                        } catch (Exception e) {
+                            log.error("Error getting native script hash. Block hash: " + metadata.getBlockHash(), e);
+                            return null;
+                        }
+                        return Script.builder()
+                                .scriptHash(scriptHash)
+                                .scriptType(ScriptType.NATIVE_SCRIPT)
+                                .content(JsonUtil.getJson(nativeScript))
+                                .build();
+                    })
+                    .filter(Objects::nonNull)
+                    .collect(Collectors.toList());
             if (nativeScripts != null && nativeScripts.size() > 0)
                 scriptStorage.saveScripts(nativeScripts);
         }
diff --git a/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRefProcessor.java b/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRefProcessor.java
index 02f014d8b..11930e763 100644
--- a/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRefProcessor.java
+++ b/stores/script/src/main/java/com/bloxbean/cardano/yaci/store/script/processor/ScriptRefProcessor.java
@@ -8,6 +8,7 @@
 import com.bloxbean.cardano.yaci.store.events.TransactionEvent;
 import com.bloxbean.cardano.yaci.store.script.domain.Script;
 import com.bloxbean.cardano.yaci.store.script.domain.ScriptType;
+import com.bloxbean.cardano.yaci.store.script.helper.ScriptUtil;
 import com.bloxbean.cardano.yaci.store.script.storage.ScriptStorage;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -38,35 +39,42 @@ public void processScriptRefInUtxo(TransactionEvent transactionEvent) {
                     .map(transactionOutput -> transactionOutput.getScriptRef())
                     .filter(Objects::nonNull)
                     .map(scriptRef -> {
-                        com.bloxbean.cardano.client.spec.Script script = null;
+                        com.bloxbean.cardano.client.spec.Script script;
                         try {
                             script = ScriptReferenceUtil.deserializeScriptRef(HexUtil.decodeHexString(scriptRef));
                         } catch (Exception e) {
                             log.error("Script deserialization failed. Block hash: " + transactionEvent.getMetadata().getBlockHash(), e);
+                            return null;
                         }
-                        return script;
-                    })
-                    .filter(Objects::nonNull)
-                    .map(script -> {
-                        ScriptType scriptType = toScriptType(script);
-                        String scriptHash = null;
 
-                        try {
-                            scriptHash = HexUtil.encodeHexString(script.getScriptHash());
-                        } catch (Exception e) {
-                            log.error("Unable to get reference script hash, Block hash: " + transactionEvent.getMetadata().getBlockHash(), e);
-                        }
+                        ScriptType scriptType = toScriptType(script);
+                        String scriptHash;
+                        String content;
 
-                        String content = null;
                         if (scriptType == ScriptType.NATIVE_SCRIPT) {
                             NativeScript nativeScript = NativeScript.builder()
                                     .type(script.getScriptType())
                                     .content(JsonUtil.getJson(script))
                                     .build();
+
                             content = JsonUtil.getJson(nativeScript);
+                            try {
+                                scriptHash = ScriptUtil.getNativeScriptHash(nativeScript);
+                            } catch (Exception e) {
+                                log.error("Error getting native script hash, Block hash: " + transactionEvent.getMetadata().getBlockHash(), e);
+                                return null;
+                            }
                         } else {
                             PlutusScript plutusScript = toPlutusScript(script);
+
                             content = JsonUtil.getJson(plutusScript);
+
+                            try {
+                                scriptHash = ScriptUtil.getPlutusScriptHash(plutusScript);
+                            } catch (Exception e) {
+                                log.error("Error getting native script hash, Block hash: " + transactionEvent.getMetadata().getBlockHash(), e);
+                                return null;
+                            }
                         }
 
                         return Script.builder()
@@ -74,8 +82,8 @@ public void processScriptRefInUtxo(TransactionEvent transactionEvent) {
                                 .scriptType(scriptType)
                                 .content(content)
                                 .build();
-
                     })
+                    .filter(Objects::nonNull)
                     .collect(Collectors.toList());
 
             //Save the scripts