Show function call position in call stack

Author: aNNiMON

ownlang-core/src/main/java/com/annimon/ownlang/lib/CallStack.java

@@ -13,8 +13,16 @@ public static synchronized void clear() {
         calls.clear();
     }
 
-    public static synchronized void enter(String name, Function function) {
-        calls.push(new CallInfo(name, function.toString()));
+    public static synchronized void enter(String name, Function function, String position) {
+        String func = function.toString();
+        if (func.contains("com.annimon.ownlang.modules")) {
+            func = func.replaceAll(
+                    "com.annimon.ownlang.modules.(\\w+)\\.?\\1?", "module $1");
+        }
+        if (func.contains("\n")) {
+            func = func.substring(0, func.indexOf("\n")).trim();
+        }
+        calls.push(new CallInfo(name, func, position));
     }
 
     public static synchronized void exit() {
@@ -25,10 +33,14 @@ public static synchronized Deque<CallInfo> getCalls() {
         return calls;
     }
 
-    public record CallInfo(String name, String function) {
+    public record CallInfo(String name, String function, String position) {
         @Override
         public String toString() {
-            return String.format("%s: %s", name, function);
+            if (position == null) {
+                return String.format("%s: %s", name, function);
+            } else {
+                return String.format("%s: %s %s", name, function, position);
+            }
         }
     }
 }

ownlang-parser/src/main/java/com/annimon/ownlang/parser/Pos.java renamed to ownlang-core/src/main/java/com/annimon/ownlang/util/Pos.java

@@ -1,4 +1,4 @@
-package com.annimon.ownlang.parser;
+package com.annimon.ownlang.util;
 
 public record Pos(int row, int col) {
     public static final Pos UNKNOWN = new Pos(-1, -1);

ownlang-parser/src/main/java/com/annimon/ownlang/parser/Range.java renamed to ownlang-core/src/main/java/com/annimon/ownlang/util/Range.java

@@ -1,4 +1,4 @@
-package com.annimon.ownlang.parser;
+package com.annimon.ownlang.util;
 
 import java.util.Objects;
 
@@ -18,8 +18,10 @@ public boolean isOnSameLine() {
     }
 
     public String format() {
-        if (isOnSameLine()) {
+        if (isEqualPosition())
             return start.format();
+        else if (isOnSameLine()) {
+            return "[%d:%d~%d]".formatted(start.row(), start.col(), end.col());
         } else {
             return start.format() + "..." + end.format();
         }

ownlang-core/src/main/java/com/annimon/ownlang/util/SourceLocation.java

@@ -0,0 +1,13 @@
+package com.annimon.ownlang.util;
+
+public interface SourceLocation {
+
+    default Range getRange() {
+        return null;
+    }
+
+    default String formatRange() {
+        final var range = getRange();
+        return range == null ? "" : range.format();
+    }
+}

ownlang-parser/src/main/java/com/annimon/ownlang/exceptions/BaseParserException.java

@@ -1,6 +1,6 @@
 package com.annimon.ownlang.exceptions;
 
-import com.annimon.ownlang.parser.Range;
+import com.annimon.ownlang.util.Range;
 
 /**
  * Base type for all lexer and parser exceptions

ownlang-parser/src/main/java/com/annimon/ownlang/exceptions/ParseException.java

@@ -1,6 +1,6 @@
 package com.annimon.ownlang.exceptions;
 
-import com.annimon.ownlang.parser.Range;
+import com.annimon.ownlang.util.Range;
 
 /**
  *

ownlang-parser/src/main/java/com/annimon/ownlang/lib/UserDefinedFunction.java

@@ -31,7 +31,7 @@ public String getArgsName(int index) {
     }
 
     @Override
-    public Value execute(Value... values) {
+    public Value execute(Value[] values) {
         final int size = values.length;
         final int requiredArgsCount = arguments.getRequiredArgumentsCount();
         if (size < requiredArgsCount) {

ownlang-parser/src/main/java/com/annimon/ownlang/parser/Lexer.java

@@ -2,6 +2,8 @@
 
 import com.annimon.ownlang.exceptions.OwnLangParserException;
 import com.annimon.ownlang.parser.error.ParseError;
+import com.annimon.ownlang.util.Pos;
+import com.annimon.ownlang.util.Range;
 import java.util.*;
 
 /**

ownlang-parser/src/main/java/com/annimon/ownlang/parser/Parser.java

@@ -8,6 +8,8 @@
 import com.annimon.ownlang.parser.ast.*;
 import com.annimon.ownlang.parser.error.ParseError;
 import com.annimon.ownlang.parser.error.ParseErrors;
+import com.annimon.ownlang.util.Pos;
+import com.annimon.ownlang.util.Range;
 import java.util.*;
 import java.util.function.Function;
 import java.util.stream.Collectors;
@@ -351,12 +353,14 @@ private Expression functionChain(Expression qualifiedNameExpr) {
 
     private FunctionalExpression function(Expression qualifiedNameExpr) {
         // function(arg1, arg2, ...)
+        final var startTokenIndex = index - 1;
         consume(TokenType.LPAREN);
         final FunctionalExpression function = new FunctionalExpression(qualifiedNameExpr);
         while (!match(TokenType.RPAREN)) {
             function.addArgument(expression());
             match(TokenType.COMMA);
         }
+        function.setRange(getRange(startTokenIndex, index - 1));
         return function;
     }
 

ownlang-parser/src/main/java/com/annimon/ownlang/parser/Token.java

@@ -1,5 +1,7 @@
 package com.annimon.ownlang.parser;
 
+import com.annimon.ownlang.util.Pos;
+
 /**
  * @author aNNiMON
  */