fixed repo_config bug on setup

pages/api/dpu/setup.ts

@@ -1,5 +1,7 @@
 import { NextApiRequest, NextApiResponse } from 'next';
 import { SetupReposArgs, removePreviousInstallations, saveSetupReposInDb } from '../../../utils/db/setupRepos';
+import { getUserIdByTopicName } from '../../../utils/db/users';
+import { insertRepoConfig } from '../../../utils/db/repos';
 
 const setupHandler = async (req: NextApiRequest, res: NextApiResponse) => {
 	console.info("[setupHandler]Saving setup info in db...");
@@ -10,17 +12,26 @@ const setupHandler = async (req: NextApiRequest, res: NextApiResponse) => {
 		return;
 	}
 	try {
-		await removePreviousInstallations(jsonBody.installationId)	
+		await removePreviousInstallations(jsonBody.installationId)
 	} catch (err) {
 		console.error(`[setupHandler] Unable to remove previous installations for ${jsonBody.installationId}`, err);
-		res.status(500).json({"error": "Internal Server Error"});
+		res.status(500).json({ "error": "Internal Server Error" });
+		return;
+	}
+	// get user_id for the given install_id
+	const userId = await getUserIdByTopicName(jsonBody.installationId).catch((error: any) => {
+		console.error("[setupHandler/getUserIdByTopicName] Failed to fetch userId from the database.", error);
+	});
+	if (!userId) {
+		console.error(`[setupHandler/getUserIdByTopicName] NO userId found for topic name: ${jsonBody.installationId} from database.`);
+		res.status(404).json({ "error": "No userId found for given installationId" });
 		return;
 	}
 	const allSetupReposPromises = [];
 	for (const ownerInfo of jsonBody.info) {
 		let setupReposArgs: SetupReposArgs = {
 			repo_owner: ownerInfo.owner,
-			repo_provider:  ownerInfo.provider,
+			repo_provider: ownerInfo.provider,
 			repo_names: ownerInfo.repos,
 			install_id: jsonBody.installationId
 		}
@@ -30,8 +41,23 @@ const setupHandler = async (req: NextApiRequest, res: NextApiResponse) => {
 			});
 		allSetupReposPromises.push(saveSetupReposPromises);
 	}
-	await Promise.all(allSetupReposPromises).then((values) => {
+	await Promise.all(allSetupReposPromises).then(async (values) => {
 		console.info("[setupHandler] All setup info saved succesfully...")
+		// for repos in ownerInfo, save default repo_config value in repo_config table
+		for (const ownerInfo of jsonBody.info) {
+			await insertRepoConfig(ownerInfo.owner, ownerInfo.repos, userId, ownerInfo.provider)
+			.then((queryResponse: any) => {
+				if (queryResponse) {
+					console.info(`[setupHandler/insertRepoConfigOnSetup] repo config info saved succesfully for repos: ${ownerInfo.repos} for owner: ${ownerInfo.owner}`);
+				}
+				else {
+					console.error(`[setupHandler/insertRepoConfigOnSetup] Failed to save repo config info for repos: ${ownerInfo.repos} for owner: ${ownerInfo.owner}`);
+				}
+			})
+			.catch((error: Error) => {
+				console.error(`[setupHandler/insertRepoConfigOnSetup] Failed to save repo config for repos: ${ownerInfo.repos} for owner: ${ownerInfo.owner}`, error);
+			})
+		}
 		res.status(200).send("Ok");
 		return;
 	}).catch((error) => {

utils/db/repos.ts

@@ -170,3 +170,29 @@ export const getRepoConfigByUserAndRepo = async (provider: string, repoName: str
 	}
 	return userRows[0].config;
 }
+export const insertRepoConfig = async (repoOwner: string, repoNames: string[], userId: string, provider: string) => {
+	// Construct the VALUES clause dynamically
+	const valuesClause = repoNames.map(repoName => `('${repoName}', '${repoOwner}', '${provider}')`).join(',');
+	const query = `
+		INSERT INTO repo_config (repo_id, user_id, auto_assign, comment_setting)
+		SELECT id, $1, true, true
+		FROM public.repos
+		WHERE (repo_name, repo_owner, repo_provider) IN (${valuesClause})
+		ON CONFLICT DO NOTHING
+		`;
+	const params = [userId];
+
+	const queryIsSuccessful = await conn.query(query, params)
+	.then((dbResponse) => {
+		if (dbResponse.rowCount == 0) {
+			return false;
+		}
+		return true;
+	})
+	.catch((err: Error) => {
+		console.error(`[db/insertRepoConfigOnSetup] Could not insert repo config for the repos: ${repoNames}`, { pg_query: query }, err);
+		return false;
+	})
+	return queryIsSuccessful;
+  }
+

utils/db/users.ts

@@ -250,4 +250,24 @@ export const getAuthInfoFromDb = async function (user_id: string): Promise<AuthI
 			return { authInfo: null };
 		});
 	return authinfo_promise.authInfo;
-}
+}
+
+export const getUserIdByTopicName = async function(topic_name: string) {
+	const query = `SELECT id FROM users WHERE topic_name = $1`;
+	const params = [topic_name];
+
+	const queryResult = await conn.query(query, params).catch((err) => {
+		console.error(`[getUserIdByTopicName] Query failed: Select from users where topic_name = ${topic_name}`, { pg_query: query }, err);
+    });
+
+	if (queryResult && queryResult.rowCount === 0) {
+		console.error(`[getUserIdByTopicName] No user found for topicName ${topic_name}`);
+		return null;
+	} else if (queryResult && queryResult.rowCount > 1) {
+		console.error(`[getUserIdByTopicName] Multiple users found for topicName ${topic_name}`);
+		return null;
+	} else {
+		return queryResult?.rows[0].id;
+	}
+  }
+