From 765d25857b0472e3f9df89ac822ad36ea0502466 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Tue, 6 Dec 2022 17:50:19 +0200
Subject: Rename stdlib -> core, allow import without .milch suffix

---
 src/Interpreter.hs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/Interpreter.hs')

diff --git a/src/Interpreter.hs b/src/Interpreter.hs
index de77fc9..f9e00b9 100644
--- a/src/Interpreter.hs
+++ b/src/Interpreter.hs
@@ -217,7 +217,10 @@ evaluateImport asts = do
     case args of
         -- non-qualified import
         [AST { astNode = ASTString path }] -> do
-            LState { stateEnv = evaledRawEnv } <- lift $ execStateT (runScriptFile path) initialState
+            let checkedPath = if (not $ ".milch" `L.isSuffixOf` path)
+                then (path ++ ".milch")
+                else path
+            LState { stateEnv = evaledRawEnv } <- lift $ execStateT (runScriptFile checkedPath) initialState
             let exportedEnvMap = envThis evaledRawEnv
 
             env <- getEnv
-- 
cgit v1.3