level 8

Author: saeedvaziry

app/Actions/Database/CreateDatabase.php

@@ -21,8 +21,14 @@ public function create(Server $server, array $input): Database
             'collation' => $input['collation'],
             'name' => $input['name'],
         ]);
+
+        $service = $server->database();
+        if (! $service instanceof \App\Models\Service) {
+            throw new \Exception('Database service not found');
+        }
+
         /** @var \App\SSH\Services\Database\Database $databaseHandler */
-        $databaseHandler = $server->database()->handler();
+        $databaseHandler = $service->handler();
         $databaseHandler->create($database->name, $database->charset, $database->collation);
         $database->status = DatabaseStatus::READY;
         $database->save();

app/Actions/Database/CreateDatabaseUser.php

@@ -26,8 +26,13 @@ public function create(Server $server, array $input, array $links = []): Databas
             'host' => (isset($input['remote']) && $input['remote']) || isset($input['host']) ? $input['host'] : 'localhost',
             'databases' => $links,
         ]);
+        $service = $server->database();
+        if (! $service instanceof \App\Models\Service) {
+            throw new \Exception('Database service not found');
+        }
+
         /** @var Database $databaseHandler */
-        $databaseHandler = $server->database()->handler();
+        $databaseHandler = $service->handler();
         $databaseHandler->createUser(
             $databaseUser->username,
             $databaseUser->password,

app/Actions/Database/DeleteDatabase.php

@@ -9,8 +9,12 @@ class DeleteDatabase
 {
     public function delete(Server $server, Database $database): void
     {
+        $service = $server->database();
+        if (! $service instanceof \App\Models\Service) {
+            throw new \Exception('Database service not found');
+        }
         /** @var \App\SSH\Services\Database\Database $handler */
-        $handler = $server->database()->handler();
+        $handler = $service->handler();
         $handler->delete($database->name);
         $database->delete();
     }

app/Actions/Database/DeleteDatabaseUser.php

@@ -9,8 +9,12 @@ class DeleteDatabaseUser
 {
     public function delete(Server $server, DatabaseUser $databaseUser): void
     {
+        $service = $server->database();
+        if (! $service instanceof \App\Models\Service) {
+            throw new \Exception('Database service not found');
+        }
         /** @var \App\SSH\Services\Database\Database $handler */
-        $handler = $server->database()->handler();
+        $handler = $service->handler();
         $handler->deleteUser($databaseUser->username, $databaseUser->host);
         $databaseUser->delete();
     }

app/Actions/Database/LinkUser.php

@@ -32,8 +32,13 @@ public function link(DatabaseUser $databaseUser, array $input): DatabaseUser
 
         $databaseUser->databases = $input['databases'];
 
+        $service = $databaseUser->server->database();
+        if (! $service) {
+            throw new \Exception('Database service not found');
+        }
+
         /** @var \App\SSH\Services\Database\Database $handler */
-        $handler = $databaseUser->server->database()->handler();
+        $handler = $service->handler();
 
         // Unlink the user from all databases
         $handler->unlink(

app/Actions/Database/RestoreBackup.php

@@ -20,8 +20,12 @@ public function restore(BackupFile $backupFile, array $input): void
         $backupFile->save();
 
         dispatch(function () use ($backupFile, $database): void {
+            $service = $database->server->database();
+            if (! $service) {
+                throw new \Exception('Database service not found');
+            }
             /** @var \App\SSH\Services\Database\Database $databaseHandler */
-            $databaseHandler = $database->server->database()->handler();
+            $databaseHandler = $service->handler();
             $databaseHandler->restoreBackup($backupFile, $database->name);
             $backupFile->status = BackupFileStatus::RESTORED;
             $backupFile->restored_at = now();

app/Actions/Database/RunBackup.php

@@ -21,8 +21,12 @@ public function run(Backup $backup): BackupFile
         $file->save();
 
         dispatch(function () use ($file, $backup): void {
+            $service = $backup->server->database();
+            if (! $service) {
+                throw new \Exception('Database service not found');
+            }
             /** @var Database $databaseHandler */
-            $databaseHandler = $file->backup->server->database()->handler();
+            $databaseHandler = $service->handler();
             $databaseHandler->runBackup($file);
             $file->status = BackupFileStatus::CREATED;
             $file->save();

app/Actions/FirewallRule/ManageRule.php

@@ -67,8 +67,12 @@ public function delete(FirewallRule $rule): void
     protected function applyRule(FirewallRule $rule): void
     {
         try {
+            $service = $rule->server->firewall();
+            if (! $service) {
+                throw new \Exception('Firewall service not found');
+            }
             /** @var Firewall $handler */
-            $handler = $rule->server->firewall()->handler();
+            $handler = $service->handler();
             $handler->applyRules();
         } catch (\Exception $e) {
             $rule->server->firewallRules()

app/Actions/Monitoring/UpdateMetricSettings.php

@@ -12,6 +12,9 @@ class UpdateMetricSettings
     public function update(Server $server, array $input): void
     {
         $service = $server->monitoring();
+        if (! $service instanceof \App\Models\Service) {
+            throw new \Exception('Monitoring service not found');
+        }
         /** @var \App\SSH\Services\ServiceInterface $handler */
         $handler = $service->handler();
         $data = $handler->data();

app/Actions/NodeJS/ChangeDefaultCli.php

@@ -18,10 +18,13 @@ public function change(Server $server, array $input): void
     {
         $this->validate($server, $input);
         $service = $server->nodejs($input['version']);
+        if (! $service instanceof \App\Models\Service) {
+            throw new \Exception('NodeJS service not found');
+        }
         /** @var NodeJS $handler */
         $handler = $service->handler();
         $handler->setDefaultCli();
-        $server->defaultService('nodejs')->update(['is_default' => 0]);
+        $server->defaultService('nodejs')?->update(['is_default' => 0]);
         $service->update(['is_default' => 1]);
         $service->update(['status' => ServiceStatus::READY]);
     }