[project @ 2000-12-18 15:18:11 by simonmar]
[ghc-hetmet.git] / ghc / compiler / main / Main.hs
1 {-# OPTIONS -fno-warn-incomplete-patterns #-}
2 -----------------------------------------------------------------------------
3 -- $Id: Main.hs,v 1.40 2000/12/18 15:18:11 simonmar Exp $
4 --
5 -- GHC Driver program
6 --
7 -- (c) Simon Marlow 2000
8 --
9 -----------------------------------------------------------------------------
10
11 -- with path so that ghc -M can find config.h
12 #include "../includes/config.h"
13
14 module Main (main) where
15
16 #include "HsVersions.h"
17
18
19 #ifdef GHCI
20 import Interpreter
21 import InteractiveUI
22 #endif
23
24 #ifndef mingw32_TARGET_OS
25 import Dynamic
26 import Posix
27 #endif
28
29 import CompManager
30 import DriverPipeline
31 import DriverState
32 import DriverFlags
33 import DriverMkDepend
34 import DriverUtil
35 import Panic
36 import DriverPhases     ( Phase(..) )
37 import CmdLineOpts      ( HscLang(..), DynFlags(..), v_Static_hsc_opts )
38 import TmpFiles
39 import Finder           ( initFinder )
40 import CmStaticInfo
41 import Config
42 import Util
43
44
45
46 import Concurrent
47 import Directory
48 import IOExts
49 import Exception
50
51 import IO
52 import Monad
53 import List
54 import System
55 import Maybe
56
57
58 -----------------------------------------------------------------------------
59 -- Changes:
60
61 -- * -fglasgow-exts NO LONGER IMPLIES -package lang!!!  (-fglasgow-exts is a
62 --   dynamic flag whereas -package is a static flag.)
63
64 -----------------------------------------------------------------------------
65 -- ToDo:
66
67 -- -nohi doesn't work
68 -- new mkdependHS doesn't support all the options that the old one did (-X et al.)
69 -- time commands when run with -v
70 -- split marker
71 -- mkDLL
72 -- java generation
73 -- user ways
74 -- Win32 support: proper signal handling
75 -- make sure OPTIONS in .hs file propogate to .hc file if -C or -keep-hc-file-too
76 -- reading the package configuration file is too slow
77 -- -H, -K, -Rghc-timing
78 -- hi-diffs
79 -- -ddump-all doesn't do anything
80
81 -----------------------------------------------------------------------------
82 -- Differences vs. old driver:
83
84 -- No more "Enter your Haskell program, end with ^D (on a line of its own):"
85 -- consistency checking removed (may do this properly later)
86 -- removed -noC
87 -- no -Ofile
88
89 -----------------------------------------------------------------------------
90 -- Main loop
91
92 main =
93   -- top-level exception handler: any unrecognised exception is a compiler bug.
94   handle (\exception -> hPutStr stderr (show (Panic (show exception)))) $ do
95
96   -- all error messages are propagated as exceptions
97   handleDyn (\dyn -> case dyn of
98                           PhaseFailed _phase code -> exitWith code
99                           Interrupted -> exitWith (ExitFailure 1)
100                           _ -> do hPutStrLn stderr (show (dyn :: GhcException))
101                                   exitWith (ExitFailure 1)
102               ) $ do
103
104    -- make sure we clean up after ourselves
105    later (do  forget_it <- readIORef v_Keep_tmp_files
106               unless forget_it $ do
107               verb <- dynFlag verbosity
108               cleanTempFiles (verb >= 2)
109      ) $ do
110         -- exceptions will be blocked while we clean the temporary files,
111         -- so there shouldn't be any difficulty if we receive further
112         -- signals.
113
114         -- install signal handlers
115    main_thread <- myThreadId
116 #ifndef mingw32_TARGET_OS
117    let sig_handler = Catch (throwTo main_thread 
118                                 (DynException (toDyn Interrupted)))
119    installHandler sigQUIT sig_handler Nothing 
120    installHandler sigINT  sig_handler Nothing
121 #endif
122
123    argv   <- getArgs
124
125         -- grab any -B options from the command line first
126    argv'  <- setTopDir argv
127    top_dir <- readIORef v_TopDir
128
129    let installed s = top_dir ++ '/':s
130        inplace s   = top_dir ++ '/':cCURRENT_DIR ++ '/':s
131
132        installed_pkgconfig = installed ("package.conf")
133        inplace_pkgconfig   = inplace (cGHC_DRIVER_DIR ++ "/package.conf.inplace")
134
135         -- discover whether we're running in a build tree or in an installation,
136         -- by looking for the package configuration file.
137    am_installed <- doesFileExist installed_pkgconfig
138
139    if am_installed
140         then writeIORef v_Path_package_config installed_pkgconfig
141         else do am_inplace <- doesFileExist inplace_pkgconfig
142                 if am_inplace
143                     then writeIORef v_Path_package_config inplace_pkgconfig
144                     else throwDyn (OtherError "can't find package.conf")
145
146         -- set the location of our various files
147    if am_installed
148         then do writeIORef v_Path_usage (installed "ghc-usage.txt")
149                 writeIORef v_Pgm_L (installed "unlit")
150                 writeIORef v_Pgm_m (installed "ghc-asm")
151                 writeIORef v_Pgm_s (installed "ghc-split")
152
153         else do writeIORef v_Path_usage (inplace (cGHC_DRIVER_DIR ++ "/ghc-usage.txt"))
154                 writeIORef v_Pgm_L (inplace cGHC_UNLIT)
155                 writeIORef v_Pgm_m (inplace cGHC_MANGLER)
156                 writeIORef v_Pgm_s (inplace cGHC_SPLIT)
157
158         -- read the package configuration
159    conf_file <- readIORef v_Path_package_config
160    contents <- readFile conf_file
161    let pkg_details = read contents      -- ToDo: faster
162    writeIORef v_Package_details pkg_details
163
164         -- find the phase to stop after (i.e. -E, -C, -c, -S flags)
165    (flags2, mode, stop_flag) <- getGhcMode argv'
166    writeIORef v_GhcMode mode
167
168         -- process all the other arguments, and get the source files
169    non_static <- processArgs static_flags flags2 []
170
171         -- find the build tag, and re-process the build-specific options
172    more_opts <- findBuildTag
173    way_non_static <- processArgs static_flags more_opts []
174
175         -- give the static flags to hsc
176    static_opts <- buildStaticHscOpts
177    writeIORef v_Static_hsc_opts static_opts
178
179         -- warnings
180    warn_level <- readIORef v_Warning_opt
181
182    let warn_opts =  case warn_level of
183                         W_default -> standardWarnings
184                         W_        -> minusWOpts
185                         W_all     -> minusWallOpts
186                         W_not     -> []
187
188         -- build the default DynFlags (these may be adjusted on a per
189         -- module basis by OPTIONS pragmas and settings in the interpreter).
190
191    core_todo <- buildCoreToDo
192    stg_todo  <- buildStgToDo
193
194    -- set the "global" HscLang.  The HscLang can be further adjusted on a module
195    -- by module basis, using only the -fvia-C and -fasm flags.  If the global
196    -- HscLang is not HscC or HscAsm, -fvia-C and -fasm have no effect.
197    opt_level  <- readIORef v_OptLevel
198    let lang = case mode of 
199                  StopBefore HCc -> HscC
200                  DoInteractive  -> HscInterpreted
201                  _other        | opt_level >= 1  -> HscC  -- -O implies -fvia-C 
202                                | otherwise       -> defaultHscLang
203
204    writeIORef v_DynFlags 
205         DynFlags{ coreToDo = core_todo,
206                   stgToDo  = stg_todo,
207                   hscLang  = lang,
208                   -- leave out hscOutName for now
209                   hscOutName = panic "Main.main:hscOutName not set",
210
211                   verbosity = case mode of
212                                 DoInteractive -> 1
213                                 DoMake        -> 1
214                                 _other        -> 0,
215
216                   flags = [] }
217
218         -- the rest of the arguments are "dynamic"
219    srcs <- processArgs dynamic_flags (way_non_static ++ 
220                                         non_static ++ warn_opts) []
221         -- save the "initial DynFlags" away
222    init_dyn_flags <- readIORef v_DynFlags
223    writeIORef v_InitDynFlags init_dyn_flags
224
225         -- complain about any unknown flags
226    mapM unknownFlagErr [ f | f@('-':_) <- srcs ]
227
228         -- save the flag state, because this could be modified by OPTIONS 
229         -- pragmas during the compilation, and we'll need to restore it
230         -- before starting the next compilation.
231    saved_driver_state <- readIORef v_Driver_state
232    writeIORef v_InitDriverState saved_driver_state
233
234    verb <- dynFlag verbosity
235
236    when (verb >= 2) 
237         (do hPutStr stderr "Glasgow Haskell Compiler, Version "
238             hPutStr stderr cProjectVersion
239             hPutStr stderr ", for Haskell 98, compiled by GHC version "
240             hPutStrLn stderr cBooterVersion)
241
242    when (verb >= 2) 
243         (hPutStrLn stderr ("Using package config file: " ++ conf_file))
244
245         -- initialise the finder
246    pkg_avails <- getPackageInfo
247    initFinder pkg_avails
248
249         -- mkdependHS is special
250    when (mode == DoMkDependHS) beginMkDependHS
251
252         -- make/interactive require invoking the compilation manager
253    if (mode == DoMake)        then beginMake srcs        else do
254    if (mode == DoInteractive) then beginInteractive srcs else do
255
256         -- for each source file, find which phases to run
257    let lang = hscLang init_dyn_flags
258    pipelines <- mapM (genPipeline mode stop_flag True lang) srcs
259    let src_pipelines = zip srcs pipelines
260
261         -- sanity checking
262    o_file <- readIORef v_Output_file
263    ohi    <- readIORef v_Output_hi
264    if length srcs > 1 && (isJust ohi || (isJust o_file && mode /= DoLink))
265         then throwDyn (UsageError "can't apply -o or -ohi options to multiple source files")
266         else do
267
268    if null srcs then throwDyn (UsageError "no input files") else do
269
270    let compileFile (src, phases) = do
271           writeIORef v_Driver_state saved_driver_state
272           writeIORef v_DynFlags init_dyn_flags
273           r <- runPipeline phases src (mode==DoLink) True
274           return r
275
276    o_files <- mapM compileFile src_pipelines
277
278    when (mode == DoMkDependHS) endMkDependHS
279    when (mode == DoLink) (doLink o_files)
280
281         -- grab the last -B option on the command line, and
282         -- set topDir to its value.
283 setTopDir :: [String] -> IO [String]
284 setTopDir args = do
285   let (minusbs, others) = partition (prefixMatch "-B") args
286   (case minusbs of
287     []   -> writeIORef v_TopDir clibdir
288     some -> writeIORef v_TopDir (drop 2 (last some)))
289   return others
290
291 beginMake :: [String] -> IO ()
292 beginMake mods
293   = do case mods of
294          []    -> throwDyn (UsageError "no input files")
295          [mod] -> do state <- cmInit Batch
296                      cmLoadModule state mod
297                      return ()
298          _     -> throwDyn (UsageError "only one module allowed with --make")
299
300
301 beginInteractive :: [String] -> IO ()
302 #ifndef GHCI
303 beginInteractive = throwDyn (OtherError "not build for interactive use")
304 #else
305 beginInteractive mods
306   = do state <- cmInit Interactive
307        let mod = case mods of
308                 []    -> Nothing
309                 [mod] -> Just mod
310                 _     -> throwDyn (UsageError 
311                                     "only one module allowed with --interactive")
312        interactiveUI state mod
313 #endif