Compare commits

...

2 commits

Author SHA1 Message Date
soup af3fa30986
Make some impls more general 2024-05-19 03:12:02 -04:00
soup 3f08af78d0
Add some additional impls 2024-05-19 02:55:02 -04:00
2 changed files with 153 additions and 80 deletions

View file

@ -19,7 +19,6 @@ pub fn execute_(out: &mut String, ast: &Ast, value: &dyn Value) {
AstKind::For(f) => for_(out, f, value),
AstKind::Has(h) => has(out, h, value),
AstKind::If(h) => if_(out, h, value),
e => todo!("{e:?}"),
}
}
}

View file

@ -1,4 +1,4 @@
use core::fmt::{Display, Write};
use core::fmt::Write;
pub mod internal;
@ -7,7 +7,7 @@ use internal::{
parse::{parse, Ast},
};
pub trait Value: core::fmt::Debug {
pub trait Value {
fn print(&self, out: &mut String);
fn lookup(&self, name: &str) -> Option<&dyn Value>;
fn has(&self) -> Option<&dyn Value>;
@ -15,9 +15,76 @@ pub trait Value: core::fmt::Debug {
fn for_(&self, index: usize) -> Option<&dyn Value>;
}
macro_rules! value_for_int {
($tp:ty) => {
impl Value for $tp {
fn print(&self, out: &mut String) {
let _ = write!(out, "{self}");
}
fn lookup(&self, _: &str) -> Option<&dyn Value> {
None
}
fn has(&self) -> Option<&dyn Value> {
if self.if_() {
Some(self)
} else {
None
}
}
fn if_(&self) -> bool {
*self != 0
}
fn for_(&self, _: usize) -> Option<&dyn Value> {
None
}
}
};
}
value_for_int!(u8);
value_for_int!(u16);
value_for_int!(u32);
value_for_int!(u64);
value_for_int!(u128);
value_for_int!(usize);
value_for_int!(i8);
value_for_int!(i16);
value_for_int!(i32);
value_for_int!(i64);
value_for_int!(i128);
value_for_int!(isize);
impl<'a, T> Value for &'a T
where
T: Value,
T: Value + ?Sized,
{
fn print(&self, out: &mut String) {
(&**self).print(out)
}
fn lookup(&self, name: &str) -> Option<&dyn Value> {
(&**self).lookup(name)
}
fn has(&self) -> Option<&dyn Value> {
(&**self).has()
}
fn if_(&self) -> bool {
(&**self).if_()
}
fn for_(&self, index: usize) -> Option<&dyn Value> {
(&**self).for_(index)
}
}
impl<'a, T> Value for &'a mut T
where
T: Value + ?Sized,
{
fn print(&self, out: &mut String) {
(&**self).print(out)
@ -70,24 +137,56 @@ impl Value for &str {
}
}
impl Value for (&str, &dyn Value) {
impl Value for String {
fn print(&self, out: &mut String) {
out.push_str(self);
}
fn lookup(&self, _: &str) -> Option<&dyn Value> {
None
}
fn has(&self) -> Option<&dyn Value> {
if self.if_() {
None
} else {
Some(self)
}
}
fn if_(&self) -> bool {
!self.is_empty()
}
fn for_(&self, index: usize) -> Option<&dyn Value> {
if index == 0 {
Some(self)
} else {
None
}
}
}
impl<A, B> Value for (A, B)
where
A: Value,
B: Value,
{
fn print(&self, out: &mut String) {
out.push('(');
self.0.print(out);
out.push_str(", ");
self.1.print(out);
}
fn lookup(&self, name: &str) -> Option<&dyn Value> {
match name {
"0" => Some(&self.0),
"1" => Some(self.1),
"1" => Some(&self.1),
_ => None,
}
}
fn has(&self) -> Option<&dyn Value> {
Some(self)
None
}
fn if_(&self) -> bool {
@ -97,53 +196,55 @@ impl Value for (&str, &dyn Value) {
fn for_(&self, index: usize) -> Option<&dyn Value> {
match index {
0 => Some(&self.0),
1 => Some(self.1),
1 => Some(&self.1),
_ => None,
}
}
}
impl Value for (&dyn Value, &dyn Value) {
fn print(&self, out: &mut String) {
self.0.print(out);
self.1.print(out);
}
fn lookup(&self, name: &str) -> Option<&dyn Value> {
match name {
"0" => Some(self.0),
"1" => Some(self.1),
_ => None,
}
}
fn has(&self) -> Option<&dyn Value> {
Some(self)
}
fn if_(&self) -> bool {
self.0.if_() && self.1.if_()
}
fn for_(&self, index: usize) -> Option<&dyn Value> {
match index {
0 => Some(self.0),
1 => Some(self.1),
_ => None,
}
}
}
impl<const N: usize> Value for [(&str, &dyn Value); N] {
pub struct ValuesListMap<'a>(pub &'a [(&'a str, &'a dyn Value)]);
impl<'a> Value for ValuesListMap<'a> {
fn print(&self, w: &mut String) {
for (name, v) in self.iter() {
for (name, v) in self.0.iter() {
let _ = write!(w, "{name}");
v.print(w);
}
}
fn lookup(&self, name: &str) -> Option<&dyn Value> {
self.iter().find(|(n, _)| *n == name).map(|(_, v)| *v)
self.0.iter().find(|(n, _)| *n == name).map(|(_, v)| *v)
}
fn has(&self) -> Option<&dyn Value> {
if self.0.is_empty() {
None
} else {
Some(self)
}
}
fn if_(&self) -> bool {
!self.0.is_empty()
}
fn for_(&self, index: usize) -> Option<&dyn Value> {
self.0.get(index).map(|v| v as &dyn Value)
}
}
impl<const N: usize, T> Value for [T; N]
where
T: Value,
{
fn print(&self, w: &mut String) {
for v in self.iter() {
v.print(w)
}
}
fn lookup(&self, name: &str) -> Option<&dyn Value> {
let idx: usize = name.parse().ok()?;
self.get(idx).map(|v| v as &dyn Value)
}
fn has(&self) -> Option<&dyn Value> {
@ -163,35 +264,6 @@ impl<const N: usize> Value for [(&str, &dyn Value); N] {
}
}
impl<const N: usize> Value for [&dyn Value; N] {
fn print(&self, w: &mut String) {
for v in self.iter() {
v.print(w)
}
}
fn lookup(&self, name: &str) -> Option<&dyn Value> {
let idx: usize = name.parse().ok()?;
self.get(idx).copied()
}
fn has(&self) -> Option<&dyn Value> {
if self.is_empty() {
None
} else {
Some(self)
}
}
fn if_(&self) -> bool {
self.is_empty()
}
fn for_(&self, index: usize) -> Option<&dyn Value> {
self.get(index).map(|v| *v)
}
}
impl<T> Value for Option<T>
where
T: Value,
@ -242,11 +314,13 @@ impl Newt {
#[macro_export]
macro_rules! values_list_map {
($($key:literal: $val:expr),* $(,)?) => {{
&[
$(
($key, $val as &dyn Value)
)*,
]
&$crate::ValuesListMap(
&[
$(
($key, $val as &dyn Value)
)*,
]
)
}}
}
@ -300,10 +374,10 @@ mod test {
"{for .items}{for .names}{.}, {/for}{/for}",
values_list_map! {
"items": values![values_list_map! {
"names": values!["Bob", "Larry"],
"names": values!["Bob", "Larry", 0],
}]
},
"Bob, Larry, ",
"Bob, Larry, 0, ",
)
}