#![no_std]
#[macro_export]
macro_rules! defmac {
($name:ident => $e:expr) => {
macro_rules! $name {
() => { $e }
}
};
($name:ident $x:pat => $e:expr) => {
macro_rules! $name {
($arg:expr) => {
match $arg { $x => $e }
}
}
};
($name:ident $x1:pat, $x2:pat => $e:expr) => {
macro_rules! $name {
($a1:expr, $a2:expr) => {
match $a1 { $x1 =>
match $a2 { $x2 => $e } }
}
}
};
($name:ident $x1:pat, $x2:pat, $x3:pat => $e:expr) => {
macro_rules! $name {
($a1:expr, $a2:expr, $a3:expr) => {
match $a1 { $x1 =>
match $a2 { $x2 =>
match $a3 { $x3 => $e } } }
}
}
};
($name:ident $x1:pat, $x2:pat, $x3:pat, $x4:pat => $e:expr) => {
macro_rules! $name {
($a1:expr, $a2:expr, $a3:expr, $a4:expr) => {
match $a1 { $x1 =>
match $a2 { $x2 =>
match $a3 { $x3 =>
match $a4 { $x4 => $e } } } }
}
}
};
}
#[cfg(test)]
mod tests {
#[test]
fn it_works() {
let value = "xyz";
defmac!(none => value);
assert_eq!(none!(), "xyz");
defmac!(one x => x);
assert_eq!(one!(2), 2);
defmac!(two x, y => x + y);
assert_eq!(two!(1., 2.), 3.);
defmac!(three x, y, z => (x, y, z));
assert_eq!(three!(1, (2, 3), (4, 5, 6)), (1, (2, 3), (4, 5, 6)));
defmac!(four w, x, y, z => (w + x, z, y));
assert_eq!(four!(3, 4, "a", "b"), (7, "b", "a"));
}
}