Handling SubItems Trait for not even Invoices added

Author: Vitexus

src/FlexiPeeHP/Adresar.php

@@ -16,6 +16,8 @@
 class Adresar extends FlexiBeeRW
 {
     use Stitky;
+    use SubItems;
+    use Firma;
 
     /**
      * Evidence užitá objektem.

src/FlexiPeeHP/Cenik.php

@@ -17,6 +17,8 @@
 class Cenik extends FlexiBeeRW
 {
     use Stitky;
+    use SubItems;
+    
     /**
      * Evidence užitá objektem.
      *

src/FlexiPeeHP/FakturaPrijata.php

@@ -1,4 +1,5 @@
 <?php
+
 /**
  * FlexiPeeHP - Invoice Incomed.
  *
@@ -13,11 +14,13 @@
  *
  * @link https://demo.flexibee.eu/c/demo/faktura-prijata/properties
  */
-class FakturaPrijata extends FlexiBeeRW
-{
+class FakturaPrijata extends FlexiBeeRW {
+
     use Stitky;
     use Firma;
     use Sum;
+    use SubItems;
+
     /**
      * Evidence použitá třídou.
      *

src/FlexiPeeHP/FakturaVydana.php

@@ -19,6 +19,8 @@ class FakturaVydana extends FlexiBeeRW
     use Stitky;
     use Firma;
     use Sum;
+    use SubItems;
+
     /**
      * Evidence užitá objektem.
      *

src/FlexiPeeHP/FlexiBeeRO.php

@@ -1579,15 +1579,6 @@ public function recordExists($data = []) {
         return $found;
     }
 
-    /**
-     * Subitems - ex. items of invoice
-     * 
-     * @return array of document items or null
-     */
-    public function getSubItems() {
-        return array_key_exists('polozkyFaktury', $this->getData()) ? $this->getDataValue('polozkyFaktury') : (array_key_exists('polozkyDokladu', $this->getData()) ? $this->getDataValue('polozkyDokladu') : null);
-    }
-
     /**
      * Vrací z FlexiBee sloupečky podle podmínek.
      *

src/FlexiPeeHP/SubItems.php

@@ -0,0 +1,34 @@
+<?php
+
+/*
+ * To change this license header, choose License Headers in Project Properties.
+ * To change this template file, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+namespace FlexiPeeHP;
+
+/**
+ *
+ * @author vitex
+ */
+trait SubItems {
+    /**
+     * Subitems - ex. items of invoice
+     * 
+     * @return array of document items or null
+     */
+    public function getSubItems() {
+        return $this->getDataValue($this->getSubmenuName());
+    }
+    
+    public function setSubitems(array $subitems) {
+        return $this->setDataValue($this->getSubmenuName(), $subitems);    
+    }
+
+    public function getSubMenuName() {
+        return array_key_exists('polozkyFaktury', $this->getData()) ? 'polozkyFaktury' : (array_key_exists('polozkyDokladu', $this->getData()) ? 'polozkyDokladu' : null);
+
+    }
+    
+}