use crate::parse::format::Format;
use crate::types::Type;
use crate::value::Value;
use crate::{OverError, OverResult, INDENT_STEP};
use std::fmt;
use std::slice::Iter;
use std::sync::Arc;
#[derive(Clone, Debug)]
struct ArrInner {
vec: Vec<Value>,
inner_t: Type,
}
#[derive(Clone, Debug)]
pub struct Arr {
inner: Arc<ArrInner>,
}
impl Arr {
pub fn from_vec(vec: Vec<Value>) -> OverResult<Arr> {
let mut tcur = Type::Any;
let mut has_any = true;
for value in &vec {
let tnew = value.get_type();
if has_any {
match Type::most_specific(&tcur, &tnew) {
Some((t, any)) => {
tcur = t;
has_any = any;
}
None => return Err(OverError::ArrTypeMismatch(tcur, tnew)),
}
} else if tcur != tnew {
return Err(OverError::ArrTypeMismatch(tcur, tnew));
}
}
Ok(Arr {
inner: Arc::new(ArrInner { vec, inner_t: tcur }),
})
}
pub fn from_vec_unchecked(vec: Vec<Value>, inner_t: Type) -> Arr {
Arr {
inner: Arc::new(ArrInner { vec, inner_t }),
}
}
pub fn vec_ref(&self) -> &Vec<Value> {
&self.inner.vec
}
pub fn with_each<F>(&self, mut f: F)
where
F: FnMut(&Value),
{
for value in &self.inner.vec {
f(value)
}
}
pub fn get(&self, index: usize) -> OverResult<Value> {
if index >= self.inner.vec.len() {
Err(OverError::ArrOutOfBounds(index))
} else {
Ok(self.inner.vec[index].clone())
}
}
pub fn inner_type(&self) -> Type {
self.inner.inner_t.clone()
}
pub fn len(&self) -> usize {
self.inner.vec.len()
}
pub fn is_empty(&self) -> bool {
self.inner.vec.is_empty()
}
pub fn ptr_eq(&self, other: &Self) -> bool {
Arc::ptr_eq(&self.inner, &other.inner)
}
pub fn iter(&self) -> Iter<Value> {
self.vec_ref().iter()
}
}
impl Default for Arr {
fn default() -> Self {
Self::from_vec_unchecked(vec![], Type::Any)
}
}
impl fmt::Display for Arr {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}", self.format(true, INDENT_STEP))
}
}
impl PartialEq for Arr {
fn eq(&self, other: &Self) -> bool {
if self.inner.inner_t != other.inner.inner_t {
return false;
}
self.inner.vec == other.inner.vec
}
}