1 -- -----------------------------------------------------------------------------
3 -- (c) The University of Glasgow, 2005
7 -- -----------------------------------------------------------------------------
12 defaultCleanupHandler,
15 Ghc, GhcT, GhcMonad(..),
16 runGhc, runGhcT, initGhcMonad,
17 gcatch, gbracket, gfinally,
18 clearWarnings, getWarnings, hasWarnings,
19 printExceptionAndWarnings, printWarnings,
20 handleSourceError, defaultCallbacks, GhcApiCallbacks(..),
23 -- * Flags and settings
24 DynFlags(..), DynFlag(..), Severity(..), HscTarget(..), dopt,
25 GhcMode(..), GhcLink(..), defaultObjectTarget,
32 Target(..), TargetId(..), Phase,
39 -- * Extending the program scope
42 extendGlobalTypeScope,
45 -- * Loading\/compiling the program
47 load, loadWithLogger, LoadHowMuch(..),
48 SuccessFlag(..), succeeded, failed,
49 defaultWarnErrLogger, WarnErrLogger,
50 workingDirectoryChanged,
51 parseModule, typecheckModule, desugarModule, loadModule,
52 ParsedModule(..), TypecheckedModule(..), DesugaredModule(..),
53 TypecheckedSource, ParsedSource, RenamedSource, -- ditto
54 TypecheckedMod, ParsedMod,
55 moduleInfo, renamedSource, typecheckedSource,
56 parsedSource, coreModule,
57 compileToCoreModule, compileToCoreSimplified,
61 -- * Inspecting the module structure of the program
62 ModuleGraph, ModSummary(..), ms_mod_name, ModLocation(..),
67 -- * Inspecting modules
74 modInfoIsExportedName,
78 mkPrintUnqualifiedForModule,
80 -- * Querying the environment
84 PrintUnqualified, alwaysQualify,
86 -- * Interactive evaluation
87 getBindings, getPrintUnqual,
91 setContext, getContext,
101 runStmt, SingleStep(..),
103 Resume(resumeStmt, resumeThreadId, resumeBreakInfo, resumeSpan,
104 resumeHistory, resumeHistoryIx),
105 History(historyBreakInfo, historyEnclosingDecl),
106 GHC.getHistorySpan, getHistoryModule,
109 InteractiveEval.back,
110 InteractiveEval.forward,
113 InteractiveEval.compileExpr, HValue, dynCompileExpr,
115 GHC.obtainTermFromId, GHC.obtainTermFromVal, reconstructType,
117 ModBreaks(..), BreakIndex,
118 BreakInfo(breakInfo_number, breakInfo_module),
119 BreakArray, setBreakOn, setBreakOff, getBreak,
122 -- * Abstract syntax elements
128 Module, mkModule, pprModule, moduleName, modulePackageId,
129 ModuleName, mkModuleName, moduleNameString,
133 isExternalName, nameModule, pprParenSymName, nameSrcSpan,
135 RdrName(Qual,Unqual),
139 isImplicitId, isDeadBinder,
140 isExportedId, isLocalId, isGlobalId,
142 isPrimOpId, isFCallId, isClassOpId_maybe,
143 isDataConWorkId, idDataCon,
144 isBottomingId, isDictonaryId,
145 recordSelectorFieldLabel,
147 -- ** Type constructors
149 tyConTyVars, tyConDataCons, tyConArity,
150 isClassTyCon, isSynTyCon, isNewTyCon, isPrimTyCon, isFunTyCon,
152 synTyConDefn, synTyConType, synTyConResKind,
158 -- ** Data constructors
160 dataConSig, dataConType, dataConTyCon, dataConFieldLabels,
161 dataConIsInfix, isVanillaDataCon,
163 StrictnessMark(..), isMarkedStrict,
167 classMethods, classSCTheta, classTvsFds,
172 instanceDFunId, pprInstance, pprInstanceHdr,
174 -- ** Types and Kinds
175 Type, splitForAllTys, funResultTy,
176 pprParendType, pprTypeApp,
179 ThetaType, pprThetaArrow,
185 module HsSyn, -- ToDo: remove extraneous bits
189 defaultFixity, maxPrecedence,
193 -- ** Source locations
195 mkSrcLoc, isGoodSrcLoc, noSrcLoc,
196 srcLocFile, srcLocLine, srcLocCol,
198 mkSrcSpan, srcLocSpan, isGoodSrcSpan, noSrcSpan,
199 srcSpanStart, srcSpanEnd,
201 srcSpanStartLine, srcSpanEndLine,
202 srcSpanStartCol, srcSpanEndCol,
207 -- *** Constructing Located
208 noLoc, mkGeneralLocated,
210 -- *** Deconstructing Located
213 -- *** Combining and comparing Located values
214 eqLocated, cmpLocated, combineLocs, addCLoc,
215 leftmost_smallest, leftmost_largest, rightmost,
219 GhcException(..), showGhcException,
221 -- * Token stream manipulations
223 getTokenStream, getRichTokenStream,
224 showRichTokenStream, addSourceToTokens,
234 * inline bits of HscMain here to simplify layering: hscTcExpr, hscStmt.
235 * what StaticFlags should we expose, if any?
238 #include "HsVersions.h"
241 import qualified Linker
242 import Linker ( HValue )
245 import InteractiveEval
250 import TcRnTypes hiding (LIE)
251 import TcRnMonad ( initIfaceCheck )
255 import qualified HsSyn -- hack as we want to reexport the whole module
256 import HsSyn hiding ((<.>))
258 import TcType hiding( typeKind )
261 import TysPrim ( alphaTyVars )
266 import Name hiding ( varName )
267 -- import OccName ( parenSymOcc )
268 import InstEnv ( Instance, instanceDFunId, pprInstance, pprInstanceHdr,
270 import FamInstEnv ( emptyFamInstEnv )
274 import DriverPipeline
275 import DriverPhases ( Phase(..), isHaskellSrcFilename, startPhase )
281 import StaticFlagParser
282 import qualified StaticFlags
283 import SysTools ( initSysTools, cleanTempFiles, cleanTempFilesExcept,
288 import qualified UniqFM as UFM
292 import Bag ( unitBag, listToBag, emptyBag, isEmptyBag )
296 import StringBuffer ( StringBuffer, hGetStringBuffer, nextChar )
299 import Maybes ( expectJust, mapCatMaybes )
303 import System.Directory ( getModificationTime, doesFileExist,
304 getCurrentDirectory )
307 import qualified Data.List as List
308 import Data.Typeable ( Typeable )
309 import Data.Word ( Word8 )
311 import System.Exit ( exitWith, ExitCode(..) )
312 import System.Time ( ClockTime, getClockTime )
315 import System.FilePath
317 import System.IO.Error ( try, isDoesNotExistError )
318 import Prelude hiding (init)
321 -- -----------------------------------------------------------------------------
322 -- Exception handlers
324 -- | Install some default exception handlers and run the inner computation.
325 -- Unless you want to handle exceptions yourself, you should wrap this around
326 -- the top level of your program. The default handlers output the error
327 -- message(s) to stderr and exit cleanly.
328 defaultErrorHandler :: (ExceptionMonad m, MonadIO m) => DynFlags -> m a -> m a
329 defaultErrorHandler dflags inner =
330 -- top-level exception handler: any unrecognised exception is a compiler bug.
331 ghandle (\exception -> liftIO $ do
333 case fromException exception of
334 -- an IO exception probably isn't our fault, so don't panic
335 Just (ioe :: IOException) ->
336 fatalErrorMsg dflags (text (show ioe))
337 _ -> case fromException exception of
338 Just StackOverflow ->
339 fatalErrorMsg dflags (text "stack overflow: use +RTS -K<size> to increase it")
340 _ -> case fromException exception of
341 Just (ex :: ExitCode) -> throw ex
344 (text (show (Panic (show exception))))
345 exitWith (ExitFailure 1)
348 -- error messages propagated as exceptions
353 PhaseFailed _ code -> exitWith code
354 Interrupted -> exitWith (ExitFailure 1)
355 Signal _ -> exitWith (ExitFailure 1)
356 _ -> do fatalErrorMsg dflags (text (show ge))
357 exitWith (ExitFailure 1)
361 -- | Install a default cleanup handler to remove temporary files deposited by
362 -- a GHC run. This is seperate from 'defaultErrorHandler', because you might
363 -- want to override the error handling, but still get the ordinary cleanup
365 defaultCleanupHandler :: (ExceptionMonad m, MonadIO m) =>
366 DynFlags -> m a -> m a
367 defaultCleanupHandler dflags inner =
368 -- make sure we clean up after ourselves
371 cleanTempFiles dflags
374 -- exceptions will be blocked while we clean the temporary files,
375 -- so there shouldn't be any difficulty if we receive further
378 -- | Print the error message and all warnings. Useful inside exception
379 -- handlers. Clears warnings after printing.
380 printExceptionAndWarnings :: GhcMonad m => SourceError -> m ()
381 printExceptionAndWarnings err = do
382 let errs = srcErrorMessages err
384 dflags <- getSessionDynFlags
386 -- Empty errors means we failed due to -Werror. (Since this function
387 -- takes a source error as argument, we know for sure _some_ error
388 -- did indeed happen.)
390 printBagOfWarnings dflags warns
391 printBagOfErrors dflags (unitBag warnIsErrorMsg)
392 else liftIO $ printBagOfErrors dflags errs
395 -- | Print all accumulated warnings using 'log_action'.
396 printWarnings :: GhcMonad m => m ()
398 dflags <- getSessionDynFlags
400 liftIO $ printBagOfWarnings dflags warns
403 -- | Run function for the 'Ghc' monad.
405 -- It initialises the GHC session and warnings via 'initGhcMonad'. Each call
406 -- to this function will create a new session which should not be shared among
409 -- Any errors not handled inside the 'Ghc' action are propagated as IO
412 runGhc :: Maybe FilePath -- ^ See argument to 'initGhcMonad'.
413 -> Ghc a -- ^ The action to perform.
415 runGhc mb_top_dir ghc = do
416 wref <- newIORef emptyBag
417 ref <- newIORef undefined
418 let session = Session ref wref
419 flip unGhc session $ do
420 initGhcMonad mb_top_dir
422 -- XXX: unregister interrupt handlers here?
424 -- | Run function for 'GhcT' monad transformer.
426 -- It initialises the GHC session and warnings via 'initGhcMonad'. Each call
427 -- to this function will create a new session which should not be shared among
430 runGhcT :: (ExceptionMonad m, Functor m, MonadIO m) =>
431 Maybe FilePath -- ^ See argument to 'initGhcMonad'.
432 -> GhcT m a -- ^ The action to perform.
434 runGhcT mb_top_dir ghct = do
435 wref <- liftIO $ newIORef emptyBag
436 ref <- liftIO $ newIORef undefined
437 let session = Session ref wref
438 flip unGhcT session $ do
439 initGhcMonad mb_top_dir
442 -- | Initialise a GHC session.
444 -- If you implement a custom 'GhcMonad' you must call this function in the
445 -- monad run function. It will initialise the session variable and clear all
448 -- The first argument should point to the directory where GHC's library files
449 -- reside. More precisely, this should be the output of @ghc --print-libdir@
450 -- of the version of GHC the module using this API is compiled with. For
451 -- portability, you should use the @ghc-paths@ package, available at
452 -- <http://hackage.haskell.org/cgi-bin/hackage-scripts/package/ghc-paths>.
454 initGhcMonad :: GhcMonad m => Maybe FilePath -> m ()
455 initGhcMonad mb_top_dir = do
457 liftIO $ installSignalHandlers
459 liftIO $ StaticFlags.initStaticOpts
461 dflags0 <- liftIO $ initDynFlags defaultDynFlags
462 dflags <- liftIO $ initSysTools mb_top_dir dflags0
463 env <- liftIO $ newHscEnv defaultCallbacks dflags
467 defaultCallbacks :: GhcApiCallbacks
470 reportModuleCompilationResult =
471 \_ mb_err -> defaultWarnErrLogger mb_err
474 -- -----------------------------------------------------------------------------
477 -- | Grabs the DynFlags from the Session
478 getSessionDynFlags :: GhcMonad m => m DynFlags
479 getSessionDynFlags = withSession (return . hsc_dflags)
481 -- | Updates the DynFlags in a Session. This also reads
482 -- the package database (unless it has already been read),
483 -- and prepares the compilers knowledge about packages. It
484 -- can be called again to load new packages: just add new
485 -- package flags to (packageFlags dflags).
487 -- Returns a list of new packages that may need to be linked in using
488 -- the dynamic linker (see 'linkPackages') as a result of new package
489 -- flags. If you are not doing linking or doing static linking, you
490 -- can ignore the list of packages returned.
492 setSessionDynFlags :: GhcMonad m => DynFlags -> m [PackageId]
493 setSessionDynFlags dflags = do
494 (dflags', preload) <- liftIO $ initPackages dflags
495 modifySession (\h -> h{ hsc_dflags = dflags' })
498 -- | If there is no -o option, guess the name of target executable
499 -- by using top-level source file name as a base.
500 guessOutputFile :: GhcMonad m => m ()
501 guessOutputFile = modifySession $ \env ->
502 let dflags = hsc_dflags env
503 mod_graph = hsc_mod_graph env
504 mainModuleSrcPath :: Maybe String
505 mainModuleSrcPath = do
506 let isMain = (== mainModIs dflags) . ms_mod
507 [ms] <- return (filter isMain mod_graph)
508 ml_hs_file (ms_location ms)
509 name = fmap dropExtension mainModuleSrcPath
511 #if defined(mingw32_HOST_OS)
512 -- we must add the .exe extention unconditionally here, otherwise
513 -- when name has an extension of its own, the .exe extension will
514 -- not be added by DriverPipeline.exeFileName. See #2248
515 name_exe = fmap (<.> "exe") name
520 case outputFile dflags of
522 Nothing -> env { hsc_dflags = dflags { outputFile = name_exe } }
524 -- -----------------------------------------------------------------------------
527 -- ToDo: think about relative vs. absolute file paths. And what
528 -- happens when the current directory changes.
530 -- | Sets the targets for this session. Each target may be a module name
531 -- or a filename. The targets correspond to the set of root modules for
532 -- the program\/library. Unloading the current program is achieved by
533 -- setting the current set of targets to be empty, followed by 'load'.
534 setTargets :: GhcMonad m => [Target] -> m ()
535 setTargets targets = modifySession (\h -> h{ hsc_targets = targets })
537 -- | Returns the current set of targets
538 getTargets :: GhcMonad m => m [Target]
539 getTargets = withSession (return . hsc_targets)
541 -- | Add another target.
542 addTarget :: GhcMonad m => Target -> m ()
544 = modifySession (\h -> h{ hsc_targets = target : hsc_targets h })
547 removeTarget :: GhcMonad m => TargetId -> m ()
548 removeTarget target_id
549 = modifySession (\h -> h{ hsc_targets = filter (hsc_targets h) })
551 filter targets = [ t | t@(Target id _ _) <- targets, id /= target_id ]
553 -- | Attempts to guess what Target a string refers to. This function
554 -- implements the @--make@/GHCi command-line syntax for filenames:
556 -- - if the string looks like a Haskell source filename, then interpret it
559 -- - if adding a .hs or .lhs suffix yields the name of an existing file,
562 -- - otherwise interpret the string as a module name
564 guessTarget :: GhcMonad m => String -> Maybe Phase -> m Target
565 guessTarget str (Just phase)
566 = return (Target (TargetFile str (Just phase)) True Nothing)
567 guessTarget str Nothing
568 | isHaskellSrcFilename file
569 = return (target (TargetFile file Nothing))
571 = do exists <- liftIO $ doesFileExist hs_file
573 then return (target (TargetFile hs_file Nothing))
575 exists <- liftIO $ doesFileExist lhs_file
577 then return (target (TargetFile lhs_file Nothing))
579 if looksLikeModuleName file
580 then return (target (TargetModule (mkModuleName file)))
583 (ProgramError (showSDoc $
584 text "target" <+> quotes (text file) <+>
585 text "is not a module name or a source file"))
588 | '*':rest <- str = (rest, False)
589 | otherwise = (str, True)
591 hs_file = file <.> "hs"
592 lhs_file = file <.> "lhs"
594 target tid = Target tid obj_allowed Nothing
596 -- -----------------------------------------------------------------------------
597 -- Extending the program scope
599 extendGlobalRdrScope :: GhcMonad m => [GlobalRdrElt] -> m ()
600 extendGlobalRdrScope rdrElts
601 = modifySession $ \hscEnv ->
602 let global_rdr = hsc_global_rdr_env hscEnv
603 in hscEnv{ hsc_global_rdr_env = foldl extendGlobalRdrEnv global_rdr rdrElts }
605 setGlobalRdrScope :: GhcMonad m => [GlobalRdrElt] -> m ()
606 setGlobalRdrScope rdrElts
607 = modifySession $ \hscEnv ->
608 hscEnv{ hsc_global_rdr_env = foldl extendGlobalRdrEnv emptyGlobalRdrEnv rdrElts }
610 extendGlobalTypeScope :: GhcMonad m => [Id] -> m ()
611 extendGlobalTypeScope ids
612 = modifySession $ \hscEnv ->
613 let global_type = hsc_global_type_env hscEnv
614 in hscEnv{ hsc_global_type_env = extendTypeEnvWithIds global_type ids }
616 setGlobalTypeScope :: GhcMonad m => [Id] -> m ()
617 setGlobalTypeScope ids
618 = modifySession $ \hscEnv ->
619 hscEnv{ hsc_global_type_env = extendTypeEnvWithIds emptyTypeEnv ids }
621 -- -----------------------------------------------------------------------------
622 -- Loading the program
624 -- | Perform a dependency analysis starting from the current targets
625 -- and update the session with the new module graph.
627 -- Dependency analysis entails parsing the @import@ directives and may
628 -- therefore require running certain preprocessors.
630 -- Note that each 'ModSummary' in the module graph caches its 'DynFlags'.
631 -- These 'DynFlags' are determined by the /current/ session 'DynFlags' and the
632 -- @OPTIONS@ and @LANGUAGE@ pragmas of the parsed module. Thus if you want to
633 -- changes to the 'DynFlags' to take effect you need to call this function
636 depanal :: GhcMonad m =>
637 [ModuleName] -- ^ excluded modules
638 -> Bool -- ^ allow duplicate roots
640 depanal excluded_mods allow_dup_roots = do
641 hsc_env <- getSession
643 dflags = hsc_dflags hsc_env
644 targets = hsc_targets hsc_env
645 old_graph = hsc_mod_graph hsc_env
647 liftIO $ showPass dflags "Chasing dependencies"
648 liftIO $ debugTraceMsg dflags 2 (hcat [
649 text "Chasing modules from: ",
650 hcat (punctuate comma (map pprTarget targets))])
652 mod_graph <- downsweep hsc_env old_graph excluded_mods allow_dup_roots
653 modifySession $ \_ -> hsc_env { hsc_mod_graph = mod_graph }
656 -- | Describes which modules of the module graph need to be loaded.
659 -- ^ Load all targets and its dependencies.
660 | LoadUpTo ModuleName
661 -- ^ Load only the given module and its dependencies.
662 | LoadDependenciesOf ModuleName
663 -- ^ Load only the dependencies of the given module, but not the module
666 -- | Try to load the program. See 'LoadHowMuch' for the different modes.
668 -- This function implements the core of GHC's @--make@ mode. It preprocesses,
669 -- compiles and loads the specified modules, avoiding re-compilation wherever
670 -- possible. Depending on the target (see 'DynFlags.hscTarget') compilating
671 -- and loading may result in files being created on disk.
673 -- Calls the 'reportModuleCompilationResult' callback after each compiling
674 -- each module, whether successful or not.
676 -- Throw a 'SourceError' if errors are encountered before the actual
677 -- compilation starts (e.g., during dependency analysis). All other errors
678 -- are reported using the callback.
680 load :: GhcMonad m => LoadHowMuch -> m SuccessFlag
682 mod_graph <- depanal [] False
683 load2 how_much mod_graph
685 -- | A function called to log warnings and errors.
686 type WarnErrLogger = GhcMonad m => Maybe SourceError -> m ()
688 defaultWarnErrLogger :: WarnErrLogger
689 defaultWarnErrLogger Nothing = printWarnings
690 defaultWarnErrLogger (Just e) = printExceptionAndWarnings e
692 -- | Try to load the program. If a Module is supplied, then just
693 -- attempt to load up to this target. If no Module is supplied,
694 -- then try to load all targets.
696 -- The first argument is a function that is called after compiling each
697 -- module to print wanrings and errors.
699 -- While compiling a module, all 'SourceError's are caught and passed to the
700 -- logger, however, this function may still throw a 'SourceError' if
701 -- dependency analysis failed (e.g., due to a parse error).
703 loadWithLogger :: GhcMonad m => WarnErrLogger -> LoadHowMuch -> m SuccessFlag
704 loadWithLogger logger how_much = do
705 -- Dependency analysis first. Note that this fixes the module graph:
706 -- even if we don't get a fully successful upsweep, the full module
707 -- graph is still retained in the Session. We can tell which modules
708 -- were successfully loaded by inspecting the Session's HPT.
709 withLocalCallbacks (\cbs -> cbs { reportModuleCompilationResult =
713 load2 :: GhcMonad m => LoadHowMuch -> [ModSummary]
715 load2 how_much mod_graph = do
717 hsc_env <- getSession
719 let hpt1 = hsc_HPT hsc_env
720 let dflags = hsc_dflags hsc_env
722 -- The "bad" boot modules are the ones for which we have
723 -- B.hs-boot in the module graph, but no B.hs
724 -- The downsweep should have ensured this does not happen
726 let all_home_mods = [ms_mod_name s
727 | s <- mod_graph, not (isBootSummary s)]
728 bad_boot_mods = [s | s <- mod_graph, isBootSummary s,
729 not (ms_mod_name s `elem` all_home_mods)]
730 ASSERT( null bad_boot_mods ) return ()
732 -- check that the module given in HowMuch actually exists, otherwise
733 -- topSortModuleGraph will bomb later.
734 let checkHowMuch (LoadUpTo m) = checkMod m
735 checkHowMuch (LoadDependenciesOf m) = checkMod m
739 | m `elem` all_home_mods = and_then
741 liftIO $ errorMsg dflags (text "no such module:" <+>
745 checkHowMuch how_much $ do
747 -- mg2_with_srcimps drops the hi-boot nodes, returning a
748 -- graph with cycles. Among other things, it is used for
749 -- backing out partially complete cycles following a failed
750 -- upsweep, and for removing from hpt all the modules
751 -- not in strict downwards closure, during calls to compile.
752 let mg2_with_srcimps :: [SCC ModSummary]
753 mg2_with_srcimps = topSortModuleGraph True mod_graph Nothing
755 -- If we can determine that any of the {-# SOURCE #-} imports
756 -- are definitely unnecessary, then emit a warning.
757 warnUnnecessarySourceImports mg2_with_srcimps
760 -- check the stability property for each module.
761 stable_mods@(stable_obj,stable_bco)
762 = checkStability hpt1 mg2_with_srcimps all_home_mods
764 -- prune bits of the HPT which are definitely redundant now,
766 pruned_hpt = pruneHomePackageTable hpt1
767 (flattenSCCs mg2_with_srcimps)
770 _ <- liftIO $ evaluate pruned_hpt
772 -- before we unload anything, make sure we don't leave an old
773 -- interactive context around pointing to dead bindings. Also,
774 -- write the pruned HPT to allow the old HPT to be GC'd.
775 modifySession $ \_ -> hsc_env{ hsc_IC = emptyInteractiveContext,
776 hsc_HPT = pruned_hpt }
778 liftIO $ debugTraceMsg dflags 2 (text "Stable obj:" <+> ppr stable_obj $$
779 text "Stable BCO:" <+> ppr stable_bco)
781 -- Unload any modules which are going to be re-linked this time around.
782 let stable_linkables = [ linkable
783 | m <- stable_obj++stable_bco,
784 Just hmi <- [lookupUFM pruned_hpt m],
785 Just linkable <- [hm_linkable hmi] ]
786 liftIO $ unload hsc_env stable_linkables
788 -- We could at this point detect cycles which aren't broken by
789 -- a source-import, and complain immediately, but it seems better
790 -- to let upsweep_mods do this, so at least some useful work gets
791 -- done before the upsweep is abandoned.
792 --hPutStrLn stderr "after tsort:\n"
793 --hPutStrLn stderr (showSDoc (vcat (map ppr mg2)))
795 -- Now do the upsweep, calling compile for each module in
796 -- turn. Final result is version 3 of everything.
798 -- Topologically sort the module graph, this time including hi-boot
799 -- nodes, and possibly just including the portion of the graph
800 -- reachable from the module specified in the 2nd argument to load.
801 -- This graph should be cycle-free.
802 -- If we're restricting the upsweep to a portion of the graph, we
803 -- also want to retain everything that is still stable.
804 let full_mg :: [SCC ModSummary]
805 full_mg = topSortModuleGraph False mod_graph Nothing
807 maybe_top_mod = case how_much of
809 LoadDependenciesOf m -> Just m
812 partial_mg0 :: [SCC ModSummary]
813 partial_mg0 = topSortModuleGraph False mod_graph maybe_top_mod
815 -- LoadDependenciesOf m: we want the upsweep to stop just
816 -- short of the specified module (unless the specified module
819 | LoadDependenciesOf _mod <- how_much
820 = ASSERT( case last partial_mg0 of
821 AcyclicSCC ms -> ms_mod_name ms == _mod; _ -> False )
822 List.init partial_mg0
828 | AcyclicSCC ms <- full_mg,
829 ms_mod_name ms `elem` stable_obj++stable_bco,
830 ms_mod_name ms `notElem` [ ms_mod_name ms' |
831 AcyclicSCC ms' <- partial_mg ] ]
833 mg = stable_mg ++ partial_mg
835 -- clean up between compilations
836 let cleanup = cleanTempFilesExcept dflags
837 (ppFilesFromSummaries (flattenSCCs mg2_with_srcimps))
839 liftIO $ debugTraceMsg dflags 2 (hang (text "Ready for upsweep")
841 (upsweep_ok, hsc_env1, modsUpswept)
842 <- upsweep (hsc_env { hsc_HPT = emptyHomePackageTable })
843 pruned_hpt stable_mods cleanup mg
845 -- Make modsDone be the summaries for each home module now
846 -- available; this should equal the domain of hpt3.
847 -- Get in in a roughly top .. bottom order (hence reverse).
849 let modsDone = reverse modsUpswept
851 -- Try and do linking in some form, depending on whether the
852 -- upsweep was completely or only partially successful.
854 if succeeded upsweep_ok
857 -- Easy; just relink it all.
858 do liftIO $ debugTraceMsg dflags 2 (text "Upsweep completely successful.")
860 -- Clean up after ourselves
861 liftIO $ cleanTempFilesExcept dflags (ppFilesFromSummaries modsDone)
863 -- Issue a warning for the confusing case where the user
864 -- said '-o foo' but we're not going to do any linking.
865 -- We attempt linking if either (a) one of the modules is
866 -- called Main, or (b) the user said -no-hs-main, indicating
867 -- that main() is going to come from somewhere else.
869 let ofile = outputFile dflags
870 let no_hs_main = dopt Opt_NoHsMain dflags
872 main_mod = mainModIs dflags
873 a_root_is_Main = any ((==main_mod).ms_mod) mod_graph
874 do_linking = a_root_is_Main || no_hs_main || ghcLink dflags == LinkDynLib
876 when (ghcLink dflags == LinkBinary
877 && isJust ofile && not do_linking) $
878 liftIO $ debugTraceMsg dflags 1 $
879 text ("Warning: output was redirected with -o, " ++
880 "but no output will be generated\n" ++
881 "because there is no " ++
882 moduleNameString (moduleName main_mod) ++ " module.")
884 -- link everything together
885 linkresult <- liftIO $ link (ghcLink dflags) dflags do_linking (hsc_HPT hsc_env1)
887 loadFinish Succeeded linkresult hsc_env1
890 -- Tricky. We need to back out the effects of compiling any
891 -- half-done cycles, both so as to clean up the top level envs
892 -- and to avoid telling the interactive linker to link them.
893 do liftIO $ debugTraceMsg dflags 2 (text "Upsweep partially successful.")
896 = map ms_mod modsDone
897 let mods_to_zap_names
898 = findPartiallyCompletedCycles modsDone_names
901 = filter ((`notElem` mods_to_zap_names).ms_mod)
904 let hpt4 = retainInTopLevelEnvs (map ms_mod_name mods_to_keep)
907 -- Clean up after ourselves
908 liftIO $ cleanTempFilesExcept dflags (ppFilesFromSummaries mods_to_keep)
910 -- there should be no Nothings where linkables should be, now
911 ASSERT(all (isJust.hm_linkable)
912 (eltsUFM (hsc_HPT hsc_env))) do
914 -- Link everything together
915 linkresult <- liftIO $ link (ghcLink dflags) dflags False hpt4
917 let hsc_env4 = hsc_env1{ hsc_HPT = hpt4 }
918 loadFinish Failed linkresult hsc_env4
920 -- Finish up after a load.
922 -- If the link failed, unload everything and return.
923 loadFinish :: GhcMonad m =>
924 SuccessFlag -> SuccessFlag -> HscEnv
926 loadFinish _all_ok Failed hsc_env
927 = do liftIO $ unload hsc_env []
928 modifySession $ \_ -> discardProg hsc_env
931 -- Empty the interactive context and set the module context to the topmost
932 -- newly loaded module, or the Prelude if none were loaded.
933 loadFinish all_ok Succeeded hsc_env
934 = do modifySession $ \_ -> hsc_env{ hsc_IC = emptyInteractiveContext }
938 -- Forget the current program, but retain the persistent info in HscEnv
939 discardProg :: HscEnv -> HscEnv
941 = hsc_env { hsc_mod_graph = emptyMG,
942 hsc_IC = emptyInteractiveContext,
943 hsc_HPT = emptyHomePackageTable }
945 -- used to fish out the preprocess output files for the purposes of
946 -- cleaning up. The preprocessed file *might* be the same as the
947 -- source file, but that doesn't do any harm.
948 ppFilesFromSummaries :: [ModSummary] -> [FilePath]
949 ppFilesFromSummaries summaries = map ms_hspp_file summaries
951 -- -----------------------------------------------------------------------------
953 class ParsedMod m where
954 modSummary :: m -> ModSummary
955 parsedSource :: m -> ParsedSource
957 class ParsedMod m => TypecheckedMod m where
958 renamedSource :: m -> Maybe RenamedSource
959 typecheckedSource :: m -> TypecheckedSource
960 moduleInfo :: m -> ModuleInfo
961 tm_internals :: m -> (TcGblEnv, ModDetails)
962 -- ToDo: improvements that could be made here:
963 -- if the module succeeded renaming but not typechecking,
964 -- we can still get back the GlobalRdrEnv and exports, so
965 -- perhaps the ModuleInfo should be split up into separate
968 class TypecheckedMod m => DesugaredMod m where
969 coreModule :: m -> ModGuts
971 -- | The result of successful parsing.
973 ParsedModule { pm_mod_summary :: ModSummary
974 , pm_parsed_source :: ParsedSource }
976 instance ParsedMod ParsedModule where
977 modSummary m = pm_mod_summary m
978 parsedSource m = pm_parsed_source m
980 -- | The result of successful typechecking. It also contains the parser
982 data TypecheckedModule =
983 TypecheckedModule { tm_parsed_module :: ParsedModule
984 , tm_renamed_source :: Maybe RenamedSource
985 , tm_typechecked_source :: TypecheckedSource
986 , tm_checked_module_info :: ModuleInfo
987 , tm_internals_ :: (TcGblEnv, ModDetails)
990 instance ParsedMod TypecheckedModule where
991 modSummary m = modSummary (tm_parsed_module m)
992 parsedSource m = parsedSource (tm_parsed_module m)
994 instance TypecheckedMod TypecheckedModule where
995 renamedSource m = tm_renamed_source m
996 typecheckedSource m = tm_typechecked_source m
997 moduleInfo m = tm_checked_module_info m
998 tm_internals m = tm_internals_ m
1000 -- | The result of successful desugaring (i.e., translation to core). Also
1001 -- contains all the information of a typechecked module.
1002 data DesugaredModule =
1003 DesugaredModule { dm_typechecked_module :: TypecheckedModule
1004 , dm_core_module :: ModGuts
1007 instance ParsedMod DesugaredModule where
1008 modSummary m = modSummary (dm_typechecked_module m)
1009 parsedSource m = parsedSource (dm_typechecked_module m)
1011 instance TypecheckedMod DesugaredModule where
1012 renamedSource m = renamedSource (dm_typechecked_module m)
1013 typecheckedSource m = typecheckedSource (dm_typechecked_module m)
1014 moduleInfo m = moduleInfo (dm_typechecked_module m)
1015 tm_internals m = tm_internals_ (dm_typechecked_module m)
1017 instance DesugaredMod DesugaredModule where
1018 coreModule m = dm_core_module m
1020 type ParsedSource = Located (HsModule RdrName)
1021 type RenamedSource = (HsGroup Name, [LImportDecl Name], Maybe [LIE Name],
1023 type TypecheckedSource = LHsBinds Id
1026 -- - things that aren't in the output of the typechecker right now:
1027 -- - the export list
1029 -- - type signatures
1030 -- - type/data/newtype declarations
1031 -- - class declarations
1033 -- - extra things in the typechecker's output:
1034 -- - default methods are turned into top-level decls.
1035 -- - dictionary bindings
1037 -- | Return the 'ModSummary' of a module with the given name.
1039 -- The module must be part of the module graph (see 'hsc_mod_graph' and
1040 -- 'ModuleGraph'). If this is not the case, this function will throw a
1043 -- This function ignores boot modules and requires that there is only one
1044 -- non-boot module with the given name.
1045 getModSummary :: GhcMonad m => ModuleName -> m ModSummary
1046 getModSummary mod = do
1047 mg <- liftM hsc_mod_graph getSession
1048 case [ ms | ms <- mg, ms_mod_name ms == mod, not (isBootSummary ms) ] of
1049 [] -> throw $ mkApiErr (text "Module not part of module graph")
1051 multiple -> throw $ mkApiErr (text "getModSummary is ambiguous: " <+> ppr multiple)
1053 -- | Parse a module.
1055 -- Throws a 'SourceError' on parse error.
1056 parseModule :: GhcMonad m => ModSummary -> m ParsedModule
1058 rdr_module <- withTempSession
1059 (\e -> e { hsc_dflags = ms_hspp_opts ms }) $
1061 return (ParsedModule ms rdr_module)
1063 -- | Typecheck and rename a parsed module.
1065 -- Throws a 'SourceError' if either fails.
1066 typecheckModule :: GhcMonad m => ParsedModule -> m TypecheckedModule
1067 typecheckModule pmod = do
1068 let ms = modSummary pmod
1069 withTempSession (\e -> e { hsc_dflags = ms_hspp_opts ms }) $ do
1070 (tc_gbl_env, rn_info)
1071 <- hscTypecheckRename ms (parsedSource pmod)
1072 details <- makeSimpleDetails tc_gbl_env
1075 tm_internals_ = (tc_gbl_env, details),
1076 tm_parsed_module = pmod,
1077 tm_renamed_source = rn_info,
1078 tm_typechecked_source = tcg_binds tc_gbl_env,
1079 tm_checked_module_info =
1081 minf_type_env = md_types details,
1082 minf_exports = availsToNameSet $ md_exports details,
1083 minf_rdr_env = Just (tcg_rdr_env tc_gbl_env),
1084 minf_instances = md_insts details
1086 ,minf_modBreaks = emptyModBreaks
1090 -- | Desugar a typechecked module.
1091 desugarModule :: GhcMonad m => TypecheckedModule -> m DesugaredModule
1092 desugarModule tcm = do
1093 let ms = modSummary tcm
1094 withTempSession (\e -> e { hsc_dflags = ms_hspp_opts ms }) $ do
1095 let (tcg, _) = tm_internals tcm
1096 guts <- hscDesugar ms tcg
1099 dm_typechecked_module = tcm,
1100 dm_core_module = guts
1103 -- | Load a module. Input doesn't need to be desugared.
1105 -- A module must be loaded before dependent modules can be typechecked. This
1106 -- always includes generating a 'ModIface' and, depending on the
1107 -- 'DynFlags.hscTarget', may also include code generation.
1109 -- This function will always cause recompilation and will always overwrite
1110 -- previous compilation results (potentially files on disk).
1112 loadModule :: (TypecheckedMod mod, GhcMonad m) => mod -> m mod
1114 let ms = modSummary tcm
1115 let mod = ms_mod_name ms
1116 let loc = ms_location ms
1117 let (tcg, _details) = tm_internals tcm
1119 withTempSession (\e -> e { hsc_dflags = ms_hspp_opts ms }) $ do
1121 let compilerBackend comp env ms' _ _mb_old_iface _ =
1122 withTempSession (\_ -> env) $
1123 hscBackend comp tcg ms' Nothing
1125 hsc_env <- getSession
1128 case ms_obj_date ms of
1129 Just t | t > ms_hs_date ms -> do
1130 l <- liftIO $ findObjectLinkable (ms_mod ms)
1133 _otherwise -> return Nothing
1135 compile' (compilerBackend hscNothingCompiler
1136 ,compilerBackend hscInteractiveCompiler
1137 ,hscCheckRecompBackend hscBatchCompiler tcg)
1138 hsc_env ms 1 1 Nothing mb_linkable
1139 -- compile' shouldn't change the environment
1140 return $ addToUFM (hsc_HPT hsc_env) mod mod_info
1141 modifySession $ \e -> e{ hsc_HPT = hpt_new }
1145 -- | This is the way to get access to the Core bindings corresponding
1146 -- to a module. 'compileToCore' parses, typechecks, and
1147 -- desugars the module, then returns the resulting Core module (consisting of
1148 -- the module name, type declarations, and function declarations) if
1150 compileToCoreModule :: GhcMonad m => FilePath -> m CoreModule
1151 compileToCoreModule = compileCore False
1153 -- | Like compileToCoreModule, but invokes the simplifier, so
1154 -- as to return simplified and tidied Core.
1155 compileToCoreSimplified :: GhcMonad m => FilePath -> m CoreModule
1156 compileToCoreSimplified = compileCore True
1158 -- | Provided for backwards-compatibility: compileToCore returns just the Core
1159 -- bindings, but for most purposes, you probably want to call
1160 -- compileToCoreModule.
1161 compileToCore :: GhcMonad m => FilePath -> m [CoreBind]
1162 compileToCore fn = do
1163 mod <- compileToCoreModule session fn
1164 return $ cm_binds mod
1166 -- | Takes a CoreModule and compiles the bindings therein
1167 -- to object code. The first argument is a bool flag indicating
1168 -- whether to run the simplifier.
1169 -- The resulting .o, .hi, and executable files, if any, are stored in the
1170 -- current directory, and named according to the module name.
1171 -- This has only so far been tested with a single self-contained module.
1172 compileCoreToObj :: GhcMonad m => Bool -> CoreModule -> m ()
1173 compileCoreToObj simplify cm@(CoreModule{ cm_module = mName }) = do
1174 dflags <- getSessionDynFlags
1175 currentTime <- liftIO $ getClockTime
1176 cwd <- liftIO $ getCurrentDirectory
1177 modLocation <- liftIO $ mkHiOnlyModLocation dflags (hiSuf dflags) cwd
1178 ((moduleNameSlashes . moduleName) mName)
1180 let modSummary = ModSummary { ms_mod = mName,
1181 ms_hsc_src = ExtCoreFile,
1182 ms_location = modLocation,
1183 -- By setting the object file timestamp to Nothing,
1184 -- we always force recompilation, which is what we
1185 -- want. (Thus it doesn't matter what the timestamp
1186 -- for the (nonexistent) source file is.)
1187 ms_hs_date = currentTime,
1188 ms_obj_date = Nothing,
1189 -- Only handling the single-module case for now, so no imports.
1194 ms_hspp_opts = dflags,
1195 ms_hspp_buf = Nothing
1198 let maybe_simplify mod_guts | simplify = hscSimplify mod_guts
1199 | otherwise = return mod_guts
1200 guts <- maybe_simplify (mkModGuts cm)
1201 (iface, changed, _details, cgguts)
1202 <- hscNormalIface guts Nothing
1203 hscWriteIface iface changed modSummary
1204 _ <- hscGenHardCode cgguts modSummary
1207 -- Makes a "vanilla" ModGuts.
1208 mkModGuts :: CoreModule -> ModGuts
1209 mkModGuts coreModule = ModGuts {
1210 mg_module = cm_module coreModule,
1213 mg_deps = noDependencies,
1214 mg_dir_imps = emptyModuleEnv,
1215 mg_used_names = emptyNameSet,
1216 mg_rdr_env = emptyGlobalRdrEnv,
1217 mg_fix_env = emptyFixityEnv,
1218 mg_types = emptyTypeEnv,
1222 mg_binds = cm_binds coreModule,
1223 mg_foreign = NoStubs,
1224 mg_warns = NoWarnings,
1226 mg_hpc_info = emptyHpcInfo False,
1227 mg_modBreaks = emptyModBreaks,
1228 mg_vect_info = noVectInfo,
1229 mg_inst_env = emptyInstEnv,
1230 mg_fam_inst_env = emptyFamInstEnv
1233 compileCore :: GhcMonad m => Bool -> FilePath -> m CoreModule
1234 compileCore simplify fn = do
1235 -- First, set the target to the desired filename
1236 target <- guessTarget fn Nothing
1238 _ <- load LoadAllTargets
1239 -- Then find dependencies
1240 modGraph <- depanal [] True
1241 case find ((== fn) . msHsFilePath) modGraph of
1242 Just modSummary -> do
1243 -- Now we have the module name;
1244 -- parse, typecheck and desugar the module
1245 mod_guts <- coreModule `fmap`
1246 -- TODO: space leaky: call hsc* directly?
1247 (desugarModule =<< typecheckModule =<< parseModule modSummary)
1248 liftM gutsToCoreModule $
1251 -- If simplify is true: simplify (hscSimplify), then tidy
1253 hsc_env <- getSession
1254 simpl_guts <- hscSimplify mod_guts
1255 tidy_guts <- liftIO $ tidyProgram hsc_env simpl_guts
1256 return $ Left tidy_guts
1258 return $ Right mod_guts
1260 Nothing -> panic "compileToCoreModule: target FilePath not found in\
1261 module dependency graph"
1262 where -- two versions, based on whether we simplify (thus run tidyProgram,
1263 -- which returns a (CgGuts, ModDetails) pair, or not (in which case
1264 -- we just have a ModGuts.
1265 gutsToCoreModule :: Either (CgGuts, ModDetails) ModGuts -> CoreModule
1266 gutsToCoreModule (Left (cg, md)) = CoreModule {
1267 cm_module = cg_module cg, cm_types = md_types md,
1268 cm_imports = cg_dir_imps cg, cm_binds = cg_binds cg
1270 gutsToCoreModule (Right mg) = CoreModule {
1271 cm_module = mg_module mg, cm_types = mg_types mg,
1272 cm_imports = moduleEnvKeys (mg_dir_imps mg), cm_binds = mg_binds mg
1275 -- ---------------------------------------------------------------------------
1278 unload :: HscEnv -> [Linkable] -> IO ()
1279 unload hsc_env stable_linkables -- Unload everthing *except* 'stable_linkables'
1280 = case ghcLink (hsc_dflags hsc_env) of
1282 LinkInMemory -> Linker.unload (hsc_dflags hsc_env) stable_linkables
1284 LinkInMemory -> panic "unload: no interpreter"
1285 -- urgh. avoid warnings:
1286 hsc_env stable_linkables
1290 -- -----------------------------------------------------------------------------
1294 Stability tells us which modules definitely do not need to be recompiled.
1295 There are two main reasons for having stability:
1297 - avoid doing a complete upsweep of the module graph in GHCi when
1298 modules near the bottom of the tree have not changed.
1300 - to tell GHCi when it can load object code: we can only load object code
1301 for a module when we also load object code fo all of the imports of the
1302 module. So we need to know that we will definitely not be recompiling
1303 any of these modules, and we can use the object code.
1305 The stability check is as follows. Both stableObject and
1306 stableBCO are used during the upsweep phase later.
1309 stable m = stableObject m || stableBCO m
1312 all stableObject (imports m)
1313 && old linkable does not exist, or is == on-disk .o
1314 && date(on-disk .o) > date(.hs)
1317 all stable (imports m)
1318 && date(BCO) > date(.hs)
1321 These properties embody the following ideas:
1323 - if a module is stable, then:
1325 - if it has been compiled in a previous pass (present in HPT)
1326 then it does not need to be compiled or re-linked.
1328 - if it has not been compiled in a previous pass,
1329 then we only need to read its .hi file from disk and
1330 link it to produce a 'ModDetails'.
1332 - if a modules is not stable, we will definitely be at least
1333 re-linking, and possibly re-compiling it during the 'upsweep'.
1334 All non-stable modules can (and should) therefore be unlinked
1335 before the 'upsweep'.
1337 - Note that objects are only considered stable if they only depend
1338 on other objects. We can't link object code against byte code.
1342 :: HomePackageTable -- HPT from last compilation
1343 -> [SCC ModSummary] -- current module graph (cyclic)
1344 -> [ModuleName] -- all home modules
1345 -> ([ModuleName], -- stableObject
1346 [ModuleName]) -- stableBCO
1348 checkStability hpt sccs all_home_mods = foldl checkSCC ([],[]) sccs
1350 checkSCC (stable_obj, stable_bco) scc0
1351 | stableObjects = (scc_mods ++ stable_obj, stable_bco)
1352 | stableBCOs = (stable_obj, scc_mods ++ stable_bco)
1353 | otherwise = (stable_obj, stable_bco)
1355 scc = flattenSCC scc0
1356 scc_mods = map ms_mod_name scc
1357 home_module m = m `elem` all_home_mods && m `notElem` scc_mods
1359 scc_allimps = nub (filter home_module (concatMap ms_home_allimps scc))
1360 -- all imports outside the current SCC, but in the home pkg
1362 stable_obj_imps = map (`elem` stable_obj) scc_allimps
1363 stable_bco_imps = map (`elem` stable_bco) scc_allimps
1367 && all object_ok scc
1370 and (zipWith (||) stable_obj_imps stable_bco_imps)
1374 | Just t <- ms_obj_date ms = t >= ms_hs_date ms
1378 same_as_prev t = case lookupUFM hpt (ms_mod_name ms) of
1379 Just hmi | Just l <- hm_linkable hmi
1380 -> isObjectLinkable l && t == linkableTime l
1382 -- why '>=' rather than '>' above? If the filesystem stores
1383 -- times to the nearset second, we may occasionally find that
1384 -- the object & source have the same modification time,
1385 -- especially if the source was automatically generated
1386 -- and compiled. Using >= is slightly unsafe, but it matches
1387 -- make's behaviour.
1390 = case lookupUFM hpt (ms_mod_name ms) of
1391 Just hmi | Just l <- hm_linkable hmi ->
1392 not (isObjectLinkable l) &&
1393 linkableTime l >= ms_hs_date ms
1396 -- -----------------------------------------------------------------------------
1398 -- | Prune the HomePackageTable
1400 -- Before doing an upsweep, we can throw away:
1402 -- - For non-stable modules:
1403 -- - all ModDetails, all linked code
1404 -- - all unlinked code that is out of date with respect to
1407 -- This is VERY IMPORTANT otherwise we'll end up requiring 2x the
1408 -- space at the end of the upsweep, because the topmost ModDetails of the
1409 -- old HPT holds on to the entire type environment from the previous
1412 pruneHomePackageTable
1415 -> ([ModuleName],[ModuleName])
1418 pruneHomePackageTable hpt summ (stable_obj, stable_bco)
1421 | is_stable modl = hmi'
1422 | otherwise = hmi'{ hm_details = emptyModDetails }
1424 modl = moduleName (mi_module (hm_iface hmi))
1425 hmi' | Just l <- hm_linkable hmi, linkableTime l < ms_hs_date ms
1426 = hmi{ hm_linkable = Nothing }
1429 where ms = expectJust "prune" (lookupUFM ms_map modl)
1431 ms_map = listToUFM [(ms_mod_name ms, ms) | ms <- summ]
1433 is_stable m = m `elem` stable_obj || m `elem` stable_bco
1435 -- -----------------------------------------------------------------------------
1437 -- Return (names of) all those in modsDone who are part of a cycle
1438 -- as defined by theGraph.
1439 findPartiallyCompletedCycles :: [Module] -> [SCC ModSummary] -> [Module]
1440 findPartiallyCompletedCycles modsDone theGraph
1444 chew ((AcyclicSCC _):rest) = chew rest -- acyclic? not interesting.
1445 chew ((CyclicSCC vs):rest)
1446 = let names_in_this_cycle = nub (map ms_mod vs)
1448 = nub ([done | done <- modsDone,
1449 done `elem` names_in_this_cycle])
1450 chewed_rest = chew rest
1452 if notNull mods_in_this_cycle
1453 && length mods_in_this_cycle < length names_in_this_cycle
1454 then mods_in_this_cycle ++ chewed_rest
1457 -- -----------------------------------------------------------------------------
1461 -- This is where we compile each module in the module graph, in a pass
1462 -- from the bottom to the top of the graph.
1464 -- There better had not be any cyclic groups here -- we check for them.
1468 HscEnv -- ^ Includes initially-empty HPT
1469 -> HomePackageTable -- ^ HPT from last time round (pruned)
1470 -> ([ModuleName],[ModuleName]) -- ^ stable modules (see checkStability)
1471 -> IO () -- ^ How to clean up unwanted tmp files
1472 -> [SCC ModSummary] -- ^ Mods to do (the worklist)
1478 -- 1. A flag whether the complete upsweep was successful.
1479 -- 2. The 'HscEnv' with an updated HPT
1480 -- 3. A list of modules which succeeded loading.
1482 upsweep hsc_env old_hpt stable_mods cleanup sccs = do
1483 (res, hsc_env, done) <- upsweep' hsc_env old_hpt [] sccs 1 (length sccs)
1484 return (res, hsc_env, reverse done)
1487 upsweep' hsc_env _old_hpt done
1489 = return (Succeeded, hsc_env, done)
1491 upsweep' hsc_env _old_hpt done
1492 (CyclicSCC ms:_) _ _
1493 = do liftIO $ fatalErrorMsg (hsc_dflags hsc_env) (cyclicModuleErr ms)
1494 return (Failed, hsc_env, done)
1496 upsweep' hsc_env old_hpt done
1497 (AcyclicSCC mod:mods) mod_index nmods
1498 = do -- putStrLn ("UPSWEEP_MOD: hpt = " ++
1499 -- show (map (moduleUserString.moduleName.mi_module.hm_iface)
1500 -- (moduleEnvElts (hsc_HPT hsc_env)))
1501 let logger = reportModuleCompilationResult (hsc_callbacks hsc_env)
1504 <- handleSourceError
1505 (\err -> do logger mod (Just err); return Nothing) $ do
1506 mod_info <- upsweep_mod hsc_env old_hpt stable_mods
1508 logger mod Nothing -- log warnings
1509 return (Just mod_info)
1511 liftIO cleanup -- Remove unwanted tmp files between compilations
1514 Nothing -> return (Failed, hsc_env, done)
1516 let this_mod = ms_mod_name mod
1518 -- Add new info to hsc_env
1519 hpt1 = addToUFM (hsc_HPT hsc_env) this_mod mod_info
1520 hsc_env1 = hsc_env { hsc_HPT = hpt1 }
1522 -- Space-saving: delete the old HPT entry
1523 -- for mod BUT if mod is a hs-boot
1524 -- node, don't delete it. For the
1525 -- interface, the HPT entry is probaby for the
1526 -- main Haskell source file. Deleting it
1527 -- would force the real module to be recompiled
1529 old_hpt1 | isBootSummary mod = old_hpt
1530 | otherwise = delFromUFM old_hpt this_mod
1534 -- fixup our HomePackageTable after we've finished compiling
1535 -- a mutually-recursive loop. See reTypecheckLoop, below.
1536 hsc_env2 <- liftIO $ reTypecheckLoop hsc_env1 mod done'
1538 upsweep' hsc_env2 old_hpt1 done' mods (mod_index+1) nmods
1540 -- | Compile a single module. Always produce a Linkable for it if
1541 -- successful. If no compilation happened, return the old Linkable.
1542 upsweep_mod :: GhcMonad m =>
1545 -> ([ModuleName],[ModuleName])
1547 -> Int -- index of module
1548 -> Int -- total number of modules
1551 upsweep_mod hsc_env old_hpt (stable_obj, stable_bco) summary mod_index nmods
1553 this_mod_name = ms_mod_name summary
1554 this_mod = ms_mod summary
1555 mb_obj_date = ms_obj_date summary
1556 obj_fn = ml_obj_file (ms_location summary)
1557 hs_date = ms_hs_date summary
1559 is_stable_obj = this_mod_name `elem` stable_obj
1560 is_stable_bco = this_mod_name `elem` stable_bco
1562 old_hmi = lookupUFM old_hpt this_mod_name
1564 -- We're using the dflags for this module now, obtained by
1565 -- applying any options in its LANGUAGE & OPTIONS_GHC pragmas.
1566 dflags = ms_hspp_opts summary
1567 prevailing_target = hscTarget (hsc_dflags hsc_env)
1568 local_target = hscTarget dflags
1570 -- If OPTIONS_GHC contains -fasm or -fvia-C, be careful that
1571 -- we don't do anything dodgy: these should only work to change
1572 -- from -fvia-C to -fasm and vice-versa, otherwise we could
1573 -- end up trying to link object code to byte code.
1574 target = if prevailing_target /= local_target
1575 && (not (isObjectTarget prevailing_target)
1576 || not (isObjectTarget local_target))
1577 then prevailing_target
1580 -- store the corrected hscTarget into the summary
1581 summary' = summary{ ms_hspp_opts = dflags { hscTarget = target } }
1583 -- The old interface is ok if
1584 -- a) we're compiling a source file, and the old HPT
1585 -- entry is for a source file
1586 -- b) we're compiling a hs-boot file
1587 -- Case (b) allows an hs-boot file to get the interface of its
1588 -- real source file on the second iteration of the compilation
1589 -- manager, but that does no harm. Otherwise the hs-boot file
1590 -- will always be recompiled
1595 Just hm_info | isBootSummary summary -> Just iface
1596 | not (mi_boot iface) -> Just iface
1597 | otherwise -> Nothing
1599 iface = hm_iface hm_info
1601 compile_it :: GhcMonad m => Maybe Linkable -> m HomeModInfo
1602 compile_it = compile hsc_env summary' mod_index nmods mb_old_iface
1604 compile_it_discard_iface :: GhcMonad m =>
1605 Maybe Linkable -> m HomeModInfo
1606 compile_it_discard_iface
1607 = compile hsc_env summary' mod_index nmods Nothing
1609 -- With the HscNothing target we create empty linkables to avoid
1610 -- recompilation. We have to detect these to recompile anyway if
1611 -- the target changed since the last compile.
1613 | Just hmi <- old_hmi, Just l <- hm_linkable hmi =
1614 null (linkableUnlinked l)
1616 -- we have no linkable, so it cannot be fake
1619 implies False _ = True
1625 -- Regardless of whether we're generating object code or
1626 -- byte code, we can always use an existing object file
1627 -- if it is *stable* (see checkStability).
1628 | is_stable_obj, Just hmi <- old_hmi -> do
1629 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1630 (text "skipping stable obj mod:" <+> ppr this_mod_name)
1632 -- object is stable, and we have an entry in the
1633 -- old HPT: nothing to do
1635 | is_stable_obj, isNothing old_hmi -> do
1636 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1637 (text "compiling stable on-disk mod:" <+> ppr this_mod_name)
1638 linkable <- liftIO $ findObjectLinkable this_mod obj_fn
1639 (expectJust "upsweep1" mb_obj_date)
1640 compile_it (Just linkable)
1641 -- object is stable, but we need to load the interface
1642 -- off disk to make a HMI.
1644 | not (isObjectTarget target), is_stable_bco,
1645 (target /= HscNothing) `implies` not is_fake_linkable ->
1646 ASSERT(isJust old_hmi) -- must be in the old_hpt
1647 let Just hmi = old_hmi in do
1648 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1649 (text "skipping stable BCO mod:" <+> ppr this_mod_name)
1651 -- BCO is stable: nothing to do
1653 | not (isObjectTarget target),
1654 Just hmi <- old_hmi,
1655 Just l <- hm_linkable hmi,
1656 not (isObjectLinkable l),
1657 (target /= HscNothing) `implies` not is_fake_linkable,
1658 linkableTime l >= ms_hs_date summary -> do
1659 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1660 (text "compiling non-stable BCO mod:" <+> ppr this_mod_name)
1662 -- we have an old BCO that is up to date with respect
1663 -- to the source: do a recompilation check as normal.
1665 -- When generating object code, if there's an up-to-date
1666 -- object file on the disk, then we can use it.
1667 -- However, if the object file is new (compared to any
1668 -- linkable we had from a previous compilation), then we
1669 -- must discard any in-memory interface, because this
1670 -- means the user has compiled the source file
1671 -- separately and generated a new interface, that we must
1672 -- read from the disk.
1674 | isObjectTarget target,
1675 Just obj_date <- mb_obj_date,
1676 obj_date >= hs_date -> do
1679 | Just l <- hm_linkable hmi,
1680 isObjectLinkable l && linkableTime l == obj_date -> do
1681 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1682 (text "compiling mod with new on-disk obj:" <+> ppr this_mod_name)
1685 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1686 (text "compiling mod with new on-disk obj2:" <+> ppr this_mod_name)
1687 linkable <- liftIO $ findObjectLinkable this_mod obj_fn obj_date
1688 compile_it_discard_iface (Just linkable)
1691 liftIO $ debugTraceMsg (hsc_dflags hsc_env) 5
1692 (text "compiling mod:" <+> ppr this_mod_name)
1697 -- Filter modules in the HPT
1698 retainInTopLevelEnvs :: [ModuleName] -> HomePackageTable -> HomePackageTable
1699 retainInTopLevelEnvs keep_these hpt
1700 = listToUFM [ (mod, expectJust "retain" mb_mod_info)
1702 , let mb_mod_info = lookupUFM hpt mod
1703 , isJust mb_mod_info ]
1705 -- ---------------------------------------------------------------------------
1706 -- Typecheck module loops
1709 See bug #930. This code fixes a long-standing bug in --make. The
1710 problem is that when compiling the modules *inside* a loop, a data
1711 type that is only defined at the top of the loop looks opaque; but
1712 after the loop is done, the structure of the data type becomes
1715 The difficulty is then that two different bits of code have
1716 different notions of what the data type looks like.
1718 The idea is that after we compile a module which also has an .hs-boot
1719 file, we re-generate the ModDetails for each of the modules that
1720 depends on the .hs-boot file, so that everyone points to the proper
1721 TyCons, Ids etc. defined by the real module, not the boot module.
1722 Fortunately re-generating a ModDetails from a ModIface is easy: the
1723 function TcIface.typecheckIface does exactly that.
1725 Picking the modules to re-typecheck is slightly tricky. Starting from
1726 the module graph consisting of the modules that have already been
1727 compiled, we reverse the edges (so they point from the imported module
1728 to the importing module), and depth-first-search from the .hs-boot
1729 node. This gives us all the modules that depend transitively on the
1730 .hs-boot module, and those are exactly the modules that we need to
1733 Following this fix, GHC can compile itself with --make -O2.
1736 reTypecheckLoop :: HscEnv -> ModSummary -> ModuleGraph -> IO HscEnv
1737 reTypecheckLoop hsc_env ms graph
1738 | not (isBootSummary ms) &&
1739 any (\m -> ms_mod m == this_mod && isBootSummary m) graph
1741 let mss = reachableBackwards (ms_mod_name ms) graph
1742 non_boot = filter (not.isBootSummary) mss
1743 debugTraceMsg (hsc_dflags hsc_env) 2 $
1744 text "Re-typechecking loop: " <> ppr (map ms_mod_name non_boot)
1745 typecheckLoop hsc_env (map ms_mod_name non_boot)
1749 this_mod = ms_mod ms
1751 typecheckLoop :: HscEnv -> [ModuleName] -> IO HscEnv
1752 typecheckLoop hsc_env mods = do
1754 fixIO $ \new_hpt -> do
1755 let new_hsc_env = hsc_env{ hsc_HPT = new_hpt }
1756 mds <- initIfaceCheck new_hsc_env $
1757 mapM (typecheckIface . hm_iface) hmis
1758 let new_hpt = addListToUFM old_hpt
1759 (zip mods [ hmi{ hm_details = details }
1760 | (hmi,details) <- zip hmis mds ])
1762 return hsc_env{ hsc_HPT = new_hpt }
1764 old_hpt = hsc_HPT hsc_env
1765 hmis = map (expectJust "typecheckLoop" . lookupUFM old_hpt) mods
1767 reachableBackwards :: ModuleName -> [ModSummary] -> [ModSummary]
1768 reachableBackwards mod summaries
1769 = [ ms | (ms,_,_) <- reachableG (transposeG graph) root ]
1770 where -- the rest just sets up the graph:
1771 (graph, lookup_node) = moduleGraphNodes False summaries
1772 root = expectJust "reachableBackwards" (lookup_node HsBootFile mod)
1774 -- ---------------------------------------------------------------------------
1775 -- Topological sort of the module graph
1777 type SummaryNode = (ModSummary, Int, [Int])
1781 -- ^ Drop hi-boot nodes? (see below)
1784 -- ^ Root module name. If @Nothing@, use the full graph.
1786 -- ^ Calculate SCCs of the module graph, possibly dropping the hi-boot nodes
1787 -- The resulting list of strongly-connected-components is in topologically
1788 -- sorted order, starting with the module(s) at the bottom of the
1789 -- dependency graph (ie compile them first) and ending with the ones at
1792 -- Drop hi-boot nodes (first boolean arg)?
1794 -- - @False@: treat the hi-boot summaries as nodes of the graph,
1795 -- so the graph must be acyclic
1797 -- - @True@: eliminate the hi-boot nodes, and instead pretend
1798 -- the a source-import of Foo is an import of Foo
1799 -- The resulting graph has no hi-boot nodes, but can be cyclic
1801 topSortModuleGraph drop_hs_boot_nodes summaries mb_root_mod
1802 = map (fmap summaryNodeSummary) $ stronglyConnCompG initial_graph
1804 (graph, lookup_node) = moduleGraphNodes drop_hs_boot_nodes summaries
1806 initial_graph = case mb_root_mod of
1809 -- restrict the graph to just those modules reachable from
1810 -- the specified module. We do this by building a graph with
1811 -- the full set of nodes, and determining the reachable set from
1812 -- the specified node.
1813 let root | Just node <- lookup_node HsSrcFile root_mod, graph `hasVertexG` node = node
1814 | otherwise = ghcError (ProgramError "module does not exist")
1815 in graphFromEdgedVertices (seq root (reachableG graph root))
1817 summaryNodeKey :: SummaryNode -> Int
1818 summaryNodeKey (_, k, _) = k
1820 summaryNodeSummary :: SummaryNode -> ModSummary
1821 summaryNodeSummary (s, _, _) = s
1823 moduleGraphNodes :: Bool -> [ModSummary]
1824 -> (Graph SummaryNode, HscSource -> ModuleName -> Maybe SummaryNode)
1825 moduleGraphNodes drop_hs_boot_nodes summaries = (graphFromEdgedVertices nodes, lookup_node)
1827 numbered_summaries = zip summaries [1..]
1829 lookup_node :: HscSource -> ModuleName -> Maybe SummaryNode
1830 lookup_node hs_src mod = lookupFM node_map (mod, hs_src)
1832 lookup_key :: HscSource -> ModuleName -> Maybe Int
1833 lookup_key hs_src mod = fmap summaryNodeKey (lookup_node hs_src mod)
1835 node_map :: NodeMap SummaryNode
1836 node_map = listToFM [ ((moduleName (ms_mod s), ms_hsc_src s), node)
1837 | node@(s, _, _) <- nodes ]
1839 -- We use integers as the keys for the SCC algorithm
1840 nodes :: [SummaryNode]
1841 nodes = [ (s, key, out_keys)
1842 | (s, key) <- numbered_summaries
1843 -- Drop the hi-boot ones if told to do so
1844 , not (isBootSummary s && drop_hs_boot_nodes)
1845 , let out_keys = out_edge_keys hs_boot_key (map unLoc (ms_home_srcimps s)) ++
1846 out_edge_keys HsSrcFile (map unLoc (ms_home_imps s)) ++
1847 (-- see [boot-edges] below
1848 if drop_hs_boot_nodes || ms_hsc_src s == HsBootFile
1850 else case lookup_key HsBootFile (ms_mod_name s) of
1854 -- [boot-edges] if this is a .hs and there is an equivalent
1855 -- .hs-boot, add a link from the former to the latter. This
1856 -- has the effect of detecting bogus cases where the .hs-boot
1857 -- depends on the .hs, by introducing a cycle. Additionally,
1858 -- it ensures that we will always process the .hs-boot before
1859 -- the .hs, and so the HomePackageTable will always have the
1860 -- most up to date information.
1862 -- Drop hs-boot nodes by using HsSrcFile as the key
1863 hs_boot_key | drop_hs_boot_nodes = HsSrcFile
1864 | otherwise = HsBootFile
1866 out_edge_keys :: HscSource -> [ModuleName] -> [Int]
1867 out_edge_keys hi_boot ms = mapCatMaybes (lookup_key hi_boot) ms
1868 -- If we want keep_hi_boot_nodes, then we do lookup_key with
1869 -- the IsBootInterface parameter True; else False
1872 type NodeKey = (ModuleName, HscSource) -- The nodes of the graph are
1873 type NodeMap a = FiniteMap NodeKey a -- keyed by (mod, src_file_type) pairs
1875 msKey :: ModSummary -> NodeKey
1876 msKey (ModSummary { ms_mod = mod, ms_hsc_src = boot }) = (moduleName mod,boot)
1878 mkNodeMap :: [ModSummary] -> NodeMap ModSummary
1879 mkNodeMap summaries = listToFM [ (msKey s, s) | s <- summaries]
1881 nodeMapElts :: NodeMap a -> [a]
1882 nodeMapElts = eltsFM
1884 -- | If there are {-# SOURCE #-} imports between strongly connected
1885 -- components in the topological sort, then those imports can
1886 -- definitely be replaced by ordinary non-SOURCE imports: if SOURCE
1887 -- were necessary, then the edge would be part of a cycle.
1888 warnUnnecessarySourceImports :: GhcMonad m => [SCC ModSummary] -> m ()
1889 warnUnnecessarySourceImports sccs =
1890 logWarnings (listToBag (concatMap (check.flattenSCC) sccs))
1892 let mods_in_this_cycle = map ms_mod_name ms in
1893 [ warn i | m <- ms, i <- ms_home_srcimps m,
1894 unLoc i `notElem` mods_in_this_cycle ]
1896 warn :: Located ModuleName -> WarnMsg
1899 (ptext (sLit "Warning: {-# SOURCE #-} unnecessary in import of ")
1900 <+> quotes (ppr mod))
1902 -----------------------------------------------------------------------------
1903 -- Downsweep (dependency analysis)
1905 -- Chase downwards from the specified root set, returning summaries
1906 -- for all home modules encountered. Only follow source-import
1909 -- We pass in the previous collection of summaries, which is used as a
1910 -- cache to avoid recalculating a module summary if the source is
1913 -- The returned list of [ModSummary] nodes has one node for each home-package
1914 -- module, plus one for any hs-boot files. The imports of these nodes
1915 -- are all there, including the imports of non-home-package modules.
1917 downsweep :: GhcMonad m =>
1919 -> [ModSummary] -- Old summaries
1920 -> [ModuleName] -- Ignore dependencies on these; treat
1921 -- them as if they were package modules
1922 -> Bool -- True <=> allow multiple targets to have
1923 -- the same module name; this is
1924 -- very useful for ghc -M
1926 -- The elts of [ModSummary] all have distinct
1927 -- (Modules, IsBoot) identifiers, unless the Bool is true
1928 -- in which case there can be repeats
1929 downsweep hsc_env old_summaries excl_mods allow_dup_roots
1930 = do -- catch error messages and return them
1931 --handleErrMsg -- should be covered by GhcMonad now
1932 -- (\err_msg -> printBagOfErrors (hsc_dflags hsc_env) (unitBag err_msg) >> return Nothing) $ do
1933 rootSummaries <- mapM getRootSummary roots
1934 let root_map = mkRootMap rootSummaries
1935 checkDuplicates root_map
1936 summs <- loop (concatMap msDeps rootSummaries) root_map
1939 roots = hsc_targets hsc_env
1941 old_summary_map :: NodeMap ModSummary
1942 old_summary_map = mkNodeMap old_summaries
1944 getRootSummary :: GhcMonad m => Target -> m ModSummary
1945 getRootSummary (Target (TargetFile file mb_phase) obj_allowed maybe_buf)
1946 = do exists <- liftIO $ doesFileExist file
1948 then summariseFile hsc_env old_summaries file mb_phase
1949 obj_allowed maybe_buf
1950 else throwOneError $ mkPlainErrMsg noSrcSpan $
1951 text "can't find file:" <+> text file
1952 getRootSummary (Target (TargetModule modl) obj_allowed maybe_buf)
1953 = do maybe_summary <- summariseModule hsc_env old_summary_map False
1954 (L rootLoc modl) obj_allowed
1956 case maybe_summary of
1957 Nothing -> packageModErr modl
1960 rootLoc = mkGeneralSrcSpan (fsLit "<command line>")
1962 -- In a root module, the filename is allowed to diverge from the module
1963 -- name, so we have to check that there aren't multiple root files
1964 -- defining the same module (otherwise the duplicates will be silently
1965 -- ignored, leading to confusing behaviour).
1966 checkDuplicates :: GhcMonad m => NodeMap [ModSummary] -> m ()
1967 checkDuplicates root_map
1968 | allow_dup_roots = return ()
1969 | null dup_roots = return ()
1970 | otherwise = liftIO $ multiRootsErr (head dup_roots)
1972 dup_roots :: [[ModSummary]] -- Each at least of length 2
1973 dup_roots = filterOut isSingleton (nodeMapElts root_map)
1975 loop :: GhcMonad m =>
1976 [(Located ModuleName,IsBootInterface)]
1977 -- Work list: process these modules
1978 -> NodeMap [ModSummary]
1979 -- Visited set; the range is a list because
1980 -- the roots can have the same module names
1981 -- if allow_dup_roots is True
1983 -- The result includes the worklist, except
1984 -- for those mentioned in the visited set
1985 loop [] done = return (concat (nodeMapElts done))
1986 loop ((wanted_mod, is_boot) : ss) done
1987 | Just summs <- lookupFM done key
1988 = if isSingleton summs then
1991 do { liftIO $ multiRootsErr summs; return [] }
1993 = do mb_s <- summariseModule hsc_env old_summary_map
1994 is_boot wanted_mod True
1997 Nothing -> loop ss done
1998 Just s -> loop (msDeps s ++ ss) (addToFM done key [s])
2000 key = (unLoc wanted_mod, if is_boot then HsBootFile else HsSrcFile)
2002 mkRootMap :: [ModSummary] -> NodeMap [ModSummary]
2003 mkRootMap summaries = addListToFM_C (++) emptyFM
2004 [ (msKey s, [s]) | s <- summaries ]
2006 msDeps :: ModSummary -> [(Located ModuleName, IsBootInterface)]
2007 -- (msDeps s) returns the dependencies of the ModSummary s.
2008 -- A wrinkle is that for a {-# SOURCE #-} import we return
2009 -- *both* the hs-boot file
2010 -- *and* the source file
2011 -- as "dependencies". That ensures that the list of all relevant
2012 -- modules always contains B.hs if it contains B.hs-boot.
2013 -- Remember, this pass isn't doing the topological sort. It's
2014 -- just gathering the list of all relevant ModSummaries
2016 concat [ [(m,True), (m,False)] | m <- ms_home_srcimps s ]
2017 ++ [ (m,False) | m <- ms_home_imps s ]
2019 home_imps :: [Located (ImportDecl RdrName)] -> [Located ModuleName]
2020 home_imps imps = [ ideclName i | L _ i <- imps, isNothing (ideclPkgQual i) ]
2022 ms_home_allimps :: ModSummary -> [ModuleName]
2023 ms_home_allimps ms = map unLoc (ms_home_srcimps ms ++ ms_home_imps ms)
2025 ms_home_srcimps :: ModSummary -> [Located ModuleName]
2026 ms_home_srcimps = home_imps . ms_srcimps
2028 ms_home_imps :: ModSummary -> [Located ModuleName]
2029 ms_home_imps = home_imps . ms_imps
2031 -----------------------------------------------------------------------------
2032 -- Summarising modules
2034 -- We have two types of summarisation:
2036 -- * Summarise a file. This is used for the root module(s) passed to
2037 -- cmLoadModules. The file is read, and used to determine the root
2038 -- module name. The module name may differ from the filename.
2040 -- * Summarise a module. We are given a module name, and must provide
2041 -- a summary. The finder is used to locate the file in which the module
2047 -> [ModSummary] -- old summaries
2048 -> FilePath -- source file name
2049 -> Maybe Phase -- start phase
2050 -> Bool -- object code allowed?
2051 -> Maybe (StringBuffer,ClockTime)
2054 summariseFile hsc_env old_summaries file mb_phase obj_allowed maybe_buf
2055 -- we can use a cached summary if one is available and the
2056 -- source file hasn't changed, But we have to look up the summary
2057 -- by source file, rather than module name as we do in summarise.
2058 | Just old_summary <- findSummaryBySourceFile old_summaries file
2060 let location = ms_location old_summary
2062 -- return the cached summary if the source didn't change
2063 src_timestamp <- case maybe_buf of
2064 Just (_,t) -> return t
2065 Nothing -> liftIO $ getModificationTime file
2066 -- The file exists; we checked in getRootSummary above.
2067 -- If it gets removed subsequently, then this
2068 -- getModificationTime may fail, but that's the right
2071 if ms_hs_date old_summary == src_timestamp
2072 then do -- update the object-file timestamp
2074 if isObjectTarget (hscTarget (hsc_dflags hsc_env))
2075 || obj_allowed -- bug #1205
2076 then liftIO $ getObjTimestamp location False
2078 return old_summary{ ms_obj_date = obj_timestamp }
2086 let dflags = hsc_dflags hsc_env
2088 (dflags', hspp_fn, buf)
2089 <- preprocessFile hsc_env file mb_phase maybe_buf
2091 (srcimps,the_imps, L _ mod_name) <- getImports dflags' buf hspp_fn file
2093 -- Make a ModLocation for this file
2094 location <- liftIO $ mkHomeModLocation dflags mod_name file
2096 -- Tell the Finder cache where it is, so that subsequent calls
2097 -- to findModule will find it, even if it's not on any search path
2098 mod <- liftIO $ addHomeModuleToFinder hsc_env mod_name location
2100 src_timestamp <- case maybe_buf of
2101 Just (_,t) -> return t
2102 Nothing -> liftIO $ getModificationTime file
2103 -- getMofificationTime may fail
2105 -- when the user asks to load a source file by name, we only
2106 -- use an object file if -fobject-code is on. See #1205.
2108 if isObjectTarget (hscTarget (hsc_dflags hsc_env))
2109 || obj_allowed -- bug #1205
2110 then liftIO $ modificationTimeIfExists (ml_obj_file location)
2113 return (ModSummary { ms_mod = mod, ms_hsc_src = HsSrcFile,
2114 ms_location = location,
2115 ms_hspp_file = hspp_fn,
2116 ms_hspp_opts = dflags',
2117 ms_hspp_buf = Just buf,
2118 ms_srcimps = srcimps, ms_imps = the_imps,
2119 ms_hs_date = src_timestamp,
2120 ms_obj_date = obj_timestamp })
2122 findSummaryBySourceFile :: [ModSummary] -> FilePath -> Maybe ModSummary
2123 findSummaryBySourceFile summaries file
2124 = case [ ms | ms <- summaries, HsSrcFile <- [ms_hsc_src ms],
2125 expectJust "findSummaryBySourceFile" (ml_hs_file (ms_location ms)) == file ] of
2129 -- Summarise a module, and pick up source and timestamp.
2133 -> NodeMap ModSummary -- Map of old summaries
2134 -> IsBootInterface -- True <=> a {-# SOURCE #-} import
2135 -> Located ModuleName -- Imported module to be summarised
2136 -> Bool -- object code allowed?
2137 -> Maybe (StringBuffer, ClockTime)
2138 -> [ModuleName] -- Modules to exclude
2139 -> m (Maybe ModSummary) -- Its new summary
2141 summariseModule hsc_env old_summary_map is_boot (L loc wanted_mod)
2142 obj_allowed maybe_buf excl_mods
2143 | wanted_mod `elem` excl_mods
2146 | Just old_summary <- lookupFM old_summary_map (wanted_mod, hsc_src)
2147 = do -- Find its new timestamp; all the
2148 -- ModSummaries in the old map have valid ml_hs_files
2149 let location = ms_location old_summary
2150 src_fn = expectJust "summariseModule" (ml_hs_file location)
2152 -- check the modification time on the source file, and
2153 -- return the cached summary if it hasn't changed. If the
2154 -- file has disappeared, we need to call the Finder again.
2156 Just (_,t) -> check_timestamp old_summary location src_fn t
2158 m <- liftIO $ System.IO.Error.try (getModificationTime src_fn)
2160 Right t -> check_timestamp old_summary location src_fn t
2161 Left e | isDoesNotExistError e -> find_it
2162 | otherwise -> liftIO $ ioError e
2164 | otherwise = find_it
2166 dflags = hsc_dflags hsc_env
2168 hsc_src = if is_boot then HsBootFile else HsSrcFile
2170 check_timestamp old_summary location src_fn src_timestamp
2171 | ms_hs_date old_summary == src_timestamp = do
2172 -- update the object-file timestamp
2173 obj_timestamp <- liftIO $
2174 if isObjectTarget (hscTarget (hsc_dflags hsc_env))
2175 || obj_allowed -- bug #1205
2176 then getObjTimestamp location is_boot
2178 return (Just old_summary{ ms_obj_date = obj_timestamp })
2180 -- source changed: re-summarise.
2181 new_summary location (ms_mod old_summary) src_fn src_timestamp
2184 -- Don't use the Finder's cache this time. If the module was
2185 -- previously a package module, it may have now appeared on the
2186 -- search path, so we want to consider it to be a home module. If
2187 -- the module was previously a home module, it may have moved.
2188 liftIO $ uncacheModule hsc_env wanted_mod
2189 found <- liftIO $ findImportedModule hsc_env wanted_mod Nothing
2192 | isJust (ml_hs_file location) ->
2194 just_found location mod
2196 -- Drop external-pkg
2197 ASSERT(modulePackageId mod /= thisPackage dflags)
2200 err -> liftIO $ noModError dflags loc wanted_mod err
2203 just_found location mod = do
2204 -- Adjust location to point to the hs-boot source file,
2205 -- hi file, object file, when is_boot says so
2206 let location' | is_boot = addBootSuffixLocn location
2207 | otherwise = location
2208 src_fn = expectJust "summarise2" (ml_hs_file location')
2210 -- Check that it exists
2211 -- It might have been deleted since the Finder last found it
2212 maybe_t <- liftIO $ modificationTimeIfExists src_fn
2214 Nothing -> noHsFileErr loc src_fn
2215 Just t -> new_summary location' mod src_fn t
2218 new_summary location mod src_fn src_timestamp
2220 -- Preprocess the source file and get its imports
2221 -- The dflags' contains the OPTIONS pragmas
2222 (dflags', hspp_fn, buf) <- preprocessFile hsc_env src_fn Nothing maybe_buf
2223 (srcimps, the_imps, L mod_loc mod_name) <- getImports dflags' buf hspp_fn src_fn
2225 when (mod_name /= wanted_mod) $
2226 throwOneError $ mkPlainErrMsg mod_loc $
2227 text "File name does not match module name:"
2228 $$ text "Saw:" <+> quotes (ppr mod_name)
2229 $$ text "Expected:" <+> quotes (ppr wanted_mod)
2231 -- Find the object timestamp, and return the summary
2232 obj_timestamp <- liftIO $
2233 if isObjectTarget (hscTarget (hsc_dflags hsc_env))
2234 || obj_allowed -- bug #1205
2235 then getObjTimestamp location is_boot
2238 return (Just (ModSummary { ms_mod = mod,
2239 ms_hsc_src = hsc_src,
2240 ms_location = location,
2241 ms_hspp_file = hspp_fn,
2242 ms_hspp_opts = dflags',
2243 ms_hspp_buf = Just buf,
2244 ms_srcimps = srcimps,
2246 ms_hs_date = src_timestamp,
2247 ms_obj_date = obj_timestamp }))
2250 getObjTimestamp :: ModLocation -> Bool -> IO (Maybe ClockTime)
2251 getObjTimestamp location is_boot
2252 = if is_boot then return Nothing
2253 else modificationTimeIfExists (ml_obj_file location)
2256 preprocessFile :: GhcMonad m =>
2259 -> Maybe Phase -- ^ Starting phase
2260 -> Maybe (StringBuffer,ClockTime)
2261 -> m (DynFlags, FilePath, StringBuffer)
2262 preprocessFile hsc_env src_fn mb_phase Nothing
2264 (dflags', hspp_fn) <- preprocess hsc_env (src_fn, mb_phase)
2265 buf <- liftIO $ hGetStringBuffer hspp_fn
2266 return (dflags', hspp_fn, buf)
2268 preprocessFile hsc_env src_fn mb_phase (Just (buf, _time))
2270 let dflags = hsc_dflags hsc_env
2271 -- case we bypass the preprocessing stage?
2273 local_opts = getOptions dflags buf src_fn
2275 (dflags', leftovers, warns)
2276 <- parseDynamicNoPackageFlags dflags local_opts
2277 checkProcessArgsResult leftovers
2278 handleFlagWarnings dflags' warns
2282 | Just (Unlit _) <- mb_phase = True
2283 | Nothing <- mb_phase, Unlit _ <- startPhase src_fn = True
2284 -- note: local_opts is only required if there's no Unlit phase
2285 | dopt Opt_Cpp dflags' = True
2286 | dopt Opt_Pp dflags' = True
2289 when needs_preprocessing $
2290 ghcError (ProgramError "buffer needs preprocesing; interactive check disabled")
2292 return (dflags', src_fn, buf)
2295 -----------------------------------------------------------------------------
2297 -----------------------------------------------------------------------------
2299 noModError :: DynFlags -> SrcSpan -> ModuleName -> FindResult -> IO ab
2300 -- ToDo: we don't have a proper line number for this error
2301 noModError dflags loc wanted_mod err
2302 = throwOneError $ mkPlainErrMsg loc $ cannotFindModule dflags wanted_mod err
2304 noHsFileErr :: GhcMonad m => SrcSpan -> String -> m a
2305 noHsFileErr loc path
2306 = throwOneError $ mkPlainErrMsg loc $ text "Can't find" <+> text path
2308 packageModErr :: GhcMonad m => ModuleName -> m a
2310 = throwOneError $ mkPlainErrMsg noSrcSpan $
2311 text "module" <+> quotes (ppr mod) <+> text "is a package module"
2313 multiRootsErr :: [ModSummary] -> IO ()
2314 multiRootsErr [] = panic "multiRootsErr"
2315 multiRootsErr summs@(summ1:_)
2316 = throwOneError $ mkPlainErrMsg noSrcSpan $
2317 text "module" <+> quotes (ppr mod) <+>
2318 text "is defined in multiple files:" <+>
2319 sep (map text files)
2322 files = map (expectJust "checkDup" . ml_hs_file . ms_location) summs
2324 cyclicModuleErr :: [ModSummary] -> SDoc
2326 = hang (ptext (sLit "Module imports form a cycle for modules:"))
2327 2 (vcat (map show_one ms))
2329 mods_in_cycle = map ms_mod_name ms
2330 imp_modname = unLoc . ideclName . unLoc
2331 just_in_cycle = filter ((`elem` mods_in_cycle) . imp_modname)
2334 vcat [ show_mod (ms_hsc_src ms) (ms_mod_name ms) <+>
2335 maybe empty (parens . text) (ml_hs_file (ms_location ms)),
2336 nest 2 $ ptext (sLit "imports:") <+> vcat [
2337 pp_imps HsBootFile (just_in_cycle $ ms_srcimps ms),
2338 pp_imps HsSrcFile (just_in_cycle $ ms_imps ms) ]
2340 show_mod hsc_src mod = ppr mod <> text (hscSourceString hsc_src)
2341 pp_imps src imps = fsep (map (show_mod src . unLoc . ideclName . unLoc) imps)
2344 -- | Inform GHC that the working directory has changed. GHC will flush
2345 -- its cache of module locations, since it may no longer be valid.
2347 -- Note: Before changing the working directory make sure all threads running
2348 -- in the same session have stopped. If you change the working directory,
2349 -- you should also unload the current program (set targets to empty,
2350 -- followed by load).
2351 workingDirectoryChanged :: GhcMonad m => m ()
2352 workingDirectoryChanged = withSession $ (liftIO . flushFinderCaches)
2354 -- -----------------------------------------------------------------------------
2355 -- inspecting the session
2357 -- | Get the module dependency graph.
2358 getModuleGraph :: GhcMonad m => m ModuleGraph -- ToDo: DiGraph ModSummary
2359 getModuleGraph = liftM hsc_mod_graph getSession
2361 -- | Determines whether a set of modules requires Template Haskell.
2363 -- Note that if the session's 'DynFlags' enabled Template Haskell when
2364 -- 'depanal' was called, then each module in the returned module graph will
2365 -- have Template Haskell enabled whether it is actually needed or not.
2366 needsTemplateHaskell :: ModuleGraph -> Bool
2367 needsTemplateHaskell ms =
2368 any (dopt Opt_TemplateHaskell . ms_hspp_opts) ms
2370 -- | Return @True@ <==> module is loaded.
2371 isLoaded :: GhcMonad m => ModuleName -> m Bool
2372 isLoaded m = withSession $ \hsc_env ->
2373 return $! isJust (lookupUFM (hsc_HPT hsc_env) m)
2375 -- | Return the bindings for the current interactive session.
2376 getBindings :: GhcMonad m => m [TyThing]
2377 getBindings = withSession $ \hsc_env ->
2378 -- we have to implement the shadowing behaviour of ic_tmp_ids here
2379 -- (see InteractiveContext) and the quickest way is to use an OccEnv.
2381 occ_env = mkOccEnv [ (nameOccName (idName id), AnId id)
2382 | id <- ic_tmp_ids (hsc_IC hsc_env) ]
2384 return (occEnvElts occ_env)
2386 getPrintUnqual :: GhcMonad m => m PrintUnqualified
2387 getPrintUnqual = withSession $ \hsc_env ->
2388 return (icPrintUnqual (hsc_dflags hsc_env) (hsc_IC hsc_env))
2390 -- | Container for information about a 'Module'.
2391 data ModuleInfo = ModuleInfo {
2392 minf_type_env :: TypeEnv,
2393 minf_exports :: NameSet, -- ToDo, [AvailInfo] like ModDetails?
2394 minf_rdr_env :: Maybe GlobalRdrEnv, -- Nothing for a compiled/package mod
2395 minf_instances :: [Instance]
2397 ,minf_modBreaks :: ModBreaks
2399 -- ToDo: this should really contain the ModIface too
2401 -- We don't want HomeModInfo here, because a ModuleInfo applies
2402 -- to package modules too.
2404 -- | Request information about a loaded 'Module'
2405 getModuleInfo :: GhcMonad m => Module -> m (Maybe ModuleInfo) -- XXX: Maybe X
2406 getModuleInfo mdl = withSession $ \hsc_env -> do
2407 let mg = hsc_mod_graph hsc_env
2408 if mdl `elem` map ms_mod mg
2409 then liftIO $ getHomeModuleInfo hsc_env (moduleName mdl)
2411 {- if isHomeModule (hsc_dflags hsc_env) mdl
2413 else -} liftIO $ getPackageModuleInfo hsc_env mdl
2414 -- getPackageModuleInfo will attempt to find the interface, so
2415 -- we don't want to call it for a home module, just in case there
2416 -- was a problem loading the module and the interface doesn't
2417 -- exist... hence the isHomeModule test here. (ToDo: reinstate)
2419 getPackageModuleInfo :: HscEnv -> Module -> IO (Maybe ModuleInfo)
2421 getPackageModuleInfo hsc_env mdl = do
2422 (_msgs, mb_avails) <- getModuleExports hsc_env mdl
2424 Nothing -> return Nothing
2426 eps <- readIORef (hsc_EPS hsc_env)
2428 names = availsToNameSet avails
2430 tys = [ ty | name <- concatMap availNames avails,
2431 Just ty <- [lookupTypeEnv pte name] ]
2433 return (Just (ModuleInfo {
2434 minf_type_env = mkTypeEnv tys,
2435 minf_exports = names,
2436 minf_rdr_env = Just $! nameSetToGlobalRdrEnv names (moduleName mdl),
2437 minf_instances = error "getModuleInfo: instances for package module unimplemented",
2438 minf_modBreaks = emptyModBreaks
2441 getPackageModuleInfo _hsc_env _mdl = do
2442 -- bogusly different for non-GHCI (ToDo)
2446 getHomeModuleInfo :: HscEnv -> ModuleName -> IO (Maybe ModuleInfo)
2447 getHomeModuleInfo hsc_env mdl =
2448 case lookupUFM (hsc_HPT hsc_env) mdl of
2449 Nothing -> return Nothing
2451 let details = hm_details hmi
2452 return (Just (ModuleInfo {
2453 minf_type_env = md_types details,
2454 minf_exports = availsToNameSet (md_exports details),
2455 minf_rdr_env = mi_globals $! hm_iface hmi,
2456 minf_instances = md_insts details
2458 ,minf_modBreaks = getModBreaks hmi
2462 -- | The list of top-level entities defined in a module
2463 modInfoTyThings :: ModuleInfo -> [TyThing]
2464 modInfoTyThings minf = typeEnvElts (minf_type_env minf)
2466 modInfoTopLevelScope :: ModuleInfo -> Maybe [Name]
2467 modInfoTopLevelScope minf
2468 = fmap (map gre_name . globalRdrEnvElts) (minf_rdr_env minf)
2470 modInfoExports :: ModuleInfo -> [Name]
2471 modInfoExports minf = nameSetToList $! minf_exports minf
2473 -- | Returns the instances defined by the specified module.
2474 -- Warning: currently unimplemented for package modules.
2475 modInfoInstances :: ModuleInfo -> [Instance]
2476 modInfoInstances = minf_instances
2478 modInfoIsExportedName :: ModuleInfo -> Name -> Bool
2479 modInfoIsExportedName minf name = elemNameSet name (minf_exports minf)
2481 mkPrintUnqualifiedForModule :: GhcMonad m =>
2483 -> m (Maybe PrintUnqualified) -- XXX: returns a Maybe X
2484 mkPrintUnqualifiedForModule minf = withSession $ \hsc_env -> do
2485 return (fmap (mkPrintUnqualified (hsc_dflags hsc_env)) (minf_rdr_env minf))
2487 modInfoLookupName :: GhcMonad m =>
2489 -> m (Maybe TyThing) -- XXX: returns a Maybe X
2490 modInfoLookupName minf name = withSession $ \hsc_env -> do
2491 case lookupTypeEnv (minf_type_env minf) name of
2492 Just tyThing -> return (Just tyThing)
2494 eps <- liftIO $ readIORef (hsc_EPS hsc_env)
2495 return $! lookupType (hsc_dflags hsc_env)
2496 (hsc_HPT hsc_env) (eps_PTE eps) name
2499 modInfoModBreaks :: ModuleInfo -> ModBreaks
2500 modInfoModBreaks = minf_modBreaks
2503 isDictonaryId :: Id -> Bool
2505 = case tcSplitSigmaTy (idType id) of { (_tvs, _theta, tau) -> isDictTy tau }
2507 -- | Looks up a global name: that is, any top-level name in any
2508 -- visible module. Unlike 'lookupName', lookupGlobalName does not use
2509 -- the interactive context, and therefore does not require a preceding
2511 lookupGlobalName :: GhcMonad m => Name -> m (Maybe TyThing)
2512 lookupGlobalName name = withSession $ \hsc_env -> do
2513 liftIO $ lookupTypeHscEnv hsc_env name
2515 findGlobalAnns :: (GhcMonad m, Typeable a) => ([Word8] -> a) -> AnnTarget Name -> m [a]
2516 findGlobalAnns deserialize target = withSession $ \hsc_env -> do
2517 ann_env <- liftIO $ prepareAnnotations hsc_env Nothing
2518 return (findAnns deserialize ann_env target)
2521 -- | get the GlobalRdrEnv for a session
2522 getGRE :: GhcMonad m => m GlobalRdrEnv
2523 getGRE = withSession $ \hsc_env-> return $ ic_rn_gbl_env (hsc_IC hsc_env)
2526 -- -----------------------------------------------------------------------------
2528 -- | Return all /external/ modules available in the package database.
2529 -- Modules from the current session (i.e., from the 'HomePackageTable') are
2531 packageDbModules :: GhcMonad m =>
2532 Bool -- ^ Only consider exposed packages.
2534 packageDbModules only_exposed = do
2535 dflags <- getSessionDynFlags
2536 let pkgs = UFM.eltsUFM (pkgIdMap (pkgState dflags))
2538 [ mkModule pid modname | p <- pkgs
2539 , not only_exposed || exposed p
2540 , let pid = packageConfigId p
2541 , modname <- exposedModules p ]
2543 -- -----------------------------------------------------------------------------
2544 -- Misc exported utils
2546 dataConType :: DataCon -> Type
2547 dataConType dc = idType (dataConWrapId dc)
2549 -- | print a 'NamedThing', adding parentheses if the name is an operator.
2550 pprParenSymName :: NamedThing a => a -> SDoc
2551 pprParenSymName a = parenSymOcc (getOccName a) (ppr (getName a))
2553 -- ----------------------------------------------------------------------------
2558 -- - Data and Typeable instances for HsSyn.
2560 -- ToDo: check for small transformations that happen to the syntax in
2561 -- the typechecker (eg. -e ==> negate e, perhaps for fromIntegral)
2563 -- ToDo: maybe use TH syntax instead of IfaceSyn? There's already a way
2564 -- to get from TyCons, Ids etc. to TH syntax (reify).
2566 -- :browse will use either lm_toplev or inspect lm_interface, depending
2567 -- on whether the module is interpreted or not.
2571 -- Extract the filename, stringbuffer content and dynflags associed to a module
2573 -- XXX: Explain pre-conditions
2574 getModuleSourceAndFlags :: GhcMonad m => Module -> m (String, StringBuffer, DynFlags)
2575 getModuleSourceAndFlags mod = do
2576 m <- getModSummary (moduleName mod)
2577 case ml_hs_file $ ms_location m of
2578 Nothing -> throw $ mkApiErr (text "No source available for module " <+> ppr mod)
2579 Just sourceFile -> do
2580 source <- liftIO $ hGetStringBuffer sourceFile
2581 return (sourceFile, source, ms_hspp_opts m)
2584 -- | Return module source as token stream, including comments.
2586 -- The module must be in the module graph and its source must be available.
2587 -- Throws a 'HscTypes.SourceError' on parse error.
2588 getTokenStream :: GhcMonad m => Module -> m [Located Token]
2589 getTokenStream mod = do
2590 (sourceFile, source, flags) <- getModuleSourceAndFlags mod
2591 let startLoc = mkSrcLoc (mkFastString sourceFile) 1 1
2592 case lexTokenStream source startLoc flags of
2593 POk _ ts -> return ts
2594 PFailed span err -> throw $ mkSrcErr (unitBag $ mkPlainErrMsg span err)
2596 -- | Give even more information on the source than 'getTokenStream'
2597 -- This function allows reconstructing the source completely with
2598 -- 'showRichTokenStream'.
2599 getRichTokenStream :: GhcMonad m => Module -> m [(Located Token, String)]
2600 getRichTokenStream mod = do
2601 (sourceFile, source, flags) <- getModuleSourceAndFlags mod
2602 let startLoc = mkSrcLoc (mkFastString sourceFile) 1 1
2603 case lexTokenStream source startLoc flags of
2604 POk _ ts -> return $ addSourceToTokens startLoc source ts
2605 PFailed span err -> throw $ mkSrcErr (unitBag $ mkPlainErrMsg span err)
2607 -- | Given a source location and a StringBuffer corresponding to this
2608 -- location, return a rich token stream with the source associated to the
2610 addSourceToTokens :: SrcLoc -> StringBuffer -> [Located Token]
2611 -> [(Located Token, String)]
2612 addSourceToTokens _ _ [] = []
2613 addSourceToTokens loc buf (t@(L span _) : ts)
2614 | not (isGoodSrcSpan span) = (t,"") : addSourceToTokens loc buf ts
2615 | otherwise = (t,str) : addSourceToTokens newLoc newBuf ts
2617 (newLoc, newBuf, str) = go "" loc buf
2618 start = srcSpanStart span
2619 end = srcSpanEnd span
2620 go acc loc buf | loc < start = go acc nLoc nBuf
2621 | start <= loc && loc < end = go (ch:acc) nLoc nBuf
2622 | otherwise = (loc, buf, reverse acc)
2623 where (ch, nBuf) = nextChar buf
2624 nLoc = advanceSrcLoc loc ch
2627 -- | Take a rich token stream such as produced from 'getRichTokenStream' and
2628 -- return source code almost identical to the original code (except for
2629 -- insignificant whitespace.)
2630 showRichTokenStream :: [(Located Token, String)] -> String
2631 showRichTokenStream ts = go startLoc ts ""
2632 where sourceFile = srcSpanFile (getLoc . fst . head $ ts)
2633 startLoc = mkSrcLoc sourceFile 1 1
2635 go loc ((L span _, str):ts)
2636 | not (isGoodSrcSpan span) = go loc ts
2637 | locLine == tokLine = ((replicate (tokCol - locCol) ' ') ++)
2640 | otherwise = ((replicate (tokLine - locLine) '\n') ++)
2641 . ((replicate tokCol ' ') ++)
2644 where (locLine, locCol) = (srcLocLine loc, srcLocCol loc)
2645 (tokLine, tokCol) = (srcSpanStartLine span, srcSpanStartCol span)
2646 tokEnd = srcSpanEnd span
2648 -- -----------------------------------------------------------------------------
2649 -- Interactive evaluation
2651 -- | Takes a 'ModuleName' and possibly a 'PackageId', and consults the
2652 -- filesystem and package database to find the corresponding 'Module',
2653 -- using the algorithm that is used for an @import@ declaration.
2654 findModule :: GhcMonad m => ModuleName -> Maybe FastString -> m Module
2655 findModule mod_name maybe_pkg = withSession $ \hsc_env -> do
2657 dflags = hsc_dflags hsc_env
2658 this_pkg = thisPackage dflags
2661 Just pkg | fsToPackageId pkg /= this_pkg && pkg /= fsLit "this" -> liftIO $ do
2662 res <- findImportedModule hsc_env mod_name maybe_pkg
2664 Found _ m -> return m
2665 err -> noModError dflags noSrcSpan mod_name err
2667 home <- lookupLoadedHomeModule mod_name
2670 Nothing -> liftIO $ do
2671 res <- findImportedModule hsc_env mod_name maybe_pkg
2673 Found loc m | modulePackageId m /= this_pkg -> return m
2674 | otherwise -> modNotLoadedError m loc
2675 err -> noModError dflags noSrcSpan mod_name err
2677 modNotLoadedError :: Module -> ModLocation -> IO a
2678 modNotLoadedError m loc = ghcError $ CmdLineError $ showSDoc $
2679 text "module is not loaded:" <+>
2680 quotes (ppr (moduleName m)) <+>
2681 parens (text (expectJust "modNotLoadedError" (ml_hs_file loc)))
2683 -- | Like 'findModule', but differs slightly when the module refers to
2684 -- a source file, and the file has not been loaded via 'load'. In
2685 -- this case, 'findModule' will throw an error (module not loaded),
2686 -- but 'lookupModule' will check to see whether the module can also be
2687 -- found in a package, and if so, that package 'Module' will be
2688 -- returned. If not, the usual module-not-found error will be thrown.
2690 lookupModule :: GhcMonad m => ModuleName -> Maybe FastString -> m Module
2691 lookupModule mod_name (Just pkg) = findModule mod_name (Just pkg)
2692 lookupModule mod_name Nothing = withSession $ \hsc_env -> do
2693 home <- lookupLoadedHomeModule mod_name
2696 Nothing -> liftIO $ do
2697 res <- findExposedPackageModule hsc_env mod_name Nothing
2699 Found _ m -> return m
2700 err -> noModError (hsc_dflags hsc_env) noSrcSpan mod_name err
2702 lookupLoadedHomeModule :: GhcMonad m => ModuleName -> m (Maybe Module)
2703 lookupLoadedHomeModule mod_name = withSession $ \hsc_env ->
2704 case lookupUFM (hsc_HPT hsc_env) mod_name of
2705 Just mod_info -> return (Just (mi_module (hm_iface mod_info)))
2706 _not_a_home_module -> return Nothing
2709 getHistorySpan :: GhcMonad m => History -> m SrcSpan
2710 getHistorySpan h = withSession $ \hsc_env ->
2711 return$ InteractiveEval.getHistorySpan hsc_env h
2713 obtainTermFromVal :: GhcMonad m => Int -> Bool -> Type -> a -> m Term
2714 obtainTermFromVal bound force ty a =
2715 withSession $ \hsc_env ->
2716 liftIO $ InteractiveEval.obtainTermFromVal hsc_env bound force ty a
2718 obtainTermFromId :: GhcMonad m => Int -> Bool -> Id -> m Term
2719 obtainTermFromId bound force id =
2720 withSession $ \hsc_env ->
2721 liftIO $ InteractiveEval.obtainTermFromId hsc_env bound force id