Added forced budget option

Author: samuelWilliams99

README.md

@@ -96,6 +96,8 @@ main = do
           , -- Dry run mode will build the tx, but skip the submit step
             pcDryRun = False
           , pcLogLevel = Debug
+          , -- | Forced budget for scripts, as optional (CPU Steps, Memory Units)
+            pcForceBudget = Nothing
           , -- Protocol params file location relative to the cardano-cli working directory (needed for the cli)
           , pcProtocolParamsFile = "./protocol.json"
           , pcEnableTxEndpoint = True

src/BotPlutusInterface/CardanoCLI.hs

@@ -319,6 +319,13 @@ signTx pabConf tx pubKeys =
         , ["--out-file", txFilePath pabConf "signed" tx]
         ]
 
+budgetFromConfig :: PABConfig -> ExBudget -> ExBudget
+budgetFromConfig pabConf derivedBudget =
+  maybe
+    derivedBudget
+    (\(steps, memory) -> ExBudget (ExCPU $ fromInteger steps) (ExMemory $ fromInteger memory))
+    pabConf.pcForceBudget
+
 -- Signs and writes a tx (uses the tx body written to disk as input)
 submitTx ::
   forall (w :: Type) (effs :: [Type -> Type]).
@@ -357,10 +364,11 @@ txInOpts pabConf txInfo =
         Just (ConsumeScriptAddress validator redeemer datum) ->
           let scriptContext = ScriptContext txInfo $ Plutus.Spending txOutRef
               exBudget =
-                fromRight mempty $
-                  calculateExBudget
-                    (Scripts.unValidatorScript validator)
-                    [Plutus.getRedeemer redeemer, Plutus.getDatum datum, Plutus.toBuiltinData scriptContext]
+                budgetFromConfig pabConf $
+                  fromRight mempty $
+                    calculateExBudget
+                      (Scripts.unValidatorScript validator)
+                      [Plutus.getRedeemer redeemer, Plutus.getDatum datum, Plutus.toBuiltinData scriptContext]
            in (,exBudget) $
                 mconcat
                   [
@@ -399,10 +407,11 @@ mintOpts pabConf txInfo mintingPolicies redeemers mintValue =
                   curSymbol = Value.mpsSymbol $ Scripts.mintingPolicyHash policy
                   scriptContext = ScriptContext txInfo $ Plutus.Minting curSymbol
                   exBudget r =
-                    fromRight mempty $
-                      calculateExBudget
-                        (Scripts.unMintingPolicyScript policy)
-                        [Plutus.getRedeemer r, Plutus.toBuiltinData scriptContext]
+                    budgetFromConfig pabConf $
+                      fromRight mempty $
+                        calculateExBudget
+                          (Scripts.unMintingPolicyScript policy)
+                          [Plutus.getRedeemer r, Plutus.toBuiltinData scriptContext]
                   toOpts r =
                     let budget = exBudget r
                      in (,budget) $

src/BotPlutusInterface/Types.hs

@@ -63,6 +63,8 @@ data PABConfig = PABConfig
   , pcLogLevel :: !LogLevel
   , pcOwnPubKeyHash :: !PubKeyHash
   , pcTipPollingInterval :: !Natural
+  , -- | Forced budget for scripts, as optional (CPU Steps, Memory Units)
+    pcForceBudget :: !(Maybe (Integer, Integer))
   , pcPort :: !Port
   , pcEnableTxEndpoint :: !Bool
   }
@@ -125,6 +127,7 @@ instance Default PABConfig where
       , pcProtocolParamsFile = "./protocol.json"
       , pcLogLevel = Info
       , pcOwnPubKeyHash = ""
+      , pcForceBudget = Nothing
       , pcPort = 9080
       , pcEnableTxEndpoint = False
       }

test/Spec/BotPlutusInterface/Contract.hs

@@ -518,7 +518,7 @@ mintTokens = do
           --tx-out ${addr2}+1000 + 5 ${curSymbol'}.74657374546F6B656E
           --mint-script-file ./result-scripts/policy-${curSymbol'}.plutus
           --mint-redeemer-file ./result-scripts/redeemer-${redeemerHash}.json
-          --mint-execution-units (387149,1400)
+          --mint-execution-units (500000,2000)
           --mint 5 ${curSymbol'}.74657374546F6B656E
           --required-signer ./signing-keys/signing-key-${pkh1'}.skey
           --fee 0
@@ -531,14 +531,14 @@ mintTokens = do
           cardano-cli transaction build-raw --alonzo-era
           --tx-in ${inTxId}#0
           --tx-in-collateral ${inTxId}#0
-          --tx-out ${addr1}+610151
+          --tx-out ${addr1}+496700
           --tx-out ${addr2}+1000 + 5 ${curSymbol'}.74657374546F6B656E
           --mint-script-file ./result-scripts/policy-${curSymbol'}.plutus
           --mint-redeemer-file ./result-scripts/redeemer-${redeemerHash}.json
-          --mint-execution-units (387149,1400)
+          --mint-execution-units (500000,2000)
           --mint 5 ${curSymbol'}.74657374546F6B656E
           --required-signer ./signing-keys/signing-key-${pkh1'}.skey
-          --fee 388849
+          --fee 502300
           --protocol-params-file ./protocol.json --out-file ./txs/tx-${outTxId}.raw
           |]
         )
@@ -702,7 +702,7 @@ redeemFromValidator = do
           --tx-in-script-file ./result-scripts/validator-${valHash'}.plutus
           --tx-in-datum-file ./result-scripts/datum-${datumHash'}.json
           --tx-in-redeemer-file ./result-scripts/redeemer-${redeemerHash}.json
-          --tx-in-execution-units (476468,1700)
+          --tx-in-execution-units (500000,2000)
           --tx-in-collateral ${inTxId}#0
           --tx-out ${addr2}+500
           --required-signer ./signing-keys/signing-key-${pkh1'}.skey
@@ -718,12 +718,12 @@ redeemFromValidator = do
           --tx-in-script-file ./result-scripts/validator-${valHash'}.plutus
           --tx-in-datum-file ./result-scripts/datum-${datumHash'}.json
           --tx-in-redeemer-file ./result-scripts/redeemer-${redeemerHash}.json
-          --tx-in-execution-units (476468,1700)
+          --tx-in-execution-units (500000,2000)
           --tx-in-collateral ${inTxId}#0
-          --tx-out ${addr1}+522182
+          --tx-out ${addr1}+498350
           --tx-out ${addr2}+500
           --required-signer ./signing-keys/signing-key-${pkh1'}.skey
-          --fee 478568
+          --fee 502400
           --protocol-params-file ./protocol.json --out-file ./txs/tx-${outTxId}.raw
           |]
         )

test/Spec/MockContract.hs

@@ -230,7 +230,7 @@ instance Monoid w => Default (MockContractState w) where
 instance Monoid w => Default (ContractEnvironment w) where
   def =
     ContractEnvironment
-      { cePABConfig = def {pcNetwork = Mainnet, pcOwnPubKeyHash = pkh1}
+      { cePABConfig = def {pcNetwork = Mainnet, pcOwnPubKeyHash = pkh1, pcForceBudget = Just (500000, 2000)}
       , ceContractInstanceId = ContractInstanceId UUID.nil
       , ceContractState = unsafePerformIO $ newTVarIO def
       }