Merge pull request #570 from mit-27/update-with-mapper-connectorUpdate

Update connectorUpdate script and fix doc
panoratech · Jul 27, 2024 · 5bd577b · 5bd577b
2 parents 14d515d + ad63865
commit 5bd577b
Showing 1 changed file with 32 additions and 3 deletions.
diff --git a/packages/api/scripts/connectorUpdate.js b/packages/api/scripts/connectorUpdate.js
@@ -231,7 +231,7 @@ function updateArrayInFile(filePath, arrayName, newArray) {
   updateFileContents(filePath, newContents);
 }
 
-function updateModuleFile(moduleFile, newServiceDirs) {
+function updateModuleFileForService(moduleFile, newServiceDirs) {
   let moduleFileContent = fs.readFileSync(moduleFile, 'utf8');
 
   // Generate and insert new service imports
@@ -258,6 +258,34 @@ function updateModuleFile(moduleFile, newServiceDirs) {
   fs.writeFileSync(moduleFile, moduleFileContent);
 }
 
+function updateModuleFileForMapper(moduleFile, newServiceDirs, objectType) {
+  let moduleFileContent = fs.readFileSync(moduleFile, 'utf8');
+  objectType = objectType.charAt(0).toUpperCase() + objectType.slice(1);
+
+  // Generate and insert new service imports
+  newServiceDirs.forEach((serviceName) => {
+    const mapperClass =
+      serviceName.charAt(0).toUpperCase() + serviceName.slice(1) + objectType + 'Mapper';
+    const importStatement = `import { ${mapperClass} } from './services/${serviceName}/mappers';\n`;
+    if (!moduleFileContent.includes(importStatement)) {
+      moduleFileContent = importStatement + moduleFileContent;
+    }
+
+    // Add new service to the providers array if not already present
+    const providerRegex = /providers: \[\n([\s\S]*?)\n  \],/;
+    const match = moduleFileContent.match(providerRegex);
+    if (match && !match[1].includes(mapperClass)) {
+      const updatedProviders = match[1] + `\n    ${mapperClass},\n`;
+      moduleFileContent = moduleFileContent.replace(
+        providerRegex,
+        `providers: [\n${updatedProviders}  ],`,
+      );
+    }
+  });
+
+  fs.writeFileSync(moduleFile, moduleFileContent);
+}
+
 function updateEnumFile(enumFilePath, newServiceDirs, vertical) {
   let fileContent = fs.readFileSync(enumFilePath, 'utf8');
   const base = vertical.substring(0, 1).toUpperCase() + vertical.substring(1);
@@ -328,7 +356,7 @@ function updateInitSQLFile(initSQLFile, newServiceDirs, vertical) {
   let newLines = newServiceDirs
     .map((serviceName) => {
       const columnName = `${vertical.toLowerCase()}_${serviceName.toLowerCase()}`;
-      return ` ${columnName} boolean NOT NULL,\n`;
+      return ` ${columnName} boolean NULL,\n`;
     })
     .join('');
 
@@ -439,7 +467,8 @@ function updateObjectTypes(baseDir, objectType, vertical) {
     `../src/${vertical}/${objectType.toLowerCase()}/${objectType.toLowerCase()}.module.ts`,
   );
 
-  updateModuleFile(moduleFile, newServiceDirs, servicesDir);
+  updateModuleFileForService(moduleFile, newServiceDirs);
+  updateModuleFileForMapper(moduleFile, newServiceDirs, objectType)
 
   // Path to the mappings file
   // const mappingsFile = path.join(