Support reading from stdin

phpmyadmin · Apr 5, 2019 · a215a30 · a215a30
1 parent d500a21
commit a215a30
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/src/Utils/CLI.php b/src/Utils/CLI.php
@@ -77,6 +77,11 @@ public function runHighlight()
 
             return 0;
         }
+        if (!isset($params['q'])) {
+            if ($stdIn = $this->readStdin()) {
+                $params['q'] = $stdIn;
+            }
+        }
         if (isset($params['q'])) {
             echo Formatter::format(
                 $params['q'],
@@ -127,6 +132,11 @@ public function runLint()
         if (isset($params['c'])) {
             Context::load($params['c']);
         }
+        if (!isset($params['q'])) {
+            if ($stdIn = $this->readStdin()) {
+                $params['q'] = $stdIn;
+            }
+        }
         if (isset($params['q'])) {
             $lexer = new Lexer($params['q'], false);
             $parser = new Parser($lexer->list);
@@ -177,6 +187,11 @@ public function runTokenize()
 
             return 0;
         }
+        if (!isset($params['q'])) {
+            if ($stdIn = $this->readStdin()) {
+                $params['q'] = $stdIn;
+            }
+        }
         if (isset($params['q'])) {
             $lexer = new Lexer($params['q'], false);
             foreach ($lexer->list->tokens as $idx => $token) {
@@ -199,4 +214,12 @@ public function runTokenize()
 
         return 1;
     }
+
+    private function readStdin() {
+        stream_set_blocking(STDIN, false);
+        $stdin = stream_get_contents(STDIN);
+        // restore-default block-mode setting
+        stream_set_blocking(STDIN, true);
+        return $stdin;
+    }
 }