This commit is contained in:
annieversary 2021-11-05 10:12:03 +00:00
commit e62f421fa6
5 changed files with 127 additions and 0 deletions

2
.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
/target
Cargo.lock

8
Cargo.toml Normal file
View file

@ -0,0 +1,8 @@
[package]
name = "bad-optics"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]

47
src/lenses/first_tuple.rs Normal file
View file

@ -0,0 +1,47 @@
use crate::{LensOver, LensView};
pub struct _0;
macro_rules! make {
($f:ident, ( $( $v:ident ),* ), ( $( $t:ident ),* ) ) => {
impl< $($t,)* > LensView<( $($t,)* )> for _0 {
type Field = T;
fn view(( $($v,)* ): ($($t,)*)) -> Self::Field {
$f
}
}
impl< $($t,)* > LensOver<( $($t,)* )> for _0 {
fn over(
mut tup: ($($t,)*),
f: &dyn Fn(Self::Field) -> Self::Field
) -> ( $($t,)* ) {
tup.0 = f(tup.0);
tup
}
}
impl<'a, $($t,)* > LensView<&'a ( $($t,)* )> for _0 {
type Field = &'a T;
fn view(( $($v,)* ): &'a ($($t,)*)) -> Self::Field {
$f
}
}
impl<'a, $($t,)* > LensView<&'a mut ( $($t,)* )> for _0 {
type Field = &'a mut T;
fn view(( $($v,)* ): &'a mut ($($t,)*)) -> Self::Field {
$f
}
}
};
}
make!(t, (t, _u), (T, U));
make!(t, (t, _u, _v), (T, U, V));
make!(t, (t, _u, _v, _w), (T, U, V, W));
make!(t, (t, _u, _v, _w, _x), (T, U, V, W, X));
make!(t, (t, _u, _v, _w, _x, _y), (T, U, V, W, X, Y));
make!(t, (t, _u, _v, _w, _x, _y, _z), (T, U, V, W, X, Y, Z));
// not doing more cause i'm lazy, open a pr if you need more :)

2
src/lenses/mod.rs Normal file
View file

@ -0,0 +1,2 @@
mod first_tuple;
pub use first_tuple::_0;

68
src/lib.rs Normal file
View file

@ -0,0 +1,68 @@
pub trait LensView<T> {
type Field;
fn view(thing: T) -> Self::Field;
}
pub trait LensOver<T>: LensView<T> {
fn over(thing: T, f: &dyn Fn(Self::Field) -> Self::Field) -> T;
}
pub fn view<T, L: LensView<T>>(_lens: L, thing: T) -> L::Field {
L::view(thing)
}
pub fn over<T, L: LensOver<T>>(_lens: L, thing: T, f: &dyn Fn(L::Field) -> L::Field) -> T {
L::over(thing, f)
}
// TODO add Fn implementation for lenses
// with one param it should be view, with two it should be over
// TODO add std::ops::Add to combine lenses
pub mod lenses;
#[cfg(test)]
mod tests {
use super::{lenses::_0, *};
#[test]
fn view_first_from_tuple() {
let a = (1, 2);
assert_eq!(view(_0, a), 1);
// you can call it both ways
let a = (1, 2);
assert_eq!(_0::view(a), 1);
let a = (1, 2);
assert_eq!(view(_0, &a), &1);
let mut a = (1, 2);
assert_eq!(view(_0, &mut a), &mut 1);
let a = (1, 2, 3);
assert_eq!(view(_0, a), 1);
let a = (1, 2, 3);
assert_eq!(view(_0, &a), &1);
}
#[test]
fn view_first_from_tuple_mut_works() {
let mut a = (1, 2);
*view(_0, &mut a) += 1;
assert_eq!(a, (2, 2));
}
#[test]
fn over_first_from_tuple_mut_works() {
let a = (1, 2);
let a = over(_0, a, &|v| v + 1);
assert_eq!(a, (2, 2));
let a = (1, 2);
let a = _0::over(a, &|v| v + 1);
assert_eq!(a, (2, 2));
}
}