From eed7a470ed3edef77c609eea520584f7017f3623 Mon Sep 17 00:00:00 2001
From: Victor Westerlund <victor.vesterlund@gmail.com>
Date: Thu, 2 Nov 2023 11:53:54 +0100
Subject: [PATCH] feat: remove table model by passing `null` to `with()` (#19)

* feat: remove table model by passing null to with()

* feat(doc): add with() in README
---
 README.md     | 12 +++++++++++-
 src/MySQL.php |  8 +++++++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 9178ecd..a94b4cb 100644
--- a/README.md
+++ b/README.md
@@ -40,7 +40,7 @@ Statement|Method
 `ORDER BY`|[`order()`](#order-by)
 `LIMIT`|[`limit()`](#limit)
 
----
+----
 
 `Example table name: beverages`
 id|beverage_type|beverage_name|beverage_size
@@ -282,3 +282,13 @@ $coffee = $db->for("beverages")->limit([3 => 2])->select(["beverage_name", "beve
   // ...etc
 ]
 ```
+
+----
+
+# Restrict affected/returned database columns to table model
+
+Chain and pass an array to `MySQL->with()` before a `select()`, `update()`, or `insert()` method to limit which columns will be returned/affected. It will use the **values** of the array so it can be either sequential or associative.
+
+**This method will cause `select()`, `update()`, and `insert()` to ignore any columns that are not present in the passed table model.**
+
+You can remove an already set table model by passing `null` to `MySQL->with()`
diff --git a/src/MySQL.php b/src/MySQL.php
index e62c7af..f4b5c7d 100644
--- a/src/MySQL.php
+++ b/src/MySQL.php
@@ -48,7 +48,13 @@
 		}
 
 		// Restrict query to array of column names
-		public function with(array $model): self {
+		public function with(?array $model = null): self {
+			// Remove table model if empty
+			if (!$model) {
+				$this->model = null;
+				return $this;
+			}
+
 			// Reset table model
 			$this->model = [];