Use own theme handling instead Colors\Color

Author: JakubOnderka

example.php

@@ -2,15 +2,9 @@
 use Colors\Color;
 use JakubOnderka\PhpConsoleHighlighter\Highlighter;
 
-$colors = new Color();
-$colors->setTheme(array(
-    Highlighter::TOKEN_STRING => 'red',
-    Highlighter::TOKEN_COMMENT => 'yellow',
-    Highlighter::TOKEN_KEYWORD => 'green',
-    Highlighter::TOKEN_DEFAULT => 'white',
-    Highlighter::TOKEN_HTML => 'cyan'
-));
-$highlighter = new Highlighter($colors);
+require __DIR__ . '/vendor/autoload.php';
+
+$highlighter = new Highlighter(new Color());
 
 $fileContent = file_get_contents(__DIR__ . '/example.php');
 echo $highlighter->getWholeFile($fileContent);

src/JakubOnderka/PhpConsoleHighlighter/Highlighter.php

@@ -14,9 +14,26 @@ class Highlighter
     /** @var Color */
     private $color;
 
-    public function __construct(Color $color)
+    /** @var array */
+    private $theme = array(
+        self::TOKEN_STRING => 'red',
+        self::TOKEN_COMMENT => 'yellow',
+        self::TOKEN_KEYWORD => 'green',
+        self::TOKEN_DEFAULT => 'white',
+        self::TOKEN_HTML => 'cyan'
+    );
+
+    /**
+     * @param Color $color
+     * @param array|null $theme
+     */
+    public function __construct(Color $color, array $theme = null)
     {
         $this->color = $color;
+
+        if ($theme) {
+            $this->setTheme($theme);
+        }
     }
 
     /**
@@ -58,6 +75,22 @@ public function getWholeFile($source)
         return $this->output($tokenLines);
     }
 
+    /**
+     * @return array
+     */
+    public function getTheme()
+    {
+        return $this->theme;
+    }
+
+    /**
+     * @param array $theme
+     */
+    public function setTheme(array $theme)
+    {
+        $this->theme = $theme;
+    }
+
     /**
      * @param string $source
      * @return array
@@ -175,7 +208,12 @@ private function colorLines(array $tokenLines)
         foreach ($tokenLines as $lineCount => $tokenLine) {
             $line = '';
             foreach ($tokenLine as $token) {
-                $line .= $this->color->apply($token[0], $token[1]);
+                list($tokenType, $tokenValue) = $token;
+                if (isset($this->theme[$tokenType])) {
+                    $line .= $this->color->apply($this->theme[$tokenType], $tokenValue);
+                } else {
+                    $line .= $tokenValue;
+                }
             }
             $lines[$lineCount] = $line;
         }