From 1c7e875f67e31c68e76c660294b67078acd4f38f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Mon, 6 May 2019 21:50:47 +0200
Subject: Multiple signatures in Signed object

---
 src/Storage.hs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'src/Storage.hs')

diff --git a/src/Storage.hs b/src/Storage.hs
index b3c2619..a306206 100644
--- a/src/Storage.hs
+++ b/src/Storage.hs
@@ -24,7 +24,7 @@ module Storage (
 
     loadBlob, loadRec, loadZero,
     loadInt, loadNum, loadText, loadBinary, loadDate, loadJson, loadRef,
-    loadMbInt, loadMbNum, loadMbText, loadMbBinary, loadMbDate, loadMbJson, loadMbRef,
+    loadMbInt, loadMbNum, loadMbText, loadMbBinary, loadMbDate, loadMbJson, loadMbRef, loadRefs,
     loadZRef,
 
     Stored,
@@ -607,12 +607,21 @@ loadMbRawRef name = asks (lookup (BC.pack name) . snd) >>= \case
     Just (RecRef x) -> return (Just x)
     Just _ -> throwError $ "Expecting type ref of record item '"++name++"'"
 
+loadRawRefs :: String -> LoadRec [Ref]
+loadRawRefs name = do
+    items <- map snd . filter ((BC.pack name ==) . fst) <$> asks snd
+    forM items $ \case RecRef x -> return x
+                       _ -> throwError $ "Expecting type ref of record item '"++name++"'"
+
 loadRef :: Storable a => String -> LoadRec a
 loadRef name = load <$> loadRawRef name
 
 loadMbRef :: Storable a => String -> LoadRec (Maybe a)
 loadMbRef name = fmap load <$> loadMbRawRef name
 
+loadRefs :: Storable a => String -> LoadRec [a]
+loadRefs name = map load <$> loadRawRefs name
+
 loadZRef :: ZeroStorable a => String -> LoadRec a
 loadZRef name = loadMbRef name >>= \case
                     Nothing -> do Ref st _ <- asks fst
-- 
cgit v1.2.3