custom_print/
into_write_fn.rs1#[cfg(feature = "alloc")]
2use alloc::string::String;
3#[cfg(feature = "std")]
4use std::ffi::{CStr, CString};
5#[cfg(feature = "std")]
6use std::os::raw::c_char;
7
8#[cfg(feature = "alloc")]
9use crate::WriteStringFn;
10use crate::{WriteBytesFn, WriteLenPtrFn, WritePtrLenFn, WriteStrFn};
11#[cfg(feature = "std")]
12use crate::{WriteCCharPtrFn, WriteCStrFn, WriteCStringFn};
13
14pub trait IntoWriteFn<Ts> {
34 type WriteFn;
36
37 fn into_write_fn(self) -> Self::WriteFn;
39}
40
41macro_rules! def {
42 ( ($F:tt, $R:tt), $func:ty, ($($ty:ty),*) ) => {
43 impl<$F, $R> IntoWriteFn<($($ty,)*)> for $F
44 where
45 $F: FnMut($($ty),*) -> $R,
46 {
47 type WriteFn = $func;
48 fn into_write_fn(self) -> Self::WriteFn {
49 Self::WriteFn::new(self)
50 }
51 }
52
53 }
54}
55
56def!((F, R), WritePtrLenFn<F, R>, (*const u8, usize));
57def!((F, R), WriteLenPtrFn<F, R>, (usize, *const u8));
58def!((F, R), WriteBytesFn<F, R>, (&[u8]));
59
60def!((F, R), WriteStrFn<F, R>, (&str));
61#[cfg(feature = "alloc")]
62def!((F, R), WriteStringFn<F, R>, (String));
63
64#[cfg(feature = "std")]
65def!((F, R), WriteCStrFn<F, R>, (&CStr));
66#[cfg(feature = "std")]
67def!((F, R), WriteCStringFn<F, R>, (CString));
68#[cfg(feature = "std")]
69def!((F, R), WriteCCharPtrFn<F, R>, (*const c_char));