From 9f0fef419afd8d7bf4213dafbd8e0ff79eb2e999 Mon Sep 17 00:00:00 2001
From: Sean Molenaar <sean@m2mobi.com>
Date: Wed, 4 Oct 2017 23:55:15 +0200
Subject: [PATCH] Out: Add material color picker

---
 src/PHPDraft/Out/HTML/default.phtml    |  2 +-
 src/PHPDraft/Out/HTML/material.phtml   |  4 ++--
 src/PHPDraft/Out/TemplateGenerator.php |  9 +++++---
 src/PHPDraft/Out/UI.php                | 32 ++++++++++++++++++++++----
 tests/statics/drafter/html_material    |  2 +-
 5 files changed, 37 insertions(+), 12 deletions(-)

diff --git a/src/PHPDraft/Out/HTML/default.phtml b/src/PHPDraft/Out/HTML/default.phtml
index ae52b288..826686e4 100644
--- a/src/PHPDraft/Out/HTML/default.phtml
+++ b/src/PHPDraft/Out/HTML/default.phtml
@@ -312,7 +312,7 @@ use Enjoy\HttpStatusCode\Statuscodes;
     </div>
 </div>
 <?php $extras = array_filter($this->base_data, function ($value) {
-    return !in_array($value, ['HOST', 'TITLE', 'ALT_HOST', 'FORMAT', 'DESC']);
+    return !in_array($value, ['HOST', 'TITLE', 'ALT_HOST', 'FORMAT', 'DESC', 'COLOR_1', 'COLOR_2']);
 }, ARRAY_FILTER_USE_KEY);
 if (!empty($extras)):
     $extras['host'] = $this->base_data['HOST']; ?>
diff --git a/src/PHPDraft/Out/HTML/material.phtml b/src/PHPDraft/Out/HTML/material.phtml
index fb527e43..b208baf2 100644
--- a/src/PHPDraft/Out/HTML/material.phtml
+++ b/src/PHPDraft/Out/HTML/material.phtml
@@ -14,7 +14,7 @@ use Enjoy\HttpStatusCode\Statuscodes;
     <title><?= $this->base_data['TITLE']; ?></title>
     <meta name="viewport" content="width=device-width, initial-scale=1">
     <link rel="stylesheet" href="https://fonts.googleapis.com/icon?family=Material+Icons">
-    <link rel="stylesheet" href="https://code.getmdl.io/1.3.0/material.blue_grey-blue.min.css" />
+    <link rel="stylesheet" href="https://code.getmdl.io/1.3.0/material.<?= $this->base_data['COLOR_1']?>-<?= $this->base_data['COLOR_2']?>.min.css" />
     <script defer src="https://code.getmdl.io/1.3.0/material.min.js"></script>
     <script src="https://code.jquery.com/jquery-2.2.4.min.js"
             integrity="sha256-BbhdlvQf/xTY9gja0Dq3HiwQF8LaCRTXxZKRutelT44="
@@ -237,7 +237,7 @@ use Enjoy\HttpStatusCode\Statuscodes;
     </main>
 </div>
 <?php $extras = array_filter($this->base_data, function ($value) {
-    return !in_array($value, ['HOST', 'TITLE', 'ALT_HOST', 'FORMAT', 'DESC']);
+    return !in_array($value, ['HOST', 'TITLE', 'ALT_HOST', 'FORMAT', 'DESC', 'COLOR_1', 'COLOR_2']);
 }, ARRAY_FILTER_USE_KEY);
 if (!empty($extras)):
     $extras['host'] = $this->base_data['HOST']; ?>
diff --git a/src/PHPDraft/Out/TemplateGenerator.php b/src/PHPDraft/Out/TemplateGenerator.php
index 01bceeaf..f9f44b94 100644
--- a/src/PHPDraft/Out/TemplateGenerator.php
+++ b/src/PHPDraft/Out/TemplateGenerator.php
@@ -79,9 +79,12 @@ class TemplateGenerator
      */
     public function __construct($template, $image)
     {
-        $this->template    = $template;
-        $this->image       = $image;
-        $this->http_status = new Httpstatus();
+        $template_parts             = explode('__', $template);
+        $this->template             = $template_parts[0];
+        $this->base_data['COLOR_1'] = isset($template_parts[1]) ? $template_parts[1] : 'green';
+        $this->base_data['COLOR_2'] = isset($template_parts[2]) ? $template_parts[2] : 'light_green';
+        $this->image                = $image;
+        $this->http_status          = new Httpstatus();
     }
 
     /**
diff --git a/src/PHPDraft/Out/UI.php b/src/PHPDraft/Out/UI.php
index 9e5f8ced..75f4b8d5 100644
--- a/src/PHPDraft/Out/UI.php
+++ b/src/PHPDraft/Out/UI.php
@@ -86,14 +86,19 @@ public static function main($argv = [])
             define('DRAFTER_ONLINE_MODE', 1);
         }
 
-        $template = (isset($options['t']) && $options['t']) ? $options['t'] : 'default';
-        $image    = (isset($options['i']) && $options['i']) ? $options['i'] : NULL;
-        $css      = (isset($options['c']) && $options['c']) ? $options['i'] : NULL;
-        $js       = (isset($options['j']) && $options['j']) ? $options['i'] : NULL;
+        $template = (new self())->var_or_default($options['t'], 'default');
+        $image    = (new self())->var_or_default($options['i']);
+        $css      = (new self())->var_or_default($options['c']);
+        $js       = (new self())->var_or_default($options['j']);
+        $color1   = getenv('COLOR_PRIMARY') === FALSE ? NULL : getenv('COLOR_PRIMARY');
+        $color1   = (new self())->var_or_default($color1);
+        $color2   = getenv('COLOR_SECONDARY') === FALSE ? NULL : getenv('COLOR_SECONDARY');
+        $color2   = (new self())->var_or_default($color2);
+        $colors   = (is_null($color1) || is_null($color2)) ? '' : '__' . $color1 . '__' . $color2;
 
         return [
             'file'     => $file,
-            'template' => $template,
+            'template' => $template . $colors,
             'image'    => $image,
             'css'      => $css,
             'js'       => $js,
@@ -119,6 +124,23 @@ public function help()
         echo "\t-h\tDisplays this text." . PHP_EOL;
     }
 
+    /**
+     * Check if a variable exists, otherwise return a default.
+     *
+     * @param mixed $var
+     * @param mixed $default
+     *
+     * @return mixed
+     */
+    private function var_or_default(&$var, $default = NULL)
+    {
+        if (!isset($var) || is_null($var)) {
+            return $default;
+        }
+
+        return $var;
+    }
+
     /**
      * Return the version.
      *
diff --git a/tests/statics/drafter/html_material b/tests/statics/drafter/html_material
index 4af10134..eec5c051 100644
--- a/tests/statics/drafter/html_material
+++ b/tests/statics/drafter/html_material
@@ -4,7 +4,7 @@
     <title>Tesla Model S JSON API</title>
     <meta name="viewport" content="width=device-width, initial-scale=1">
     <link rel="stylesheet" href="https://fonts.googleapis.com/icon?family=Material+Icons">
-    <link rel="stylesheet" href="https://code.getmdl.io/1.3.0/material.blue_grey-blue.min.css" />
+    <link rel="stylesheet" href="https://code.getmdl.io/1.3.0/material.green-light_green.min.css" />
     <script defer src="https://code.getmdl.io/1.3.0/material.min.js"></script>
     <script src="https://code.jquery.com/jquery-2.2.4.min.js"
             integrity="sha256-BbhdlvQf/xTY9gja0Dq3HiwQF8LaCRTXxZKRutelT44="