module UHC.Util.CompileRun2
( CompileRunner
, CompileRunState(..)
, CompileRun(..)
, crCUCache, crCompileOrder, crTopModNm, crState, crStateInfo
, CompilePhase
, CompilePhaseT(runCompilePhaseT)
, CompileUnit(..)
, CompileUnitState(..)
, CompileRunError(..)
, CompileModName(..)
, CompileRunStateInfo(..)
, CompileParticipation(..)
, FileLocatable(..)
, mkEmptyCompileRun
, crCU, crMbCU
, ppCR
, cpUpdStateInfo, cpUpdSI
, cpUpdCU, cpUpdCUWithKey
, cpMbCU
, cpSetFail, cpSetStop, cpSetStopSeq, cpSetStopAllSeq
, cpSetOk, cpSetErrs, cpSetLimitErrs, cpSetLimitErrsWhen, cpSetInfos, cpSetCompileOrder
, cpSeq, cpSeqWhen
, cpEmpty
, cpFindFileForNameOrFPath
, cpFindFilesForFPathInLocations, cpFindFilesForFPath, cpFindFileForFPath
, cpImportGather, cpImportGatherFromMods, cpImportGatherFromModsWithImp
, cpPP, cpPPMsg
, forgetM
)
where
import Data.Maybe
import System.Exit
import Control.Monad
import Control.Monad.Fix
import Control.Applicative(Applicative(..))
import UHC.Util.Error as ME
import Control.Monad.State
import qualified Control.Exception as CE
import Control.Monad.Identity
import System.IO
import qualified Data.Map as Map
import UHC.Util.Pretty
import UHC.Util.Utils
import UHC.Util.FPath
import UHC.Util.Lens
forgetM :: Monad m => m a -> m ()
forgetM m
= do { _ <- m
; return ()
}
data CompileParticipation
= CompileParticipation_NoImport
deriving (Eq, Ord)
class CompileModName n where
mkCMNm :: String -> n
class CompileUnitState s where
cusDefault :: s
cusUnk :: s
cusIsUnk :: s -> Bool
cusIsImpKnown :: s -> Bool
class CompileUnit u n l s | u -> n l s where
cuDefault :: u
cuFPath :: u -> FPath
cuUpdFPath :: FPath -> u -> u
cuLocation :: u -> l
cuUpdLocation :: l -> u -> u
cuKey :: u -> n
cuUpdKey :: n -> u -> u
cuState :: u -> s
cuUpdState :: s -> u -> u
cuImports :: u -> [n]
cuParticipation :: u -> [CompileParticipation]
cuParticipation _ = []
class CompileRunError e p | e -> p where
crePPErrL :: [e] -> PP_Doc
creMkNotFoundErrL :: p -> String -> [String] -> [FileSuffix] -> [e]
creAreFatal :: [e] -> Bool
crePPErrL _ = empty
creMkNotFoundErrL _ _ _ _ = []
creAreFatal _ = True
class CompileRunStateInfo i n p where
crsiImportPosOfCUKey :: n -> i -> p
class
( CompileModName nm
, CompileUnitState state
, CompileUnit unit nm loc state
, CompileRunError err pos
, CompileRunStateInfo info nm pos
, MonadState (CompileRun nm unit info err) m
, MonadIO m
, Monad m
) => CompileRunner state nm pos loc unit info err m
where
instance CompileRunError String ()
class FileLocatable x loc | loc -> x where
fileLocation :: x -> loc
noFileLocation :: loc
data CompileRunState err
= CRSOk
| CRSFail
| CRSFailMsg String
| CRSStopSeq
| CRSStopAllSeq
| CRSStop
| CRSFailErrL String [err] (Maybe Int)
| CRSErrInfoL String Bool [err]
data CompileRun nm unit info err
= CompileRun
{ _crCUCache :: Map.Map nm unit
, _crCompileOrder :: [[nm]]
, _crTopModNm :: nm
, _crState :: CompileRunState err
, _crStateInfo :: info
}
mkLabel ''CompileRun
instance Error (CompileRunState err) where
noMsg = CRSOk
strMsg = CRSFailMsg
instance Show (CompileRunState err) where
show CRSOk = "CRSOk"
show CRSFail = "CRSFail"
show (CRSFailMsg s) = "CRSFail: " ++ s
show CRSStopSeq = "CRSStopSeq"
show CRSStopAllSeq = "CRSStopAllSeq"
show CRSStop = "CRSStop"
show (CRSFailErrL _ _ _) = "CRSFailErrL"
show (CRSErrInfoL _ _ _) = "CRSErrInfoL"
mkEmptyCompileRun :: n -> i -> CompileRun n u i e
mkEmptyCompileRun nm info
= CompileRun
{ _crCUCache = Map.empty
, _crCompileOrder = []
, _crTopModNm = nm
, _crState = CRSOk
, _crStateInfo = info
}
newtype CompilePhaseT n u i e m a
= CompilePhaseT {runCompilePhaseT :: m a}
type CompilePhase n u i e a = CompilePhaseT n u i e Identity a
instance CompileRunner state n pos loc u i e m => Functor (CompilePhaseT n u i e m) where
fmap = liftM
instance CompileRunner state n pos loc u i e m => Applicative (CompilePhaseT n u i e m) where
pure = return
(<*>) = ap
instance CompileRunner state n pos loc u i e m => Monad (CompilePhaseT n u i e m) where
return x = CompilePhaseT $ return x
cp >>= f = CompilePhaseT $ do
x <- runCompilePhaseT cp
let modf f = do {modify f ; return x}
cr <- get
case _crState cr of
CRSFailErrL about es mbLim
-> do { let (showErrs,omitErrs) = maybe (es,[]) (flip splitAt es) mbLim
; liftIO (unless (null about) (hPutPPLn stderr (pp about)))
; liftIO $ unless (null showErrs) $
do { hPutPPLn stderr (crePPErrL showErrs)
; unless (null omitErrs) $ hPutStrLn stderr "... and more errors"
; hFlush stderr
}
; if creAreFatal es then liftIO exitFailure else modf crSetOk
}
CRSErrInfoL about doPrint is
-> do { if null is
then return x
else liftIO (do { hFlush stdout
; hPutPPLn stderr (about >#< "found errors" >-< e)
; return x
})
; if not (null is) then liftIO exitFailure else return x
}
where e = empty
CRSFailMsg msg
-> do { liftIO $ hPutStrLn stderr msg
; liftIO exitFailure
}
CRSFail
-> do { liftIO exitFailure
}
CRSStop
-> do { liftIO $ exitWith ExitSuccess
}
_ -> return x
cr <- get
case _crState cr of
CRSOk -> runCompilePhaseT (f x)
CRSStopSeq -> do { modf crSetOk ; return $ panic "Monad.CompilePhaseT.CRSStopSeq" }
CRSStopAllSeq -> do { modf crSetStopAllSeq ; return $ panic "Monad.CompilePhaseT.CRSStopAllSeq" }
crs -> return $ panic "Monad.CompilePhaseT._"
instance MonadTrans (CompilePhaseT n u i e) where
lift = CompilePhaseT
instance (CompileRunner state n pos loc u i e m, MonadState s m) => MonadState s (CompilePhaseT n u i e m) where
get = lift get
put = lift . put
instance (CompileRunner state n pos loc u i e m, MonadIO m) => MonadIO (CompilePhaseT n u i e m) where
liftIO = lift . liftIO
instance (CompileRunner state n pos loc u i e m, MonadError e' m) => MonadError e' (CompilePhaseT n u i e m) where
throwError = lift . throwError
catchError m hdl = lift $ catchError (runCompilePhaseT m) (runCompilePhaseT . hdl)
instance (CompileRunner state n pos loc u i e m, MonadFix m) => MonadFix (CompilePhaseT n u i e m) where
mfix f = CompilePhaseT $ mfix $ \ ~a -> runCompilePhaseT (f a)
ppCR :: (PP n,PP u) => CompileRun n u i e -> PP_Doc
ppCR cr
= "CR" >#< show (_crState cr) >|< ":" >#<
( (ppBracketsCommasBlock $ map (\(n,u) -> pp n >#< "->" >#< pp u) $ Map.toList $ _crCUCache cr)
>-< ppBracketsCommas (map ppBracketsCommas $ _crCompileOrder $ cr)
)
crPP :: (PP n,PP u) => String -> CompileRun n u i e -> IO (CompileRun n u i e)
crPP m cr = do { hPutStrLn stderr (m ++ ":") ; hPutPPLn stderr (ppCR cr) ; hFlush stderr ; return cr }
crPPMsg :: (PP m) => m -> CompileRun n u i e -> IO (CompileRun n u i e)
crPPMsg m cr = do { hPutPPLn stdout (pp m) ; return cr }
cpPP :: (PP n, PP u, CompileRunner s n p l u i e m) => String -> CompilePhaseT n u i e m ()
cpPP m
= do { liftIO (hPutStrLn stderr (m ++ ":"))
; cr <- get
; liftIO (hPutPPLn stderr (ppCR cr))
; liftIO (hFlush stderr)
; return ()
}
cpPPMsg :: (PP msg, CompileRunner s n p l u i e m) => msg -> CompilePhaseT n u i e m ()
cpPPMsg m
= do { liftIO (hPutPPLn stdout (pp m))
; return ()
}
crMbCU :: Ord n => n -> CompileRun n u i e -> Maybe u
crMbCU modNm cr = Map.lookup modNm (_crCUCache cr)
crCU :: (Show n,Ord n) => n -> CompileRun n u i e -> u
crCU modNm = panicJust ("crCU: " ++ show modNm) . crMbCU modNm
crSetOk :: CompileRun n u i e -> CompileRun n u i e
crSetOk = crState ^= CRSOk
crSetFail :: CompileRun n u i e -> CompileRun n u i e
crSetFail = crState ^= CRSFail
crSetStop :: CompileRun n u i e -> CompileRun n u i e
crSetStop = crState ^= CRSStop
crSetStopSeq :: CompileRun n u i e -> CompileRun n u i e
crSetStopSeq = crState ^= CRSStopSeq
crSetStopAllSeq :: CompileRun n u i e -> CompileRun n u i e
crSetStopAllSeq = crState ^= CRSStopAllSeq
crSetErrs' :: Maybe Int -> String -> [e] -> CompileRun n u i e -> CompileRun n u i e
crSetErrs' limit about es cr
= case es of
[] -> cr
_ -> cr {_crState = CRSFailErrL about es limit}
crSetInfos' :: String -> Bool -> [e] -> CompileRun n u i e -> CompileRun n u i e
crSetInfos' msg dp is cr
= case is of
[] -> cr
_ -> cr {_crState = CRSErrInfoL msg dp is}
crCUState :: (Ord n,CompileUnit u n l s,CompileUnitState s) => n -> CompileRun n u i e -> s
crCUState modNm cr = maybe cusUnk cuState (crMbCU modNm cr)
crCUFPath :: (Ord n,CompileUnit u n l s) => n -> CompileRun n u i e -> FPath
crCUFPath modNm cr = maybe emptyFPath cuFPath (crMbCU modNm cr)
crCULocation :: (Ord n,FileLocatable u loc) => n -> CompileRun n u i e -> loc
crCULocation modNm cr = maybe noFileLocation fileLocation (crMbCU modNm cr)
cpFindFileForNameOrFPath :: (FPATH n) => String -> n -> FPath -> [(String,FPath)]
cpFindFileForNameOrFPath loc _ fp = searchFPathFromLoc loc fp
cpFindFilesForFPathInLocations
:: ( Ord n
, FPATH n, FileLocatable u loc, Show loc
, CompileRunner s n p loc u i e m
) => (loc -> n -> FPath -> [(loc,FPath,[e])])
-> ((FPath,loc,s,[e]) -> res)
-> Bool
-> [(FileSuffixWith s)]
-> [loc]
-> Maybe n
-> Maybe FPath
-> CompilePhaseT n u i e m [res]
cpFindFilesForFPathInLocations getfp putres stopAtFirst suffs locs mbModNm mbFp
= do { cr <- get
; let cus = maybe cusUnk (flip crCUState cr) mbModNm
; if cusIsUnk cus
then do { let fp = maybe (mkFPath $ panicJust ("cpFindFileForFPath") $ mbModNm) id mbFp
modNm = maybe (mkCMNm $ fpathBase $ fp) id mbModNm
suffs' = map fst suffs
; fpsFound <- liftIO $ searchLocationsForReadableFilesWith (\l f -> getfp l modNm f) stopAtFirst locs suffs fp
; case fpsFound of
[]
-> do { cpSetErrs (creMkNotFoundErrL (crsiImportPosOfCUKey modNm (_crStateInfo cr)) (fpathToStr fp) (map show locs) suffs')
; return []
}
((_,_,_,e@(_:_)):_)
-> do { cpSetErrs e
; return []
}
ffs@((ff,loc,s,_):_)
-> do { cpUpdCU modNm (cuUpdLocation loc . cuUpdFPath ff . cuUpdState cus . cuUpdKey modNm)
; return (map putres ffs)
}
where cus = case lookup (Just $ fpathSuff ff) suffs of
Just c -> c
Nothing -> case lookup (Just "*") suffs of
Just c -> c
Nothing -> cusUnk
}
else return (maybe [] (\nm -> [putres (crCUFPath nm cr, crCULocation nm cr, crCUState nm cr, [])]) mbModNm)
}
cpFindFilesForFPath
:: forall e n u p i s m .
( Ord n
, FPATH n, FileLocatable u String
, CompileRunner s n p String u i e m
) => Bool -> [FileSuffixWith s] -> [String] -> Maybe n -> Maybe FPath -> CompilePhaseT n u i e m [FPath]
cpFindFilesForFPath
= cpFindFilesForFPathInLocations (\l n f -> map (tup12to123 ([]::[e])) $ cpFindFileForNameOrFPath l n f) tup1234to1
cpFindFileForFPath
:: ( Ord n
, FPATH n, FileLocatable u String
, CompileRunner s n p String u i e m
) => [(FileSuffix,s)] -> [String] -> Maybe n -> Maybe FPath -> CompilePhaseT n u i e m (Maybe FPath)
cpFindFileForFPath suffs sp mbModNm mbFp
= do { fps <- cpFindFilesForFPath True suffs sp mbModNm mbFp
; return (listToMaybe fps)
}
cpImportGatherFromModsWithImp
:: (Show n, Ord n, CompileRunner s n p l u i e m)
=> (u -> [n])
-> (Maybe prev -> n -> CompilePhaseT n u i e m (x,Maybe prev))
-> [n]
-> CompilePhaseT n u i e m ()
cpImportGatherFromModsWithImp getImports imp1Mod modNmL
= do { cr <- get
; cpSeq ( [ one Nothing modNm | modNm <- modNmL ]
++ [ cpImportScc ]
)
}
where one prev modNm
= do { (_,new) <- imp1Mod prev modNm
; cr <- get
; if CompileParticipation_NoImport `elem` cuParticipation (crCU modNm cr)
then cpDelCU modNm
else imps new modNm
}
imps prev m
= do { cr <- get
; let impL m = [ i | i <- getImports (crCU m cr), not (cusIsImpKnown (crCUState i cr)) ]
; cpSeq (map (\n -> one prev n) (impL m))
}
cpImportGatherFromMods
:: (Show n, Ord n, CompileRunner s n p l u i e m)
=> (Maybe prev -> n -> CompilePhaseT n u i e m (x,Maybe prev))
-> [n]
-> CompilePhaseT n u i e m ()
cpImportGatherFromMods = cpImportGatherFromModsWithImp cuImports
cpImportGather
:: (Show n,Ord n,CompileRunner s n p l u i e m)
=> (n -> CompilePhaseT n u i e m ()) -> n -> CompilePhaseT n u i e m ()
cpImportGather imp1Mod modNm
= cpImportGatherFromMods
(\_ n -> do { r <- imp1Mod n
; return (r,Nothing)
}
)
[modNm]
crImportDepL :: (CompileUnit u n l s) => CompileRun n u i e -> [(n,[n])]
crImportDepL = map (\cu -> (cuKey cu,cuImports cu)) . Map.elems . _crCUCache
cpImportScc :: (Ord n, CompileRunner s n p l u i e m) => CompilePhaseT n u i e m ()
cpImportScc = modify (\cr -> (cr {_crCompileOrder = scc (crImportDepL cr)}))
cpUpdStateInfo, cpUpdSI :: CompileRunner s n p l u i e m => (i -> i) -> CompilePhaseT n u i e m ()
cpUpdStateInfo upd = crStateInfo =$: upd
cpUpdSI = cpUpdStateInfo
cpUpdCUM :: (Ord n, CompileRunner s n p l u i e m) => n -> (u -> IO u) -> CompilePhaseT n u i e m ()
cpUpdCUM modNm upd
= do { cr <- get
; cu <- liftIO $ maybe (upd cuDefault) upd (crMbCU modNm cr)
; crCUCache =$: Map.insert modNm cu
}
cpUpdCUWithKey :: (Ord n, CompileRunner s n p l u i e m) => n -> (n -> u -> (n,u)) -> CompilePhaseT n u i e m n
cpUpdCUWithKey modNm upd
= do { cr <- get
; let (modNm',cu) = (maybe (upd modNm cuDefault) (upd modNm) (crMbCU modNm cr))
; crCUCache =$: Map.insert modNm' cu . Map.delete modNm
; return modNm'
}
cpUpdCU :: (Ord n, CompileRunner s n p l u i e m) => n -> (u -> u) -> CompilePhaseT n u i e m ()
cpUpdCU modNm upd
= do { cpUpdCUWithKey modNm (\k u -> (k, upd u))
; return ()
}
cpMbCU :: (Ord n,CompileRunner s n p l u i e m) => n -> CompilePhaseT n u i e m (Maybe u)
cpMbCU modNm = liftM (crMbCU modNm) get
cpDelCU :: (Ord n,CompileRunner s n p l u i e m) => n -> CompilePhaseT n u i e m ()
cpDelCU modNm = crCUCache =$: Map.delete modNm
cpSetErrs :: CompileRunner s n p l u i e m => [e] -> CompilePhaseT n u i e m ()
cpSetErrs es
= modify (crSetErrs' Nothing "" es)
cpSetInfos :: CompileRunner s n p l u i e m => String -> Bool -> [e] -> CompilePhaseT n u i e m ()
cpSetInfos msg dp is
= modify (crSetInfos' msg dp is)
cpSetFail :: CompileRunner s n p l u i e m => CompilePhaseT n u i e m ()
cpSetFail
= modify crSetFail
cpSetStop :: CompileRunner s n p l u i e m => CompilePhaseT n u i e m ()
cpSetStop
= modify crSetStop
cpSetStopSeq :: CompileRunner s n p l u i e m => CompilePhaseT n u i e m ()
cpSetStopSeq
= modify crSetStopSeq
cpSetStopAllSeq :: CompileRunner s n p l u i e m => CompilePhaseT n u i e m ()
cpSetStopAllSeq
= modify crSetStopAllSeq
cpSetOk :: CompileRunner s n p l u i e m => CompilePhaseT n u i e m ()
cpSetOk = crState =: CRSOk
cpSetCompileOrder :: CompileRunner s n p l u i e m => [[n]] -> CompilePhaseT n u i e m ()
cpSetCompileOrder nameLL = crCompileOrder =: nameLL
cpSetLimitErrs, cpSetLimitErrsWhen :: CompileRunner s n p l u i e m => Int -> String -> [e] -> CompilePhaseT n u i e m ()
cpSetLimitErrs l a e
= modify (crSetErrs' (Just l) a e)
cpSetLimitErrsWhen l a e
= do { when (not (null e))
(cpSetLimitErrs l a e)
}
cpEmpty :: CompileRunner s n p l u i e m => CompilePhaseT n u i e m ()
cpEmpty = return ()
cpSeq :: CompileRunner s n p l u i e m => [CompilePhaseT n u i e m ()] -> CompilePhaseT n u i e m ()
cpSeq = sequence_
cpSeqWhen :: CompileRunner s n p l u i e m => Bool -> [CompilePhaseT n u i e m ()] -> CompilePhaseT n u i e m ()
cpSeqWhen True as = cpSeq as
cpSeqWhen _ _ = return ()