1 files changed, 26 insertions, 14 deletions
diff --git a/src/NXP/Classes/TokenFactory.php b/src/NXP/Classes/TokenFactory.php
index 6956f31..74b5789 100644
--- a/src/NXP/Classes/TokenFactory.php
+++ b/src/NXP/Classes/TokenFactory.php
@@ -16,14 +16,15 @@ use NXP\Classes\Token\TokenFunction;
 use NXP\Classes\Token\TokenLeftBracket;
 use NXP\Classes\Token\TokenNumber;
 use NXP\Classes\Token\TokenRightBracket;
+use NXP\Classes\Token\TokenStringSingleQuoted;
 use NXP\Classes\Token\TokenVariable;
-use NXP\Classes\Token\TokenString;
+use NXP\Classes\Token\TokenStringDoubleQuoted;
 use NXP\Exception\UnknownFunctionException;
 use NXP\Exception\UnknownOperatorException;
 use NXP\Exception\UnknownTokenException;
 
 /**
- * @author Alexander Kiryukhin <alexander@symdev.org>
+ * @author Alexander Kiryukhin <a.kiryukhin@mail.ru>
  */
 class TokenFactory
 {
@@ -50,12 +51,17 @@ class TokenFactory
 
     /**
      * Add function
-     * @param string   $name
+     * @param string $name
      * @param callable $function
-     * @param int      $places
+     * @param int $places
+     * @throws \ReflectionException
      */
-    public function addFunction($name, callable $function, $places = 1)
+    public function addFunction($name, callable $function, $places = null)
     {
+        if ($places === null) {
+            $reflector = new \ReflectionFunction($function);
+            $places = $reflector->getNumberOfParameters();
+        }
         $this->functions[$name] = [$places, $function];
     }
 
@@ -72,8 +78,9 @@ class TokenFactory
 
     /**
      * Add operator
-     * @param  string                                  $operatorClass
-     * @throws \NXP\Exception\UnknownOperatorException
+     * @param  string $operatorClass
+     * @throws UnknownOperatorException
+     * @throws \ReflectionException
      */
     public function addOperator($operatorClass)
     {
@@ -83,8 +90,7 @@ class TokenFactory
             throw new UnknownOperatorException($operatorClass);
         }
 
-        $this->operators[] = $operatorClass;
-        $this->operators = array_unique($this->operators);
+        $this->operators[$operatorClass::getRegex()] = $operatorClass;
     }
 
     /**
@@ -131,9 +137,10 @@ class TokenFactory
         }
 
         return sprintf(
-            '/(%s)|(%s)|([%s])|(%s)|(%s)|([%s%s%s])/i',
+            '/(%s)|(%s)|(%s)|([%s])|(%s)|(%s)|([%s%s%s])/i',
             TokenNumber::getRegex(),
-            TokenString::getRegex(),
+            TokenStringDoubleQuoted::getRegex(),
+            TokenStringSingleQuoted::getRegex(),
             $operatorsRegex,
             TokenFunction::getRegex(),
             TokenVariable::getRegex(),
@@ -144,9 +151,10 @@ class TokenFactory
     }
 
     /**
-     * @param  string                $token
+     * @param  string $token
      * @return InterfaceToken
      * @throws UnknownTokenException
+     * @throws UnknownFunctionException
      */
     public function createToken($token)
     {
@@ -163,7 +171,11 @@ class TokenFactory
         }
 
         if ($token[0] == '"') {
-            return new TokenString(str_replace('"', '', $token));
+            return new TokenStringDoubleQuoted(str_replace('"', '', $token));
+        }
+
+        if ($token[0] == "'") {
+            return new TokenStringSingleQuoted(str_replace("'", '', $token));
         }
 
         if ($token == ',') {
@@ -180,7 +192,7 @@ class TokenFactory
 
         $regex = sprintf('/%s/i', TokenVariable::getRegex());
         if (preg_match($regex, $token)) {
-            return new TokenVariable(substr($token,1));
+            return new TokenVariable(substr($token, 1));
         }
 
         $regex = sprintf('/%s/i', TokenFunction::getRegex());