Display dependency graph progress

src/cli/progress.js

@@ -1,3 +1,5 @@
+const logger = require('./logger');
+
 let currentSpinner;
 
 const getStartMessage = (stage) => {
@@ -44,7 +46,7 @@ const getEndMessage = (stage) => {
 
 const onProgress =
   (ora) =>
-  ({type, stage, message}) => {
+  ({type, stage, message, progress}) => {
     switch (type) {
       case 'start':
         currentSpinner = ora().start(getStartMessage(stage));
@@ -55,6 +57,11 @@ const onProgress =
       case 'update':
         currentSpinner.text = message;
         break;
+      case 'progress':
+        currentSpinner.text = `${getStartMessage(stage)} ${logger.colors.DIM}${progress}${
+          logger.colors.RESET
+        }`;
+        break;
       default:
         break;
     }

src/graph/index.js

@@ -6,7 +6,7 @@ const {postProcessGraph, addDependencyGraphData, getRegistryDataMultiple} = requ
 
 const generateGraphPromise = async (
   appPath,
-  {packageData, loadDataFrom = false, rootIsShell = false, includeDev = false} = {},
+  {packageData, loadDataFrom = false, rootIsShell = false, includeDev = false, onProgress} = {},
 ) => {
   const lockfile = await loadLockfile(appPath);
 
@@ -68,6 +68,7 @@ const generateGraphPromise = async (
     } else if (loadDataFrom === 'registry') {
       const {data, errors: registryErrors} = await getRegistryDataMultiple(
         includeDev ? allConnectedPackages : prodDependencies,
+        onProgress,
       );
       additionalPackageData = data;
       errors = [...errors, ...registryErrors];
@@ -98,14 +99,24 @@ const generateGraphAsync = (appPath, options, done = () => {}) => {
 
 const generateGraph = (
   appPath,
-  {packageData, loadDataFrom = false, rootIsShell = false, includeDev = false} = {},
+  {packageData, loadDataFrom = false, rootIsShell = false, includeDev = false, onProgress} = {},
   done = undefined,
 ) => {
   if (typeof done === 'function') {
-    return generateGraphAsync(appPath, {packageData, loadDataFrom, rootIsShell, includeDev}, done);
+    return generateGraphAsync(
+      appPath,
+      {packageData, loadDataFrom, rootIsShell, includeDev, onProgress},
+      done,
+    );
   }
 
-  return generateGraphPromise(appPath, {packageData, loadDataFrom, rootIsShell, includeDev});
+  return generateGraphPromise(appPath, {
+    packageData,
+    loadDataFrom,
+    rootIsShell,
+    includeDev,
+    onProgress,
+  });
 };
 
 module.exports = generateGraph;

src/graph/utils.js

@@ -298,7 +298,9 @@ const getRegistryData = (packageName, packageVersion) =>
     req.end();
   });
 
-const getRegistryDataMultiple = async (packages) => {
+const getRegistryDataMultiple = async (packages, onProgress = () => {}) => {
+  const totalCount = packages.length;
+  let currentCount = 0;
   const errors = [];
   const threadCount = 10;
   const jobCount = Math.ceil(packages.length / threadCount);
@@ -312,6 +314,9 @@ const getRegistryDataMultiple = async (packages) => {
           try {
             const {name, version} = packages[globalJobIndex];
             const packageData = await getRegistryData(name, version);
+
+            currentCount += 1;
+            onProgress?.(`${currentCount}/${totalCount}`);
             return [...prevData, packageData];
           } catch (error) {
             errors.push(error);

src/index.js

@@ -44,7 +44,13 @@ const getReport = async ({
   // Generate the dependency graph
   onProgress({type: 'start', stage: 'graph'});
   const dGraph =
-    dependencyGraph || (await getDependencyGraph(appPath, {loadDataFrom, rootIsShell, includeDev}));
+    dependencyGraph ||
+    (await getDependencyGraph(appPath, {
+      loadDataFrom,
+      rootIsShell,
+      includeDev,
+      onProgress: (progress) => onProgress({type: 'progress', stage: 'graph', progress}),
+    }));
   const packageGraph = dGraph.root;
   errors = [...errors, ...(dGraph.errors || [])];
   onProgress({type: 'end', stage: 'graph'});