use crate::{JsonValue, TomlValue, extension::TomlTableExt};
use serde_json::Number;
pub trait TomlValueExt {
fn as_u8(&self) -> Option<u8>;
fn as_u16(&self) -> Option<u16>;
fn as_u32(&self) -> Option<u32>;
fn as_usize(&self) -> Option<usize>;
fn as_i8(&self) -> Option<i8>;
fn as_i16(&self) -> Option<i16>;
fn as_i32(&self) -> Option<i32>;
fn as_isize(&self) -> Option<isize>;
fn as_f32(&self) -> Option<f32>;
fn to_json_value(&self) -> JsonValue;
}
impl TomlValueExt for TomlValue {
#[inline]
fn as_u8(&self) -> Option<u8> {
self.as_integer().and_then(|i| u8::try_from(i).ok())
}
#[inline]
fn as_u16(&self) -> Option<u16> {
self.as_integer().and_then(|i| u16::try_from(i).ok())
}
#[inline]
fn as_u32(&self) -> Option<u32> {
self.as_integer().and_then(|i| u32::try_from(i).ok())
}
#[inline]
fn as_usize(&self) -> Option<usize> {
self.as_integer().and_then(|i| usize::try_from(i).ok())
}
#[inline]
fn as_i8(&self) -> Option<i8> {
self.as_integer().and_then(|i| i8::try_from(i).ok())
}
#[inline]
fn as_i16(&self) -> Option<i16> {
self.as_integer().and_then(|i| i16::try_from(i).ok())
}
#[inline]
fn as_i32(&self) -> Option<i32> {
self.as_integer().and_then(|i| i32::try_from(i).ok())
}
#[inline]
fn as_isize(&self) -> Option<isize> {
self.as_integer().and_then(|i| isize::try_from(i).ok())
}
#[inline]
fn as_f32(&self) -> Option<f32> {
self.as_float().map(|f| f as f32)
}
fn to_json_value(&self) -> JsonValue {
match self {
TomlValue::String(s) => JsonValue::String(s.to_owned()),
TomlValue::Integer(i) => JsonValue::Number((*i).into()),
TomlValue::Float(f) => {
if let Some(number) = Number::from_f64(*f) {
JsonValue::Number(number)
} else {
JsonValue::Null
}
}
TomlValue::Boolean(b) => JsonValue::Bool(*b),
TomlValue::Datetime(dt) => JsonValue::String(dt.to_string()),
TomlValue::Array(vec) => {
let vec = vec.iter().map(|v| v.to_json_value()).collect();
JsonValue::Array(vec)
}
TomlValue::Table(table) => JsonValue::Object(table.to_map()),
}
}
}