Add a download function (#6)

cable8mm · Feb 14, 2025 · 02d4a84 · 02d4a84
1 parent adf04f3
commit 02d4a84
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 12 deletions.
diff --git a/src/Waybill.php b/src/Waybill.php
@@ -95,6 +95,31 @@ private function __construct(
         $this->mpdf->useActiveForms = $config['use_active_forms'];
     }
 
+    /**
+     * Run WriteHTML function
+     *
+     * @param  ?array<string,Tvalue>  $args  The array of arguments for a model or table fields
+     */
+    private function write(?array $args = null): void
+    {
+        $data = [];
+
+        if (! is_null($args)) {
+            $data = $args;
+        } elseif (! empty($this->state)) {
+            $data = $this->parcelService->factoryClass()::make()->state($this->state)->definition();
+        } else {
+            $data = $this->parcelService->factoryClass()::make()->definition();
+        }
+
+        $this->mpdf->WriteHTML(
+            Stub::of(
+                $this->parcelService->stub(),
+                $data
+            )->render()
+        );
+    }
+
     /**
      * Setter for $state
      *
@@ -134,7 +159,8 @@ public function path(string $path): static
      */
     public function toArray(): array
     {
-        return $this->parcelService->factoryClass()::make()->state($this->state)->create();
+        return $this->parcelService->factoryClass()::make()
+            ->state($this->state)->create();
     }
 
     /**
@@ -146,16 +172,25 @@ public function toArray(): array
      */
     public function save(string $filename = 'waybills.pdf'): mixed
     {
-        $this->mpdf->WriteHTML(
-            $html = Stub::of(
-                $this->parcelService->stub(),
-                $this->parcelService->factoryClass()::make()->definition()
-            )->render()
-        );
+        $this->write();
 
         $path = empty($this->path) ? '' : $this->path.DIRECTORY_SEPARATOR;
 
-        return $this->mpdf->Output($path.$filename, 'F');
+        return $this->mpdf->Output($path.$filename, false);
+    }
+
+    /**
+     * Download the waybills data to PDF
+     *
+     * @param  string  $filename  A filename to create
+     *
+     * @example $waybill = Waybill::of(ParcelService::Cj)->path(realpath(__DIR__.'/../dist'))->save('test.pdf');
+     */
+    public function download(string $filename = 'waybills.pdf'): mixed
+    {
+        $this->write();
+
+        return $this->mpdf->Output($filename, true);
     }
 
     /**

diff --git a/tests/WaybillTest.php b/tests/WaybillTest.php
@@ -23,15 +23,24 @@ public function test_path_exists(): void
         $this->assertStringContainsString(DIRECTORY_SEPARATOR.'dist', $path->getValue($waybill));
     }
 
-    public function test_it_export_to_array(): void
+    public function test_to_array(): void
     {
-        $orderSheet = Waybill::of(ParcelService::Cj)
+        $waybill = Waybill::of(ParcelService::Cj)
+            ->toArray();
+
+        $this->assertIsArray($waybill);
+    }
+
+    public function test_state(): void
+    {
+        $waybill = Waybill::of(ParcelService::Cj)
+            ->state(['6' => 'new value'])
             ->toArray();
 
-        $this->assertIsArray($orderSheet);
+        $this->assertSame('new value', $waybill['6']);
     }
 
-    public function test_it_export_to_file(): void
+    public function test_save(): void
     {
         $waybill = Waybill::of(ParcelService::Cj)
             ->path(realpath(__DIR__.'/../dist'))