From 96fdf787dee73bcbfd51fab9838c987066c062d9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Benjamin=20Beganovi=C4=87?= <k1pstabug@gmail.com>
Date: Tue, 4 Aug 2020 13:37:51 +0200
Subject: [PATCH] add comments

---
 app/Services/PdfMaker/Designs/Plain.php | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/app/Services/PdfMaker/Designs/Plain.php b/app/Services/PdfMaker/Designs/Plain.php
index de38aa1817..4e71d91112 100644
--- a/app/Services/PdfMaker/Designs/Plain.php
+++ b/app/Services/PdfMaker/Designs/Plain.php
@@ -19,12 +19,16 @@ class Plain
 {
     use MakesInvoiceValues, BuildTableHeader;
 
+    /** Global list of table elements, @var array */
     public $elements;
 
+    /** @var App\Models\Client */
     public $client;
 
+    /** @var App\Models\Invoice */
     public $invoice;
 
+    /** Global state of the design, @var string */
     public $context;
 
     public function html(): ?string
@@ -36,11 +40,13 @@ class Plain
 
     public function setup(): void
     {
-        $this->client = $this->context['client'];
+        if (isset($this->context['client'])) {
+            $this->client = $this->context['client'];
+        }
 
-        $this->invoice = $this->context['invoice'];
-
-        /** @todo: Wrap the elements with correct checks & exceptions. */
+        if (isset($this->context['invoice'])) {
+            $this->invoice = $this->context['invoice'];
+        }
     }
 
     public function elements(array $context): array
@@ -72,7 +78,6 @@ class Plain
                         ['element' => 'td', 'content' => '$subtotal_label', 'properties' => ['class' => 'px-4 py-4 text-right', 'colspan' => '2']],
                         ['element' => 'td', 'content' => '$subtotal', 'properties' => ['class' => 'px-4 py-2 text-right']],
                     ]],
-
                     ['element' => 'tr', 'content' => '', 'elements' => [
                         ['element' => 'td', 'content' => '$discount_label', 'properties' => ['class' => 'border-l-4 border-white px-4 text-right', 'colspan' => '4']],
                         ['element' => 'td', 'content' => '$discount', 'properties' => ['class' => 'px-4 py-2 text-right']],