1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
--
-- Haddock - A Haskell Documentation Tool
--
-- (c) Simon Marlow 2003
--
module Haddock.GHC.Typecheck (
typecheckFiles
) where
import Haddock.Exception
import Haddock.Types
import Haddock.GHC.Utils
import Data.Maybe
import Control.Monad
import GHC
import Digraph
import BasicTypes
import SrcLoc
import Data.List
type CheckedMod = (Module, FilePath, FullyCheckedMod)
type FullyCheckedMod = (ParsedSource,
RenamedSource,
TypecheckedSource,
ModuleInfo)
-- TODO: make it handle cleanup
typecheckFiles :: [FilePath] -> Ghc [GhcModule]
typecheckFiles files = do
-- load all argument files
targets <- mapM (\f -> guessTarget f Nothing) files
setTargets targets
flag <- load LoadAllTargets
when (failed flag) $
throwE "Failed to load all needed modules"
modgraph <- getModuleGraph
let mods = concatMap flattenSCC $ topSortModuleGraph False modgraph Nothing
getModFile = fromJust . ml_hs_file . ms_location
mods'= [ (ms_mod modsum, ms_hspp_opts modsum, getModFile modsum) |
modsum <- mods ]
-- typecheck the argument modules
ghcMods <- forM mods' $ \(mod, flags, file) ->
handleSourceError
(\err -> do
printExceptionAndWarnings err
throwE ("Failed to check module: " ++ moduleString mod)) $
do tc_mod <- typecheckModule =<< parseModule (moduleName mod)
let Just renamed_src = renamedSource tc_mod
return $ mkGhcModule (mod, file, (parsedSource tc_mod,
renamed_src,
typecheckedSource tc_mod,
moduleInfo tc_mod))
flags
return ghcMods
-- | Dig out what we want from the typechecker output
mkGhcModule :: CheckedMod -> DynFlags -> GhcModule
mkGhcModule (mod, file, checkedMod) dynflags = GhcModule {
ghcModule = mod,
ghcFilename = file,
ghcMbDocOpts = mbOpts,
ghcHaddockModInfo = info,
ghcMbDoc = mbDoc,
ghcGroup = group,
ghcMbExports = mbExports,
ghcExportedNames = modInfoExports modInfo,
ghcDefinedNames = map getName $ modInfoTyThings modInfo,
ghcNamesInScope = fromJust $ modInfoTopLevelScope modInfo,
ghcInstances = modInfoInstances modInfo
}
where
#if __GLASGOW_HASKELL__ == 608 && __GHC_PATCHLEVEL__ == 2
HsModule _ _ _ _ _ mbOpts _ _ = unLoc parsed
#else
mbOpts = haddockOptions dynflags
#endif
(group, _, mbExports, mbDoc, info) = renamed
(parsed, renamed, _, modInfo) = checkedMod
|
