109 lines
3.1 KiB
Haskell
109 lines
3.1 KiB
Haskell
{-# LANGUAGE FlexibleInstances #-}
|
|
|
|
module Types where
|
|
|
|
import Control.Monad
|
|
import CoreMonad
|
|
import DynFlags
|
|
import ErrMsg
|
|
import Exception
|
|
import GHC
|
|
import GHC.Paths (libdir)
|
|
|
|
----------------------------------------------------------------
|
|
|
|
data OutputStyle = LispStyle | PlainStyle
|
|
|
|
data Options = Options {
|
|
outputStyle :: OutputStyle
|
|
, hlintOpts :: [String]
|
|
, ghcOpts :: [String]
|
|
, operators :: Bool
|
|
}
|
|
|
|
----------------------------------------------------------------
|
|
convert :: ToString a => Options -> a -> String
|
|
convert Options{ outputStyle = LispStyle } = toLisp
|
|
convert Options{ outputStyle = PlainStyle } = toPlain
|
|
|
|
class ToString a where
|
|
toLisp :: a -> String
|
|
toPlain :: a -> String
|
|
|
|
instance ToString [String] where
|
|
toLisp = addNewLine . toSexp True
|
|
toPlain = unlines
|
|
|
|
instance ToString [((Int,Int,Int,Int),String)] where
|
|
toLisp = addNewLine . toSexp False . map toS
|
|
where
|
|
toS x = "(" ++ tupToString x ++ ")"
|
|
toPlain = unlines . map tupToString
|
|
|
|
toSexp :: Bool -> [String] -> String
|
|
toSexp False ss = "(" ++ unwords ss ++ ")"
|
|
toSexp True ss = "(" ++ unwords (map quote ss) ++ ")"
|
|
|
|
tupToString :: ((Int,Int,Int,Int),String) -> String
|
|
tupToString ((a,b,c,d),s) = show a ++ " "
|
|
++ show b ++ " "
|
|
++ show c ++ " "
|
|
++ show d ++ " "
|
|
++ quote s
|
|
|
|
quote :: String -> String
|
|
quote x = "\"" ++ x ++ "\""
|
|
|
|
addNewLine :: String -> String
|
|
addNewLine = (++ "\n")
|
|
|
|
----------------------------------------------------------------
|
|
|
|
withGHC :: (MonadPlus m) => Ghc (m a) -> IO (m a)
|
|
withGHC body = ghandle ignore $ runGhc (Just libdir) body
|
|
where
|
|
ignore :: (MonadPlus m) => SomeException -> IO (m a)
|
|
ignore _ = return mzero
|
|
|
|
----------------------------------------------------------------
|
|
|
|
initSession0 :: Options -> Ghc [PackageId]
|
|
initSession0 opt = getSessionDynFlags >>=
|
|
(>>= setSessionDynFlags) . setGhcFlags opt
|
|
|
|
initSession :: Options -> [String] -> [FilePath] -> Bool -> Ghc LogReader
|
|
initSession opt cmdOpts idirs logging = do
|
|
dflags <- getSessionDynFlags
|
|
let opts = map noLoc cmdOpts
|
|
(dflags',_,_) <- parseDynamicFlags dflags opts
|
|
(dflags'',readLog) <- liftIO . (>>= setLogger logging) . setGhcFlags opt . setFlags dflags' $ idirs
|
|
setSessionDynFlags dflags''
|
|
return readLog
|
|
|
|
----------------------------------------------------------------
|
|
|
|
setFlags :: DynFlags -> [FilePath] -> DynFlags
|
|
setFlags d idirs = d'
|
|
where
|
|
d' = d {
|
|
packageFlags = ghcPackage : packageFlags d
|
|
, importPaths = idirs
|
|
, ghcLink = NoLink
|
|
, hscTarget = HscInterpreted
|
|
}
|
|
|
|
ghcPackage :: PackageFlag
|
|
ghcPackage = ExposePackage "ghc"
|
|
|
|
setGhcFlags :: Monad m => Options -> DynFlags -> m DynFlags
|
|
setGhcFlags opt flagset =
|
|
do (flagset',_,_) <- parseDynamicFlags flagset (map noLoc (ghcOpts opt))
|
|
return flagset'
|
|
|
|
----------------------------------------------------------------
|
|
|
|
setTargetFile :: (GhcMonad m) => String -> m ()
|
|
setTargetFile file = do
|
|
target <- guessTarget file Nothing
|
|
setTargets [target]
|