considerate data-based urls to get images

Author: oscarotero

src/ImageInfo/Curl.php

@@ -41,43 +41,46 @@ public static function getImagesInfo(array $images, array $config = null)
             return [];
         }
 
-        if (count($images) === 1) {
-            $info = self::getImageInfo($images[0], $config);
-
-            return empty($info) ? [] : [array_merge($images[0], $info)];
-        }
-
         $finfo = finfo_open(FILEINFO_MIME_TYPE);
         $connections = [];
         $curl = curl_multi_init();
+        $result = [];
 
         foreach ($images as $k => $image) {
+            if (strpos($image['value'], 'data:') === 0) {
+                if ($info = static::getEmbeddedImageInfo($image['value'])) {
+                    $result[] = array_merge($image, $info);
+                }
+
+                continue;
+            }
+
             $connections[$k] = new static($image['value'], $finfo, $config);
 
             curl_multi_add_handle($curl, $connections[$k]->getConnection());
         }
 
-        do {
-            $return = curl_multi_exec($curl, $active);
-        } while ($return === CURLM_CALL_MULTI_PERFORM);
-
-        while ($active && $return === CURLM_OK) {
-            if (curl_multi_select($curl) === -1) {
-                usleep(100);
-            }
-
+        if ($connections) {
             do {
                 $return = curl_multi_exec($curl, $active);
             } while ($return === CURLM_CALL_MULTI_PERFORM);
-        }
 
-        $result = [];
+            while ($active && $return === CURLM_OK) {
+                if (curl_multi_select($curl) === -1) {
+                    usleep(100);
+                }
 
-        foreach ($connections as $k => $connection) {
-            curl_multi_remove_handle($curl, $connection->getConnection());
+                do {
+                    $return = curl_multi_exec($curl, $active);
+                } while ($return === CURLM_CALL_MULTI_PERFORM);
+            }
 
-            if (($info = $connection->getInfo())) {
-                $result[] = array_merge($images[$k], $info);
+            foreach ($connections as $k => $connection) {
+                curl_multi_remove_handle($curl, $connection->getConnection());
+
+                if (($info = $connection->getInfo())) {
+                    $result[] = array_merge($images[$k], $info);
+                }
             }
         }
 
@@ -87,30 +90,6 @@ public static function getImagesInfo(array $images, array $config = null)
         return $result;
     }
 
-    /**
-     * Get the info of only one image.
-     *
-     * @param string     $image
-     * @param null|array $config
-     *
-     * @return array|null
-     */
-    private static function getImageInfo($image, array $config = null)
-    {
-        $finfo = finfo_open(FILEINFO_MIME_TYPE);
-        $img = new static($image['value'], $finfo, $config);
-
-        $curl = $img->getConnection();
-        curl_exec($curl);
-        curl_close($curl);
-
-        $info = $img->getInfo();
-
-        finfo_close($finfo);
-
-        return $info;
-    }
-
     /**
      * Init the curl connection.
      *
@@ -190,4 +169,22 @@ public function writeCallback($connection, $string)
 
         return -1;
     }
+
+    protected static function getEmbeddedImageInfo($content)
+    {
+        $pieces = explode(';', $content, 2);
+
+        if ((count($pieces) !== 2) || (strpos($pieces[0], 'image/') === false) || (strpos($pieces[1], 'base64,') !== 0)) {
+            return false;
+        }
+
+        $info = getimagesizefromstring(base64_decode(substr($pieces[1], 7)));
+
+        return [
+            'width' => $info[0],
+            'height' => $info[1],
+            'size' => $info[0] * $info[1],
+            'mime' => $info['mime'],
+        ];
+    }
 }

src/Url.php

@@ -65,6 +65,16 @@ public function match($patterns)
         return false;
     }
 
+    /**
+     * Return the content of the url (for embedded images).
+     *
+     * @return string The content or null
+     */
+    public function getContent()
+    {
+        return isset($this->info['content']) ? $this->info['content'] : null;
+    }
+
     /**
      * Return the extension of the url (html, php, jpg, etc).
      *
@@ -395,9 +405,14 @@ protected function buildUrl()
     {
         $url = '';
 
+        if (isset($this->info['content'])) {
+            return 'data:'.$this->info['content'];
+        }
+
         if (isset($this->info['scheme'])) {
             $url .= $this->info['scheme'].'://';
         }
+
         if (isset($this->info['host'])) {
             $url .= $this->info['host'];
         }
@@ -481,6 +496,12 @@ public function getAbsolute($url)
      */
     private function setPath($path)
     {
+        if ($this->getScheme() === 'data') {
+            $this->info['content'] = $path;
+            $this->info['path'] = $this->info['file'] = $this->info['extension'] = null;
+            return;
+        }
+
         $parts = pathinfo($path);
 
         $this->info['path'] = [];
@@ -495,5 +516,6 @@ private function setPath($path)
 
         $this->info['file'] = isset($parts['filename']) ? $parts['filename'] : null;
         $this->info['extension'] = isset($parts['extension']) ? $parts['extension'] : null;
+        $this->info['content'] = null;
     }
 }