From cd452fe0d1d709730b46114aedb49e4e2476f34e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment?= <clement@desmidt.fr>
Date: Thu, 25 Feb 2021 17:33:40 +0100
Subject: [PATCH] :art: Factorize some code and add the file deletion

---
 src/Assembler.php           | 13 +++++++
 src/Assembler/Mosaic.php    | 53 +++++++++++++++++++++-----
 src/Assembler/Simple.php    | 66 +++++++++++++++++++++-----------
 src/Assembler/Superpose.php | 75 +++++++++++++++++++++++++++----------
 src/Pool.php                | 18 +++++++--
 5 files changed, 171 insertions(+), 54 deletions(-)

diff --git a/src/Assembler.php b/src/Assembler.php
index 27f3492..ec67d38 100644
--- a/src/Assembler.php
+++ b/src/Assembler.php
@@ -4,6 +4,7 @@ namespace Shikiryu\WebGobbler;
 
 abstract class Assembler
 {
+    protected $files = [];
     /**
      * @var \Shikiryu\WebGobbler\Pool
      */
@@ -26,6 +27,18 @@ abstract class Assembler
         $this->config = $config;
     }
 
+    /**
+     * @return void
+     * @link https://php.net/manual/en/language.oop5.decon.php
+     */
+    public function __destruct()
+    {
+        foreach ($this->files as $file) {
+            unlink($file);
+        }
+    }
+
+
     /**
      * @param string $file
      *
diff --git a/src/Assembler/Mosaic.php b/src/Assembler/Mosaic.php
index ecb0330..b4cd2f8 100644
--- a/src/Assembler/Mosaic.php
+++ b/src/Assembler/Mosaic.php
@@ -14,14 +14,51 @@ class Mosaic extends Assembler
      */
     public function saveTo($file)
     {
-        $final_image = new \Imagick();
-        $final_image->setColorspace(\Imagick::COLORSPACE_RGB);
-        $final_image->newImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'), 'none');
+        try {
+            $final_image = new \Imagick();
+            $final_image->setColorspace(\Imagick::COLORSPACE_RGB);
+            $final_image->newImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'), 'none');
+            $final_image = $this->prepareImage($final_image);
+
+            $final_image->writeImage($file);
+        } catch (\ImagickException $e) {
+            echo $e->getMessage();
+        }
+    }
+
+    /**
+     *
+     */
+    public function display()
+    {
+        try {
+            $final_image = new \Imagick();
+            $final_image->setColorspace(\Imagick::COLORSPACE_RGB);
+            $final_image->newImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'), 'none');
+            $final_image = $this->prepareImage($final_image);
+
+            header('Content-type: image/jpeg');
+            echo $final_image;
+        } catch (\ImagickException $e) {
+            echo $e->getMessage();
+        }
+    }
+
+    /**
+     * @param \Imagick $final_image
+     *
+     * @return \Imagick
+     * @throws \ImagickException
+     */
+    private function prepareImage(\Imagick $final_image)
+    {
         $image_size_x = $this->config->get('assembler.sizex') / $this->config->get('assembler.nbx');
         $image_size_y = $this->config->get('assembler.sizey') / $this->config->get('assembler.nby');
         for ($y = 0; $y < $this->config->get('assembler.nby'); $y++) {
             for ($x = 0; $x < $this->config->get('assembler.nbx'); $x++) {
-                $image = new \Imagick($this->pool->getImage());
+                $file = $this->pool->getImage();
+                $this->files[] = $file;
+                $image = new \Imagick($file);
                 if ($image->getColorspace() !== \Imagick::COLORSPACE_RGB) {
                     $image->setColorspace(\Imagick::COLORSPACE_RGB);
                 }
@@ -38,11 +75,7 @@ class Mosaic extends Assembler
         if (true === $this->config->get('assembler.invert')) {
             $final_image->negateImage(false);
         }
-        $final_image->writeImage($file);
-    }
 
-    public function display()
-    {
-        // TODO: Implement display() method.
+        return $final_image;
     }
-}
\ No newline at end of file
+}
diff --git a/src/Assembler/Simple.php b/src/Assembler/Simple.php
index e786787..0a3150e 100644
--- a/src/Assembler/Simple.php
+++ b/src/Assembler/Simple.php
@@ -12,33 +12,57 @@ class Simple extends Assembler
     public function saveTo($file)
     {
         try {
-            $image = new \Imagick($this->pool->getImage());
-            if ($image->getColorspace() !== \Imagick::COLORSPACE_RGB) {
-                $image->setColorspace(\Imagick::COLORSPACE_RGB);
-            }
-            $size = $image->getSize();
-            $imagex = $size['columns'];
-            $imagey = $size['rows'];
-            if ($imagex !== $this->config->get('assembler.sizex') || $imagey !== $this->config->get('assembler.sizey')) {
-                $image->thumbnailImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'));
-            }
-            if (true === $this->config->get('assembler.mirror')) {
-                $image->flopImage();
-            }
-            if (true === $this->config->get('assembler.emboss')) {
-                $image->embossImage(0, 1);
-            }
-            if (true === $this->config->get('assembler.invert')) {
-                $image->flipImage();
-            }
+            $image = $this->prepareImage();
+
             $image->writeImage($file);
         } catch (\ImagickException $e) {
             echo $e->getMessage();
         }
     }
 
-    public function display()
+    /**
+     * @param string $file
+     */
+    public function display($file)
     {
-        // TODO: Implement display() method.
+        try {
+            $image = $this->prepareImage();
+
+            header('Content-type: image/jpeg');
+            echo $image;
+        } catch (\ImagickException $e) {
+            echo $e->getMessage();
+        }
+    }
+
+    /**
+     * @return \Imagick
+     * @throws \ImagickException
+     */
+    private function prepareImage()
+    {
+        $file = $this->pool->getImage();
+        $this->files[] = $file;
+        $image = new \Imagick($file);
+        if ($image->getColorspace() !== \Imagick::COLORSPACE_RGB) {
+            $image->setColorspace(\Imagick::COLORSPACE_RGB);
+        }
+        $size = $image->getSize();
+        $imagex = $size['columns'];
+        $imagey = $size['rows'];
+        if ($imagex !== $this->config->get('assembler.sizex') || $imagey !== $this->config->get('assembler.sizey')) {
+            $image->thumbnailImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'));
+        }
+        if (true === $this->config->get('assembler.mirror')) {
+            $image->flopImage();
+        }
+        if (true === $this->config->get('assembler.emboss')) {
+            $image->embossImage(0, 1);
+        }
+        if (true === $this->config->get('assembler.invert')) {
+            $image->flipImage();
+        }
+
+        return $image;
     }
 }
\ No newline at end of file
diff --git a/src/Assembler/Superpose.php b/src/Assembler/Superpose.php
index 924f047..00e95d3 100644
--- a/src/Assembler/Superpose.php
+++ b/src/Assembler/Superpose.php
@@ -4,6 +4,8 @@ namespace Shikiryu\WebGobbler\Assembler;
 
 use Imagick;
 use Shikiryu\WebGobbler\Assembler;
+use Shikiryu\WebGobbler\Config;
+use Shikiryu\WebGobbler\Pool;
 
 class Superpose extends Assembler
 {
@@ -12,6 +14,22 @@ class Superpose extends Assembler
      */
     private $current_image;
 
+    private $base_image;
+
+    /**
+     * Superpose constructor.
+     *
+     * @param \Shikiryu\WebGobbler\Pool   $pool
+     * @param \Shikiryu\WebGobbler\Config $config
+     * @param string                      $from_file
+     */
+    public function __construct(Pool $pool, Config $config, $from_file = null)
+    {
+        $this->base_image = $from_file;
+
+        parent::__construct($pool, $config);
+    }
+
     /**
      * @param string $file
      *
@@ -21,20 +39,35 @@ class Superpose extends Assembler
      */
     public function saveTo($file)
     {
-        $this->current_image = new \Imagick();
-        $this->current_image->setSize($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'));
-        $this->current_image->newImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'), 'none');
-
-        $nb_images = $this->config->get('assembler.superpose.min_num_images', 5);
-        for ($i = 0; $i < $nb_images; $i++) {
-            $this->current_image = $this->superpose();
-        }
+        $this->prepareImage();
         $this->current_image->writeImage($file);
     }
 
+    /**
+     *
+     */
+    public function display()
+    {
+        try {
+            $this->prepareImage();
+
+            header('Content-type: image/jpeg');
+            echo $this->current_image;
+        } catch (\ImagickException $e) {
+            echo $e->getMessage();
+        }
+    }
+
+    /**
+     * @return \Imagick
+     * @throws \ImagickException
+     */
     public function superpose()
     {
-        return $this->superposeOneImage($this->current_image, new \Imagick($this->pool->getImage()));
+        $file = $this->pool->getImage();
+        $this->files[] = $file;
+
+        return $this->superposeOneImage($this->current_image, new \Imagick($file));
     }
 
     /**
@@ -79,12 +112,6 @@ class Superpose extends Assembler
         $image_x = $image_to_superpose->getImageWidth();
         $image_y = $image_to_superpose->getImageHeight();
 
-        # Compensate for poorly-contrasted images on the web
-        /*try:
-            imageToSuperpose = ImageOps.autocontrast(imageToSuperpose)
-        except TypeError:  # Aaron tells me that this exception occurs with PNG images.
-            raise BadImage*/
-
         # Some image are too white.
         # For example, the photo of a coin on a white background.
         # These picture degrad the quality of the final image.
@@ -113,8 +140,8 @@ class Superpose extends Assembler
             $image_to_superpose->negateImage(false);
         }
 
-        $paste_coords_x = random_int(-($image_x/2), $this->config->get('assembler.sizex')-$image_x);
-        $paste_coords_y = random_int(-($image_y/2), $this->config->get('assembler.sizey')-$image_y);
+        $paste_coords_x = random_int(-($image_x/2), max(0, $this->config->get('assembler.sizex')-$image_x));
+        $paste_coords_y = random_int(-($image_y/2), max(0, $this->config->get('assembler.sizey')-$image_y));
 
         # Darken image borders
         $image_to_superpose = $this->darkenImageBorder($image_to_superpose);
@@ -175,8 +202,18 @@ class Superpose extends Assembler
         return $image;
     }
 
-    public function display()
+    /**
+     * @throws \ImagickException
+     */
+    private function prepareImage()
     {
-        // TODO: Implement display() method.
+        $this->current_image = null === $this->base_image ? new \Imagick() : new \Imagick($this->base_image);
+        $this->current_image->setSize($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'));
+        $this->current_image->newImage($this->config->get('assembler.sizex'), $this->config->get('assembler.sizey'), 'none');
+
+        $nb_images = $this->config->get('assembler.superpose.min_num_images', 5);
+        for ($i = 0; $i < $nb_images; $i++) {
+            $this->current_image = $this->superpose();
+        }
     }
 }
\ No newline at end of file
diff --git a/src/Pool.php b/src/Pool.php
index f2e8558..19e721b 100644
--- a/src/Pool.php
+++ b/src/Pool.php
@@ -4,6 +4,9 @@ namespace Shikiryu\WebGobbler;
 
 class Pool
 {
+    /**
+     * @var string[]
+     */
     protected $file_list = [];
 
     /**
@@ -58,7 +61,7 @@ class Pool
 
         $this->pool_directory = $pool_dir;
 
-        $this->nb_images = $pool_config['nb_images']; // FIXME
+        $this->nb_images = $pool_config['nb_images'];
 
         $this->prepareCollectors();
     }
@@ -110,6 +113,11 @@ class Pool
         return $file_list;
     }
 
+    /**
+     * Check if collector folder exists or create it
+     *
+     * Download images if configured to
+     */
     private function prepareCollectors()
     {
         foreach ($this->collectors as $collector) {
@@ -117,9 +125,11 @@ class Pool
             if (!is_dir($directory) && !mkdir($directory) && !is_dir($directory)) {
                 throw new \RuntimeException(sprintf('Directory "%s" was not created', $directory));
             }
-            $images = glob($directory.'/*.{jpg,gif,png}', GLOB_BRACE);
-            if (count($images) < $this->nb_images) {
-                $collector->getRandomImages($this->nb_images - count($images));
+            if ($this->config->get('pool.pre_download', false) === true) {
+                $images = glob($directory . '/*.{jpg,gif,png}', GLOB_BRACE);
+                if (count($images) < $this->nb_images) {
+                    $collector->getRandomImages($this->nb_images - count($images));
+                }
             }
         }
     }