1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234
/*! # Universal Suffix Calls
This module provides a single trait, `Pipe`, which provides a number of methods
useful for placing functions in suffix position. The most common method, `pipe`,
forwards a value `T` into any function `T -> R`, returning `R`. The other
methods all apply some form of borrowing to the value before passing the borrow
into the piped function. These are of less value, but provided to maintain a
similar API to the `tap` module’s methods, and for convenience in the event that
you do have a use for them.
This module is as much of a [UFCS] method syntax that can be provided as a
library, rather than in the language grammar.
[UFCS]: https://en.wikipedia.org/wiki/Uniform_Function_Call_Syntax
!*/
use core::{
borrow::{Borrow, BorrowMut},
ops::{Deref, DerefMut},
};
/** Provides universal suffix-position call syntax for any function.
This trait provides methods that allow any closure or free function to be placed
as a suffix-position call, by writing them as
```rust
# use tap::pipe::Pipe;
# let receiver = 5;
fn not_a_method(x: i32) -> u8 { x as u8 }
receiver.pipe(not_a_method);
```
Piping into functions that take more than one argument still requires writing a
closure with ordinary function-call syntax. This is after all only a library,
not a syntax transformation:
```rust
use tap::pipe::Pipe;
fn add(x: i32, y: i32) -> i32 { x + y }
let out = 5.pipe(|x| add(x, 10));
assert_eq!(out, 15);
```
Like tapping, piping is useful for cases where you want to write a sequence of
processing steps without introducing many intermediate bindings, and your steps
contain functions which are not eligible for dot-call syntax.
The main difference between piping and tapping is that tapping always returns
the value that was passed into the tap, while piping forwards the value into the
effect function, and returns the output of evaluating the effect function with
the value. Piping is a transformation, not merely an inspection or modification.
**/
pub trait Pipe {
/// Pipes by value. This is generally the method you want to use.
///
/// # Examples
///
/// ```rust
/// use tap::pipe::Pipe;
///
/// fn triple(x: i32) -> i64 {
/// x as i64 * 3
/// }
///
/// assert_eq!(
/// 10.pipe(triple),
/// 30,
/// );
/// ```
#[inline(always)]
fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> R
where
Self: Sized,
R: Sized,
{
func(self)
}
/// Borrows `self` and passes that borrow into the pipe function.
///
/// # Examples
///
/// ```rust
/// use tap::pipe::Pipe;
///
/// fn fold(v: &Vec<i32>) -> i32 {
/// v.iter().copied().sum()
/// }
/// let vec = vec![1, 2, 3, 4, 5];
/// let sum = vec.pipe_ref(fold);
/// assert_eq!(sum, 15);
/// assert_eq!(vec.len(), 5);
/// ```
#[inline(always)]
fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> R
where
R: 'a + Sized,
{
func(self)
}
/// Mutably borrows `self` and passes that borrow into the pipe function.
///
/// # Examples
///
/// ```rust
/// use tap::pipe::Pipe;
///
/// let mut vec = vec![false, true];
/// let last = vec
/// .pipe_ref_mut(Vec::pop)
/// .pipe(Option::unwrap);
/// assert!(last);
/// ```
///
/// Both of these functions are eligible for method-call syntax, and should
/// not be piped. Writing out non-trivial examples for these is a lot of
/// boilerplate.
#[inline(always)]
fn pipe_ref_mut<'a, R>(
&'a mut self,
func: impl FnOnce(&'a mut Self) -> R,
) -> R
where
R: 'a + Sized,
{
func(self)
}
/// Borrows `self`, then passes `self.borrow()` into the pipe function.
///
/// # Examples
///
/// ```rust
/// use std::borrow::Cow;
/// use tap::pipe::Pipe;
///
/// let len = Cow::<'static, str>::from("hello, world")
/// .pipe_borrow(str::len);
/// assert_eq!(len, 12);
/// ```
#[inline(always)]
fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
where
Self: Borrow<B>,
B: 'a + ?Sized,
R: 'a + Sized,
{
func(Borrow::<B>::borrow(self))
}
/// Mutably borrows `self`, then passes `self.borrow_mut()` into the pipe
/// function.
///
/// ```rust
/// use tap::pipe::Pipe;
///
/// let mut txt = "hello, world".to_string();
/// let ptr = txt
/// .pipe_borrow_mut(str::as_mut_ptr);
/// ```
///
/// This is a very contrived example, but the `BorrowMut` trait has almost
/// no implementors in the standard library, and of the implementations
/// available, there are almost no methods that fit this API.
#[inline(always)]
fn pipe_borrow_mut<'a, B, R>(
&'a mut self,
func: impl FnOnce(&'a mut B) -> R,
) -> R
where
Self: BorrowMut<B>,
B: 'a + ?Sized,
R: 'a + Sized,
{
func(BorrowMut::<B>::borrow_mut(self))
}
/// Borrows `self`, then passes `self.as_ref()` into the pipe function.
#[inline(always)]
fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
where
Self: AsRef<U>,
U: 'a + ?Sized,
R: 'a + Sized,
{
func(AsRef::<U>::as_ref(self))
}
/// Mutably borrows `self`, then passes `self.as_mut()` into the pipe
/// function.
#[inline(always)]
fn pipe_as_mut<'a, U, R>(
&'a mut self,
func: impl FnOnce(&'a mut U) -> R,
) -> R
where
Self: AsMut<U>,
U: 'a + ?Sized,
R: 'a + Sized,
{
func(AsMut::<U>::as_mut(self))
}
/// Borrows `self`, then passes `self.deref()` into the pipe function.
#[inline(always)]
fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
where
Self: Deref<Target = T>,
T: 'a + ?Sized,
R: 'a + Sized,
{
func(Deref::deref(self))
}
/// Mutably borrows `self`, then passes `self.deref_mut()` into the pipe
/// function.
#[inline(always)]
fn pipe_deref_mut<'a, T, R>(
&'a mut self,
func: impl FnOnce(&'a mut T) -> R,
) -> R
where
Self: DerefMut + Deref<Target = T>,
T: 'a + ?Sized,
R: 'a + Sized,
{
func(DerefMut::deref_mut(self))
}
}
impl<T> Pipe for T where T: ?Sized {}