Add in PHP opening tag

Author: AydinHassan

README.md

@@ -101,3 +101,11 @@ Every keyword inside each type will be coloured by that colour.
  * `(`
  * `)`
 
+#### TYPE_CLASS
+  
+ * `SyntaxHighlighterConfig`
+ 
+#### TYPE_OPEN_TAG
+ 
+  * `<?php`
+ 

src/SyntaxHighlightPrinter.php

@@ -46,6 +46,23 @@ public function __construct(
         $this->config = $config;
     }
 
+    /**
+     * Pretty prints a file of statements (includes the opening <?php tag if it is required).
+     *
+     * @param Node[] $stmts Array of statements
+     *
+     * @return string Pretty printed statements
+     */
+    public function prettyPrintFile(array $stmts) {
+        $p = rtrim($this->prettyPrint($stmts));
+        $p = preg_replace('/^\?>\n?/', '', $p, -1, $count);
+        $p = preg_replace('/<\?php$/', '', $p);
+        if (!$count) {
+            $p = sprintf("%s\n\n%s", $this->color('<?php', SyntaxHighlighterConfig::TYPE_OPEN_TAG), $p);
+        }
+        return $p;
+    }
+
     /**
      * @param Stmt\Echo_ $node
      *

src/SyntaxHighlighter.php

@@ -17,6 +17,7 @@ class SyntaxHighlighter
      * @var \PhpParser\Parser
      */
     private $parser;
+
     /**
      * @var SyntaxHighlightPrinter
      */
@@ -47,6 +48,6 @@ public function highlight($code)
             throw new \InvalidArgumentException('PHP could not be parsed');
         }
 
-        return $this->printer->prettyPrint($statements);
+        return $this->printer->prettyPrintFile($statements);
     }
 }

src/SyntaxHighlighterConfig.php

@@ -18,6 +18,7 @@ class SyntaxHighlighterConfig
     const TYPE_CALL_PARENTHESIS = 'call_parenthesis';
     const TYPE_LHS              = 'lhs';
     const TYPE_CLASS            = 'class';
+    const TYPE_OPEN_TAG         = 'open_tag';
 
     /**
      * @var array
@@ -32,6 +33,7 @@ class SyntaxHighlighterConfig
         self::TYPE_CALL_PARENTHESIS,
         self::TYPE_LHS,
         self::TYPE_CLASS,
+        self::TYPE_OPEN_TAG,
     ];
 
     /**
@@ -47,6 +49,7 @@ class SyntaxHighlighterConfig
         self::TYPE_CALL_PARENTHESIS => Colours::LIGHT_GRAY,
         self::TYPE_LHS              => Colours::YELLOW,
         self::TYPE_CLASS            => Colours::LIGHT_GRAY,
+        self::TYPE_OPEN_TAG         => Colours::CYAN,
     ];
 
     /**

test/SyntaxHighlighterTest.php

@@ -42,7 +42,7 @@ public function testSyntaxHighlighter()
     {
         $code = '<?php echo "hello world!";';
         $highlighter = $this->getHighlighter();
-        $expected = "[33mecho[0m [32m'hello world!'[0m;";
+        $expected = "[36m<?php[0m\n\n[33mecho[0m [32m'hello world!'[0m;";
         $this->assertEquals($expected, $highlighter->highlight($code));
     }