Add builtin function 'as_int'

2 files changed, 56 insertions, 0 deletions

diff --git a/main/core/runtime/builtins/AsIntBuiltinFunction.java b/main/core/runtime/builtins/AsIntBuiltinFunction.java
new file mode 100644
index 0000000..3e1c71c
--- /dev/null
+++ b/main/core/runtime/builtins/AsIntBuiltinFunction.java
@@ -0,0 +1,55 @@
+package com.jantuomi.interpreter.main.core.runtime.builtins;
+
+import com.jantuomi.interpreter.main.core.parser.datatype.BooleanDataContainer;
+import com.jantuomi.interpreter.main.core.parser.datatype.DataContainer;
+import com.jantuomi.interpreter.main.core.parser.datatype.IntegerDataContainer;
+import com.jantuomi.interpreter.main.core.parser.datatype.StringDataContainer;
+import com.jantuomi.interpreter.main.core.runtime.Function;
+import com.jantuomi.interpreter.main.exception.ExceptionManager;
+import com.jantuomi.interpreter.main.exception.InterpreterException;
+
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * Created by jan on 21.6.2016.
+ */
+public class AsIntBuiltinFunction extends Function {
+    public AsIntBuiltinFunction() {
+        super(
+                Arrays.asList("expression"),
+                null
+        );
+    }
+
+    @Override
+    public String getName() {
+        return "as_int";
+    }
+
+    @Override
+    public DataContainer evaluate(List<DataContainer> params) throws InterpreterException {
+        if (params.size() != 1) {
+            ExceptionManager.raise(InterpreterException.ExceptionType.SyntaxError, -1, "amount of parameters to as_int");
+        }
+        DataContainer param = params.get(0);
+        int i = 0;
+        if (param instanceof StringDataContainer) {
+            try {
+                i = Integer.parseInt(((StringDataContainer) param).getData());
+            }
+            catch (NumberFormatException ex) {
+                ExceptionManager.raise(InterpreterException.ExceptionType.SyntaxError, -1,
+                        String.format("parameter %s to as_int", param.toString()));
+                return null;
+            }
+        }
+        else if (param instanceof IntegerDataContainer) {
+            i = ((IntegerDataContainer) param).getData();
+        }
+        else if (param instanceof BooleanDataContainer) {
+            i = ((BooleanDataContainer) param).getData() ? 1 : 0;
+        }
+        return new IntegerDataContainer(i);
+    }
+}
diff --git a/main/core/runtime/builtins/BuiltinManager.java b/main/core/runtime/builtins/BuiltinManager.java
index a16d8e9..c6c9e26 100644
--- a/main/core/runtime/builtins/BuiltinManager.java
+++ b/main/core/runtime/builtins/BuiltinManager.java
@@ -21,6 +21,7 @@ public class BuiltinManager {
         builtins.add(new OutBuiltinFunction());
         builtins.add(new InBuiltinFunction());
         builtins.add(new ConcatBuiltinFunction());
+        builtins.add(new AsIntBuiltinFunction());
     }
 
     public List<Function> getBuiltins() {