From d14abdb248794bb7e0dd5ace90dcbe0d294401ae Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jlamothe1980@gmail.com>
Date: Mon, 11 Nov 2019 20:52:56 -0500
Subject: [PATCH] implemented menuControllerWith

---
 src/Mtlstats/Menu.hs | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/Mtlstats/Menu.hs b/src/Mtlstats/Menu.hs
index 1d586c3..b812ae2 100644
--- a/src/Mtlstats/Menu.hs
+++ b/src/Mtlstats/Menu.hs
@@ -59,12 +59,7 @@ import Mtlstats.Util
 
 -- | Generates a simple 'Controller' for a Menu
 menuController :: Menu () -> Controller
-menuController menu = Controller
-  { drawController   = const $ drawMenu menu
-  , handleController = \e -> do
-    menuHandler menu e
-    return True
-  }
+menuController = menuControllerWith $ const $ return ()
 
 -- | Generate a simple 'Controller' for a 'Menu' with a header
 menuControllerWith
@@ -74,7 +69,14 @@ menuControllerWith
   -- ^ The menu
   -> Controller
   -- ^ The resulting controller
-menuControllerWith = undefined
+menuControllerWith header menu = Controller
+  { drawController = \s -> do
+    header s
+    drawMenu menu
+  , handleController = \e -> do
+    menuHandler menu e
+    return True
+  }
 
 -- | The draw function for a 'Menu'
 drawMenu :: Menu a -> C.Update C.CursorMode