clock functionality

1 files changed, 10 insertions, 10 deletions

diff --git a/countdown.hs b/countdown.hs
index 13269ff..0960a85 100755
--- a/countdown.hs
+++ b/countdown.hs
@@ -52,11 +52,11 @@ billion = 1000 * 1000 * 1000
 ageOfUniverseInYears :: Integer
 ageOfUniverseInYears = 13 * billion
 
-data CustomEvent = Counter deriving Show
+data CustomEvent = TimeChanged UTCTime deriving Show
 
 data St =
     St { _stLastBrickEvent :: Maybe (BrickEvent () CustomEvent)
-       , _stCounter :: Int
+       , _stCurrentTime :: UTCTime
        }
 
 makeLenses ''St
@@ -66,21 +66,21 @@ drawUI st = [a]
     where
         a = (str $ "Last event: " <> (show $ st^.stLastBrickEvent))
             <=>
-            (str $ "Counter value is: " <> (show $ st^.stCounter))
+            (str $ "Time now is: " <> (show $ st^.stCurrentTime))
 
 appEvent :: St -> BrickEvent () CustomEvent -> EventM () (Next St)
 appEvent st e =
     case e of
         VtyEvent (V.EvKey V.KEsc []) -> halt st
         VtyEvent _ -> continue $ st & stLastBrickEvent .~ (Just e)
-        AppEvent Counter -> continue $ st & stCounter %~ (+1)
-                                          & stLastBrickEvent .~ (Just e)
+        AppEvent (TimeChanged t) -> continue $ st & stCurrentTime .~ t
+                                                  & stLastBrickEvent .~ (Just e)
         _ -> continue st
 
-initialState :: St
-initialState =
+initialState :: UTCTime -> St
+initialState t =
     St { _stLastBrickEvent = Nothing
-       , _stCounter = 0
+       , _stCurrentTime = t
        }
 
 theApp :: App St CustomEvent ()
@@ -98,9 +98,9 @@ main = do
     chan <- newBChan 10
 
     void $ forkIO $ forever $ do
-        writeBChan chan Counter
+        getCurrentTime >>= writeBChan chan . TimeChanged
         threadDelay 1000000
 
     let buildVty = V.mkVty V.defaultConfig
     initialVty <- buildVty
-    void $ customMain initialVty buildVty (Just chan) theApp initialState
+    void $ customMain initialVty buildVty (Just chan) theApp (initialState now)