From ddb3c79657b074b9e3428249a44a66a4f2f06c6d Mon Sep 17 00:00:00 2001
From: Morten Olsen <fbtijfdq@void.black>
Date: Mon, 11 Aug 2025 14:35:15 +0200
Subject: [PATCH] fix pg db

---
 .../postgres-database/portgres-database.schemas.ts            | 1 +
 .../postgres-database/postgres-database.resource.ts           | 4 +++-
 src/services/postgres/postgres.instance.ts                    | 2 ++
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/custom-resouces/postgres-database/portgres-database.schemas.ts b/src/custom-resouces/postgres-database/portgres-database.schemas.ts
index edef139..9b1f1e0 100644
--- a/src/custom-resouces/postgres-database/portgres-database.schemas.ts
+++ b/src/custom-resouces/postgres-database/portgres-database.schemas.ts
@@ -9,6 +9,7 @@ const postgresDatabaseSecretSchema = z.object({
   port: z.string(),
   user: z.string(),
   password: z.string(),
+  database: z.string().optional(),
 });
 
 const postgresDatabaseConnectionSecretSchema = z.object({
diff --git a/src/custom-resouces/postgres-database/postgres-database.resource.ts b/src/custom-resouces/postgres-database/postgres-database.resource.ts
index 9620e6b..24a0dc9 100644
--- a/src/custom-resouces/postgres-database/postgres-database.resource.ts
+++ b/src/custom-resouces/postgres-database/postgres-database.resource.ts
@@ -95,6 +95,7 @@ class PostgresDatabaseResource extends CustomResource<typeof postgresDatabaseSpe
       port: serverSecretData.data.port,
       user: this.#userName,
       database: this.#dbName,
+      ...databaseSecretData.data,
     };
 
     if (!isDeepSubset(databaseSecretData.data, expectedSecret)) {
@@ -132,7 +133,7 @@ class PostgresDatabaseResource extends CustomResource<typeof postgresDatabaseSpe
       };
     }
 
-    const secretData = postgresDatabaseConnectionSecretSchema.safeParse(decodeSecret(this.#serverSecret.current?.data));
+    const secretData = postgresDatabaseConnectionSecretSchema.safeParse(decodeSecret(this.#databaseSecret.data));
     if (!secretData.success || !secretData.data) {
       return {
         ready: false,
@@ -145,6 +146,7 @@ class PostgresDatabaseResource extends CustomResource<typeof postgresDatabaseSpe
     const database = postgresService.get({
       ...connectionSecretData.data,
       port: connectionSecretData.data.port ? Number(connectionSecretData.data.port) : 5432,
+      database: connectionSecretData.data.database,
     });
     await database.upsertRole({
       name: secretData.data.user,
diff --git a/src/services/postgres/postgres.instance.ts b/src/services/postgres/postgres.instance.ts
index d40d000..f347ad9 100644
--- a/src/services/postgres/postgres.instance.ts
+++ b/src/services/postgres/postgres.instance.ts
@@ -10,6 +10,7 @@ type PostgresInstanceOptions = {
   port?: number;
   user: string;
   password: string;
+  database?: string;
 };
 
 class PostgresInstance {
@@ -23,6 +24,7 @@ class PostgresInstance {
         user: process.env.FORCE_PG_USER ?? options.user,
         password: process.env.FORCE_PG_PASSWORD ?? options.password,
         port: process.env.FORCE_PG_PORT ? parseInt(process.env.FORCE_PG_PORT) : options.port,
+        database: options.database,
       },
     });
   }