Completed untested Flattening implementation

1 files changed, 15 insertions, 1 deletions

diff --git a/src/Data/JLD/Model/NodeMap.hs b/src/Data/JLD/Model/NodeMap.hs
index d0fb2f9..f76c662 100644
--- a/src/Data/JLD/Model/NodeMap.hs
+++ b/src/Data/JLD/Model/NodeMap.hs
@@ -10,13 +10,18 @@ module Data.JLD.Model.NodeMap (
     hasKey2,
     hasKey3,
     memberArray,
+    propsToKeyMap,
 ) where
 
 import Data.JLD.Prelude hiding (modify)
 
 import Data.Aeson (Array, Value (..))
+import Data.Aeson.Key qualified as K
+import Data.Aeson.KeyMap (KeyMap)
+import Data.Aeson.KeyMap qualified as KM
+import Data.Foldable.WithIndex (FoldableWithIndex (..))
 import Data.JLD.Util (valueToArray)
-import Data.Map.Strict qualified as M (alter, insert, lookup, member)
+import Data.Map.Strict qualified as M (alter, insert, lookup, member, toList)
 
 type PropertyKey = Maybe Text
 type PropertyMap = Map PropertyKey Value
@@ -55,3 +60,12 @@ memberArray :: GraphKey -> SubjectKey -> PropertyKey -> Value -> NodeMap -> Bool
 memberArray graphName subject property value nodeMap = case lookup3 graphName subject property nodeMap of
     Just (Array a) -> value `elem` a
     _ -> False
+
+propsToKeyMap :: PropertyMap -> KeyMap Value
+propsToKeyMap =
+    ifoldl'
+        ( \maybeKey km value -> case maybeKey of
+            Just key -> KM.insert (K.fromText key) value km
+            Nothing -> km
+        )
+        mempty