From 16fcab85bb158bc491970e2430d3622b00aec4a8 Mon Sep 17 00:00:00 2001
From: Vitalii Yarmus <mr.yarmus@gmail.com>
Date: Tue, 21 Apr 2026 17:38:12 +0300
Subject: [PATCH] HCK-15809: extend dbtProvider to support certain config
 options

---
 forward_engineering/dbtProvider.js | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/forward_engineering/dbtProvider.js b/forward_engineering/dbtProvider.js
index f66c6fb..a9171c6 100644
--- a/forward_engineering/dbtProvider.js
+++ b/forward_engineering/dbtProvider.js
@@ -80,6 +80,17 @@ class DbtProvider {
 	getColumnConstraints({ columnDefinition, jsonSchema }) {
 		return getColumnConstraints({ columnDefinition });
 	}
+
+	/**
+	 * @param {{ modelData: object[]; containerData: object[]; entityData: object[];}}
+	 * @returns {{ databaseName?: string, schemaName?: string }}
+	 */
+	getEntityProperties({ modelData, containerData, entityData }) {
+		return {
+			databaseName: containerData?.[0]?.databaseName,
+			schemaName: containerData?.[0]?.code ?? containerData?.[0]?.name,
+		};
+	}
 }
 
 module.exports = DbtProvider;