Add budget multiplier field to config

Author: samuelWilliams99

src/BotPlutusInterface/Config.hs

@@ -33,6 +33,7 @@ import Data.String.ToString (toString)
 import Data.Text (Text)
 import Data.Text qualified as Text
 import PlutusConfig.Base (
+  customRationalSpec,
   enumToAtom,
   filepathSpec,
   maybeSpec,
@@ -95,6 +96,7 @@ instance ToValue PABConfig where
         pcPort
         pcEnableTxEndpoint
         pcCollectStats
+        pcBudgetMultiplier
       ) =
       Sections
         ()
@@ -116,6 +118,7 @@ instance ToValue PABConfig where
         , Section () "port"               $ toValue pcPort
         , Section () "enableTxEndpoint"   $ toValue pcEnableTxEndpoint
         , Section () "collectStats"       $ toValue pcCollectStats
+        , Section () "budgetMultiplier"   $ toValue pcBudgetMultiplier
         ]
 {- ORMOLU_ENABLE -}
 
@@ -206,6 +209,13 @@ pabConfigSpec = sectionsSpec "PABConfig" $ do
       trueOrFalseSpec
       "Save some stats during contract run (only transactions execution budgets supported atm)"
 
+  pcBudgetMultiplier <-
+    sectionWithDefault'
+      (pcBudgetMultiplier def)
+      "budgetMultiplier"
+      customRationalSpec
+      "Multiplier on the budgets automatically calculated"
+
   pure PABConfig {..}
 
 docPABConfig :: String

src/BotPlutusInterface/ExBudget.hs

@@ -46,11 +46,18 @@ estimateBudget pabConf txFile = do
   let txBudget = do
         body <- txBody
         budget <- budgetRes
-        (spendingBudgets, policyBudgets) <- mkBudgetMaps budget body
+        let scaledBudget = fmap (fmap $ scaleBudget pabConf.pcBudgetMultiplier) budget
+        (spendingBudgets, policyBudgets) <- mkBudgetMaps scaledBudget body
         Right $ TxBudget spendingBudgets policyBudgets
 
   return txBudget
 
+-- | Scale the budget by the multipliers in config
+scaleBudget :: Rational -> CAPI.ExecutionUnits -> CAPI.ExecutionUnits
+scaleBudget scaler (CAPI.ExecutionUnits steps mem) = CAPI.ExecutionUnits (scale steps) (scale mem)
+  where
+    scale x = round $ toRational x * scaler
+
 -- | Deserialize transaction body from ".signed" file
 deserialiseSigned :: FilePath -> IO (Either BudgetEstimationError (CAPI.Tx CAPI.AlonzoEra))
 deserialiseSigned txFile = do

src/BotPlutusInterface/Types.hs

@@ -83,6 +83,7 @@ data PABConfig = PABConfig
   , pcPort :: !Port
   , pcEnableTxEndpoint :: !Bool
   , pcCollectStats :: !Bool
+  , pcBudgetMultiplier :: !Rational
   }
   deriving stock (Show, Eq)
 
@@ -221,6 +222,7 @@ instance Default PABConfig where
       , pcPort = 9080
       , pcEnableTxEndpoint = False
       , pcCollectStats = False
+      , pcBudgetMultiplier = 1
       }
 
 data RawTx = RawTx