ast: Make functio remember their runtime state when used with '::'

Author: hexaredecimal

src/main/java/org/piccode/ast/CCOperationAst.java

@@ -6,6 +6,7 @@
 import org.piccode.piccodescript.TargetEnvironment;
 import org.piccode.rt.Context;
 import org.piccode.rt.PiccodeArray;
+import org.piccode.rt.PiccodeClosure;
 import org.piccode.rt.PiccodeException;
 import org.piccode.rt.PiccodeModule;
 import org.piccode.rt.PiccodeNumber;
@@ -129,10 +130,18 @@ private PiccodeValue process(IdentifierAst id, PiccodeModule mod, Integer frame)
 				return result;
 			}
 			if (node instanceof FunctionAst func && func.name.equals(_id.text)) {
-				var result = Ast.safeExecute(frame, func, (expr) -> {
-					node.execute(frame);
-					return call.execute(frame);
-				});
+				PiccodeValue result = null;
+				if (func.rtObject == null) {
+					result = Ast.safeExecute(frame, func, (expr) -> {
+						func.setRtObject((PiccodeClosure) node.execute(frame));
+						return call.execute(frame);
+					});
+				} else {
+					result = Ast.safeExecute(frame, func, (expr) -> {
+						ctx.putLocal(func.name, func.rtObject);
+						return call.execute(frame);
+					});
+				}
 				ctx.deleteLocal(func.name);
 				return result;
 			}

src/main/java/org/piccode/ast/FunctionAst.java

@@ -22,13 +22,18 @@ public class FunctionAst extends Ast {
 	public Ast body;
 	public List<ClauseAst> clauses = new ArrayList<>();
 	public List<String> annotations = new ArrayList<>();
+	public PiccodeClosure rtObject = null;
 
 	public FunctionAst(String name, List<Ast> arg, Ast body) {
 		this.name = name;
 		this.arg = arg;
 		this.body = body;
 	}
 
+	public void setRtObject(PiccodeClosure value) {
+		rtObject = value;
+	}
+
 	@Override
 	public String toString() {
 		StringBuilder sb = new StringBuilder();