From e7fd6cb9d1a856fb02417aba34deb7d7ba3261ed Mon Sep 17 00:00:00 2001
From: root <root@localhost.localdomain>
Date: Sun, 2 Sep 2018 23:19:03 +0800
Subject: [PATCH] Fix battery image issue for under warranty

---
 src/Controller/APIController.php | 10 +++++++++-
 src/Service/InvoiceCreator.php   |  3 ++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/Controller/APIController.php b/src/Controller/APIController.php
index c900f1d9..74e1f8cb 100644
--- a/src/Controller/APIController.php
+++ b/src/Controller/APIController.php
@@ -50,6 +50,12 @@ class APIController extends Controller
         $this->session = null;
     }
 
+    protected function debugRequest(Request $req)
+    {
+        $all = $req->request->all();
+        error_log(print_r($all, true));
+    }
+
     protected function checkMissingParameters(Request $req, $params = [])
     {
         $missing = [];
@@ -939,6 +945,8 @@ class APIController extends Controller
 
     public function getEstimate(Request $req, InvoiceCreator $ic)
     {
+        // $this->debugRequest($req);
+
         // check required parameters and api key
         $required_params = [
             'service_type',
@@ -1066,7 +1074,7 @@ class APIController extends Controller
 
         $data['items'] = $items_data;
 
-        error_log(print_r($data, true));
+        // error_log(print_r($data, true));
 
         // set data
         $res->setData($data);
diff --git a/src/Service/InvoiceCreator.php b/src/Service/InvoiceCreator.php
index 5f8dac9d..858b3132 100644
--- a/src/Service/InvoiceCreator.php
+++ b/src/Service/InvoiceCreator.php
@@ -253,7 +253,8 @@ class InvoiceCreator
             $item->setInvoice($invoice)
                 ->setTitle($batt->getModel()->getName() . ' ' . $batt->getSize()->getName() . ' - Service Unit')
                 ->setQuantity(1)
-                ->setPrice(0.00);
+                ->setPrice(0.00)
+                ->setBattery($batt);
             $invoice->addItem($item);
         }
     }