From 484117f6cd24b9d99bf8b7a2ccca776e34df5b56 Mon Sep 17 00:00:00 2001
From: Brandon <brandon@tendency.me>
Date: Tue, 20 Aug 2024 06:18:12 -0500
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=91=E2=80=8D=F0=9F=92=BB=20Add=20suppo?=
 =?UTF-8?q?rt=20for=20overriding=20the=20plugin=20models?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Livewire/MenuItems.php | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/Livewire/MenuItems.php b/src/Livewire/MenuItems.php
index 9cb1700..9aa188a 100644
--- a/src/Livewire/MenuItems.php
+++ b/src/Livewire/MenuItems.php
@@ -7,7 +7,6 @@
 use Datlechin\FilamentMenuBuilder\Enums\LinkTarget;
 use Datlechin\FilamentMenuBuilder\FilamentMenuBuilderPlugin;
 use Datlechin\FilamentMenuBuilder\Models\Menu;
-use Datlechin\FilamentMenuBuilder\Models\MenuItem;
 use Filament\Actions\Action;
 use Filament\Actions\Concerns\InteractsWithActions;
 use Filament\Actions\Contracts\HasActions;
@@ -48,7 +47,7 @@ public function reorder(array $order, ?string $parentId = null): void
             return;
         }
 
-        MenuItem::query()
+        FilamentMenuBuilderPlugin::get()->getMenuItemModel()::query()
             ->whereIn('id', $order)
             ->update([
                 'order' => DB::raw(
@@ -80,7 +79,7 @@ public function editAction(): Action
             ->size(ActionSize::Small)
             ->modalHeading(fn (array $arguments): string => __('filament-actions::edit.single.modal.heading', ['label' => $arguments['title']]))
             ->icon('heroicon-m-pencil-square')
-            ->fillForm(fn (array $arguments): array => MenuItem::query()
+            ->fillForm(fn (array $arguments): array => FilamentMenuBuilderPlugin::get()->getMenuItemModel()::query()
                 ->where('id', $arguments['id'])
                 ->with('linkable')
                 ->first()
@@ -108,7 +107,7 @@ public function editAction(): Action
                     ->schema(FilamentMenuBuilderPlugin::get()->getMenuItemFields()),
             ])
             ->action(
-                fn (array $data, array $arguments) => MenuItem::query()
+                fn (array $data, array $arguments) => FilamentMenuBuilderPlugin::get()->getMenuItemModel()::query()
                     ->where('id', $arguments['id'])
                     ->update($data),
             )
@@ -130,7 +129,7 @@ public function deleteAction(): Action
             ->modalSubmitActionLabel(__('filament-actions::delete.single.modal.actions.delete.label'))
             ->modalIcon(FilamentIcon::resolve('actions::delete-action.modal') ?? 'heroicon-o-trash')
             ->action(function (array $arguments): void {
-                $menuItem = MenuItem::query()->where('id', $arguments['id'])->first();
+                $menuItem = FilamentMenuBuilderPlugin::get()->getMenuItemModel()::query()->where('id', $arguments['id'])->first();
 
                 $menuItem?->delete();
             });