refactor: extract code to display execution budget

Author: Unisay

doc/docusaurus/docs/using-plinth/evaluating-plinth.md

@@ -46,7 +46,7 @@ import PlutusTx.Test (
   EvalResult,
   applyLifted,
   evaluateCompiledCode,
-  prettyEvalResult
+  displayEvalResult
  )
 ```
 
@@ -129,7 +129,7 @@ EvalResult
 ```
 
 The output is quite verbose and not very readable.
-To make it more readable, we can use the `prettyEvalResult` function, 
+To make it more readable, we can use the `displayEvalResult` function, 
 which formats the output in a prettier way:
 
 ```

doc/docusaurus/static/code/Example/Evaluation/Main.hs

@@ -16,7 +16,7 @@ import Data.Text.IO qualified as Text
 import PlutusTx (CompiledCode, applyCode, compile, liftCodeDef, unsafeApplyCode)
 import Prelude (IO, String)
 
-import PlutusTx.Test (EvalResult, evaluateCompiledCode, prettyEvalResult)
+import PlutusTx.Test (EvalResult, displayEvalResult, evaluateCompiledCode)
 
 -- BEGIN Plinth
 
@@ -88,6 +88,6 @@ result = evaluateCompiledCode appliedUnsafely
 -- BEGIN main
 
 main :: IO ()
-main = Text.putStrLn $ prettyEvalResult result
+main = Text.putStrLn $ displayEvalResult result
 
 -- END main

plutus-tx/testlib/PlutusTx/Test/Run/Code.hs

@@ -41,54 +41,55 @@ data EvalResult = EvalResult
   deriving stock (Show)
 
 instance Pretty EvalResult where
-  pretty
-    EvalResult
-      { evalResultBudget =
-        ExBudget
-          { exBudgetCPU = ExCPU cpu
-          , exBudgetMemory = ExMemory mem
-          }
-      , ..
-      } =
-      vsep
-        [ case evalResult of
-            Left err ->
-              vsep
-                [ "Evaluation FAILED:"
-                , indent 2 $ prettyPlcClassicSimple err
-                ]
-            Right term ->
-              vsep
-                [ "Evaluation was SUCCESSFUL, result is:"
-                , indent 2 $ prettyPlcReadableSimple term
-                ]
-        , mempty
-        , "Execution budget spent:"
-        , indent 2 $
+  pretty EvalResult{..} =
+    vsep
+      [ case evalResult of
+          Left err ->
             vsep
-              [ "CPU" <+> pretty (format commas (unSatInt cpu))
-              , "MEM" <+> pretty (format commas (unSatInt mem))
+              [ "Evaluation FAILED:"
+              , indent 2 $ prettyPlcClassicSimple err
               ]
-        , mempty
-        , if null evalResultTraces
-            then "No traces were emitted"
-            else
-              vsep
-                [ "Evaluation"
-                    <+> plural "trace" "traces" (length evalResultTraces)
-                    <> ":"
-                , indent 2 $
-                    vsep $
-                      zipWith
-                        (\idx trace -> pretty idx <> dot <+> pretty trace)
-                        [1 :: Int ..]
-                        evalResultTraces
-                ]
-        , mempty
-        ]
-
-prettyEvalResult :: EvalResult -> Text
-prettyEvalResult = display
+          Right term ->
+            vsep
+              [ "Evaluation was SUCCESSFUL, result is:"
+              , indent 2 $ prettyPlcReadableSimple term
+              ]
+      , mempty
+      , vsep
+          [ "Execution budget spent:"
+          , indent 2 $ prettyExBudget evalResultBudget
+          ]
+      , mempty
+      , if null evalResultTraces
+          then "No traces were emitted"
+          else
+            vsep
+              [ "Evaluation"
+                  <+> plural "trace" "traces" (length evalResultTraces)
+                  <> ":"
+              , indent 2 $
+                  vsep $
+                    zipWith
+                      (\idx trace -> pretty idx <> dot <+> pretty trace)
+                      [1 :: Int ..]
+                      evalResultTraces
+              ]
+      , mempty
+      ]
+
+displayEvalResult :: EvalResult -> Text
+displayEvalResult = display
+
+displayExBudget :: ExBudget -> Text
+displayExBudget = render . prettyExBudget
+
+prettyExBudget :: ExBudget -> Doc ann
+prettyExBudget
+  ExBudget{exBudgetCPU = ExCPU cpu, exBudgetMemory = ExMemory mem} =
+    vsep
+      [ "CPU" <+> pretty (format commas (unSatInt cpu))
+      , "MEM" <+> pretty (format commas (unSatInt mem))
+      ]
 
 evaluateCompiledCode :: CompiledCode a -> EvalResult
 evaluateCompiledCode code = EvalResult{..}