moved file management to the proper place

1 files changed, 3 insertions, 2 deletions

diff --git a/src/Trivmix.hs b/src/Trivmix.hs
index 5dc213c..69fcea2 100644
--- a/src/Trivmix.hs
+++ b/src/Trivmix.hs
@@ -115,7 +115,8 @@ main = execParser opts >>= trivmix
 trivmix :: Options -> IO ()
 trivmix Options{..} = do
   level <- newMVar initialLevel
-  withINotify $ \inotify -> do
+  let withFiles = foldl (.) id $ map (\f -> onStateFile f (show initialLevel ++ "\n")) levelFiles
+  withFiles $ withINotify $ \inotify -> do
     handleFiles inotify level levelFiles
     Jack.handleExceptions $
       Jack.withClientDefault client $ \client' ->
@@ -148,7 +149,7 @@ handleFiles inotify level files = do
       forkIO $ forever $ do -- Broadcast level changes and update all files
         readChan levelChanges' >>= writeLevel file
       addWatch inotify watchedAttrs file (const $ readLevel levelChanges level file)
-  sequence [onStateFile f (show initLevel ++ "\n") $ handleFile f | f <- files]
+  mapM handleFile files
   forkIO $ forever $ do
     readChan levelChanges >>= swapMVar level
     return ()