Refactor GFI to treat command line files and the i command uniformly. Disable CSE by default in GFI.

3 files changed, 33 insertions, 29 deletions

diff --git a/src-3.0/GF/Command/Importing.hs b/src-3.0/GF/Command/Importing.hs
index 48f07969d..c3ad9d746 100644
--- a/src-3.0/GF/Command/Importing.hs
+++ b/src-3.0/GF/Command/Importing.hs
@@ -14,6 +14,7 @@ import System.FilePath
 
 -- import a grammar in an environment where it extends an existing grammar
 importGrammar :: PGF -> Options -> [FilePath] -> IO PGF
+importGrammar pgf0 _ [] = return pgf0
 importGrammar pgf0 opts files =
   case takeExtensions (last files) of
     s | elem s [".gf",".gfo"] -> do
diff --git a/src-3.0/GF/Command/Interpreter.hs b/src-3.0/GF/Command/Interpreter.hs
index a5da51f7e..fa0de5ec8 100644
--- a/src-3.0/GF/Command/Interpreter.hs
+++ b/src-3.0/GF/Command/Interpreter.hs
@@ -1,5 +1,6 @@
 module GF.Command.Interpreter (
   CommandEnv (..),
+  mkCommandEnv,
   interpretCommandLine
   ) where
 
@@ -21,6 +22,9 @@ data CommandEnv = CommandEnv {
   commands     :: Map.Map String CommandInfo
   }
 
+mkCommandEnv :: PGF -> CommandEnv
+mkCommandEnv pgf = CommandEnv pgf (allCommands pgf)
+
 interpretCommandLine :: CommandEnv -> String -> IO ()
 interpretCommandLine env line = case (pCommandLine (myLexer line)) of
   Ok CEmpty -> return ()
diff --git a/src-3.0/GFI.hs b/src-3.0/GFI.hs
index 9c38c69b0..46c6be9c9 100644
--- a/src-3.0/GFI.hs
+++ b/src-3.0/GFI.hs
@@ -20,36 +20,34 @@ import Paths_gf
 mainGFI :: Options -> [FilePath] -> IO ()
 mainGFI opts files = do
   putStrLn welcome
-  env <- importInEnv emptyPGF opts files
-  loop (GFEnv emptyGrammar env [] 0)
+  gfenv <- importInEnv emptyGFEnv opts files
+  loop opts gfenv
   return ()
 
-loop :: GFEnv -> IO GFEnv
-loop gfenv0 = do
+loop :: Options -> GFEnv -> IO GFEnv
+loop opts gfenv0 = do
   let env = commandenv gfenv0
   let sgr = sourcegrammar gfenv0
   s <- fetchCommand (prompt env)
   let gfenv = gfenv0 {history = s : history gfenv0}
+  let loopNewCPU gfenv' = do cpu' <- getCPUTime
+                             putStrLn (show ((cpu' - cputime gfenv') `div` 1000000000) ++ " msec")
+                             loop opts $ gfenv' {cputime = cpu'}
   case words s of
   -- special commands, requiring source grammar in env
     "cc":ws -> do
        -- FIXME: add options parsing for cc arguments
-       let (opts,term) = (TermPrintDefault, ws)
+       let (style,term) = (TermPrintDefault, ws)
        case pTerm (unwords term) >>= checkTerm sgr >>= computeTerm sgr of   ---- make pipable
-         Ok  x -> putStrLn (showTerm opts x)
+         Ok  x -> putStrLn (showTerm style x)
          Bad s -> putStrLn s
        loopNewCPU gfenv
     "i":args -> do
-      case parseOptions args of
-        Ok (opts,files) 
-         | flag optRetainResource opts -> 
-             do src <- importSource sgr opts files
-                loopNewCPU $ gfenv {sourcegrammar = src}
-         | otherwise ->
-             do env1 <- importInEnv (multigrammar env) opts files
-                loopNewCPU $ gfenv {commandenv = env1}
-        Bad err -> do putStrLn $ "Command parse error: " ++ err
-                      loopNewCPU gfenv
+        gfenv' <- case parseOptions args of
+                    Ok (opts',files) -> importInEnv gfenv (addOptions opts opts') files
+                    Bad err -> do putStrLn $ "Command parse error: " ++ err
+                                  return gfenv
+        loopNewCPU gfenv
 
   -- other special commands, working on GFEnv
     "e":_ -> loopNewCPU $ gfenv {commandenv=env{multigrammar=emptyPGF}}
@@ -61,19 +59,17 @@ loop gfenv0 = do
       interpretCommandLine env s
       loopNewCPU gfenv
 
-loopNewCPU gfenv = do
-  cpu' <- getCPUTime
-  putStrLn (show ((cpu' - cputime gfenv) `div` 1000000000) ++ " msec")
-  loop $ gfenv {cputime = cpu'}
-
-importInEnv :: PGF -> Options -> [FilePath] -> IO CommandEnv
-importInEnv pgf0 opts files = do
-  pgf1 <- case files of
-    [] -> return pgf0
-    _  -> importGrammar pgf0 opts files
-  let env = CommandEnv pgf1 (allCommands pgf1)
-  putStrLn $ unwords $ "\nLanguages:" : languages pgf1
-  return env
+importInEnv :: GFEnv -> Options -> [FilePath] -> IO GFEnv
+importInEnv gfenv opts files
+    | flag optRetainResource opts =
+        do src <- importSource (sourcegrammar gfenv) opts files
+           return $ gfenv {sourcegrammar = src}
+    | otherwise =
+        do let opts' = addOptions (setOptimization OptCSE False) opts
+               pgf0 = multigrammar (commandenv gfenv)
+           pgf1 <- importGrammar pgf0 opts' files
+           putStrLn $ unwords $ "\nLanguages:" : languages pgf1
+           return $ gfenv { commandenv = mkCommandEnv pgf1 }
 
 welcome = unlines [
   "                              ",
@@ -103,3 +99,6 @@ data GFEnv = GFEnv {
   history    :: [String],
   cputime    :: Integer
   }
+
+emptyGFEnv :: GFEnv
+emptyGFEnv = GFEnv emptyGrammar (mkCommandEnv emptyPGF) [] 0