Initial JS runtime and Arborix Implementation

src/Main.hs

@@ -1,6 +1,6 @@
 module Main where
 
-import ContentStore (initContentStore, termNames, hashToTerm, parseNameList)
+import ContentStore (getContentStorePath, initContentStore, termNames, hashToTerm, loadEnvironment, parseNameList)
 import Eval                   (evalTricu, mainResult, result)
 import FileEval
 import Parser                 (parseTricu)
@@ -16,6 +16,7 @@ import qualified Data.Text    as T
 import Data.Version           (showVersion)
 import Paths_tricu            (version)
 import System.Console.CmdArgs
+import System.Environment     (lookupEnv)
 import System.IO              (hPutStrLn, stderr)
 import System.Exit            (die)
 import Text.Megaparsec        ()
@@ -30,6 +31,7 @@ data TricuArgs
   = Repl
   | Evaluate { file :: [FilePath], form :: EvaluatedForm }
   | TDecode { file :: [FilePath] }
+  | Compile { inputFile :: FilePath, outFile :: FilePath, exportNameOpt :: String }
   | Export { hash :: String, exportNameOpt :: String, outFile :: FilePath }
   | Import { inFile :: FilePath }
   deriving (Show, Data, Typeable)
@@ -86,10 +88,23 @@ importMode = Import
   &= explicit
   &= name "import"
 
+compileMode :: TricuArgs
+compileMode = Compile
+  { inputFile = def &= help "Path to the tricu source file (.tri) to compile."
+                  &= name "f" &= typ "FILE"
+  , outFile = def &= help "Output bundle file path (.tri.bundle)."
+                &= name "o" &= typ "FILE"
+  , exportNameOpt = def &= help "Definition name to use as the bundle root. Defaults to 'main'."
+                      &= name "x" &= typ "NAME"
+  }
+  &= help "Compile a tricu source file into a standalone Arborix portable bundle."
+  &= explicit
+  &= name "compile"
+
 main :: IO ()
 main = do
   let versionStr = "tricu Evaluator and REPL " ++ showVersion version
-  cmdArgsParsed <- cmdArgs $ modes [replMode, evaluateMode, decodeMode, exportMode, importMode]
+  cmdArgsParsed <- cmdArgs $ modes [replMode, evaluateMode, decodeMode, compileMode, exportMode, importMode]
     &= help "tricu: Exploring Tree Calculus"
     &= program "tricu"
     &= summary versionStr
@@ -100,10 +115,26 @@ main = do
       putStrLn "You may exit with `CTRL+D` or the `!exit` command."
       repl
     Evaluate { file = filePaths, form = outputForm } -> do
+      maybeDbPath <- lookupEnv "TRICU_DB_PATH"
       evalResult <- case filePaths of
-        [] -> runTricuT <$> getContents
+        [] -> do
+            initialEnv <- case maybeDbPath of
+              Just dbPath -> do
+                conn <- initContentStore
+                env <- loadEnvironment conn
+                close conn
+                return env
+              Nothing -> return Map.empty
+            input <- getContents
+            pure $ runTricuTEnv initialEnv input
         (filePath:restFilePaths) -> do
-            initialEnv <- evaluateFile filePath
+            initialEnv <- case maybeDbPath of
+              Just _ -> do
+                conn <- initContentStore
+                env <- loadEnvironment conn
+                close conn
+                return env
+              Nothing -> return Map.empty
             finalEnv <- foldM evaluateFileWithContext initialEnv restFilePaths
             pure $ mainResult finalEnv
       let fRes = formatT outputForm evalResult
@@ -128,6 +159,9 @@ main = do
       putStrLn $ "Imported " ++ show (length roots) ++ " root(s):"
       mapM_ (\r -> putStrLn $ "  " ++ unpack r) roots
       close conn
+    Compile { inputFile = inputFile', outFile = outFile', exportNameOpt = exportNameArg } ->
+      let exportName = if null exportNameArg then Nothing else Just (T.pack exportNameArg)
+      in compileFile inputFile' outFile' exportName
 
 runTricu :: String -> String
 runTricu = formatT TreeCalculus . runTricuT