ghc-mod/GHCApi.hs

153 lines
4.5 KiB
Haskell
Raw Normal View History

2013-03-04 04:41:56 +00:00
module GHCApi (
withGHC
, withGHCDummyFile
, initializeFlags
, initializeFlagsWithCradle
, setTargetFile
, getDynamicFlags
, setSlowDynFlags
2013-03-15 05:40:36 +00:00
, checkSlowAndSet
2013-03-04 04:41:56 +00:00
) where
2012-02-14 07:09:53 +00:00
2013-03-03 06:47:03 +00:00
import CabalApi
2012-02-15 05:52:48 +00:00
import Control.Applicative
import Control.Exception
2013-03-04 04:41:56 +00:00
import Control.Monad
2012-02-14 07:09:53 +00:00
import CoreMonad
2013-03-02 07:14:55 +00:00
import Data.Maybe (isJust)
2012-02-14 07:09:53 +00:00
import DynFlags
import ErrMsg
import Exception
import GHC
import GHC.Paths (libdir)
import System.Exit
import System.IO
2012-02-14 07:09:53 +00:00
import Types
----------------------------------------------------------------
2013-03-04 04:41:56 +00:00
withGHCDummyFile :: Alternative m => Ghc (m a) -> IO (m a)
withGHCDummyFile = withGHC "Dummy"
2012-12-07 05:27:02 +00:00
2013-03-04 04:41:56 +00:00
withGHC :: Alternative m => FilePath -> Ghc (m a) -> IO (m a)
withGHC file body = ghandle ignore $ runGhc (Just libdir) $ do
2012-02-14 10:14:21 +00:00
dflags <- getSessionDynFlags
defaultCleanupHandler dflags body
2012-02-14 07:09:53 +00:00
where
2012-02-15 05:52:48 +00:00
ignore :: Alternative m => SomeException -> IO (m a)
ignore e = do
2012-12-07 05:27:02 +00:00
hPutStr stderr $ file ++ ":0:0:Error:"
hPrint stderr e
2012-06-04 06:40:26 +00:00
exitSuccess
2012-02-14 07:09:53 +00:00
----------------------------------------------------------------
importDirs :: [IncludeDir]
importDirs = [".","..","../..","../../..","../../../..","../../../../.."]
2013-03-13 04:17:22 +00:00
initializeFlagsWithCradle :: Options -> Cradle -> [GHCOption] -> Bool -> Ghc LogReader
initializeFlagsWithCradle opt cradle ghcOptions logging
2013-03-04 01:40:33 +00:00
| cabal = do
2013-03-13 04:17:22 +00:00
(gopts,idirs,depPkgs) <- liftIO $ fromCabalFile ghcOptions cradle
initSession opt gopts idirs (Just depPkgs) logging
2013-03-04 01:40:33 +00:00
| otherwise =
2013-03-13 04:17:22 +00:00
initSession opt ghcOptions importDirs Nothing logging
where
2013-03-02 07:14:55 +00:00
cabal = isJust $ cradleCabalFile cradle
2013-03-04 04:41:56 +00:00
----------------------------------------------------------------
initSession :: Options
-> [GHCOption]
-> [IncludeDir]
-> Maybe [Package]
-> Bool
-> Ghc LogReader
2013-03-13 04:17:22 +00:00
initSession opt cmdOpts idirs mDepPkgs logging = do
2013-03-04 02:21:41 +00:00
dflags0 <- getSessionDynFlags
2013-03-04 04:11:04 +00:00
(dflags1,readLog) <- setupDynamicFlags dflags0
_ <- setSessionDynFlags dflags1
2012-02-14 07:09:53 +00:00
return readLog
2013-03-04 04:11:04 +00:00
where
setupDynamicFlags df0 = do
df1 <- modifyFlagsWithOpts df0 cmdOpts
let df2 = modifyFlags df1 idirs mDepPkgs (expandSplice opt)
2013-03-05 07:16:27 +00:00
df3 <- modifyFlagsWithOpts df2 $ ghcOpts opt
2013-03-04 04:11:04 +00:00
liftIO $ setLogger logging df3
2012-02-14 07:09:53 +00:00
2013-03-04 02:21:41 +00:00
----------------------------------------------------------------
2013-03-04 04:41:56 +00:00
initializeFlags :: Options -> Ghc ()
initializeFlags opt = do
dflags0 <- getSessionDynFlags
2013-03-05 07:16:27 +00:00
dflags1 <- modifyFlagsWithOpts dflags0 $ ghcOpts opt
2013-03-04 04:41:56 +00:00
void $ setSessionDynFlags dflags1
----------------------------------------------------------------
2013-03-04 03:53:28 +00:00
-- FIXME removing Options
modifyFlags :: DynFlags -> [IncludeDir] -> Maybe [Package] -> Bool -> DynFlags
modifyFlags d0 idirs mDepPkgs splice
2013-03-04 03:53:28 +00:00
| splice = setSplice d3
| otherwise = d3
2012-02-14 07:09:53 +00:00
where
2013-03-04 03:53:28 +00:00
d1 = d0 { importPaths = idirs }
d2 = setFastOrNot d1 Fast
2013-03-04 03:53:28 +00:00
d3 = maybe d2 (addDevPkgs d2) mDepPkgs
setSplice :: DynFlags -> DynFlags
2013-03-04 04:11:04 +00:00
setSplice dflag = dopt_set dflag Opt_D_dump_splices
2013-03-04 03:53:28 +00:00
2013-03-13 04:17:22 +00:00
addDevPkgs :: DynFlags -> [Package] -> DynFlags
addDevPkgs df pkgs = df''
where
df' = dopt_set df Opt_HideAllPackages
df'' = df' {
packageFlags = map ExposePackage pkgs ++ packageFlags df
}
----------------------------------------------------------------
setFastOrNot :: DynFlags -> CheckSpeed -> DynFlags
setFastOrNot dflags Slow = dflags {
2013-03-04 04:11:04 +00:00
ghcLink = LinkInMemory
, hscTarget = HscInterpreted
2013-03-04 03:53:28 +00:00
}
setFastOrNot dflags Fast = dflags {
2013-03-04 04:11:04 +00:00
ghcLink = NoLink
, hscTarget = HscNothing
2013-03-04 03:53:28 +00:00
}
2012-02-14 07:09:53 +00:00
2013-03-13 04:17:22 +00:00
setSlowDynFlags :: Ghc ()
setSlowDynFlags = (flip setFastOrNot Slow <$> getSessionDynFlags)
2013-03-13 04:17:22 +00:00
>>= void . setSessionDynFlags
2013-03-04 02:21:41 +00:00
2013-03-15 05:40:36 +00:00
-- To check TH, a session module graph is necessary.
-- "load" sets a session module graph using "depanal".
-- But we have to set "-fno-code" to DynFlags before "load".
-- So, this is necessary redundancy.
checkSlowAndSet :: Ghc ()
checkSlowAndSet = do
slow <- needsTemplateHaskell <$> depanal [] False
when slow setSlowDynFlags
2013-03-04 02:21:41 +00:00
----------------------------------------------------------------
2012-02-14 07:09:53 +00:00
2013-03-05 07:16:27 +00:00
modifyFlagsWithOpts :: DynFlags -> [String] -> Ghc DynFlags
modifyFlagsWithOpts dflags cmdOpts =
tfst <$> parseDynamicFlags dflags (map noLoc cmdOpts)
where
tfst (a,_,_) = a
2012-02-14 07:09:53 +00:00
----------------------------------------------------------------
setTargetFile :: (GhcMonad m) => String -> m ()
setTargetFile file = do
target <- guessTarget file Nothing
setTargets [target]
2013-03-04 02:21:41 +00:00
----------------------------------------------------------------
2013-03-04 04:41:56 +00:00
getDynamicFlags :: IO DynFlags
getDynamicFlags = runGhc (Just libdir) getSessionDynFlags