From 080fcd8dc32e18bf395b73aa6a5f14fd9bcf1958 Mon Sep 17 00:00:00 2001
From: jantuomi <jans.tuomi@gmail.com>
Date: Wed, 27 Jul 2016 17:00:15 +0300
Subject: Implement contains function for checking if element is in a list.

---
 .../core/runtime/builtins/BuiltinManager.java      |  1 +
 .../runtime/builtins/ContainsBuiltinFunction.java  | 44 ++++++++++++++++++++++
 2 files changed, 45 insertions(+)
 create mode 100644 src/main/com/jantuomi/tunkki/core/runtime/builtins/ContainsBuiltinFunction.java

diff --git a/src/main/com/jantuomi/tunkki/core/runtime/builtins/BuiltinManager.java b/src/main/com/jantuomi/tunkki/core/runtime/builtins/BuiltinManager.java
index 2ceefff..94c4fce 100644
--- a/src/main/com/jantuomi/tunkki/core/runtime/builtins/BuiltinManager.java
+++ b/src/main/com/jantuomi/tunkki/core/runtime/builtins/BuiltinManager.java
@@ -24,6 +24,7 @@ public class BuiltinManager {
         builtins.add(new AsIntBuiltinFunction());
         builtins.add(new IncludeBuiltinFunction());
         builtins.add(new ListBuiltinFunction());
+        builtins.add(new ContainsBuiltinFunction());
     }
 
     public List<Function> getBuiltins() {
diff --git a/src/main/com/jantuomi/tunkki/core/runtime/builtins/ContainsBuiltinFunction.java b/src/main/com/jantuomi/tunkki/core/runtime/builtins/ContainsBuiltinFunction.java
new file mode 100644
index 0000000..ec8d6c3
--- /dev/null
+++ b/src/main/com/jantuomi/tunkki/core/runtime/builtins/ContainsBuiltinFunction.java
@@ -0,0 +1,44 @@
+package com.jantuomi.tunkki.core.runtime.builtins;
+
+import com.jantuomi.tunkki.core.parser.datatype.BooleanDataContainer;
+import com.jantuomi.tunkki.core.parser.datatype.DataContainer;
+import com.jantuomi.tunkki.core.parser.datatype.ListDataContainer;
+import com.jantuomi.tunkki.exception.TunkkiError;
+
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * Created by jan on 27.7.2016.
+ */
+public class ContainsBuiltinFunction extends BuiltinFunction {
+    public ContainsBuiltinFunction() {
+        super(Arrays.asList("list", "element"));
+    }
+
+    @Override
+    public DataContainer evaluate(List<DataContainer> params) throws TunkkiError {
+        if (params.size() != 2) {
+            throw new TunkkiError(TunkkiError.ExceptionType.ArgumentError, -1, "Function 'contains' expects two arguments.");
+        }
+
+        if (params.get(0).getType() != DataContainer.Type.List) {
+            throw new TunkkiError(TunkkiError.ExceptionType.ArgumentError, -1, "Function 'contains' expects a list as its first argument.");
+        }
+
+        ListDataContainer list = (ListDataContainer) params.get(0);
+        DataContainer comp = params.get(1);
+
+        for (DataContainer elem : list.getData()) {
+            if (elem.equals(comp)) {
+                return new BooleanDataContainer(true);
+            }
+        }
+        return new BooleanDataContainer(false);
+    }
+
+    @Override
+    public String getName() {
+        return "contains";
+    }
+}
-- 
cgit v1.3