Read params from file, update examples

Author: samuelWilliams99

examples/plutus-game/app/Main.hs

@@ -67,6 +67,7 @@ main = do
           , pcDryRun = True
           , pcLogLevel = Debug
           , pcProtocolParamsFile = "./protocol.json"
+          , pcForceBudget = Just (1000, 1000)
           , pcEnableTxEndpoint = True
           }
   BotPlutusInterface.runPAB @GameContracts pabConf

examples/plutus-nft/app/Main.hs

@@ -67,6 +67,7 @@ main = do
           , pcDryRun = True
           , pcLogLevel = Debug
           , pcProtocolParamsFile = "./protocol.json"
+          , pcForceBudget = Just (1000, 1000)
           , pcEnableTxEndpoint = True
           }
   BotPlutusInterface.runPAB @MintNFTContracts pabConf

examples/plutus-transfer/app/Main.hs

@@ -66,6 +66,7 @@ main = do
           , pcDryRun = True
           , pcLogLevel = Debug
           , pcProtocolParamsFile = "./protocol.json"
+          , pcForceBudget = Nothing
           , pcEnableTxEndpoint = True
           }
   BotPlutusInterface.runPAB @TransferContracts pabConf

src/BotPlutusInterface/CardanoCLI.hs

@@ -1,6 +1,5 @@
 {-# LANGUAGE AllowAmbiguousTypes #-}
 {-# LANGUAGE NamedFieldPuns #-}
-{-# OPTIONS_GHC -w #-}
 
 module BotPlutusInterface.CardanoCLI (
   submitTx,
@@ -27,7 +26,8 @@ import BotPlutusInterface.Files (
  )
 import BotPlutusInterface.Types (PABConfig (pcSlotConfig), Tip)
 import BotPlutusInterface.UtxoParser qualified as UtxoParser
-import Cardano.Api.Shelley (NetworkId (Mainnet, Testnet), NetworkMagic (..), serialiseAddress)
+import Cardano.Api (AnyPlutusScriptVersion (AnyPlutusScriptVersion), CostModel (CostModel), PlutusScriptVersion (PlutusScriptV1))
+import Cardano.Api.Shelley (NetworkId (Mainnet, Testnet), NetworkMagic (..), ProtocolParameters (protocolParamCostModels), serialiseAddress)
 import Codec.Serialise qualified as Codec
 import Control.Monad (join)
 import Control.Monad.Freer (Eff, Member)
@@ -367,6 +367,7 @@ txInOpts pabConf txInfo =
                 budgetFromConfig pabConf $
                   fromRight mempty $
                     calculateExBudget
+                      pabConf
                       (Scripts.unValidatorScript validator)
                       [Plutus.getRedeemer redeemer, Plutus.getDatum datum, Plutus.toBuiltinData scriptContext]
            in (,exBudget) $
@@ -410,6 +411,7 @@ mintOpts pabConf txInfo mintingPolicies redeemers mintValue =
                     budgetFromConfig pabConf $
                       fromRight mempty $
                         calculateExBudget
+                          pabConf
                           (Scripts.unMintingPolicyScript policy)
                           [Plutus.getRedeemer r, Plutus.toBuiltinData scriptContext]
                   toOpts r =
@@ -497,20 +499,22 @@ unsafeSerialiseAddress network address =
     Right a -> a
     Left _ -> error "Couldn't create address"
 
-calculateExBudget :: Script -> [BuiltinData] -> Either Text ExBudget
-calculateExBudget script builtinData = do
-  -- TODO, pull this from the protocol, they're the same for now but may not always be
-  modelParams <- maybeToRight "Cost model params invalid." Plutus.defaultCostModelParams
+extractCostModel :: PABConfig -> Maybe Plutus.CostModelParams
+extractCostModel pabConf =
+  unCostModel <$> Map.lookup (AnyPlutusScriptVersion PlutusScriptV1) (protocolParamCostModels pabConf.pcProtocolParams)
+  where
+    unCostModel :: CostModel -> Map Text Integer
+    unCostModel (CostModel m) = m
+
+calculateExBudget :: PABConfig -> Script -> [BuiltinData] -> Either Text ExBudget
+calculateExBudget pabConf script builtinData = do
+  modelParams <- maybeToRight "Cost model params invalid." $ extractCostModel pabConf
   let serialisedScript = ShortByteString.toShort $ LazyByteString.toStrict $ Codec.serialise script
       pData = map Plutus.builtinDataToData builtinData
   mapLeft showText $
     snd $
       Plutus.evaluateScriptCounting Plutus.Verbose modelParams serialisedScript pData
 
--- calculateExBudget :: Script -> [BuiltinData] -> Either Text ExBudget
--- calculateExBudget script builtinData = do
---   mapLeft showText $ fst <$> Scripts.evaluateScript (Scripts.applyArguments script $ Plutus.builtinDataToData <$> builtinData)
-
 exBudgetToCliArg :: ExBudget -> Text
 exBudgetToCliArg (ExBudget (ExCPU steps) (ExMemory memory)) =
   "(" <> showText steps <> "," <> showText memory <> ")"