run session build in mask_

Author: soulomoon

ghcide/session-loader/Development/IDE/Session.hs

@@ -799,7 +799,8 @@ loadSessionWithOptions recorder SessionLoadingOptions{..} rootDir = do
     IdeOptions{ optCheckProject = getCheckProject} <- getIdeOptions
     returnWithVersion $ \file -> do
         -- only one cradle consult at a time
-        async <- UnliftIO.async $ UnliftIO.withMVar cradleLock $ const $ do
+
+        UnliftIO.mask_ $ UnliftIO.withMVar cradleLock $ const $ do
                 -- we need to find a way to get rid of the (files, keys)
             _opts@(a, b, _files, _keys) <- use_ HieYaml file
             files <- liftIO $ atomically $ swapTVar targetFiles []
@@ -808,9 +809,12 @@ loadSessionWithOptions recorder SessionLoadingOptions{..} rootDir = do
             liftIO $ when (notNull files || notNull keys) $ do
                 checkProject <- getCheckProject
                 -- think of not to restart a second time
-                restartShakeSession VFSUnmodified "new component" (if checkProject then return (typecheckAll files) else mempty) $ pure keys
+                async <- UnliftIO.async $ restartShakeSession VFSUnmodified "new component" (if checkProject then return (typecheckAll files) else mempty) $ pure keys
+                UnliftIO.wait async
             pure $ (fmap . fmap) toAbsolutePath (a, b)
-        UnliftIO.wait async)
+
+        -- UnliftIO.wait async
+        )
 
 
 -- | Run the specific cradle on a specific FilePath via hie-bios.