2022-07-24 14:49:02 -05:00
|
|
|
use ordered_float::OrderedFloat;
|
|
|
|
use std::{
|
|
|
|
borrow::Cow,
|
|
|
|
fmt::{Display, Write},
|
2022-07-24 15:21:34 -05:00
|
|
|
hash::Hash,
|
2022-07-24 14:49:02 -05:00
|
|
|
};
|
|
|
|
|
2022-07-24 15:21:34 -05:00
|
|
|
pub type Span = std::ops::Range<usize>;
|
|
|
|
|
|
|
|
/// A spanned item
|
|
|
|
#[derive(Clone, Debug)]
|
|
|
|
pub struct Spanned<T> {
|
|
|
|
pub item: T,
|
|
|
|
pub span: Span,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T> Spanned<T> {
|
|
|
|
pub fn new(item: T, span: Span) -> Self {
|
|
|
|
Self { item, span }
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T: Display> Display for Spanned<T> {
|
|
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
|
|
write!(f, "{} \x1B[2m@ {:?}\x1B[22m", self.item, self.span)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T: PartialEq> PartialEq for Spanned<T> {
|
|
|
|
fn eq(&self, other: &Self) -> bool {
|
|
|
|
self.item == other.item
|
|
|
|
}
|
|
|
|
}
|
2022-07-24 15:24:53 -05:00
|
|
|
|
2022-07-24 15:21:34 -05:00
|
|
|
impl<T: Hash> Hash for Spanned<T> {
|
|
|
|
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
|
|
|
|
self.item.hash(state);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-24 14:49:02 -05:00
|
|
|
/// A Wisp AST
|
2022-07-24 15:21:34 -05:00
|
|
|
#[derive(Debug, Clone, Hash, PartialEq)]
|
2022-07-24 14:49:02 -05:00
|
|
|
pub enum Expr<'a> {
|
2022-07-24 15:21:34 -05:00
|
|
|
List(Vec<Spanned<Self>>),
|
|
|
|
Vector(Vec<Spanned<Self>>),
|
|
|
|
Map(Vec<(Spanned<Self>, Spanned<Self>)>),
|
|
|
|
Quote(Box<Spanned<Self>>),
|
2022-07-24 14:49:02 -05:00
|
|
|
Symbol(Cow<'a, str>),
|
|
|
|
Keyword(Cow<'a, str>),
|
|
|
|
Number(OrderedFloat<f64>),
|
|
|
|
String(Cow<'a, str>),
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a> Display for Expr<'a> {
|
|
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
|
|
match self {
|
|
|
|
Self::List(list) => write_seq(f, list, "(", ")"),
|
|
|
|
Self::Vector(vec) => write_seq(f, vec, "[", "]"),
|
2022-07-24 15:00:25 -05:00
|
|
|
Self::Map(map) => write_seq(f, map.iter().map(|(k, v)| format!("{k} {v}")), "{", "}"),
|
2022-07-24 15:21:34 -05:00
|
|
|
Self::Quote(expr) => write!(f, "'{expr}"),
|
2022-07-24 14:49:02 -05:00
|
|
|
Self::Symbol(sym) => write!(f, "{sym}"),
|
|
|
|
Self::Keyword(kw) => write!(f, ":{kw}"),
|
|
|
|
Self::Number(n) => write!(f, "{n}"),
|
|
|
|
Self::String(s) => write!(f, "\"{s}\""),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-24 14:56:20 -05:00
|
|
|
fn write_seq(
|
2022-07-24 14:49:02 -05:00
|
|
|
f: &mut impl Write,
|
|
|
|
iterable: impl IntoIterator<Item = impl Display>,
|
|
|
|
delimiter_left: &str,
|
|
|
|
delimiter_right: &str,
|
|
|
|
) -> std::fmt::Result {
|
|
|
|
let mut iter = iterable.into_iter();
|
|
|
|
write!(f, "{delimiter_left}")?;
|
|
|
|
if let Some(x) = iter.next() {
|
|
|
|
write!(f, "{x}")?;
|
|
|
|
}
|
|
|
|
for x in iter {
|
|
|
|
write!(f, " {x}")?;
|
|
|
|
}
|
|
|
|
write!(f, "{delimiter_right}")
|
|
|
|
}
|