- Add function in and out parameters

2026-09-08 16:06:07 +02:00 · 2022-01-11 19:08:15 +01:00
parent 2a7e8074ba
commit 895eb51c8d
4 changed files with 165 additions and 94 deletions
@@ -0,0 +1,118 @@
+using System;
+
+using YesNt.Interpreter.Attributes;
+using YesNt.Interpreter.Enums;
+using YesNt.Interpreter.Runtime;
+
+namespace YesNt.Interpreter.Statements
+{
+    internal class FunctionStatements : StatementRuntimeInformation
+    {
+        [Statement("fnc", SearchMode.StartOfLine, SpaceAround.End, ConsoleColor.DarkYellow, ExecuteInSearchMode = true)]
+        public void FindFunction(string args)
+        {
+            if (RuntimeInfo.InternalIsInFunction)
+            {
+                RuntimeInfo.Exit("Nested functions are not allowed", true);
+                return;
+            }
+
+            string key = args.Trim();
+            if (RuntimeInfo.Functions.ContainsKey(key))
+            {
+                RuntimeInfo.Functions[key] = RuntimeInfo.LineNumber;
+            }
+            else
+            {
+                RuntimeInfo.Functions.Add(key, RuntimeInfo.LineNumber);
+            }
+
+            if (!string.IsNullOrWhiteSpace(RuntimeInfo.SearchFunction) && RuntimeInfo.SearchFunction == key)
+            {
+                RuntimeInfo.SearchFunction = string.Empty;
+            }
+
+            RuntimeInfo.IsInFunction = true;
+        }
+
+        [Statement("in", SearchMode.StartOfLine, SpaceAround.End, ConsoleColor.DarkYellow)]
+        public void AddInParameter(string args)
+        {
+            RuntimeInfo.InParametersStack.Push(args);
+        }
+
+        [Statement("out", SearchMode.StartOfLine, SpaceAround.End, ConsoleColor.DarkYellow)]
+        public void GetOutParameter(string name)
+        {
+            if (RuntimeInfo.OutParametersStack.Count == 0)
+            {
+                RuntimeInfo.Exit("No out parameter in stack", true);
+                return;
+            }
+
+            RuntimeInfo.Variables.Add(name, RuntimeInfo.OutParametersStack.Pop());
+        }
+
+        [Statement("get", SearchMode.StartOfLine, SpaceAround.End, ConsoleColor.DarkYellow)]
+        public void GetInParameter(string args)
+        {
+            if (RuntimeInfo.FunctionCallStack.Count == 0)
+            {
+                RuntimeInfo.Exit("No function in stack", true);
+                return;
+            }
+
+            if (RuntimeInfo.FunctionCallStack.Peek().Arguemtns.Count == 0)
+            {
+                RuntimeInfo.Exit("No in parameter in stack", true);
+                return;
+            }
+
+            RuntimeInfo.Variables.TryAdd(args, RuntimeInfo.FunctionCallStack.Peek().Arguemtns.Pop());
+        }
+
+        [Statement("put", SearchMode.StartOfLine, SpaceAround.End, ConsoleColor.DarkYellow)]
+        public void AddOutParameter(string args)
+        {
+            if (RuntimeInfo.FunctionCallStack.Count == 0)
+            {
+                RuntimeInfo.Exit("No function in stack", true);
+                return;
+            }
+
+            RuntimeInfo.FunctionCallStack.Peek().Results.Push(args);
+        }
+
+        [Statement("ret", SearchMode.Exact, SpaceAround.None, ConsoleColor.DarkYellow, ExecuteInSearchMode = true)]
+        public void Return(string _)
+        {
+            if (!RuntimeInfo.IsInFunction)
+            {
+                RuntimeInfo.Exit("Not in function", true);
+                return;
+            }
+
+            if (RuntimeInfo.IsSearching)
+            {
+                RuntimeInfo.IsInFunction = false;
+                return;
+            }
+            else
+            {
+                RuntimeInfo.IsInFunction = false;
+            }
+
+            if (RuntimeInfo.FunctionCallStack.Count > 0)
+            {
+                FunctionScope functionScope = RuntimeInfo.FunctionCallStack.Pop();
+
+                RuntimeInfo.OutParametersStack = functionScope.Results;
+                RuntimeInfo.LineNumber = functionScope.CallerLine;
+            }
+            else
+            {
+                RuntimeInfo.Exit("No function in stack", true);
+            }
+        }
+    }
+}