Merge pull request #99 from mlabs-haskell/misha/potentially-unknownTns-fix

Fix risk of getting `UnknownTxIns` during statistics saving

Author: mikekeke

src/BotPlutusInterface/Contract.hs

@@ -274,6 +274,9 @@ writeBalancedTx contractEnv (Right tx) = do
           , "Signatories (pkh): " <> Text.unwords (map pkhToText requiredSigners)
           ]
 
+    when (pabConf.pcCollectStats && signable) $
+      collectBudgetStats (Tx.txId tx) pabConf
+
     when (not pabConf.pcDryRun && signable) $ do
       newEitherT $ CardanoCLI.submitTx @w pabConf tx
 
@@ -284,9 +287,6 @@ writeBalancedTx contractEnv (Right tx) = do
     mvFiles (Files.txFilePath pabConf "raw" (Tx.txId tx)) (Files.txFilePath pabConf "raw" cardanoTxId)
     when signable $ mvFiles signedSrcPath signedDstPath
 
-    when contractEnv.cePABConfig.pcCollectStats $
-      collectBudgetStats cardanoTxId signedDstPath
-
     pure cardanoTx
   where
     mvFiles :: Text -> Text -> EitherT Text (Eff effs) ()
@@ -299,10 +299,17 @@ writeBalancedTx contractEnv (Right tx) = do
             , cmdOutParser = const ()
             }
 
-    collectBudgetStats txId txPath = do
-      let path = Text.unpack txPath
-      b <- firstEitherT (Text.pack . show) $ newEitherT $ estimateBudget @w (Signed path)
-      void $ newEitherT (Right <$> saveBudget @w txId b)
+    collectBudgetStats txId pabConf = do
+      let path = Text.unpack (Files.txFilePath pabConf "signed" (Tx.txId tx))
+      txBudget <-
+        firstEitherT toBudgetSaveError $
+          newEitherT $ estimateBudget @w (Signed path)
+      void $ newEitherT (Right <$> saveBudget @w txId txBudget)
+
+    toBudgetSaveError =
+      Text.pack
+        . ("Failed to save Tx budgets statistics: " ++)
+        . show
 
 pkhToText :: Ledger.PubKey -> Text
 pkhToText = encodeByteString . fromBuiltin . Ledger.getPubKeyHash . Ledger.pubKeyHash