Trivial solution to identier quotes in SQL Mode ANSI_QUOTES, with CLI tools ansi mode update

Author: smarek

src/Context.php

@@ -423,7 +423,7 @@ public static function isSymbol($str)
 
         if ($str[0] === '@') {
             return Token::FLAG_SYMBOL_VARIABLE;
-        } elseif ($str[0] === '`') {
+        } elseif ($str[0] === self::getIdentifierQuote()) {
             return Token::FLAG_SYMBOL_BACKTICK;
         } elseif ($str[0] === ':' || $str[0] === '?') {
             return Token::FLAG_SYMBOL_PARAMETER;
@@ -450,6 +450,8 @@ public static function isString($str)
 
         if ($str[0] === '\'') {
             return Token::FLAG_STRING_SINGLE_QUOTES;
+        } elseif (self::hasMode(self::SQL_MODE_ANSI_QUOTES) && $str[0] === '"') {
+            return null;
         } elseif ($str[0] === '"') {
             return Token::FLAG_STRING_DOUBLE_QUOTES;
         }
@@ -604,6 +606,29 @@ public static function escape($str, $quote = '`')
 
         return $quote . str_replace($quote, $quote . $quote, $str) . $quote;
     }
+
+    /**
+     * Returns char used to quote identifiers based on currently set SQL Mode (ie. standard or ANSI_QUOTES)
+     * @return string either " (double quote, ansi_quotes mode) or ` (backtick, standard mode)
+     */
+    public static function getIdentifierQuote()
+    {
+        return self::hasMode(self::SQL_MODE_ANSI_QUOTES) ? '"' : '`';
+    }
+
+    /**
+     * Function verifies that given SQL Mode constant is currently set
+     *
+     * @return boolean false on empty param, true/false on given constant/int value
+     * @param $flag int for example Context::SQL_MODE_ANSI_QUOTES
+     */
+    public static function hasMode($flag = null)
+    {
+        if (empty($flag)) {
+            return false;
+        }
+        return (self::$MODE & $flag) === $flag;
+    }
 }
 
 // Initializing the default context.

src/Lexer.php

@@ -917,7 +917,7 @@ public function parseSymbol()
         $str = null;
 
         if ($this->last < $this->len) {
-            if (($str = $this->parseString('`')) === null) {
+            if (($str = $this->parseString(Context::getIdentifierQuote())) === null) {
                 if (($str = $this->parseUnknown()) === null) {
                     $this->error(
                         'Variable name was expected.',

src/Utils/CLI.php

@@ -36,7 +36,7 @@ public function mergeLongOpts(&$params, &$longopts)
 
     public function usageHighlight()
     {
-        echo "Usage: highlight-query --query SQL [--format html|cli|text]\n";
+        echo "Usage: highlight-query --query SQL [--format html|cli|text] [--ansi]\n";
         echo "       cat file.sql | highlight-query\n";
     }
 
@@ -51,9 +51,10 @@ public function parseHighlight()
             'help',
             'query:',
             'format:',
+            'ansi'
         ];
         $params = $this->getopt(
-            'hq:f:',
+            'hq:f:a',
             $longopts
         );
         if ($params === false) {
@@ -93,6 +94,9 @@ public function runHighlight()
             }
         }
 
+        if (isset($params['a'])) {
+            Context::setMode('ANSI_QUOTES');
+        }
         if (isset($params['q'])) {
             echo Formatter::format(
                 $params['q'],
@@ -111,7 +115,7 @@ public function runHighlight()
 
     public function usageLint()
     {
-        echo "Usage: lint-query --query SQL\n";
+        echo "Usage: lint-query --query SQL [--ansi]\n";
         echo "       cat file.sql | lint-query\n";
     }
 
@@ -121,9 +125,10 @@ public function parseLint()
             'help',
             'query:',
             'context:',
+            'ansi'
         ];
         $params = $this->getopt(
-            'hq:c:',
+            'hq:c:a',
             $longopts
         );
         $this->mergeLongOpts($params, $longopts);
@@ -137,38 +142,35 @@ public function runLint()
         if ($params === false) {
             return 1;
         }
-
         if (isset($params['h'])) {
             $this->usageLint();
 
             return 0;
         }
-
         if (isset($params['c'])) {
             Context::load($params['c']);
         }
-
         if (! isset($params['q'])) {
             if ($stdIn = $this->readStdin()) {
                 $params['q'] = $stdIn;
             }
         }
-
+        if (isset($params['a'])) {
+            Context::setMode('ANSI_QUOTES');
+        }
         if (isset($params['q'])) {
             $lexer = new Lexer($params['q'], false);
             $parser = new Parser($lexer->list);
             $errors = Error::get([$lexer, $parser]);
             if (count($errors) === 0) {
                 return 0;
             }
-
             $output = Error::format($errors);
             echo implode("\n", $output);
             echo "\n";
 
             return 10;
         }
-
         echo "ERROR: Missing parameters!\n";
         $this->usageLint();
 
@@ -177,7 +179,7 @@ public function runLint()
 
     public function usageTokenize()
     {
-        echo "Usage: tokenize-query --query SQL\n";
+        echo "Usage: tokenize-query --query SQL [--ansi]\n";
         echo "       cat file.sql | tokenize-query\n";
     }
 
@@ -186,9 +188,10 @@ public function parseTokenize()
         $longopts = [
             'help',
             'query:',
+            'ansi'
         ];
         $params = $this->getopt(
-            'hq:',
+            'hq:a',
             $longopts
         );
         $this->mergeLongOpts($params, $longopts);
@@ -215,6 +218,9 @@ public function runTokenize()
             }
         }
 
+        if (isset($params['a'])) {
+            Context::setMode('ANSI_QUOTES');
+        }
         if (isset($params['q'])) {
             $lexer = new Lexer($params['q'], false);
             foreach ($lexer->list->tokens as $idx => $token) {