Writer functor

Smallibs · Sep 17, 2023 · 65b0a35 · 65b0a35
1 parent 1c839d1
commit 65b0a35
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/standard/writer.rs b/src/standard/writer.rs
@@ -5,6 +5,7 @@
  */
 
 use crate::core::hkp::HKP;
+use crate::specs::functor::Functor;
 use std::marker::PhantomData;
 
 pub struct Writer<'a, E: 'a, F: HKP<'a>, A: 'a>(F::T<(A, E)>);
@@ -14,3 +15,13 @@ pub struct WriterK<'a, E, F: HKP<'a>>(PhantomData<&'a E>, PhantomData<F>);
 impl<'a, E, F: HKP<'a>> HKP<'a> for WriterK<'a, E, F> {
     type T<B: 'a> = Writer<'a, E, F, B>;
 }
+
+impl<'e, E, F: Functor<'e>> Functor<'e> for WriterK<'e, E, F> {
+    fn map<A, B, MAP>(f: MAP, ma: Self::T<A>) -> Self::T<B>
+    where
+        MAP: Fn(A) -> B + 'e,
+    {
+        let Writer(va) = ma;
+        Writer(F::map(move |(a, e)| (f(a), e), va))
+    }
+}