From 817316de60c03582d5cb481546beed3cba5fc053 Mon Sep 17 00:00:00 2001
From: Vitalii Yarmus <mr.yarmus@gmail.com>
Date: Tue, 21 Apr 2026 17:37:26 +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 63c538d..0b5342e 100644
--- a/forward_engineering/dbtProvider.js
+++ b/forward_engineering/dbtProvider.js
@@ -64,6 +64,17 @@ class DbtProvider {
 	getColumnConstraints({ columnDefinition, jsonSchema }) {
 		return getColumnConstraints({ columnDefinition, jsonSchema });
 	}
+
+	/**
+	 * @param {{ modelData: object[]; containerData: object[]; entityData: object[];}}
+	 * @returns {{ databaseName?: string, schemaName?: string }}
+	 */
+	getEntityProperties({ modelData, containerData, entityData }) {
+		return {
+			databaseName: modelData?.[0]?.database_name,
+			schemaName: containerData?.[0]?.code ?? containerData?.[0]?.name,
+		};
+	}
 }
 
 module.exports = DbtProvider;