From 058781081b811b65bf14bf2b673de22d0761fbad Mon Sep 17 00:00:00 2001
From: Simon Bihel <simon.bihel@spruceid.com>
Date: Mon, 12 Jun 2023 16:53:06 +0100
Subject: Add support for enums in top level

Partially address #71
---
 src/de/mod.rs   | 13 +++++++------
 src/de/parse.rs |  4 ++--
 2 files changed, 9 insertions(+), 8 deletions(-)

(limited to 'src/de')

diff --git a/src/de/mod.rs b/src/de/mod.rs
index ef9cd76..150f6fa 100644
--- a/src/de/mod.rs
+++ b/src/de/mod.rs
@@ -45,6 +45,7 @@ use serde::de::IntoDeserializer;
 
 use std::borrow::Cow;
 use std::collections::btree_map::{BTreeMap, Entry, IntoIter};
+use std::iter::Peekable;
 
 /// To override the default serialization parameters, first construct a new
 /// Config.
@@ -190,7 +191,7 @@ pub fn from_str<'de, T: de::Deserialize<'de>>(input: &'de str) -> Result<T> {
 ///
 /// Supported top-level outputs are structs and maps.
 pub(crate) struct QsDeserializer<'a> {
-    iter: IntoIter<Cow<'a, str>, Level<'a>>,
+    iter: Peekable<IntoIter<Cow<'a, str>, Level<'a>>>,
     value: Option<Level<'a>>,
 }
 
@@ -207,7 +208,7 @@ enum Level<'a> {
 impl<'a> QsDeserializer<'a> {
     fn with_map(map: BTreeMap<Cow<'a, str>, Level<'a>>) -> Self {
         QsDeserializer {
-            iter: map.into_iter(),
+            iter: map.into_iter().peekable(),
             value: None,
         }
     }
@@ -225,11 +226,11 @@ impl<'de> de::Deserializer<'de> for QsDeserializer<'de> {
     where
         V: de::Visitor<'de>,
     {
-        if self.iter.next().is_none() {
-            return visitor.visit_unit();
+        if self.iter.peek().is_none() {
+            visitor.visit_unit()
+        } else {
+            self.deserialize_map(visitor)
         }
-
-        Err(Error::top_level("primitive"))
     }
 
     fn deserialize_map<V>(self, visitor: V) -> Result<V::Value>
diff --git a/src/de/parse.rs b/src/de/parse.rs
index a76f2c5..0080d02 100644
--- a/src/de/parse.rs
+++ b/src/de/parse.rs
@@ -280,8 +280,8 @@ impl<'a> Parser<'a> {
         // Parses all top level nodes into the `root` map.
         while self.parse(&mut root)? {}
         let iter = match root {
-            Level::Nested(map) => map.into_iter(),
-            _ => BTreeMap::default().into_iter(),
+            Level::Nested(map) => map.into_iter().peekable(),
+            _ => BTreeMap::default().into_iter().peekable(),
         };
         Ok(QsDeserializer { iter, value: None })
     }
-- 
cgit v1.2.3