From 68f6039fbe8e87e3a13f745c78140315f18a5cf4 Mon Sep 17 00:00:00 2001
From: Simon Friis Vindum <simonfv@gmail.com>
Date: Tue, 20 Aug 2019 13:47:03 +0200
Subject: [PATCH] Fix flatMap and flatten types

---
 src/component.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/component.ts b/src/component.ts
index c2614d4..84da745 100644
--- a/src/component.ts
+++ b/src/component.ts
@@ -58,10 +58,10 @@ export abstract class Component<A, O> implements Monad<O> {
   of<P>(p: P): Component<{}, P> {
     return new OfComponent(p);
   }
-  flatMap<P>(f: (o: O) => Component<A, P>): Component<A, P> {
+  flatMap<B, P>(f: (o: O) => Component<B, P>): Component<A, P> {
     return new FlatMapComponent(this, f);
   }
-  chain<P>(f: (o: O) => Component<A, P>): Component<A, P> {
+  chain<B, P>(f: (o: O) => Component<B, P>): Component<A, P> {
     return new FlatMapComponent(this, f);
   }
   output<P>(f: (a: A) => P): Component<A, O & P>;
@@ -91,7 +91,7 @@ export abstract class Component<A, O> implements Monad<O> {
   multi: boolean = false;
   abstract run(parent: DomApi, destroyed: Future<boolean>): Out<A, O>;
   // Definitions below are inserted by Jabz
-  flatten: <B>() => Component<O, B>;
+  flatten: <B, P>(this: Component<A, Component<B, P>>) => Component<A, P>;
   map: <P>(f: (a: O) => P) => Component<A, P>;
   mapTo: <P>(b: P) => Component<A, P>;
   ap: <P>(a: Component<A, (o: O) => P>) => Component<A, P>;
@@ -177,10 +177,10 @@ export function output<A>(
  */
 export const emptyComponent = Component.of({});
 
-class FlatMapComponent<A, O, P> extends Component<A, P> {
+class FlatMapComponent<A, O, B, P> extends Component<A, P> {
   constructor(
     private readonly component: Component<A, O>,
-    private readonly f: (o: O) => Component<A, P>
+    private readonly f: (o: O) => Component<B, P>
   ) {
     super();
   }