feat: add convenience function for the high-level interface

anoma · Nov 10, 2024 · 4daf003 · 4daf003
1 parent 92e70e3
commit 4daf003
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/Anoma/Transaction/AppData.juvix b/Anoma/Transaction/AppData.juvix
@@ -4,6 +4,7 @@ import Stdlib.Prelude open;
 import Stdlib.Trait.Ord.Eq open using {fromOrdToEq};
 import Stdlib.Data.Map as Map open using {Map};
 import Anoma.Builtin.ByteArray open using {ByteArray};
+import Anoma.Builtin.System open using {anomaEncode; anomaDecode};
 import Anoma.Utils open;
 
 --- A data type encoding the deletion criterion.
@@ -74,3 +75,14 @@ module AppDataValueInternal;
   instance
   AppDataValue-Eq : Eq AppDataValue := fromOrdToEq;
 end;
+
+lookupAppData
+  {Key Value : Type} (key : Key) (appData : AppData) : Maybe Value :=
+  case
+    Map.lookup@{
+      key := anomaEncode key |> mkAppDataKey;
+      map := appData;
+    }
+  of
+    | nothing := nothing
+    | just value := just (value |> AppDataValue.data |> anomaDecode);
diff --git a/Anoma/Transaction/CustomInputs.juvix b/Anoma/Transaction/CustomInputs.juvix
@@ -4,6 +4,7 @@ import Stdlib.Prelude open;
 import Stdlib.Trait.Ord.Eq open using {fromOrdToEq};
 import Stdlib.Data.Map as Map open using {Map};
 import Anoma.Builtin.ByteArray open using {ByteArray};
+import Anoma.Builtin.System open using {anomaEncode; anomaDecode};
 import Anoma.Utils open;
 
 --- A data type encoding the lookup key of custom inputs .
@@ -50,3 +51,15 @@ module CustomInputsValueInternal;
   instance
   CustomInputsValue-Eq : Eq CustomInputsValue := fromOrdToEq;
 end;
+
+lookupCustomInputs
+  {Key Value : Type} (key : Key) (customInputs : CustomInputs) : Maybe Value :=
+  case
+    Map.lookup@{
+      key := anomaEncode key |> mkCustomInputsKey;
+      map := customInputs;
+    }
+  of
+    | nothing := nothing
+    | just value :=
+      just (value |> CustomInputsValue.unCustomInputsValue |> anomaDecode);