+{-# OPTIONS -#include "Linker.h" -#include "SchedAPI.h" #-}
-----------------------------------------------------------------------------
--- $Id: InteractiveUI.hs,v 1.92 2001/10/15 15:05:17 simonpj Exp $
+-- $Id: InteractiveUI.hs,v 1.100 2001/10/23 22:20:33 sof Exp $
--
-- GHC Interactive User Interface
--
-- (c) The GHC Team 2000
--
-----------------------------------------------------------------------------
-
-{-# OPTIONS -#include "Linker.h" #-}
-{-# OPTIONS -#include "SchedAPI.h" #-}
module InteractiveUI ( interactiveUI, ghciWelcomeMsg ) where
#include "../includes/config.h"
import Packages
import CompManager
-import HscTypes ( GhciMode(..), TyThing(..) )
-import MkIface ( ifaceTyCls )
+import HscTypes ( TyThing(..) )
+import MkIface
import ByteCodeLink
import DriverFlags
import DriverState
#if HAVE_READLINE_HEADERS && HAVE_READLINE_LIBS
import Readline
#endif
+import Concurrent
import IOExts
+import SystemExts
import Numeric
import List
import Directory
import IO
import Char
-import Monad ( when )
+import Monad ( when, join )
import PrelGHC ( unsafeCoerce# )
import Foreign ( nullPtr )
("load", keepGoing loadModule),
("module", keepGoing setContext),
("reload", keepGoing reloadModule),
- ("set", keepGoing setOptions),
+ ("set", keepGoing setCmd),
("type", keepGoing typeOfExpr),
("unset", keepGoing unsetOptions),
("undef", keepGoing undefineMacro),
\ :module <mod> set the context for expression evaluation to <mod>\n\
\ :reload reload the current module set\n\
\ :set <option> ... set options\n\
+\ :set args <arg> ... set the arguments returned by System.getArgs\n\
+\ :set prog <progname> set the value returned by System.getProgName\n\
\ :undef <cmd> undefine user-defined command :<cmd>\n\
\ :type <expr> show the type of <expr>\n\
\ :unset <option> ... unset options\n\
dflags <- getDynFlags
(cmstate, maybe_hval)
+ <- cmCompileExpr cmstate dflags "IO.hSetBuffering IO.stdout IO.NoBuffering Prelude.>> IO.hSetBuffering IO.stderr IO.NoBuffering"
+ case maybe_hval of
+ Just hval -> unsafeCoerce# hval :: IO ()
+ _ -> panic "interactiveUI:buffering"
+
+ (cmstate, maybe_hval)
<- cmCompileExpr cmstate dflags "IO.hFlush PrelHandle.stderr"
case maybe_hval of
Just hval -> writeIORef flush_stderr (unsafeCoerce# hval :: IO ())
Just hval -> writeIORef flush_stdout (unsafeCoerce# hval :: IO ())
_ -> panic "interactiveUI:stdout"
- startGHCi runGHCi GHCiState{ targets = paths,
+ startGHCi runGHCi GHCiState{ progname = "<interactive>",
+ args = [],
+ targets = paths,
cmstate = cmstate,
options = [] }
Left e -> return ()
Right hdl -> fileLoop hdl False
+ interactiveLoop
+
+ -- and finally, exit
+ io $ do putStrLn "Leaving GHCi."
+
+
+interactiveLoop = do
+ -- ignore ^C exceptions caught here
+ ghciHandleDyn (\e -> case e of Interrupted -> ghciUnblock interactiveLoop
+ _other -> return ()) $ do
-- read commands from stdin
#if HAVE_READLINE_HEADERS && HAVE_READLINE_LIBS
readlineLoop
fileLoop stdin True
#endif
- -- and finally, exit
- io $ do putStrLn "Leaving GHCi."
-
-- NOTE: We only read .ghci files if they are owned by the current user,
-- and aren't world writable. Otherwise, we could be accidentally
readlineLoop = do
st <- getGHCiState
mod <- io (cmGetContext (cmstate st))
+ io yield
l <- io (readline (mod ++ "> "))
case l of
Nothing -> return ()
doCommand (':' : command) = specialCommand command
doCommand stmt
- = do timeIt (do stuff <- runStmt stmt; finishEvalExpr stuff)
+ = do timeIt (do nms <- runStmt stmt; finishEvalExpr nms)
return False
--- Returns True if the expr was successfully parsed, renamed and
--- typechecked.
-runStmt :: String -> GHCi (Maybe [Name])
+runStmt :: String -> GHCi [Name]
runStmt stmt
- | null (filter (not.isSpace) stmt)
- = return Nothing
+ | null (filter (not.isSpace) stmt) = return []
| otherwise
= do st <- getGHCiState
dflags <- io getDynFlags
let dflags' = dopt_unset dflags Opt_WarnUnusedBinds
- (new_cmstate, names) <- io (cmRunStmt (cmstate st) dflags' stmt)
+ (new_cmstate, result) <-
+ io $ withProgName (progname st) $ withArgs (args st) $
+ cmRunStmt (cmstate st) dflags' stmt
setGHCiState st{cmstate = new_cmstate}
- return (Just names)
+ case result of
+ CmRunFailed -> return []
+ CmRunException e -> showException e >> return []
+ CmRunDeadlocked -> io (putStrLn "Deadlocked.") >> return []
+ CmRunOk names -> return names
-- possibly print the type and revert CAFs after evaluating an expression
-finishEvalExpr Nothing = return False
-finishEvalExpr (Just names)
+finishEvalExpr names
= do b <- isOptionSet ShowType
st <- getGHCiState
when b (mapM_ (showTypeOfName (cmstate st)) names)
flushEverything :: GHCi ()
flushEverything
- = io $ do flush_so <- readIORef flush_stdout
- flush_so
- flush_se <- readIORef flush_stdout
- flush_se
+ = io $ do Monad.join (readIORef flush_stdout)
+ Monad.join (readIORef flush_stderr)
return ()
specialCommand :: String -> GHCi Bool
-- This is pretty fragile: most options won't work as expected. ToDo:
-- figure out which ones & disallow them.
-setOptions :: String -> GHCi ()
-setOptions ""
+setCmd :: String -> GHCi ()
+setCmd ""
= do st <- getGHCiState
let opts = options st
io $ putStrLn (showSDoc (
then text "none."
else hsep (map (\o -> char '+' <> text (optToStr o)) opts)
))
-setOptions str
- = do -- first, deal with the GHCi opts (+s, +t, etc.)
- let (plus_opts, minus_opts) = partition isPlus (words str)
+setCmd str
+ = case words str of
+ ("args":args) -> setArgs args
+ ("prog":prog) -> setProg prog
+ wds -> setOptions wds
+
+setArgs args = do
+ st <- getGHCiState
+ setGHCiState st{ args = args }
+
+setProg [prog] = do
+ st <- getGHCiState
+ setGHCiState st{ progname = prog }
+setProg _ = do
+ io (hPutStrLn stderr "syntax: :set prog <progname>")
+
+setOptions wds =
+ do -- first, deal with the GHCi opts (+s, +t, etc.)
+ let (plus_opts, minus_opts) = partition isPlus wds
mapM setOpt plus_opts
-- now, the GHC flags
flushPackageCache pkgs
new_pkg_info <- getPackageDetails new_pkgs
- mapM_ (linkPackage False) (reverse new_pkg_info)
+ mapM_ linkPackage (reverse new_pkg_info)
-----------------------------------------------------------------------------
-- GHCi monad
data GHCiState = GHCiState
{
+ progname :: String,
+ args :: [String],
targets :: [FilePath],
cmstate :: CmState,
options :: [GHCiOption]
(GHCi m) >>= k = GHCi $ \s -> m s >>= \a -> unGHCi (k a) s
return a = GHCi $ \s -> return a
+ghciHandleDyn :: Typeable t => (t -> GHCi a) -> GHCi a -> GHCi a
+ghciHandleDyn h (GHCi m) = GHCi $ \s ->
+ Exception.catchDyn (m s) (\e -> unGHCi (h e) s)
+
getGHCiState = GHCi $ \r -> readIORef r
setGHCiState s = GHCi $ \r -> writeIORef r s
linkPackages :: [LibrarySpec] -> [PackageConfig] -> IO ()
linkPackages cmdline_lib_specs pkgs
- = do sequence_ [ linkPackage (name p `elem` loaded) p | p <- reverse pkgs ]
+ = do mapM_ linkPackage (reverse pkgs)
lib_paths <- readIORef v_Library_paths
mapM_ (preloadLib lib_paths) cmdline_lib_specs
if (null cmdline_lib_specs)
if ok then putStrLn "done."
else throwDyn (InstallationError "linking extra libraries/objects failed")
where
- -- Packages that are already linked into GHCi. For mingw32, we only
- -- skip gmp and rts, since std and after need to load the msvcrt.dll
- -- library which std depends on.
- loaded
-# ifndef mingw32_TARGET_OS
- = [ "gmp", "rts", "std", "concurrent", "posix", "text", "util" ]
-# else
- = [ "gmp", "rts" ]
-# endif
-
preloadLib :: [String] -> LibrarySpec -> IO ()
preloadLib lib_paths lib_spec
= do putStr ("Loading object " ++ showLS lib_spec ++ " ... ")
= (throwDyn . CmdLineError)
"user specified .o/.so/.DLL could not be loaded."
+-- Packages that don't need loading, because the compiler shares them with
+-- the interpreted program.
+dont_load_these = [ "gmp", "rts" ]
+
+-- Packages that are already linked into GHCi. For mingw32, we only
+-- skip gmp and rts, since std and after need to load the msvcrt.dll
+-- library which std depends on.
+loaded_in_ghci
+# ifndef mingw32_TARGET_OS
+ = [ "std", "concurrent", "posix", "text", "util" ]
+# else
+ = [ ]
+# endif
-linkPackage :: Bool -> PackageConfig -> IO ()
--- ignore rts and gmp for now (ToDo; better?)
-linkPackage loaded_in_ghci pkg
- | name pkg `elem` ["rts", "gmp"]
- = return ()
+linkPackage :: PackageConfig -> IO ()
+linkPackage pkg
+ | name pkg `elem` dont_load_these = return ()
| otherwise
- = do putStr ("Loading package " ++ name pkg ++ " ... ")
+ = do
-- For each obj, try obj.o and if that fails, obj.so.
-- Complication: all the .so's must be loaded before any of the .o's.
let dirs = library_dirs pkg
-- Don't load the .so libs if this is a package GHCi is already
-- linked against, because we'll already have the .so linked in.
let (so_libs, obj_libs) = partition isRight classifieds
- let sos_first | loaded_in_ghci = obj_libs
- | otherwise = so_libs ++ obj_libs
+ let sos_first | name pkg `elem` loaded_in_ghci = obj_libs
+ | otherwise = so_libs ++ obj_libs
+ putStr ("Loading package " ++ name pkg ++ " ... ")
mapM loadClassified sos_first
putStr "linking ... "
ok <- resolveObjs