From b17be37aee22ae807da978d77cd98dafa4e40738 Mon Sep 17 00:00:00 2001
From: Andras Bacsai <andras.bacsai@gmail.com>
Date: Fri, 14 Jun 2024 14:09:56 +0200
Subject: [PATCH] fix: db proxy status shown better in the UI

---
 app/Actions/Database/StopDatabaseProxy.php    |  2 ++
 app/Livewire/Project/Database/Heading.php     |  1 -
 .../Project/Database/Postgresql/General.php   | 27 ++++++++++++++-----
 3 files changed, 22 insertions(+), 8 deletions(-)

diff --git a/app/Actions/Database/StopDatabaseProxy.php b/app/Actions/Database/StopDatabaseProxy.php
index 984225435..1b262c898 100644
--- a/app/Actions/Database/StopDatabaseProxy.php
+++ b/app/Actions/Database/StopDatabaseProxy.php
@@ -2,6 +2,7 @@
 
 namespace App\Actions\Database;
 
+use App\Events\DatabaseStatusChanged;
 use App\Models\ServiceDatabase;
 use App\Models\StandaloneClickhouse;
 use App\Models\StandaloneDragonfly;
@@ -28,5 +29,6 @@ class StopDatabaseProxy
         instant_remote_process(["docker rm -f {$uuid}-proxy"], $server);
         $database->is_public = false;
         $database->save();
+        DatabaseStatusChanged::dispatch();
     }
 }
diff --git a/app/Livewire/Project/Database/Heading.php b/app/Livewire/Project/Database/Heading.php
index 61dafa76f..ae88ac12b 100644
--- a/app/Livewire/Project/Database/Heading.php
+++ b/app/Livewire/Project/Database/Heading.php
@@ -12,7 +12,6 @@ use App\Actions\Database\StartPostgresql;
 use App\Actions\Database\StartRedis;
 use App\Actions\Database\StopDatabase;
 use App\Actions\Docker\GetContainersStatus;
-use App\Jobs\ContainerStatusJob;
 use Livewire\Component;
 
 class Heading extends Component
diff --git a/app/Livewire/Project/Database/Postgresql/General.php b/app/Livewire/Project/Database/Postgresql/General.php
index 38cac2e5c..a950e5ef9 100644
--- a/app/Livewire/Project/Database/Postgresql/General.php
+++ b/app/Livewire/Project/Database/Postgresql/General.php
@@ -25,7 +25,17 @@ class General extends Component
 
     public ?string $db_url_public = null;
 
-    protected $listeners = ['refresh', 'save_init_script', 'delete_init_script'];
+    public function getListeners()
+    {
+        $userId = auth()->user()->id;
+
+        return [
+            "echo-private:user.{$userId},DatabaseStatusChanged" => 'database_stopped',
+            "refresh",
+            "save_init_script",
+            "delete_init_script",
+        ];
+    }
 
     protected $rules = [
         'database.name' => 'required',
@@ -68,11 +78,14 @@ class General extends Component
         }
         $this->server = data_get($this->database, 'destination.server');
     }
-
+    public function database_stopped()
+    {
+        $this->dispatch('success', 'Database proxy stopped. Database is no longer publicly accessible.');
+    }
     public function instantSaveAdvanced()
     {
         try {
-            if (! $this->server->isLogDrainEnabled()) {
+            if (!$this->server->isLogDrainEnabled()) {
                 $this->database->is_log_drain_enabled = false;
                 $this->dispatch('error', 'Log drain is not enabled on the server. Please enable it first.');
 
@@ -89,14 +102,14 @@ class General extends Component
     public function instantSave()
     {
         try {
-            if ($this->database->is_public && ! $this->database->public_port) {
+            if ($this->database->is_public && !$this->database->public_port) {
                 $this->dispatch('error', 'Public port is required.');
                 $this->database->is_public = false;
 
                 return;
             }
             if ($this->database->is_public) {
-                if (! str($this->database->status)->startsWith('running')) {
+                if (!str($this->database->status)->startsWith('running')) {
                     $this->dispatch('error', 'Database must be started to be publicly accessible.');
                     $this->database->is_public = false;
 
@@ -112,7 +125,7 @@ class General extends Component
             }
             $this->database->save();
         } catch (\Throwable $e) {
-            $this->database->is_public = ! $this->database->is_public;
+            $this->database->is_public = !$this->database->is_public;
 
             return handleError($e, $this);
         }
@@ -157,7 +170,7 @@ class General extends Component
 
             return;
         }
-        if (! isset($this->database->init_scripts)) {
+        if (!isset($this->database->init_scripts)) {
             $this->database->init_scripts = [];
         }
         $this->database->init_scripts = array_merge($this->database->init_scripts, [