From 2913c698a573ccf77f0e8c171d49d5972568aa70 Mon Sep 17 00:00:00 2001
From: Glynn Forrest <me@glynnforrest.com>
Date: Thu, 23 Apr 2015 23:50:03 +0100
Subject: [PATCH] Add support for 3 character HTML colors

---
 library/ZendPdf/Color/Html.php   | 20 ++++++++++----
 tests/ZendPdf/Color/HtmlTest.php | 46 ++++++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+), 5 deletions(-)
 create mode 100644 tests/ZendPdf/Color/HtmlTest.php

diff --git a/library/ZendPdf/Color/Html.php b/library/ZendPdf/Color/Html.php
index 94bf783..a9839e4 100644
--- a/library/ZendPdf/Color/Html.php
+++ b/library/ZendPdf/Color/Html.php
@@ -72,16 +72,26 @@ public function getComponents()
      * Creates a ColorInterface object from the HTML representation.
      *
      * @param string $color May either be a hexidecimal number of the form
-     *    #rrggbb or one of the 140 well-known names (black, white, blue, etc.)
+     *    #rrggbb, #rgb or one of the 140 well-known names (black, white, blue, etc.)
      * @return ColorInterface
      */
     public static function color($color)
     {
-        $pattern = '/^#([A-Fa-f0-9]{2})([A-Fa-f0-9]{2})([A-Fa-f0-9]{2})$/';
+        $pattern = '/^#([A-Fa-f0-9]{2})([A-Fa-f0-9]{2})([A-Fa-f0-9]{2})$|^#([A-Fa-f0-9])([A-Fa-f0-9])([A-Fa-f0-9])$/';
         if (preg_match($pattern, $color, $matches)) {
-            $r = round((hexdec($matches[1]) / 255), 3);
-            $g = round((hexdec($matches[2]) / 255), 3);
-            $b = round((hexdec($matches[3]) / 255), 3);
+            if (strlen($matches[1]) === 2) {
+                $r = $matches[1];
+                $g = $matches[2];
+                $b = $matches[3];
+            } else {
+                $r = $matches[4].$matches[4];
+                $g = $matches[5].$matches[5];
+                $b = $matches[6].$matches[6];
+            }
+            $r = round((hexdec($r) / 255), 3);
+            $g = round((hexdec($g) / 255), 3);
+            $b = round((hexdec($b) / 255), 3);
+
             if (($r == $g) && ($g == $b)) {
                 return new GrayScale($r);
             } else {
diff --git a/tests/ZendPdf/Color/HtmlTest.php b/tests/ZendPdf/Color/HtmlTest.php
new file mode 100644
index 0000000..825c338
--- /dev/null
+++ b/tests/ZendPdf/Color/HtmlTest.php
@@ -0,0 +1,46 @@
+<?php
+/**
+ * Zend Framework (http://framework.zend.com/)
+ *
+ * @link      http://github.com/zendframework/zf2 for the canonical source repository
+ * @copyright Copyright (c) 2005-2015 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license   http://framework.zend.com/license/new-bsd New BSD License
+ * @package   Zend_Pdf
+ */
+
+namespace ZendPdfTest\Color;
+
+use ZendPdf\Color\Html;
+
+/**
+ * @category   Zend
+ * @package    Zend_PDF
+ * @subpackage UnitTests
+ * @group      Zend_PDF
+ */
+class HtmlTest extends \PHPUnit_Framework_TestCase
+{
+    public function colorProvider()
+    {
+        return array(
+            array('#cccccc', array(0.8)),
+            array('#CCCCCC', array(0.8)),
+            array('#ffcc66', array(1.0, 0.8, 0.4)),
+            array('#FFCC66', array(1.0, 0.8, 0.4)),
+            array('#123456', array(0.071, 0.204, 0.337)),
+
+            array('#ccc', array(0.8)),
+            array('#CCC', array(0.8)),
+            array('#fc6', array(1.0, 0.8, 0.4)),
+            array('#FC6', array(1.0, 0.8, 0.4)),
+        );
+    }
+
+    /**
+     * @dataProvider colorProvider
+     */
+    public function testColor($color, $components)
+    {
+        $this->assertSame($components, Html::color($color)->getComponents());
+    }
+}