From 3a25d0f976050494fba85e20fd0b213d416789cf Mon Sep 17 00:00:00 2001
From: MartinOscar <40749467+RMartinOscar@users.noreply.github.com>
Date: Thu, 16 Jan 2025 11:50:08 +0100
Subject: [PATCH] Actually use nodeUpdateService not only for keys (#914)

* Actually use nodeUpdateService not only for keys

* Add behind proxy & ignore panel config updates

* Don't Halt

* Prevent double notification

* Revert "Add behind proxy & ignore panel config updates"

This reverts commit 0147888c6cbda54be3375ee91ba8600e0d238b78.
---
 .../Resources/NodeResource/Pages/EditNode.php | 48 ++++++++++++++++++-
 1 file changed, 46 insertions(+), 2 deletions(-)

diff --git a/app/Filament/Admin/Resources/NodeResource/Pages/EditNode.php b/app/Filament/Admin/Resources/NodeResource/Pages/EditNode.php
index d5e1c4e2b..7078b15b5 100644
--- a/app/Filament/Admin/Resources/NodeResource/Pages/EditNode.php
+++ b/app/Filament/Admin/Resources/NodeResource/Pages/EditNode.php
@@ -7,6 +7,7 @@ use App\Models\Node;
 use App\Services\Helpers\SoftwareVersionService;
 use App\Services\Nodes\NodeAutoDeployService;
 use App\Services\Nodes\NodeUpdateService;
+use Exception;
 use Filament\Actions;
 use Filament\Forms;
 use Filament\Forms\Components\Actions as FormActions;
@@ -25,6 +26,7 @@ use Filament\Forms\Set;
 use Filament\Notifications\Notification;
 use Filament\Resources\Pages\EditRecord;
 use Filament\Support\Enums\Alignment;
+use Illuminate\Database\Eloquent\Model;
 use Illuminate\Support\HtmlString;
 use Webbingbrasil\FilamentCopyActions\Forms\Actions\CopyAction;
 
@@ -32,6 +34,15 @@ class EditNode extends EditRecord
 {
     protected static string $resource = NodeResource::class;
 
+    private bool $errored = false;
+
+    private NodeUpdateService $nodeUpdateService;
+
+    public function boot(NodeUpdateService $nodeUpdateService): void
+    {
+        $this->nodeUpdateService = $nodeUpdateService;
+    }
+
     public function form(Forms\Form $form): Forms\Form
     {
         return $form->schema([
@@ -552,8 +563,8 @@ class EditNode extends EditRecord
                                             ->requiresConfirmation()
                                             ->modalHeading('Reset Daemon Token?')
                                             ->modalDescription('Resetting the daemon token will void any request coming from the old token. This token is used for all sensitive operations on the daemon including server creation and deletion. We suggest changing this token regularly for security.')
-                                            ->action(function (NodeUpdateService $nodeUpdateService, Node $node) {
-                                                $nodeUpdateService->handle($node, [], true);
+                                            ->action(function (Node $node) {
+                                                $this->nodeUpdateService->handle($node, [], true);
                                                 Notification::make()->success()->title('Daemon Key Reset')->send();
                                                 $this->fillForm();
                                             }),
@@ -583,6 +594,39 @@ class EditNode extends EditRecord
         return $data;
     }
 
+    protected function handleRecordUpdate(Model $record, array $data): Model
+    {
+        if (!$record instanceof Node) {
+            return $record;
+        }
+
+        try {
+            unset($data['unlimited_mem'], $data['unlimited_disk'], $data['unlimited_cpu']);
+
+            return $this->nodeUpdateService->handle($record, $data);
+        } catch (Exception $exception) {
+            $this->errored = true;
+            Notification::make()
+                ->title('Error connecting to the node')
+                ->body('The configuration could not be automatically updated on Wings, you will need to manually update the configuration file.')
+                ->color('warning')
+                ->icon('tabler-database')
+                ->warning()
+                ->send();
+
+            return parent::handleRecordUpdate($record, $data);
+        }
+    }
+
+    protected function getSavedNotification(): ?Notification
+    {
+        if ($this->errored) {
+            return null;
+        }
+
+        return parent::getSavedNotification();
+    }
+
     protected function getFormActions(): array
     {
         return [];