Skip to content

Commit

Permalink
WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
epage committed Jan 13, 2022
1 parent 68c6c70 commit a79471b
Show file tree
Hide file tree
Showing 13 changed files with 57 additions and 150 deletions.
2 changes: 1 addition & 1 deletion src/array.rs
Original file line number Diff line number Diff line change
Expand Up @@ -340,7 +340,7 @@ impl IntoIterator for Array {
self.values
.into_iter()
.filter(|v| v.is_value())
.map(|v| v.into_value().unwrap()),
.map(|v| v.into_value()),
)
}
}
Expand Down
1 change: 0 additions & 1 deletion src/de/item.rs
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,6 @@ impl<'de, 'a> serde::Deserializer<'de> for crate::Item {
V: serde::de::Visitor<'de>,
{
match self {
crate::Item::None => visitor.visit_none(),
crate::Item::Value(v) => v.deserialize_any(visitor),
crate::Item::Table(v) => visitor.visit_map(crate::de::TableMapAccess::new(v)),
crate::Item::ArrayOfTables(v) => {
Expand Down
34 changes: 7 additions & 27 deletions src/index.rs
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
use std::ops;

use crate::document::Document;
use crate::key::Key;
use crate::table::TableKeyValue;
use crate::{value, InlineTable, InternalString, Item, Table, Value};
use crate::{InlineTable, Item, Table, Value};

// copied from
// https://github.com/serde-rs/json/blob/master/src/value/index.rs
Expand Down Expand Up @@ -39,34 +37,16 @@ impl Index for str {
Item::Value(ref v) => v
.as_inline_table()
.and_then(|t| t.items.get(self))
.and_then(|kv| {
if !kv.value.is_none() {
Some(&kv.value)
} else {
None
}
}),
.map(|kv| &kv.value),
_ => None,
}
}
fn index_mut<'v>(&self, v: &'v mut Item) -> Option<&'v mut Item> {
if let Item::None = *v {
let mut t = InlineTable::default();
t.items.insert(
InternalString::from(self),
TableKeyValue::new(Key::new(self), Item::None),
);
*v = value(Value::InlineTable(t));
}
match *v {
Item::Table(ref mut t) => Some(t.entry(self).or_insert(Item::None)),
Item::Value(ref mut v) => v.as_inline_table_mut().map(|t| {
&mut t
.items
.entry(InternalString::from(self))
.or_insert_with(|| TableKeyValue::new(Key::new(self), Item::None))
.value
}),
Item::Table(ref mut t) => t.get_mut(self),
Item::Value(ref mut v) => v
.as_inline_table_mut()
.and_then(|t| t.items.get_mut(self).map(|kv| &mut kv.value)),
_ => None,
}
}
Expand Down Expand Up @@ -123,7 +103,7 @@ impl<'s> ops::Index<&'s str> for Table {

impl<'s> ops::IndexMut<&'s str> for Table {
fn index_mut(&mut self, key: &'s str) -> &mut Item {
self.entry(key).or_insert(Item::None)
self.get_mut(key).expect("index not found")
}
}

Expand Down
48 changes: 20 additions & 28 deletions src/inline_table.rs
Original file line number Diff line number Diff line change
Expand Up @@ -205,14 +205,8 @@ impl InlineTable {
match self.items.entry(key.into()) {
indexmap::map::Entry::Occupied(mut entry) => {
// Ensure it is a `Value` to simplify `InlineOccupiedEntry`'s code.
let scratch = std::mem::take(&mut entry.get_mut().value);
let scratch = Item::Value(
scratch
.into_value()
// HACK: `Item::None` is a corner case of a corner case, let's just pick a
// "safe" value
.unwrap_or_else(|_| Value::InlineTable(Default::default())),
);
let scratch = std::mem::replace(&mut entry.get_mut().value, crate::value(0));
let scratch = Item::Value(scratch.into_value());
entry.get_mut().value = scratch;

InlineEntry::Occupied(InlineOccupiedEntry { entry })
Expand All @@ -229,14 +223,8 @@ impl InlineTable {
match self.items.entry(key.get().into()) {
indexmap::map::Entry::Occupied(mut entry) => {
// Ensure it is a `Value` to simplify `InlineOccupiedEntry`'s code.
let scratch = std::mem::take(&mut entry.get_mut().value);
let scratch = Item::Value(
scratch
.into_value()
// HACK: `Item::None` is a corner case of a corner case, let's just pick a
// "safe" value
.unwrap_or_else(|_| Value::InlineTable(Default::default())),
);
let scratch = std::mem::replace(&mut entry.get_mut().value, crate::value(0));
let scratch = Item::Value(scratch.into_value());
entry.get_mut().value = scratch;

InlineEntry::Occupied(InlineOccupiedEntry { entry })
Expand Down Expand Up @@ -285,7 +273,7 @@ impl InlineTable {
let kv = TableKeyValue::new(Key::new(key), Item::Value(value));
self.items
.insert(InternalString::from(key), kv)
.and_then(|kv| kv.value.into_value().ok())
.map(|kv| kv.value.into_value())
}

/// Inserts a key-value pair into the map.
Expand All @@ -294,21 +282,20 @@ impl InlineTable {
self.items
.insert(InternalString::from(key.get()), kv)
.filter(|kv| kv.value.is_value())
.map(|kv| kv.value.into_value().unwrap())
.map(|kv| kv.value.into_value())
}

/// Removes an item given the key.
pub fn remove(&mut self, key: &str) -> Option<Value> {
self.items
.shift_remove(key)
.and_then(|kv| kv.value.into_value().ok())
self.items.shift_remove(key).map(|kv| kv.value.into_value())
}

/// Removes a key from the map, returning the stored key and value if the key was previously in the map.
pub fn remove_entry(&mut self, key: &str) -> Option<(Key, Value)> {
self.items.shift_remove(key).and_then(|kv| {
self.items.shift_remove(key).map(|kv| {
let key = kv.key;
kv.value.into_value().ok().map(|value| (key, value))
let value = kv.value.into_value();
(key, value)
})
}
}
Expand Down Expand Up @@ -351,7 +338,7 @@ impl IntoIterator for InlineTable {
self.items
.into_iter()
.filter(|(_, kv)| kv.value.is_value())
.map(|(k, kv)| (k, kv.value.into_value().unwrap())),
.map(|(k, kv)| (k, kv.value.into_value())),
)
}
}
Expand Down Expand Up @@ -395,6 +382,12 @@ impl TableLike for InlineTable {
.map(|(_, kv)| (kv.key.as_mut(), &mut kv.value)),
)
}
fn len(&self) -> usize {
self.len()
}
fn is_empty(&self) -> bool {
self.is_empty()
}
fn get<'s>(&'s self, key: &str) -> Option<&'s Item> {
self.items.get(key).map(|kv| &kv.value)
}
Expand All @@ -405,8 +398,7 @@ impl TableLike for InlineTable {
self.contains_key(key)
}
fn insert(&mut self, key: &str, value: Item) -> Option<Item> {
self.insert(key, value.into_value().unwrap())
.map(Item::Value)
self.insert(key, value.into_value()).map(Item::Value)
}
fn remove(&mut self, key: &str) -> Option<Item> {
self.remove(key).map(Item::Value)
Expand Down Expand Up @@ -526,12 +518,12 @@ impl<'a> InlineOccupiedEntry<'a> {
pub fn insert(&mut self, value: Value) -> Value {
let mut value = Item::Value(value);
std::mem::swap(&mut value, &mut self.entry.get_mut().value);
value.into_value().unwrap()
value.into_value()
}

/// Takes the value out of the entry, and returns it
pub fn remove(self) -> Value {
self.entry.shift_remove().value.into_value().unwrap()
self.entry.shift_remove().value.into_value()
}
}

Expand Down
40 changes: 7 additions & 33 deletions src/item.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,6 @@ use crate::{Array, InlineTable, Table, Value};
/// Type representing either a value, a table, an array of tables, or none.
#[derive(Debug, Clone)]
pub enum Item {
/// Type representing none.
None,
/// Type representing value.
Value(Value),
/// Type representing table.
Expand All @@ -18,24 +16,12 @@ pub enum Item {
ArrayOfTables(ArrayOfTables),
}

impl Item {
/// Sets `self` to the given item iff `self` is none and
/// returns a mutable reference to `self`.
pub fn or_insert(&mut self, item: Item) -> &mut Item {
if self.is_none() {
*self = item
}
self
}
}

// TODO: This should be generated by macro or derive
/// Downcasting
impl Item {
/// Text description of value type
pub fn type_name(&self) -> &'static str {
match self {
Item::None => "none",
Item::Value(v) => v.type_name(),
Item::Table(..) => "table",
Item::ArrayOfTables(..) => "array of tables",
Expand Down Expand Up @@ -113,24 +99,23 @@ impl Item {
}
}
/// Casts `self` to value.
pub fn into_value(self) -> Result<Value, Self> {
pub fn into_value(self) -> Value {
match self {
Item::None => Err(self),
Item::Value(v) => Ok(v),
Item::Value(v) => v,
Item::Table(v) => {
let v = v.into_inline_table();
Ok(Value::InlineTable(v))
Value::InlineTable(v)
}
Item::ArrayOfTables(v) => {
let v = v.into_array();
Ok(Value::Array(v))
Value::Array(v)
}
}
}
/// In-place convert to a value
pub fn make_value(&mut self) {
let other = std::mem::take(self);
let other = other.into_value().map(Item::Value).unwrap_or(Item::None);
let other = std::mem::replace(self, value(0));
let other = Item::Value(other.into_value());
*self = other;
}
/// Casts `self` to table.
Expand Down Expand Up @@ -164,7 +149,7 @@ impl Item {
}
// Starting private because the name is unclear
pub(crate) fn make_item(&mut self) {
let other = std::mem::take(self);
let other = std::mem::replace(self, value(0));
let other = match other.into_table().map(crate::Item::Table) {
Ok(i) => i,
Err(i) => i,
Expand All @@ -187,10 +172,6 @@ impl Item {
pub fn is_array_of_tables(&self) -> bool {
self.as_array_of_tables().is_some()
}
/// Returns true iff `self` is `None`.
pub fn is_none(&self) -> bool {
matches!(*self, Item::None)
}

// Duplicate Value downcasting API

Expand Down Expand Up @@ -296,12 +277,6 @@ impl Item {
}
}

impl Default for Item {
fn default() -> Self {
Item::None
}
}

impl FromStr for Item {
type Err = crate::TomlError;

Expand All @@ -315,7 +290,6 @@ impl FromStr for Item {
impl std::fmt::Display for Item {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match &self {
Item::None => Ok(()),
Item::Value(v) => v.fmt(f),
Item::Table(v) => v.fmt(f),
Item::ArrayOfTables(v) => v.fmt(f),
Expand Down
1 change: 0 additions & 1 deletion src/parser/table.rs
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,6 @@ impl TomlParser {
Item::Table(ref mut sweet_child_of_mine) => {
table = sweet_child_of_mine;
}
_ => unreachable!(),
}
}
Ok(table)
Expand Down
1 change: 0 additions & 1 deletion src/ser/pretty.rs
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,6 @@ impl crate::visit_mut::VisitMut for Pretty {

fn remove_table_prefix(node: &mut crate::Item) {
match node {
crate::Item::None => {}
crate::Item::Value(_) => {}
crate::Item::Table(t) => t.decor_mut().set_prefix(""),
crate::Item::ArrayOfTables(a) => {
Expand Down
26 changes: 8 additions & 18 deletions src/ser/table.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use super::{Error, ErrorKind, KeySerializer};
use super::{Error, KeySerializer};

#[doc(hidden)]
pub struct SerializeItemTable {
Expand Down Expand Up @@ -111,17 +111,12 @@ impl serde::ser::SerializeMap for SerializeKeyValuePairs {
let item = match res {
Ok(item) => item,
Err(e) => {
if e.kind != ErrorKind::UnsupportedNone {
return Err(e);
}
crate::Item::None
return Err(e);
}
};
if !item.is_none() {
let key = self.key.take().unwrap();
let kv = crate::table::TableKeyValue::new(crate::Key::new(&key), item);
self.items.insert(key, kv);
}
let key = self.key.take().unwrap();
let kv = crate::table::TableKeyValue::new(crate::Key::new(&key), item);
self.items.insert(key, kv);
Ok(())
}

Expand All @@ -146,16 +141,11 @@ impl serde::ser::SerializeStruct for SerializeKeyValuePairs {
let item = match res {
Ok(item) => item,
Err(e) => {
if e.kind != ErrorKind::UnsupportedNone {
return Err(e);
}
crate::Item::None
return Err(e);
}
};
if !item.is_none() {
let kv = crate::table::TableKeyValue::new(crate::Key::new(key), item);
self.items.insert(crate::InternalString::from(key), kv);
}
let kv = crate::table::TableKeyValue::new(crate::Key::new(key), item);
self.items.insert(crate::InternalString::from(key), kv);
Ok(())
}

Expand Down
Loading

0 comments on commit a79471b

Please sign in to comment.