Network.hs

addMBox :: RethinkDBHandle -> Token -> Term -> IO (TVar Response)
addMBox h tok term = do
  chan <- newChan
  -- TVar doesn't have an empty state?
  mbox <- newTVarIO $ ResponseSingle Nothing Null
  weak <- mkWeakTVar mbox $ do
    closeToken h tok -- TODO: don't close if already closed
    atomicModifyIORef' (rdbWait h) $ \mboxes ->
      (M.delete tok mboxes, ())
  atomicModifyIORef' (rdbWait h) $ \mboxes ->
    (M.insert tok (chan, term, finalize weak) mboxes, ())
  _ <- forkIO $ fix $ \loop -> do
    response <- readChan chan
    -- This prints the proper response
    putStrLn "Response"
    print response
    -- Doesn't seem to get written at the right time or something because I get null in the repl when testing.
    atomically $ writeTVar mbox response
    when (not $ isLastResponse response) $ do
      nextResponse response
      loop
  return mbox