last-minute bug fixes

1 files changed, 15 insertions, 3 deletions

diff --git a/src/GF/Source/TestGF.hs b/src/GF/Source/TestGF.hs
index 89700bda6..b236f1347 100644
--- a/src/GF/Source/TestGF.hs
+++ b/src/GF/Source/TestGF.hs
@@ -1,5 +1,9 @@
 -- automatically generated by BNF Converter
-module TestGF where 
+module Main where
+
+
+import IO ( stdin, hGetContents )
+import System ( getArgs, getProgName )
 
 import LexGF
 import ParGF
@@ -13,13 +17,21 @@ type ParseFun a = [Token] -> Err a
 
 myLLexer = myLexer
 
-runFile :: (Print a, Show a) => ParseFun a -> FilePath -> IO()
+runFile :: (Print a, Show a) => ParseFun a -> FilePath -> IO ()
 runFile p f = readFile f >>= run p
 
-run :: (Print a, Show a) => ParseFun a -> String -> IO()
+run :: (Print a, Show a) => ParseFun a -> String -> IO ()
 run p s = case (p (myLLexer s)) of
            Bad s    -> do  putStrLn "\nParse Failed...\n"
                            putStrLn s
            Ok  tree -> do putStrLn "\nParse Successful!"
                           putStrLn $ "\n[Abstract Syntax]\n\n" ++ show tree
                           putStrLn $ "\n[Linearized tree]\n\n" ++ printTree tree
+
+main :: IO ()
+main = do args <- getArgs
+          case args of
+            []  -> hGetContents stdin >>= run pGrammar
+            [f] -> runFile pGrammar f
+            _   -> do progName <- getProgName
+                      putStrLn $ progName ++ ": excess arguments."