use serde::{Deserialize, Serialize};
use std::borrow::Cow;
use std::collections::HashMap;
use std::hash::{Hash, Hasher};
use std::sync::Arc;
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub enum Value {
#[default]
Undef,
Bool(bool),
Int(i64),
Float(f64),
Str(Arc<String>),
Array(Vec<Value>),
Hash(HashMap<String, Value>),
Status(i32),
Ref(Box<Value>),
NativeFn(u16),
}
impl Value {
pub fn int(n: i64) -> Self {
Value::Int(n)
}
pub fn float(f: f64) -> Self {
Value::Float(f)
}
pub fn str(s: impl Into<String>) -> Self {
Value::Str(Arc::new(s.into()))
}
pub fn bool(b: bool) -> Self {
Value::Bool(b)
}
pub fn array(v: Vec<Value>) -> Self {
Value::Array(v)
}
pub fn hash(m: HashMap<String, Value>) -> Self {
Value::Hash(m)
}
pub fn status(code: i32) -> Self {
Value::Status(code)
}
pub fn is_truthy(&self) -> bool {
match self {
Value::Undef => false,
Value::Bool(b) => *b,
Value::Int(n) => *n != 0,
Value::Float(f) => *f != 0.0,
Value::Str(s) => !s.is_empty() && s.as_str() != "0",
Value::Array(a) => !a.is_empty(),
Value::Hash(h) => !h.is_empty(),
Value::Status(c) => *c == 0, Value::Ref(_) => true,
Value::NativeFn(_) => true,
}
}
pub fn to_int(&self) -> i64 {
match self {
Value::Int(n) => *n,
Value::Float(f) => *f as i64,
Value::Bool(b) => *b as i64,
Value::Str(s) => s.parse().unwrap_or(0),
Value::Status(c) => *c as i64,
Value::Array(a) => a.len() as i64,
_ => 0,
}
}
pub fn to_float(&self) -> f64 {
match self {
Value::Float(f) => *f,
Value::Int(n) => *n as f64,
Value::Bool(b) if *b => 1.0,
Value::Str(s) => s.parse().unwrap_or(0.0),
Value::Status(c) => *c as f64,
_ => 0.0,
}
}
pub fn to_str(&self) -> String {
self.as_str_cow().into_owned()
}
pub fn as_str_cow(&self) -> Cow<'_, str> {
match self {
Value::Str(s) => Cow::Borrowed(s.as_str()),
Value::Int(n) => Cow::Owned(n.to_string()),
Value::Float(f) => Cow::Owned(f.to_string()),
Value::Bool(b) => Cow::Borrowed(if *b { "1" } else { "" }),
Value::Undef => Cow::Borrowed(""),
Value::Status(c) => Cow::Owned(c.to_string()),
Value::Array(a) => {
Cow::Owned(a.iter().map(|v| v.to_str()).collect::<Vec<_>>().join(" "))
}
Value::Hash(_) => Cow::Borrowed("(hash)"),
Value::Ref(_) => Cow::Borrowed("(ref)"),
Value::NativeFn(id) => Cow::Owned(format!("(builtin:{})", id)),
}
}
pub fn len(&self) -> usize {
match self {
Value::Str(s) => s.len(),
Value::Array(a) => a.len(),
Value::Hash(h) => h.len(),
_ => self.to_str().len(),
}
}
pub fn is_empty(&self) -> bool {
self.len() == 0
}
}
impl Hash for Value {
fn hash<H: Hasher>(&self, state: &mut H) {
std::mem::discriminant(self).hash(state);
match self {
Value::Undef => {}
Value::Bool(b) => b.hash(state),
Value::Int(n) => n.hash(state),
Value::Float(f) => f.to_bits().hash(state),
Value::Str(s) => s.hash(state),
Value::Array(a) => a.hash(state),
Value::Hash(h) => {
h.len().hash(state);
for (k, v) in h {
k.hash(state);
v.hash(state);
}
}
Value::Status(c) => c.hash(state),
Value::Ref(b) => b.hash(state),
Value::NativeFn(id) => id.hash(state),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_truthiness() {
assert!(!Value::Undef.is_truthy());
assert!(!Value::Int(0).is_truthy());
assert!(Value::Int(1).is_truthy());
assert!(!Value::str("").is_truthy());
assert!(!Value::str("0").is_truthy());
assert!(Value::str("hello").is_truthy());
assert!(Value::Status(0).is_truthy()); assert!(!Value::Status(1).is_truthy());
}
#[test]
fn test_coercions() {
assert_eq!(Value::str("42").to_int(), 42);
assert_eq!(Value::Int(42).to_str(), "42");
assert_eq!(Value::Float(3.14).to_int(), 3);
assert_eq!(Value::Bool(true).to_int(), 1);
}
}