use crate::{Dictionary, FromIon, IonError, Row};
use std::str::FromStr;
#[derive(Clone, Debug, PartialEq)]
pub enum Value {
String(String),
Integer(i64),
Float(f64),
Boolean(bool),
Array(Row),
Dictionary(Dictionary),
}
impl Value {
pub fn new_string(value: &str) -> Self {
Value::String(value.to_owned())
}
pub fn new_string_array(value: &str) -> Self {
Self::new_array(Self::new_string(value))
}
pub fn new_array(value: Value) -> Self {
Value::Array(vec![value])
}
pub fn type_str(&self) -> &'static str {
match self {
Value::String(..) => "string",
Value::Integer(..) => "integer",
Value::Float(..) => "float",
Value::Boolean(..) => "boolean",
Value::Array(..) => "array",
Value::Dictionary(..) => "dictionary",
}
}
pub fn as_string(&self) -> Option<&String> {
match self {
Value::String(v) => Some(v),
_ => None,
}
}
pub fn is_string(&self) -> bool {
matches!(self, Value::String(_))
}
pub fn as_str(&self) -> Option<&str> {
match self {
Value::String(v) => Some(v.as_str()),
_ => None,
}
}
pub fn as_integer(&self) -> Option<i64> {
match self {
Value::Integer(v) => Some(*v),
_ => None,
}
}
pub fn as_float(&self) -> Option<f64> {
match self {
Value::Float(v) => Some(*v),
_ => None,
}
}
pub fn as_boolean(&self) -> Option<bool> {
match self {
Value::Boolean(v) => Some(*v),
_ => None,
}
}
pub fn as_array(&self) -> Option<&Vec<Value>> {
match self {
Value::Array(v) => Some(v),
_ => None,
}
}
pub fn as_dictionary(&self) -> Option<&Dictionary> {
match self {
Value::Dictionary(v) => Some(v),
_ => None,
}
}
pub fn get(&self, name: &str) -> Option<&Value> {
match self {
Value::Dictionary(v) => v.get(name),
_ => None,
}
}
pub fn from_ion<F>(&self) -> Result<F, F::Err>
where
F: FromIon<Value>,
{
F::from_ion(self)
}
pub fn parse<F>(&self) -> Result<F, F::Err>
where
F: FromStr,
{
match self.as_string() {
Some(s) => s.parse(),
None => self.to_string().parse(),
}
}
}
impl FromStr for Value {
type Err = IonError;
fn from_str(s: &str) -> Result<Value, IonError> {
Ok(Value::String(s.to_owned()))
}
}
#[cfg(test)]
mod tests {
use crate::Value;
#[test]
fn integer() {
let v: Value = "1".parse().unwrap();
assert_eq!(1, v.parse().unwrap());
}
#[test]
fn float() {
let v: Value = "4.0".parse().unwrap();
assert_eq!(4.0f64, v.parse().unwrap());
}
}