From 1c577ce9aadbe20886114cf6d66c5d2d91d25ffa Mon Sep 17 00:00:00 2001
From: Gautier DELEGLISE <gautier.deleglise@gmail.com>
Date: Wed, 30 Apr 2025 14:20:10 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20base=20control=20exists=20check?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md                          | 9 +++++++++
 src/Console/ControlMakeCommand.php | 5 ++---
 tests/Support/Models/Model.php     | 2 +-
 3 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index aa7347c..1b46085 100644
--- a/README.md
+++ b/README.md
@@ -46,6 +46,15 @@ class PostControl extends Control
         // ...
 ```
 
+Specify the control in your model:
+
+```php
+class Post extends Model
+{
+    use HasControl;
+}
+```
+
 Then set up your policy:
 
 ```php
diff --git a/src/Console/ControlMakeCommand.php b/src/Console/ControlMakeCommand.php
index 0c0a1fb..9d61b7b 100644
--- a/src/Console/ControlMakeCommand.php
+++ b/src/Console/ControlMakeCommand.php
@@ -91,10 +91,9 @@ protected function buildClass($name)
         $replace = $this->buildPerimetersReplacements($replace, $this->option('perimeters'));
 
         if ($baseControlExists) {
-            $replace["use {$controlNamespace}\Control;\n"] = '';
+            $replace['use Lomkit\Access\Controls\Control;'] = '';
         } else {
-            $replace[' extends Control'] = '';
-            $replace["use {$rootNamespace}Access\Controls\Control;\n"] = '';
+            $replace["use {$controlNamespace}\Control;\n"] = '';
         }
 
         return str_replace(
diff --git a/tests/Support/Models/Model.php b/tests/Support/Models/Model.php
index a105e12..0d4dc34 100644
--- a/tests/Support/Models/Model.php
+++ b/tests/Support/Models/Model.php
@@ -15,7 +15,7 @@ class Model extends BaseModel
     use HasControl;
     use Searchable;
 
-    public static $control = ModelControl::class;
+    protected static string $control = ModelControl::class;
 
     protected static function newFactory()
     {