From bc28b0c2ed4bd985588d08ad4a536df2c30f3e8c Mon Sep 17 00:00:00 2001
From: lukinovec <lukinovec@gmail.com>
Date: Wed, 7 Jun 2023 09:37:32 +0200
Subject: [PATCH] Add RLSModel interface, update BelongsToTenant

---
 src/Database/Concerns/BelongsToTenant.php |  5 +++--
 src/Database/Concerns/RLSModel.php        | 15 +++++++++++++++
 2 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 src/Database/Concerns/RLSModel.php

diff --git a/src/Database/Concerns/BelongsToTenant.php b/src/Database/Concerns/BelongsToTenant.php
index c5323065..e8c2156d 100644
--- a/src/Database/Concerns/BelongsToTenant.php
+++ b/src/Database/Concerns/BelongsToTenant.php
@@ -21,8 +21,9 @@ trait BelongsToTenant
 
     public static function bootBelongsToTenant(): void
     {
-        // If tenancy.database.rls is true, scope queries using Postgres RLS instead of TenantScope
-        if (! config('tenancy.database.rls')) {
+        // If tenancy.database.rls is true or this model implements RLSModel
+        // Scope queries using Postgres RLS instead of TenantScope
+        if (! (config('tenancy.database.rls') || in_array(RLSModel::class, class_implements(static::class)))) {
             static::addGlobalScope(new TenantScope);
         }
 
diff --git a/src/Database/Concerns/RLSModel.php b/src/Database/Concerns/RLSModel.php
new file mode 100644
index 00000000..11c64c65
--- /dev/null
+++ b/src/Database/Concerns/RLSModel.php
@@ -0,0 +1,15 @@
+<?php
+
+namespace Stancl\Tenancy\Database\Concerns;
+
+/**
+ * Interface indicating that the queries of the model it's used on
+ * get scoped using RLS instead of the global TenantScope.
+ *
+ * Used with Postgres RLS (single-database tenancy).
+ *
+ * @see \Stancl\Tenancy\Database\Concerns\BelongsToTenant
+ */
+interface RLSModel
+{
+}