1 -----------------------------------------------------------------------------
3 -- | Parsing the top of a Haskell source file to get its module name,
4 -- imports and options.
6 -- (c) Simon Marlow 2005
9 -----------------------------------------------------------------------------
11 module HeaderInfo ( getImports
12 , getOptionsFromFile, getOptions
14 checkProcessArgsResult ) where
16 #include "HsVersions.h"
20 import Parser ( parseHeader )
23 import HsSyn ( ImportDecl(..), HsModule(..) )
24 import Module ( ModuleName, moduleName )
25 import PrelNames ( gHC_PRIM, mAIN_NAME )
34 import Bag ( emptyBag, listToBag, unitBag )
36 import MonadUtils ( MonadIO )
40 import System.IO.Unsafe
43 ------------------------------------------------------------------------------
45 -- | Parse the imports of a source file.
47 -- Throws a 'SourceError' if parsing fails.
48 getImports :: GhcMonad m =>
50 -> StringBuffer -- ^ Parse this.
51 -> FilePath -- ^ Filename the buffer came from. Used for
52 -- reporting parse error locations.
53 -> FilePath -- ^ The original source filename (used for locations
54 -- in the function result)
55 -> m ([Located (ImportDecl RdrName)], [Located (ImportDecl RdrName)], Located ModuleName)
56 -- ^ The source imports, normal imports, and the module name.
57 getImports dflags buf filename source_filename = do
58 let loc = mkSrcLoc (mkFastString filename) 1 0
59 case unP parseHeader (mkPState buf loc dflags) of
60 PFailed span err -> parseError span err
61 POk pst rdr_module -> do
62 let ms@(_warns, errs) = getMessages pst
63 -- don't log warnings: they'll be reported when we parse the file
64 -- for real. See #2500.
66 if errorsFound dflags ms
67 then liftIO $ throwIO $ mkSrcErr errs
70 L _ (HsModule mb_mod _ imps _ _ _ _) ->
72 main_loc = mkSrcLoc (mkFastString source_filename) 1 0
73 mod = mb_mod `orElse` L (srcLocSpan main_loc) mAIN_NAME
74 (src_idecls, ord_idecls) = partition (ideclSource.unLoc) imps
75 ordinary_imps = filter ((/= moduleName gHC_PRIM) . unLoc . ideclName . unLoc)
77 -- GHC.Prim doesn't exist physically, so don't go looking for it.
79 return (src_idecls, ordinary_imps, mod)
81 parseError :: GhcMonad m => SrcSpan -> Message -> m a
82 parseError span err = throwOneError $ mkPlainErrMsg span err
84 --------------------------------------------------------------
86 --------------------------------------------------------------
88 -- | Parse OPTIONS and LANGUAGE pragmas of the source file.
90 -- Throws a 'SourceError' if flag parsing fails (including unsupported flags.)
91 getOptionsFromFile :: DynFlags
92 -> FilePath -- ^ Input file
93 -> IO [Located String] -- ^ Parsed options, if any.
94 getOptionsFromFile dflags filename
96 (openBinaryFile filename ReadMode)
99 opts <- fmap getOptions' $ lazyGetToks dflags filename handle
100 seqList opts $ return opts)
103 -- blockSize = 17 -- for testing :-)
106 lazyGetToks :: DynFlags -> FilePath -> Handle -> IO [Located Token]
107 lazyGetToks dflags filename handle = do
108 buf <- hGetStringBufferBlock handle blockSize
109 unsafeInterleaveIO $ lazyLexBuf handle (pragState dflags buf loc) False
111 loc = mkSrcLoc (mkFastString filename) 1 0
113 lazyLexBuf :: Handle -> PState -> Bool -> IO [Located Token]
114 lazyLexBuf handle state eof = do
115 case unP (lexer return) state of
117 -- pprTrace "lazyLexBuf" (text (show (buffer state'))) (return ())
118 if atEnd (buffer state') && not eof
119 -- if this token reached the end of the buffer, and we haven't
120 -- necessarily read up to the end of the file, then the token might
121 -- be truncated, so read some more of the file and lex it again.
122 then getMore handle state
124 L _ ITeof -> return [t]
125 _other -> do rest <- lazyLexBuf handle state' eof
127 _ | not eof -> getMore handle state
128 | otherwise -> return [L (last_loc state) ITeof]
129 -- parser assumes an ITeof sentinel at the end
131 getMore :: Handle -> PState -> IO [Located Token]
132 getMore handle state = do
133 -- pprTrace "getMore" (text (show (buffer state))) (return ())
134 nextbuf <- hGetStringBufferBlock handle blockSize
135 if (len nextbuf == 0) then lazyLexBuf handle state True else do
136 newbuf <- appendStringBuffers (buffer state) nextbuf
137 unsafeInterleaveIO $ lazyLexBuf handle state{buffer=newbuf} False
140 getToks :: DynFlags -> FilePath -> StringBuffer -> [Located Token]
141 getToks dflags filename buf = lexAll (pragState dflags buf loc)
143 loc = mkSrcLoc (mkFastString filename) 1 0
145 lexAll state = case unP (lexer return) state of
146 POk _ t@(L _ ITeof) -> [t]
147 POk state' t -> t : lexAll state'
148 _ -> [L (last_loc state) ITeof]
151 -- | Parse OPTIONS and LANGUAGE pragmas of the source file.
153 -- Throws a 'SourceError' if flag parsing fails (including unsupported flags.)
154 getOptions :: DynFlags
155 -> StringBuffer -- ^ Input Buffer
156 -> FilePath -- ^ Source filename. Used for location info.
157 -> [Located String] -- ^ Parsed options.
158 getOptions dflags buf filename
159 = getOptions' (getToks dflags filename buf)
161 -- The token parser is written manually because Happy can't
162 -- return a partial result when it encounters a lexer error.
163 -- We want to extract options before the buffer is passed through
164 -- CPP, so we can't use the same trick as 'getImports'.
165 getOptions' :: [Located Token] -- Input buffer
166 -> [Located String] -- Options.
170 getToken (L _loc tok) = tok
171 getLoc (L loc _tok) = loc
173 parseToks (open:close:xs)
174 | IToptions_prag str <- getToken open
175 , ITclose_prag <- getToken close
176 = map (L (getLoc open)) (words str) ++
178 parseToks (open:close:xs)
179 | ITinclude_prag str <- getToken open
180 , ITclose_prag <- getToken close
181 = map (L (getLoc open)) ["-#include",removeSpaces str] ++
183 parseToks (open:close:xs)
184 | ITdocOptions str <- getToken open
185 , ITclose_prag <- getToken close
186 = map (L (getLoc open)) ["-haddock-opts", removeSpaces str]
189 | ITdocOptionsOld str <- getToken open
190 = map (L (getLoc open)) ["-haddock-opts", removeSpaces str]
193 | ITlanguage_prag <- getToken open
196 parseLanguage (L loc (ITconid fs):rest)
197 = checkExtension (L loc fs) :
199 (L _loc ITcomma):more -> parseLanguage more
200 (L _loc ITclose_prag):more -> parseToks more
201 (L loc _):_ -> languagePragParseError loc
202 [] -> panic "getOptions'.parseLanguage(1) went past eof token"
203 parseLanguage (tok:_)
204 = languagePragParseError (getLoc tok)
206 = panic "getOptions'.parseLanguage(2) went past eof token"
208 -----------------------------------------------------------------------------
210 -- | Complain about non-dynamic flags in OPTIONS pragmas.
212 -- Throws a 'SourceError' if the input list is non-empty claiming that the
213 -- input flags are unknown.
214 checkProcessArgsResult :: MonadIO m => [Located String] -> m ()
215 checkProcessArgsResult flags
216 = when (notNull flags) $
217 liftIO $ throwIO $ mkSrcErr $ listToBag $ map mkMsg flags
218 where mkMsg (L loc flag)
219 = mkPlainErrMsg loc $
220 (text "unknown flag in {-# OPTIONS #-} pragma:" <+>
223 -----------------------------------------------------------------------------
225 checkExtension :: Located FastString -> Located String
226 checkExtension (L l ext)
227 -- Checks if a given extension is valid, and if so returns
228 -- its corresponding flag. Otherwise it throws an exception.
229 = let ext' = unpackFS ext in
230 if ext' `elem` supportedLanguages
231 || ext' `elem` (map ("No"++) supportedLanguages)
232 then L l ("-X"++ext')
233 else unsupportedExtnError l ext'
235 languagePragParseError :: SrcSpan -> a
236 languagePragParseError loc =
237 throw $ mkSrcErr $ unitBag $
239 text "cannot parse LANGUAGE pragma: comma-separated list expected")
241 unsupportedExtnError :: SrcSpan -> String -> a
242 unsupportedExtnError loc unsup =
243 throw $ mkSrcErr $ unitBag $
245 text "unsupported extension: " <> text unsup
248 optionsErrorMsgs :: [String] -> [Located String] -> FilePath -> Messages
249 optionsErrorMsgs unhandled_flags flags_lines _filename
250 = (emptyBag, listToBag (map mkMsg unhandled_flags_lines))
251 where unhandled_flags_lines = [ L l f | f <- unhandled_flags,
252 L l f' <- flags_lines, f == f' ]
253 mkMsg (L flagSpan flag) =
254 ErrUtils.mkPlainErrMsg flagSpan $
255 text "unknown flag in {-# OPTIONS #-} pragma:" <+> text flag