131 lines
4.4 KiB
Haskell
131 lines
4.4 KiB
Haskell
-- ghc-mod: Making Haskell development *more* fun
|
|
-- Copyright (C) 2015 Daniel Gröber <dxld ÄT darkboxed DOT org>
|
|
--
|
|
-- This program is free software: you can redistribute it and/or modify
|
|
-- it under the terms of the GNU Affero General Public License as published by
|
|
-- the Free Software Foundation, either version 3 of the License, or
|
|
-- (at your option) any later version.
|
|
--
|
|
-- This program is distributed in the hope that it will be useful,
|
|
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
-- GNU Affero General Public License for more details.
|
|
--
|
|
-- You should have received a copy of the GNU Affero General Public License
|
|
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
{-# LANGUAGE CPP #-}
|
|
module Language.Haskell.GhcMod.Monad (
|
|
runGmOutT
|
|
, runGmOutT'
|
|
, runGhcModT
|
|
, runGhcModT'
|
|
, hoistGhcModT
|
|
, runGmlT
|
|
, runGmlT'
|
|
, runGmlTWith
|
|
, runGmPkgGhc
|
|
, withGhcModEnv
|
|
, withGhcModEnv'
|
|
, module Language.Haskell.GhcMod.Monad.Types
|
|
) where
|
|
|
|
import Language.Haskell.GhcMod.Types
|
|
import Language.Haskell.GhcMod.Monad.Types
|
|
import Language.Haskell.GhcMod.Error
|
|
import Language.Haskell.GhcMod.Logging
|
|
import Language.Haskell.GhcMod.Cradle
|
|
import Language.Haskell.GhcMod.Target
|
|
import Language.Haskell.GhcMod.Output
|
|
|
|
import Control.Arrow (first)
|
|
import Control.Applicative
|
|
|
|
import Control.Concurrent
|
|
|
|
import Control.Monad.Reader (runReaderT)
|
|
import Control.Monad.State.Strict (runStateT)
|
|
import Control.Monad.Trans.Journal (runJournalT)
|
|
|
|
import Exception
|
|
|
|
import System.Directory
|
|
import Prelude
|
|
|
|
withGhcModEnv :: (IOish m, GmOut m) => FilePath -> Options -> (GhcModEnv -> m a) -> m a
|
|
withGhcModEnv = withGhcModEnv' withCradle
|
|
where
|
|
withCradle dir =
|
|
gbracket (findCradle' dir) (liftIO . cleanupCradle)
|
|
|
|
withGhcModEnv' :: (IOish m, GmOut m) => (FilePath -> (Cradle -> m a) -> m a) -> FilePath -> Options -> (GhcModEnv -> m a) -> m a
|
|
withGhcModEnv' withCradle dir opts f =
|
|
withCradle dir $ \crdl ->
|
|
withCradleRootDir crdl $
|
|
f $ GhcModEnv opts crdl
|
|
where
|
|
withCradleRootDir (cradleRootDir -> projdir) a = do
|
|
cdir <- liftIO $ getCurrentDirectory
|
|
eq <- liftIO $ pathsEqual projdir cdir
|
|
if not eq
|
|
then throw $ GMEWrongWorkingDirectory projdir cdir
|
|
else a
|
|
|
|
pathsEqual a b = do
|
|
ca <- canonicalizePath a
|
|
cb <- canonicalizePath b
|
|
return $ ca == cb
|
|
|
|
runGmOutT :: IOish m => Options -> GmOutT m a -> m a
|
|
runGmOutT opts ma = do
|
|
gmo@GhcModOut{..} <- GhcModOut (optOutput opts) <$> liftIO newChan
|
|
let action = runGmOutT' gmo ma
|
|
case ooptLinePrefix $ optOutput opts of
|
|
Nothing -> action
|
|
Just pfxs ->
|
|
gbracket_ (liftIO $ forkIO $ stdoutGateway pfxs gmoChan)
|
|
(const $ liftIO $ flushStdoutGateway gmoChan)
|
|
action
|
|
|
|
runGmOutT' :: IOish m => GhcModOut -> GmOutT m a -> m a
|
|
runGmOutT' gmo ma = flip runReaderT gmo $ unGmOutT ma
|
|
|
|
-- | Run a @GhcModT m@ computation.
|
|
runGhcModT :: IOish m
|
|
=> Options
|
|
-> GhcModT m a
|
|
-> m (Either GhcModError a, GhcModLog)
|
|
runGhcModT opt action = liftIO (getCurrentDirectory >>= canonicalizePath) >>= \dir' -> do
|
|
runGmOutT opt $
|
|
withGhcModEnv dir' opt $ \env ->
|
|
first (fst <$>) <$> runGhcModT' env defaultGhcModState
|
|
(gmSetLogLevel (ooptLogLevel $ optOutput opt) >> action)
|
|
|
|
-- | @hoistGhcModT result@. Embed a GhcModT computation's result into a GhcModT
|
|
-- computation. Note that if the computation that returned @result@ modified the
|
|
-- state part of GhcModT this cannot be restored.
|
|
hoistGhcModT :: IOish m
|
|
=> (Either GhcModError a, GhcModLog)
|
|
-> GhcModT m a
|
|
hoistGhcModT (r,l) = do
|
|
gmlJournal l >> case r of
|
|
Left e -> throwError e
|
|
Right a -> return a
|
|
|
|
|
|
-- | Run a computation inside @GhcModT@ providing the RWST environment and
|
|
-- initial state. This is a low level function, use it only if you know what to
|
|
-- do with 'GhcModEnv' and 'GhcModState'.
|
|
--
|
|
-- You should probably look at 'runGhcModT' instead.
|
|
runGhcModT' :: IOish m
|
|
=> GhcModEnv
|
|
-> GhcModState
|
|
-> GhcModT m a
|
|
-> GmOutT m (Either GhcModError (a, GhcModState), GhcModLog)
|
|
runGhcModT' r s a = do
|
|
flip runReaderT r $ runJournalT $ runErrorT $ runStateT (unGmT a) s
|
|
|
|
gbracket_ :: ExceptionMonad m => m a -> (a -> m b) -> m c -> m c
|
|
gbracket_ ma mb mc = gbracket ma mb (const mc)
|