From 77edc427ea1a314a2651b360230955e990d336dd Mon Sep 17 00:00:00 2001
From: Nazar Kovtun <nazar.kovtun@hackolade.com>
Date: Tue, 5 Nov 2024 15:54:47 +0200
Subject: [PATCH] HCK-8622: fixed comments

---
 .../databaseService/databaseService.js        | 19 +++++-------
 .../databaseService/helpers/connection.js     | 30 +++++++++++--------
 reverse_engineering/helpers/logInfo.js        | 14 ++++++++-
 3 files changed, 39 insertions(+), 24 deletions(-)

diff --git a/reverse_engineering/databaseService/databaseService.js b/reverse_engineering/databaseService/databaseService.js
index b7bba22..9cc8866 100644
--- a/reverse_engineering/databaseService/databaseService.js
+++ b/reverse_engineering/databaseService/databaseService.js
@@ -2,7 +2,6 @@ const axios = require('axios');
 const sql = require('mssql');
 const https = require('https');
 const { getObjectsFromDatabase, getNewConnectionClientByDb } = require('./helpers');
-const msal = require('@azure/msal-node');
 const getSampleDocSize = require('../helpers/getSampleDocSize');
 const { logAuthTokenInfo } = require('../helpers/logInfo');
 const { getConnection } = require('./helpers/connection');
@@ -34,16 +33,14 @@ const getConnectionClient = async (connectionInfo, logger) => {
 	const redirectUri = 'http://localhost:8080';
 
 	const connection = getConnection({
-		type: connectionInfo.authMethod,
-		data: {
-			connectionInfo,
-			commonConfig,
-			credentialsConfig,
-			tenantId,
-			clientId,
-			redirectUri,
-			logger,
-		},
+		authMethod: connectionInfo.authMethod,
+		connectionInfo,
+		commonConfig,
+		credentialsConfig,
+		tenantId,
+		clientId,
+		redirectUri,
+		logger,
 	});
 
 	return connection.connect();
diff --git a/reverse_engineering/databaseService/helpers/connection.js b/reverse_engineering/databaseService/helpers/connection.js
index 6c6b684..8eae7e5 100644
--- a/reverse_engineering/databaseService/helpers/connection.js
+++ b/reverse_engineering/databaseService/helpers/connection.js
@@ -2,17 +2,9 @@ const axios = require('axios');
 const sql = require('mssql');
 const https = require('https');
 const msal = require('@azure/msal-node');
-const { logAuthTokenInfo } = require('../../helpers/logInfo');
+const { logAuthTokenInfo, logConnectionHostAndUsername } = require('../../helpers/logInfo');
 const { prepareError } = require('./errorService');
 
-const logConnectionHostAndUsername = ({ hostname, username, authMethod, logger }) => {
-	logger.log(
-		'info',
-		`hostname: ${hostname ?? 'absent'}, username: ${username ?? 'absent'}, auth method: ${authMethod}`,
-		'Auth info',
-	);
-};
-
 class Connection {
 	constructor({ logger }) {
 		this.logger = logger;
@@ -187,7 +179,7 @@ class AzureActiveDirectoryMFAConnection extends Connection {
 		}
 	}
 
-	async #getAuthConfig() {
+	#getAuthConfig() {
 		return {
 			system: {
 				loggerOptions: {
@@ -243,8 +235,22 @@ class AzureActiveDirectoryUsernamePasswordConnection extends Connection {
 	}
 }
 
-const getConnection = ({ type, data }) => {
-	switch (type) {
+/**
+ *
+ * @param {{
+ * 	authMethod,
+ * 	connectionInfo,
+ *  commonConfig,
+ *	credentialsConfig,
+ *	tenantId,
+ * 	clientId,
+ * 	redirectUri,
+ *	logger,
+ * }} param
+ * @returns {Promise<object>}
+ */
+const getConnection = ({ authMethod, ...data }) => {
+	switch (authMethod) {
 		case 'Username / Password':
 			return new UsernamePasswordConnection(data);
 		case 'Username / Password (Windows)':
diff --git a/reverse_engineering/helpers/logInfo.js b/reverse_engineering/helpers/logInfo.js
index a0740a8..00ee977 100644
--- a/reverse_engineering/helpers/logInfo.js
+++ b/reverse_engineering/helpers/logInfo.js
@@ -55,7 +55,19 @@ const logAuthTokenInfo = ({ token, logger }) => {
 	logger.log('info', { token }, `MFA token is of type ${tokenType}${tokenLength}`, ['token']);
 };
 
+const logConnectionHostAndUsername = ({ hostname, username, authMethod, logger }) => {
+	const hostnameToDisplay = hostname ?? 'absent';
+	const usernameToDisplay = username ?? 'absent';
+
+	logger.log(
+		'info',
+		`hostname: ${hostnameToDisplay}, username: ${usernameToDisplay}, auth method: ${authMethod}`,
+		'Auth info',
+	);
+};
+
 module.exports = {
-	logAuthTokenInfo,
 	logInfo,
+	logAuthTokenInfo,
+	logConnectionHostAndUsername,
 };