Source location for unknown class/function exceptions

Author: aNNiMON

ownlang-core/src/main/java/com/annimon/ownlang/exceptions/UnknownClassException.java

@@ -1,5 +1,7 @@
 package com.annimon.ownlang.exceptions;
 
+import com.annimon.ownlang.util.Range;
+
 public final class UnknownClassException extends OwnLangRuntimeException {
 
     private final String className;
@@ -9,6 +11,11 @@ public UnknownClassException(String name) {
         this.className = name;
     }
 
+    public UnknownClassException(String name, Range range) {
+        super("Unknown class " + name, range);
+        this.className = name;
+    }
+
     public String getClassName() {
         return className;
     }

ownlang-core/src/main/java/com/annimon/ownlang/exceptions/UnknownFunctionException.java

@@ -1,5 +1,7 @@
 package com.annimon.ownlang.exceptions;
 
+import com.annimon.ownlang.util.Range;
+
 public final class UnknownFunctionException extends OwnLangRuntimeException {
 
     private final String functionName;
@@ -9,6 +11,11 @@ public UnknownFunctionException(String name) {
         this.functionName = name;
     }
 
+    public UnknownFunctionException(String name, Range range) {
+        super("Unknown function " + name, range);
+        this.functionName = name;
+    }
+
     public String getFunctionName() {
         return functionName;
     }

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

@@ -746,15 +746,18 @@ private Expression multiplicative() {
     }
 
     private Expression objectCreation() {
-       if (match(TokenType.NEW)) {
+        if (match(TokenType.NEW)) {
+            final var startTokenIndex = index - 1;
             final String className = consume(TokenType.WORD).text();
             final List<Expression> args = new ArrayList<>();
             consume(TokenType.LPAREN);
             while (!match(TokenType.RPAREN)) {
                 args.add(expression());
                 match(TokenType.COMMA);
             }
-            return new ObjectCreationExpression(className, args);
+            final var expr = new ObjectCreationExpression(className, args);
+            expr.setRange(getRange(startTokenIndex, index - 1));
+            return expr;
         }
 
         return unary();

ownlang-parser/src/main/java/com/annimon/ownlang/parser/ast/FunctionalExpression.java

@@ -75,7 +75,7 @@ private Function getFunction(String key) {
                 return ((FunctionValue)variable).getValue();
             }
         }
-        throw new UnknownFunctionException(key);
+        throw new UnknownFunctionException(key, getRange());
     }
 
     @Override

ownlang-parser/src/main/java/com/annimon/ownlang/parser/ast/ObjectCreationExpression.java

@@ -2,18 +2,30 @@
 
 import com.annimon.ownlang.exceptions.UnknownClassException;
 import com.annimon.ownlang.lib.*;
+import com.annimon.ownlang.util.Range;
+import com.annimon.ownlang.util.SourceLocation;
 import java.util.Iterator;
 import java.util.List;
 
-public final class ObjectCreationExpression implements Expression {
+public final class ObjectCreationExpression implements Expression, SourceLocation {
 
     public final String className;
     public final List<Expression> constructorArguments;
+    private Range range;
 
     public ObjectCreationExpression(String className, List<Expression> constructorArguments) {
         this.className = className;
         this.constructorArguments = constructorArguments;
     }
+
+    public void setRange(Range range) {
+        this.range = range;
+    }
+
+    @Override
+    public Range getRange() {
+        return range;
+    }
 
     @Override
     public Value eval() {
@@ -26,7 +38,7 @@ public Value eval() {
                     return instantiable.newInstance(ctorArgs());
                 }
             }
-            throw new UnknownClassException(className);
+            throw new UnknownClassException(className, getRange());
         }
 
         // Create an instance and put evaluated fields with method declarations