-----------------------------------------------------------------------------
--- $Id: InteractiveUI.hs,v 1.40 2001/02/09 17:29:59 simonmar Exp $
+-- $Id: InteractiveUI.hs,v 1.47 2001/02/13 17:13:39 sewardj Exp $
--
-- GHC Interactive User Interface
--
--
-----------------------------------------------------------------------------
-module InteractiveUI (interactiveUI) where
+{-# OPTIONS -#include "Linker.h" #-}
+module InteractiveUI ( interactiveUI, ghciWelcomeMsg ) where
#include "HsVersions.h"
import Directory
import IO
import Char
-import Monad ( when )
+import Monad ( when )
+import PrelGHC ( unsafeCoerce# )
+import PrelPack ( packString )
+import PrelByteArr
+import Foreign ( Ptr, nullPtr )
-----------------------------------------------------------------------------
\/ /_\\\\/ __ / /___| | http://www.haskell.org/ghc/\n\
\\\____/\\/ /_/\\____/|_| Type :? for help.\n"
-commands :: [(String, String -> GHCi Bool)]
-commands = [
+GLOBAL_VAR(commands, builtin_commands, [(String, String -> GHCi Bool)])
+
+builtin_commands :: [(String, String -> GHCi Bool)]
+builtin_commands = [
("add", keepGoing addModule),
("cd", keepGoing changeDirectory),
+ ("def", keepGoing defineMacro),
("help", keepGoing help),
("?", keepGoing help),
("load", keepGoing loadModule),
("set", keepGoing setOptions),
("type", keepGoing typeOfExpr),
("unset", keepGoing unsetOptions),
+ ("undef", keepGoing undefineMacro),
("quit", quit)
]
\ (eg. -v2, -fglasgow-exts, etc.)\n\
\"
-interactiveUI :: CmState -> Maybe FilePath -> IO ()
-interactiveUI cmstate mod = do
- hPutStrLn stdout ghciWelcomeMsg
+interactiveUI :: CmState -> Maybe FilePath -> [String] -> IO ()
+interactiveUI cmstate mod cmdline_libs = do
hFlush stdout
hSetBuffering stdout NoBuffering
-- link in the available packages
pkgs <- getPackageInfo
- linkPackages (reverse pkgs)
+ initLinker
+ linkPackages cmdline_libs (reverse pkgs)
(cmstate, ok, mods) <-
case mod of
l -> do quit <- runCommand l
if quit then return () else fileLoop hdl prompt
+stringLoop :: [String] -> GHCi ()
+stringLoop [] = return ()
+stringLoop (s:ss) = do
+ st <- getGHCiState
+ case remove_spaces s of
+ "" -> stringLoop ss
+ l -> do quit <- runCommand l
+ if quit then return () else stringLoop ss
+
#ifndef NO_READLINE
readlineLoop :: GHCi ()
readlineLoop = do
io (when b revertCAFs)
return True
--- Returned Bool indicates whether or not the expr was successfully
+-- Returned Maybe indicates whether or not the expr was successfully
-- parsed, renamed and typechecked.
evalExpr :: String -> GHCi (Maybe (PrintUnqualified,Type))
evalExpr expr
specialCommand ('!':str) = shellEscape (dropWhile isSpace str)
specialCommand str = do
let (cmd,rest) = break isSpace str
- case [ (s,f) | (s,f) <- commands, prefixMatch cmd s ] of
+ cmds <- io (readIORef commands)
+ case [ (s,f) | (s,f) <- cmds, prefixMatch cmd s ] of
[] -> io (hPutStr stdout ("unknown command `:" ++ cmd ++ "'\n"
++ shortHelpText) >> return False)
[(_,f)] -> f (dropWhile isSpace rest)
changeDirectory :: String -> GHCi ()
changeDirectory d = io (setCurrentDirectory d)
+defineMacro :: String -> GHCi ()
+defineMacro s = do
+ let (macro_name, definition) = break isSpace s
+ cmds <- io (readIORef commands)
+ if (null macro_name)
+ then throwDyn (OtherError "invalid macro name")
+ else do
+ if (macro_name `elem` map fst cmds)
+ then throwDyn (OtherError
+ ("command `" ++ macro_name ++ "' already defined"))
+ else do
+
+ -- give the expression a type signature, so we can be sure we're getting
+ -- something of the right type.
+ let new_expr = '(' : definition ++ ") :: String -> IO String"
+
+ -- compile the expression
+ st <- getGHCiState
+ dflags <- io (getDynFlags)
+ (new_cmstate, maybe_stuff) <-
+ io (cmGetExpr (cmstate st) dflags False (current_module st) new_expr)
+ setGHCiState st{cmstate = new_cmstate}
+ case maybe_stuff of
+ Nothing -> return ()
+ Just (hv, unqual, ty)
+ -> io (writeIORef commands
+ ((macro_name, keepGoing (runMacro hv)) : cmds))
+
+runMacro :: HValue{-String -> IO String-} -> String -> GHCi ()
+runMacro fun s = do
+ str <- io ((unsafeCoerce# fun :: String -> IO String) s)
+ stringLoop (lines str)
+
+undefineMacro :: String -> GHCi ()
+undefineMacro macro_name = do
+ cmds <- io (readIORef commands)
+ if (macro_name `elem` map fst builtin_commands)
+ then throwDyn (OtherError
+ ("command `" ++ macro_name ++ "' cannot be undefined"))
+ else do
+ if (macro_name `notElem` map fst cmds)
+ then throwDyn (OtherError
+ ("command `" ++ macro_name ++ "' not defined"))
+ else do
+ io (writeIORef commands (filter ((/= macro_name) . fst) cmds))
+
loadModule :: String -> GHCi ()
loadModule path = timeIt (loadModule' path)
-----------------------------------------------------------------------------
-- package loader
-linkPackages :: [Package] -> IO ()
-linkPackages pkgs = mapM_ linkPackage pkgs
+linkPackages :: [String] -> [Package] -> IO ()
+linkPackages cmdline_libs pkgs
+ = do mapM preloadLib cmdline_libs
+ mapM_ linkPackage pkgs
+ where
+ preloadLib orig_name
+ = do putStr ("Loading object " ++ orig_name ++ " ... ")
+ case classify orig_name of
+ Left static_ish
+ -> do b <- doesFileExist static_ish
+ if not b
+ then do putStr "not found.\n"
+ croak
+ else do loadObj static_ish
+ putStr "done.\n"
+ Right dll_unadorned
+ -> do dll_ok <- ocAddDLL (packString dll_unadorned)
+ if dll_ok == 1
+ then putStr "done.\n"
+ else do putStr "not found.\n"
+ croak
+
+ croak = throwDyn (OtherError "user specified .o/.so/.DLL cannot be found.")
+
+ classify a_lib
+ = let a_libr = reverse a_lib
+ in
+ case map toLower a_libr of
+ ('o':'.':_)
+ -> Left a_lib
+ ('o':'s':'.':_)
+ -> (Right . zap_leading_lib
+ . reverse . drop 3 . reverse) a_lib
+ ('l':'l':'d':'.':_)
+ -> (Right . reverse . drop 4 . reverse) a_lib
+ other
+ -> -- Main.beginInteractive should not have let this through
+ pprPanic "linkPackages" (text (show a_lib))
+
+ zap_leading_lib str
+ = if take 3 str == "lib" then drop 3 str else str
+
linkPackage :: Package -> IO ()
-- ignore rts and gmp for now (ToDo; better?)
-linkPackage pkg | name pkg `elem` ["rts", "gmp"] = return ()
-linkPackage pkg = do
- putStr ("Loading package " ++ name pkg ++ " ... ")
- let dirs = library_dirs pkg
- let objs = map (++".o") (hs_libraries pkg ++ extra_libraries pkg)
- mapM (linkOneObj dirs) objs
- putStr "resolving ... "
- resolveObjs
- putStrLn "done."
-
-linkOneObj dirs obj = do
- filename <- findFile dirs obj
- loadObj filename
-
-findFile [] obj = throwDyn (OtherError ("can't find " ++ obj))
-findFile (d:ds) obj = do
- let path = d ++ '/':obj
- b <- doesFileExist path
- if b then return path else findFile ds obj
+linkPackage pkg
+ | name pkg `elem` ["rts", "gmp"]
+ = return ()
+ | otherwise
+ = do putStr ("Loading package " ++ name pkg ++ " ... ")
+ -- 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
+ let objs = hs_libraries pkg ++ extra_libraries pkg
+ classifieds <- mapM (locateOneObj dirs) objs
+ let sos_first = filter isRight classifieds
+ ++ filter (not.isRight) classifieds
+ mapM loadClassified sos_first
+ putStr "linking ... "
+ resolveObjs
+ putStrLn "done."
+ where
+ isRight (Right _) = True
+ isRight (Left _) = False
+
+loadClassified :: Either FilePath String -> IO ()
+loadClassified (Left obj_absolute_filename)
+ = do loadObj obj_absolute_filename
+loadClassified (Right dll_unadorned)
+ = do dll_ok <- ocAddDLL (packString dll_unadorned)
+ if dll_ok == 1
+ then return ()
+ else throwDyn (OtherError ("can't find .o or .so/.DLL for: "
+ ++ dll_unadorned))
+
+locateOneObj :: [FilePath] -> String -> IO (Either FilePath String)
+locateOneObj [] obj
+ = return (Right obj) -- we assume
+locateOneObj (d:ds) obj
+ = do let path = d ++ '/':obj ++ ".o"
+ b <- doesFileExist path
+ if b then return (Left path) else locateOneObj ds obj
+
+
+type PackedString = ByteArray Int
+foreign import "ocAddDLL" unsafe ocAddDLL :: PackedString -> IO Int
-----------------------------------------------------------------------------
-- timing & statistics
projects / ghc-hetmet.git / blobdiff