Added Query Builder Trait

Codexshaper · Sep 24, 2019 · 2e468fb · 2e468fb
1 parent 2029cb0
commit 2e468fb
Show file tree

Hide file tree

Showing 4 changed files with 77 additions and 34 deletions.
diff --git a/src/Models/Customer.php b/src/Models/Customer.php
@@ -3,9 +3,12 @@
 namespace Codexshaper\WooCommerce\Models;
 
 use Codexshaper\WooCommerce\Facades\WooCommerce;
+use Codexshaper\WooCommerce\Traits\QueryBuilderTrait;
 
 class Customer
 {
+    use QueryBuilderTrait;
+
     public function all($options = [])
     {
         return WooCommerce::all('customers', $options);

diff --git a/src/Models/Order.php b/src/Models/Order.php
@@ -3,11 +3,11 @@
 namespace Codexshaper\WooCommerce\Models;
 
 use Codexshaper\WooCommerce\Facades\WooCommerce;
+use Codexshaper\WooCommerce\Traits\QueryBuilderTrait;
 
 class Order
 {
-    protected $options = [];
-    protected $results = [];
+    use QueryBuilderTrait;
 
     public function all($options = [])
     {
@@ -16,7 +16,7 @@ public function all($options = [])
 
     public function find($id, $options = [])
     {
-        return WooCommerce::find("orders/{$id}", $options);
+        return collect(WooCommerce::find("orders/{$id}", $options));
     }
 
     public function create($data)
@@ -86,34 +86,4 @@ public function deleteRefund($order_id, $refund_id, $options = [])
     {
         return WooCommerce::delete("orders/{$order_id}/refunds/{$refund_id}", $options);
     }
-
-    /* Custom Query */
-    public function where(...$parameters)
-    {
-        if (count($parameters) == 2) {
-            $this->options[$parameters[0]] = $parameters[1];
-        }
-
-        if (count($parameters) == 1) {
-
-            foreach ($parameters as $parameter) {
-
-                foreach ($parameter as $key => $value) {
-                    $this->options[$key] = $value;
-                }
-            }
-        }
-        return $this;
-    }
-
-    public function get()
-    {
-        return WooCommerce::all('orders', $this->options);
-    }
-
-    public function first()
-    {
-        $order = $this->get()[0];
-        return ['order' => $order];
-    }
 }
diff --git a/src/Models/Product.php b/src/Models/Product.php
@@ -3,17 +3,20 @@
 namespace Codexshaper\WooCommerce\Models;
 
 use Codexshaper\WooCommerce\Facades\WooCommerce;
+use Codexshaper\WooCommerce\Traits\QueryBuilderTrait;
 
 class Product
 {
+    use QueryBuilderTrait;
+
     public function all($options = [])
     {
         return WooCommerce::all('products', $options);
     }
 
     public function find($id, $options = [])
     {
-        return WooCommerce::find("products/{$id}", $options);
+        return collect(WooCommerce::find("products/{$id}", $options));
     }
 
     public function create($data)

diff --git a/src/Traits/QueryBuilderTrait.php b/src/Traits/QueryBuilderTrait.php
@@ -0,0 +1,67 @@
+<?php
+
+namespace Codexshaper\WooCommerce\Traits;
+
+use Codexshaper\WooCommerce\Facades\WooCommerce;
+
+trait QueryBuilderTrait
+{
+
+    protected $options;
+
+    /* Custom Query */
+
+    public function get()
+    {
+        return WooCommerce::all('products', $this->options);
+    }
+
+    public function first()
+    {
+        return collect($this->get()[0]);
+    }
+
+    public function options(array $parameters)
+    {
+        if (!is_array($parameters)) {
+            throw new \Exception("Options must be an array", 1);
+        }
+
+        if (empty($parameters)) {
+            throw new \Exception("Options must be pass at least one element", 1);
+        }
+
+        foreach ($parameters as $key => $value) {
+
+            $this->options[$key] = $value;
+        }
+
+        return $this;
+    }
+
+    public function where(...$parameters)
+    {
+        if (count($parameters) == 2) {
+            $this->options[$parameters[0]] = $parameters[1];
+        }
+
+        if (count($parameters) == 1) {
+
+            foreach ($parameters as $parameter) {
+
+                foreach ($parameter as $key => $value) {
+                    $this->options[$key] = $value;
+                }
+            }
+        }
+        return $this;
+    }
+
+    public function orderBy($name, $direction = 'desc')
+    {
+        $this->options['orderby'] = $name;
+        $this->options['order']   = $direction;
+
+        return $this;
+    }
+}