From 9f7adcf977c94cfeebd308a852f9d4404f2b99cc Mon Sep 17 00:00:00 2001
From: alexpozzi <alexpozzi@users.noreply.github.com>
Date: Wed, 16 Oct 2024 14:39:25 +0200
Subject: [PATCH] feat: implement wkhtmltopdf backend

---
 src/Backend/WkHtmlToPdf/ExtraOption.php       | 10 +++
 .../WkHtmlToPdf/ExtraOption/CollateOption.php | 17 +++++
 .../ExtraOption/CookieJarOption.php           | 17 +++++
 .../WkHtmlToPdf/ExtraOption/CopiesOption.php  | 17 +++++
 .../WkHtmlToPdf/ExtraOption/DpiOption.php     | 17 +++++
 .../ExtraOption/GrayscaleOption.php           | 15 ++++
 .../ExtraOption/ImageDpiOption.php            | 10 +++
 .../ExtraOption/NoCollateOption.php           | 10 +++
 .../WkHtmlToPdf/ExtraOption/Orientation.php   | 10 +++
 .../WkHtmlToPdf/WkHtmlToPdfAdapter.php        | 71 +++++++++++++++++--
 10 files changed, 190 insertions(+), 4 deletions(-)
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/CollateOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/CookieJarOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/CopiesOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/DpiOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/GrayscaleOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/ImageDpiOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/NoCollateOption.php
 create mode 100644 src/Backend/WkHtmlToPdf/ExtraOption/Orientation.php

diff --git a/src/Backend/WkHtmlToPdf/ExtraOption.php b/src/Backend/WkHtmlToPdf/ExtraOption.php
new file mode 100644
index 00000000..6e0f4931
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption.php
@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf;
+
+interface ExtraOption {
+    /** @return array<string|int|float> */
+    public function compile(): array;
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/CollateOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/CollateOption.php
new file mode 100644
index 00000000..0b3f70c3
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/CollateOption.php
@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+use KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class CollateOption implements ExtraOption
+{
+    public function __construct() {}
+
+    public function compile(): array
+    {
+        return ['--collate'];
+    }
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/CookieJarOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/CookieJarOption.php
new file mode 100644
index 00000000..8c489377
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/CookieJarOption.php
@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+use KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class CookieJarOption implements ExtraOption
+{
+    public function __construct(public readonly string $path) {}
+
+    public function compile(): array
+    {
+        return ['--no-collate'];
+    }
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/CopiesOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/CopiesOption.php
new file mode 100644
index 00000000..d2ba0ed0
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/CopiesOption.php
@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+use KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class CopiesOption implements ExtraOption
+{
+    public function __construct(private readonly int $number) {}
+
+    public function compile(): array
+    {
+        return ['--copies', $this->number];
+    }
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/DpiOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/DpiOption.php
new file mode 100644
index 00000000..65502d9f
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/DpiOption.php
@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+use KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class DpiOptions implements ExtraOption
+{
+    public function __construct(private readonly int $dpi) {}
+
+    public function compile(): array
+    {
+        return ['--dpi', $this->dpi];
+    }
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/GrayscaleOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/GrayscaleOption.php
new file mode 100644
index 00000000..21ed0645
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/GrayscaleOption.php
@@ -0,0 +1,15 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class GrayscaleOption
+{
+    public function __construct() {}
+
+    public function compile(): array
+    {
+        return ['--grayscale'];
+    }
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/ImageDpiOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/ImageDpiOption.php
new file mode 100644
index 00000000..d1ec313d
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/ImageDpiOption.php
@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class ImageDpiOption
+{
+    public function __construct(public readonly int $dpi) {}
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/NoCollateOption.php b/src/Backend/WkHtmlToPdf/ExtraOption/NoCollateOption.php
new file mode 100644
index 00000000..eb0e740c
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/NoCollateOption.php
@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class NoCollateOption
+{
+    public function __construct() {}
+}
diff --git a/src/Backend/WkHtmlToPdf/ExtraOption/Orientation.php b/src/Backend/WkHtmlToPdf/ExtraOption/Orientation.php
new file mode 100644
index 00000000..d2798026
--- /dev/null
+++ b/src/Backend/WkHtmlToPdf/ExtraOption/Orientation.php
@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption;
+
+final class Orientation
+{
+    public function __construct() {}
+}
diff --git a/src/Backend/WkHtmlToPdf/WkHtmlToPdfAdapter.php b/src/Backend/WkHtmlToPdf/WkHtmlToPdfAdapter.php
index ad659441..c678a81e 100644
--- a/src/Backend/WkHtmlToPdf/WkHtmlToPdfAdapter.php
+++ b/src/Backend/WkHtmlToPdf/WkHtmlToPdfAdapter.php
@@ -4,14 +4,21 @@
 
 namespace KNPLabs\Snappy\Backend\WkHtmlToPdf;
 
-use KNPLabs\Snappy\Core\Backend\Adapter;
+use KNPLabs\Snappy\Backend\WkHtmlToPdf\ExtraOption\Orientation;
 use KNPLabs\Snappy\Core\Backend\Adapter\HtmlFileToPdf;
 use KNPLabs\Snappy\Core\Backend\Adapter\Reconfigurable;
+use KNPLabs\Snappy\Core\Backend\Adapter\UriToPdf;
 use KNPLabs\Snappy\Core\Backend\Options;
+use KNPLabs\Snappy\Core\Backend\Options\PageOrientation;
+use KNPLabs\Snappy\Core\Stream\FileStream;
+use Psr\Http\Message\StreamFactoryInterface;
 use Psr\Http\Message\StreamInterface;
+use Psr\Http\Message\UriFactoryInterface;
+use Psr\Http\Message\UriInterface;
+use Symfony\Component\Process\Process;
 use SplFileInfo;
 
-final class WkHtmlToPdfAdapter implements HtmlFileToPdf
+final class WkHtmlToPdfAdapter implements HtmlFileToPdf, UriToPdf
 {
     /**
      * @use Reconfigurable<self>
@@ -26,14 +33,70 @@ public function __construct(
         private string $binary,
         private int $timeout,
         WkHtmlToPdfFactory $factory,
-        Options $options
+        Options $options,
+        private readonly StreamFactoryInterface $streamFactory,
+        private readonly UriFactoryInterface $uriFactory,
     ) {
         $this->factory = $factory;
+
+        foreach ($options->extraOptions as $extraOption) {
+            if (!$extraOption instanceof ExtraOption) {
+                throw new \InvalidArgumentException("Invalid option type");
+            }
+        }
+
         $this->options = $options;
     }
 
     public function generateFromHtmlFile(SplFileInfo $file): StreamInterface
     {
-        throw new \Exception("Not implemented for {$this->binary} with timeout {$this->timeout}.");
+        $filepath = $file->getRealPath();
+
+        if ($filepath === false) {
+            throw new \RuntimeException("File not found: {$file->getPathname()}.");
+        }
+
+        return $this->generateFromUri(
+            $this->uriFactory->createUri($filepath)->withScheme('file')
+        );
+    }
+
+    public function generateFromUri(UriInterface $uri): StreamInterface
+    {
+        $outputStream = FileStream::createTmpFile($this->streamFactory);
+
+        $process = new Process(
+            command: [
+                $this->binary,
+                ...$this->compileOptions(),
+                $uri->toString(),
+                $outputStream->file->getPathname(),
+            ],
+            timeout: $this->timeout,
+        );
+
+        return $outputStream;
+    }
+
+    /**
+     * @return array<string|int|float>
+     */
+    private function compileOptions(): array
+    {
+        return array_reduce(
+            $this->options->extraOptions,
+            fn (array $carry, ExtraOption $extraOption) =>
+                $this->options->pageOrientation !== null && $extraOption instanceof Orientation
+                    ? [
+                        ...$carry,
+                        ...(new Orientation($this->options->pageOrientation->value))->compile(),
+                    ]
+                    : [
+                        ...$carry,
+                        ...$extraOption->compile(),
+                    ]
+            ,
+            [],
+        );
     }
 }