From 4004703cb65f96a5c9dc16f4bf1660c8ee54de98 Mon Sep 17 00:00:00 2001
From: Victor <melnikukr@mail.ru>
Date: Fri, 8 Jan 2016 21:32:20 +0200
Subject: [PATCH] =?UTF-8?q?=D0=98=D1=81=D0=BF=D1=80=D0=B0=D0=B2=D0=BB?=
 =?UTF-8?q?=D0=B5=D0=BD=20=D0=B2=D1=8B=D0=B7=D0=BE=D0=B2=20=D0=BF=D0=BE?=
 =?UTF-8?q?=D0=BB=D1=8C=D0=B7=D0=BE=D0=B2=D0=B0=D1=82=D0=B5=D0=BB=D1=8C?=
 =?UTF-8?q?=D1=81=D0=BA=D0=BE=D0=B9=20=D1=84=D1=83=D0=BD=D0=BA=D1=86=D0=B8?=
 =?UTF-8?q?=D0=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../annimon/ownlang/lib/UserDefinedFunction.java | 11 ++++++++++-
 .../ownlang/parser/ast/FunctionalExpression.java | 16 +---------------
 2 files changed, 11 insertions(+), 16 deletions(-)

diff --git a/src/com/annimon/ownlang/lib/UserDefinedFunction.java b/src/com/annimon/ownlang/lib/UserDefinedFunction.java
index d8686ee..fabe27b 100644
--- a/src/com/annimon/ownlang/lib/UserDefinedFunction.java
+++ b/src/com/annimon/ownlang/lib/UserDefinedFunction.java
@@ -28,12 +28,21 @@ public final class UserDefinedFunction implements Function {
     }
 
     @Override
-    public Value execute(Value... args) {
+    public Value execute(Value... values) {
+        final int size = values.length;
+        if (size != getArgsCount()) throw new RuntimeException("Args count mismatch");
+
         try {
+            Variables.push();
+            for (int i = 0; i < size; i++) {
+                Variables.set(getArgsName(i), values[i]);
+            }
             body.execute();
             return NumberValue.ZERO;
         } catch (ReturnStatement rt) {
             return rt.getResult();
+        } finally {
+            Variables.pop();
         }
     }
 
diff --git a/src/com/annimon/ownlang/parser/ast/FunctionalExpression.java b/src/com/annimon/ownlang/parser/ast/FunctionalExpression.java
index 554017a..f86681e 100644
--- a/src/com/annimon/ownlang/parser/ast/FunctionalExpression.java
+++ b/src/com/annimon/ownlang/parser/ast/FunctionalExpression.java
@@ -40,21 +40,7 @@ public final class FunctionalExpression implements Expression {
         for (int i = 0; i < size; i++) {
             values[i] = arguments.get(i).eval();
         }
-        
-        final Function function = getFunction(name);
-        if (function instanceof UserDefinedFunction) {
-            final UserDefinedFunction userFunction = (UserDefinedFunction) function;
-            if (size != userFunction.getArgsCount()) throw new RuntimeException("Args count mismatch");
-            
-            Variables.push();
-            for (int i = 0; i < size; i++) {
-                Variables.set(userFunction.getArgsName(i), values[i]);
-            }
-            final Value result = userFunction.execute(values);
-            Variables.pop();
-            return result;
-        }
-        return function.execute(values);
+        return getFunction(name).execute(values);
     }
     
     private Function getFunction(String key) {