Wrap into main method

coveo · Sep 22, 2023 · 9a07d73 · 9a07d73
1 parent f4a80f4
commit 9a07d73
Show file tree

Hide file tree

Showing 3 changed files with 117 additions and 93 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/ui/atomic/create-atomic-component/index.js b/packages/ui/atomic/create-atomic-component/index.js
@@ -27,6 +27,20 @@ class SerializableAggregateError extends AggregateError {
     };
   }
 }
+class InvalidProjectDirectory extends Error {
+  name = 'Invalid project directory';
+  constructor(message) {
+    super(message);
+  }
+
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+    };
+  }
+}
 
 class InvalidProjectDirectory extends Error {
   name = 'Invalid project directory';
@@ -140,52 +154,56 @@ const __dirname = dirname(fileURLToPath(import.meta.url));
 const templateRelativeDir = 'template';
 const templateDirPath = resolve(__dirname, templateRelativeDir);
 
-const cwdFiles = readdirSync(cwd(), {withFileTypes: true});
-if (cwdFiles.length > 0) {
-  if (cwdFiles.some((dirent) => dirent.name === 'package.json')) {
-    return;
-  } else {
-    handleErrors(
-      new InvalidProjectDirectory(
-        'Current working directory is either not empty or not an npm project (no package.json found). Please try again in an empty directory.'
-      )
-    );
+const main = () => {
+  const cwdFiles = readdirSync(cwd(), {withFileTypes: true});
+  if (cwdFiles.length > 0) {
+    if (cwdFiles.some((dirent) => dirent.name === 'package.json')) {
+      return;
+    } else {
+      handleErrors(
+        new InvalidProjectDirectory(
+          'Current working directory is either not empty or not an npm project (no package.json found). Please try again in an empty directory.'
+        )
+      );
+    }
   }
-}
 
-cpSync(templateDirPath, cwd(), {
-  recursive: true,
-});
-
-let componentName = process.argv[2];
-if (componentName) {
-  ensureComponentValidity(componentName);
-  const transformers = [
-    {
-      srcPath: 'src/components/sample-component',
-      destPath: `src/components/${componentName}`,
-    },
-    {
-      srcPath: `src/components/${componentName}/src/sample-component.tsx`,
-      destPath: `src/components/${componentName}/src/${componentName}.tsx`,
-      transform: (text) =>
-        text
-          .replaceAll(/sample-component/g, componentName)
-          .replaceAll(/SampleComponent/g, camelize(componentName)),
-    },
-    {
-      srcPath: `src/components/${componentName}/src/sample-component.css`,
-      destPath: `src/components/${componentName}/src/${componentName}.css`,
-    },
-    {
-      srcPath: `src/components/${componentName}/package.json`,
-      destPath: `src/components/${componentName}/package.json`,
-      transform: (text) =>
-        text.replaceAll(/(@coveo\/)?sample-component/g, componentName),
-    },
-  ];
-
-  transform(transformers);
-}
+  cpSync(templateDirPath, cwd(), {
+    recursive: true,
+  });
+
+  let componentName = process.argv[2];
+  if (componentName) {
+    ensureComponentValidity(componentName);
+    const transformers = [
+      {
+        srcPath: 'src/components/sample-component',
+        destPath: `src/components/${componentName}`,
+      },
+      {
+        srcPath: `src/components/${componentName}/src/sample-component.tsx`,
+        destPath: `src/components/${componentName}/src/${componentName}.tsx`,
+        transform: (text) =>
+          text
+            .replaceAll(/sample-component/g, componentName)
+            .replaceAll(/SampleComponent/g, camelize(componentName)),
+      },
+      {
+        srcPath: `src/components/${componentName}/src/sample-component.css`,
+        destPath: `src/components/${componentName}/src/${componentName}.css`,
+      },
+      {
+        srcPath: `src/components/${componentName}/package.json`,
+        destPath: `src/components/${componentName}/package.json`,
+        transform: (text) =>
+          text.replaceAll(/(@coveo\/)?sample-component/g, componentName),
+      },
+    ];
+
+    transform(transformers);
+  }
+
+  successMessage();
+};
 
-successMessage();
+main();
diff --git a/packages/ui/atomic/create-atomic-result-component/index.js b/packages/ui/atomic/create-atomic-result-component/index.js
@@ -140,52 +140,56 @@ const __dirname = dirname(fileURLToPath(import.meta.url));
 const templateRelativeDir = 'template';
 const templateDirPath = resolve(__dirname, templateRelativeDir);
 
-/*const cwdFiles = readdirSync(cwd(), {withFileTypes: true});
-if (cwdFiles.length > 0) {
-  if (cwdFiles.some((dirent) => dirent.name === 'package.json')) {
-    return;
-  } else {
-    handleErrors(
-      new InvalidProjectDirectory(
-        'Current working directory is either not empty or not an npm project (no package.json found). Please try again in an empty directory.'
-      )
-    );
+const main = () => {
+  const cwdFiles = readdirSync(cwd(), {withFileTypes: true});
+  if (cwdFiles.length > 0) {
+    if (cwdFiles.some((dirent) => dirent.name === 'package.json')) {
+      return;
+    } else {
+      handleErrors(
+        new InvalidProjectDirectory(
+          'Current working directory is either not empty or not an npm project (no package.json found). Please try again in an empty directory.'
+        )
+      );
+    }
   }
-}*/
-
-cpSync(templateDirPath, cwd(), {
-  recursive: true,
-});
-
-let componentName = process.argv[2];
-if (componentName) {
-  ensureComponentValidity(componentName);
-  const transformers = [
-    {
-      srcPath: 'src/components/sample-result-component',
-      destPath: `src/components/${componentName}`,
-    },
-    {
-      srcPath: `src/components/${componentName}/src/sample-result-component.tsx`,
-      destPath: `src/components/${componentName}/src/${componentName}.tsx`,
-      transform: (text) =>
-        text
-          .replaceAll(/sample-result-component/g, componentName)
-          .replaceAll(/SampleResultComponent/g, camelize(componentName)),
-    },
-    {
-      srcPath: `src/components/${componentName}/src/sample-result-component.css`,
-      destPath: `src/components/${componentName}/src/${componentName}.css`,
-    },
-    {
-      srcPath: `src/components/${componentName}/package.json`,
-      destPath: `src/components/${componentName}/package.json`,
-      transform: (text) =>
-        text.replaceAll(/(@coveo\/)?sample-result-component/g, componentName),
-    },
-  ];
-
-  transform(transformers);
-}
 
-successMessage();
+  cpSync(templateDirPath, cwd(), {
+    recursive: true,
+  });
+
+  let componentName = process.argv[2];
+  if (componentName) {
+    ensureComponentValidity(componentName);
+    const transformers = [
+      {
+        srcPath: 'src/components/sample-result-component',
+        destPath: `src/components/${componentName}`,
+      },
+      {
+        srcPath: `src/components/${componentName}/src/sample-result-component.tsx`,
+        destPath: `src/components/${componentName}/src/${componentName}.tsx`,
+        transform: (text) =>
+          text
+            .replaceAll(/sample-result-component/g, componentName)
+            .replaceAll(/SampleResultComponent/g, camelize(componentName)),
+      },
+      {
+        srcPath: `src/components/${componentName}/src/sample-result-component.css`,
+        destPath: `src/components/${componentName}/src/${componentName}.css`,
+      },
+      {
+        srcPath: `src/components/${componentName}/package.json`,
+        destPath: `src/components/${componentName}/package.json`,
+        transform: (text) =>
+          text.replaceAll(/(@coveo\/)?sample-result-component/g, componentName),
+      },
+    ];
+
+    transform(transformers);
+  }
+
+  successMessage();
+};
+
+main();