piccode: Fixed bugs on the repl and clean up the API

Author: hexaredecimal

src/main/java/org/piccode/backend/Compiler.java

@@ -26,6 +26,8 @@
 import org.piccode.rt.modules.PiccodeSystemModule;
 import org.piccode.rt.modules.PiccodeTimeModule;
 import org.piccode.rt.modules.PiccodeTupleModule;
+import org.piccode.rt.modules.PiccodeTypesModule;
+import org.piccode.rt.modules.PiccodeVirtualModule;
 
 /**
  *
@@ -40,13 +42,7 @@ public static PiccodeValue compile(String file, String code) {
 	public static PiccodeValue compile(String file, String code, List<PiccodeValue> args) {
 		try {
 			var result = program(file, code);
-
-			var scope_id = new IdentifierAst("globalScope");
-			scope_id.column = 0;
-			scope_id.line = 1;
-			scope_id.file = file;
-			
-			prepareGlobalScope(scope_id);
+			prepareGlobalScope(file);
 			addGlobalFunctions();
 
 			PiccodeValue res = new PiccodeUnit();
@@ -59,17 +55,20 @@ public static PiccodeValue compile(String file, String code, List<PiccodeValue>
 			}
 
 			if (has_main) {
-				return new CallAst(new IdentifierAst("main"), List.of()).execute();
+				var _result = new CallAst(new IdentifierAst("main"), List.of()).execute();
+				Context.top.dropStackFrame();
+				return _result;
 			}
 
 			Context.top.dropStackFrame();
 			return res;
 		} catch (PiccodeException e) {
+			Context.top.dropStackFrame();
 			e.reportError();
 			//e.printStackTrace();
 			return new PiccodeUnit();
 		} catch (Exception rte) {
-			//Context.top.dropStackFrame();
+			Context.top.dropStackFrame();
 			rte.printStackTrace();
 			return new PiccodeUnit();
 		}
@@ -102,13 +101,22 @@ private static String sanitizeSourceFile(String code) {
 		return String.join("\n", lines);
 	}
 
-	public static void prepareGlobalScope(Ast parent) {
-		Context.top.pushStackFrame(parent);
-		Context.top.addGlobal("true", new PiccodeBoolean("true"));
+	public static void prepareGlobalScope(String file) {
+		prepareGlobalScope(file, "globalScope");
+	}
+
+	public static void prepareGlobalScope(String file, String globalScopeName) {
+		var scope_id = new IdentifierAst(globalScopeName);
+		scope_id.column = 0;
+		scope_id.line = 1;
+		scope_id.file = file;
+		Context.top.pushStackFrame(scope_id);
+		Context.top.putLocal("true", new PiccodeBoolean("true"));
     Context.top.putLocal("false", new PiccodeBoolean("false"));
 		addGlobalFunctions();
 	}
 
+
 	private static void addGlobalFunctions() {
 		PiccodeIOModule.addFunctions();
 		PiccodeArrayModule.addFunctions();
@@ -117,6 +125,8 @@ private static void addGlobalFunctions() {
 		PiccodeMathModule.addFunctions();
 		PiccodeSystemModule.addFunctions();
 		PiccodeTimeModule.addFunctions();
+		PiccodeTypesModule.addFunctions();
+		PiccodeVirtualModule.addFunctions();
 	}
 
 }

src/main/java/org/piccode/piccodescript/Piccode.java

@@ -35,6 +35,8 @@
 import org.jline.utils.AttributedStyle;
 import org.piccode.ast.IdentifierAst;
 import org.piccode.platf.Platforms;
+import org.piccode.rt.Context;
+import org.piccode.rt.PiccodeException;
 import org.piccode.rt.ReplState;
 
 /**
@@ -43,7 +45,7 @@
  */
 public class Piccode {
 
-	private static double VERSION = 0.2;
+	private static double VERSION = 0.3;
 
 	public static void main(String[] args) {
 		if (Platforms.isWindows()) {
@@ -300,6 +302,8 @@ public AttributedString highlight(LineReader reader, String buffer) {
 
 			var is_inner = false;
 			ReplState.ACTIVE = true;
+
+			Compiler.prepareGlobalScope("repl", "REPL");
 			outer:
 			while (true) {
 				StringBuilder inputBlock = new StringBuilder();
@@ -331,11 +335,25 @@ public AttributedString highlight(LineReader reader, String buffer) {
 					continue;
 				}
 
-				var result = Compiler.compile("repl", code, user_args);
-				terminal.writer().println(result + " : " + result.type());
-				terminal.flush();
+				var result = Compiler.program("repl", code);
+				PiccodeValue res = null;
+				try {
+					for (var stmt : result.nodes) {
+						res = stmt.execute();
+					}
+
+					terminal.writer().println(res + " : " + res.type());
+					terminal.flush();
+				} catch (PiccodeException e) {
+					e.reportError();
+					var stack_size = Context.top.getFramesCount();
+					if (stack_size > 1) {
+						Context.top.dropStackFrame();
+					}
+				}
 			}
 
+			Context.top.dropStackFrame();
 			ReplState.ACTIVE = false;
 			terminal.writer().println("Exiting REPL");
 		} catch (IOException | EndOfFileException | UserInterruptException e) {