Merge pull request #13 from Portabase/feat/postgres-cluster

Feat/postgres cluster

Author: Asuniia

cc/.env

@@ -126,6 +126,7 @@ def agent(
                     choices=[
                         "back",
                         "postgresql",
+                        "postgresql-cluster",
                         "mysql",
                         "mariadb",
                         "sqlite",
@@ -144,6 +145,14 @@ def agent(
                 if not db_type:
                     raise typer.Exit()
 
+                if db_type == "postgresql-cluster":
+                    console.print(
+                        "[warning]⚠ Postgres Cluster requires a superuser. "
+                        "Cluster backup/restore uses pg_dumpall, which dumps all "
+                        "databases and global objects (roles, tablespaces). "
+                        "The provided user must be a Postgres superuser.[/warning]"
+                    )
+
                 friendly_name = Prompt.ask("Display Name", default="External DB")
 
                 if db_type == "sqlite":
@@ -169,7 +178,7 @@ def agent(
                     port = IntPrompt.ask(
                         "Port",
                         default=5432
-                        if db_type == "postgresql"
+                        if db_type in ["postgresql", "postgresql-cluster"]
                         else (
                             3050
                             if db_type == "firebird"
@@ -208,6 +217,7 @@ def agent(
                     choices=[
                         "back",
                         "postgresql",
+                        "postgresql-cluster",
                         "mysql",
                         "mariadb",
                         "sqlite",
@@ -265,7 +275,7 @@ def agent(
                         f"[success]✔ Added SQLite database ({db_name})[/success]"
                     )
 
-                elif db_engine == "postgresql":
+                elif db_engine in ["postgresql", "postgresql-cluster"]:
                     pg_port = get_free_port()
                     db_user = "admin"
                     db_pass = generate_password(16)
@@ -295,7 +305,7 @@ def agent(
                         {
                             "name": db_name,
                             "database": db_name,
-                            "type": "postgresql",
+                            "type": db_engine,
                             "username": db_user,
                             "password": db_pass,
                             "port": 5432,
@@ -304,8 +314,13 @@ def agent(
                         },
                     )
 
+                    label = (
+                        "Postgres Cluster"
+                        if db_engine == "postgresql-cluster"
+                        else "Postgres"
+                    )
                     console.print(
-                        f"[success]✔ Added Postgres container (Port {pg_port})[/success]"
+                        f"[success]✔ Added {label} container (Port {pg_port})[/success]"
                     )
 
                 elif db_engine == "mariadb" or db_engine == "mysql":

cc/databases.json

@@ -93,6 +93,7 @@ def add_db(name: str = typer.Argument(..., help="Name of the agent")):
                 choices=[
                     "back",
                     "postgresql",
+                    "postgresql-cluster",
                     "mysql",
                     "mariadb",
                     "sqlite",
@@ -109,6 +110,14 @@ def add_db(name: str = typer.Argument(..., help="Name of the agent")):
             if not db_type:
                 raise typer.Exit()
 
+            if db_type == "postgresql-cluster":
+                console.print(
+                    "[warning]⚠ Postgres Cluster requires a superuser. "
+                    "Cluster backup/restore uses pg_dumpall, which dumps all "
+                    "databases and global objects (roles, tablespaces). "
+                    "The provided user must be a Postgres superuser.[/warning]"
+                )
+
             friendly_name = Prompt.ask("Display Name", default="External DB")
 
             if db_type == "sqlite":
@@ -125,7 +134,7 @@ def add_db(name: str = typer.Argument(..., help="Name of the agent")):
                 port = IntPrompt.ask(
                     "Port",
                     default=5432
-                    if db_type == "postgresql"
+                    if db_type in ["postgresql", "postgresql-cluster"]
                     else (
                         3050
                         if db_type == "firebird"
@@ -158,6 +167,7 @@ def add_db(name: str = typer.Argument(..., help="Name of the agent")):
                 choices=[
                     "back",
                     "postgresql",
+                    "postgresql-cluster",
                     "mysql",
                     "mariadb",
                     "sqlite",
@@ -246,7 +256,7 @@ def add_db(name: str = typer.Argument(..., help="Name of the agent")):
                 )
                 console.print(f"[success]✔ Added SQLite database ({db_name})[/success]")
 
-            elif db_engine == "postgresql":
+            elif db_engine in ["postgresql", "postgresql-cluster"]:
                 db_port = get_free_port()
                 db_user = "admin"
                 db_pass = generate_password(16)
@@ -465,7 +475,7 @@ def add_db(name: str = typer.Argument(..., help="Name of the agent")):
                     "username": "sa" if db_engine == "mssql" else db_user,
                     "password": db_pass,
                     "port": 5432
-                    if db_engine == "postgresql"
+                    if db_engine in ["postgresql", "postgresql-cluster"]
                     else (
                         3050
                         if db_engine == "firebird"