1 files changed, 15 insertions, 2 deletions
diff --git a/main/core/parser/ast/FunctionDefineNode.java b/main/core/parser/ast/FunctionDefineNode.java
index e867f78..2402bed 100644
--- a/main/core/parser/ast/FunctionDefineNode.java
+++ b/main/core/parser/ast/FunctionDefineNode.java
@@ -1,8 +1,14 @@
 package com.jantuomi.interpreter.main.core.parser.ast;
 
 import com.jantuomi.interpreter.main.core.parser.datatype.DataContainer;
+import com.jantuomi.interpreter.main.core.parser.datatype.StringDataContainer;
+import com.jantuomi.interpreter.main.core.runtime.Function;
+import com.jantuomi.interpreter.main.core.runtime.State;
 import com.jantuomi.interpreter.main.core.tokenizer.token.Token;
 
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * Created by jan on 17.6.2016.
  */
@@ -26,8 +32,15 @@ public class FunctionDefineNode extends VarargOperatorNode {
 
     @Override
     public DataContainer evaluate() {
-        // TODO add function in State
-        return null;
+        List<String> argumentNames = new ArrayList<>();
+        for (ASTNode arg : args) {
+            SymbolNode argSym = (SymbolNode) arg;
+            argumentNames.add(argSym.getName());
+        }
+        Function function = new Function(argumentNames, (FunctionBodyNode) body);
+        State.getInstance().addFunctionToScope(name, function);
+
+        return new StringDataContainer("function " + name);
     }
 
     @Override