From fa5facc5fcf173b2a223ed011b089f9c25504cc9 Mon Sep 17 00:00:00 2001
From: Noah Metz <noah@metznet.ca>
Date: Tue, 5 Sep 2023 10:48:04 -0600
Subject: [PATCH] Added deserialize for map

---
 context.go | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/context.go b/context.go
index 77d8f44..c050445 100644
--- a/context.go
+++ b/context.go
@@ -764,7 +764,30 @@ func NewContext(db * badger.DB, log Logger) (*Context, error) {
         reflect_value := reflect.MakeMap(reflect_type)
         return reflect_type, &reflect_value, new_value, nil
       } else {
-        return nil, nil, value, fmt.Errorf("deserialize map with elements unimplemented")
+        // TODO: basically copy above except instead of getting the key/elem type once, get key/elem values for map_size
+        tmp_value := value
+        var map_value reflect.Value
+        var map_type reflect.Type = nil
+        for i := 0; i < int(map_size); i += 1 {
+          tmp_value.TypeStack = value.TypeStack
+          var key_type, elem_type reflect.Type
+          var key_value, elem_value *reflect.Value
+          var err error
+          key_type, key_value, tmp_value, err = DeserializeValue(ctx, tmp_value)
+          if err != nil {
+            return nil, nil, value, err
+          }
+          elem_type, elem_value, tmp_value, err = DeserializeValue(ctx, tmp_value)
+          if err != nil {
+            return nil, nil, value, err
+          }
+          if map_type == nil {
+            map_type = reflect.MapOf(key_type, elem_type)
+            map_value = reflect.MakeMap(map_type)
+          }
+          map_value.SetMapIndex(*key_value, *elem_value)
+        }
+        return map_type, &map_value, tmp_value, nil
       }
     }
   })