Merge pull request #2399 from MAVRICK-1/feature/add-progress-bar-to-d…

…ir-upload Enhancement: Add progress tracking to DirToUSSDirHandler for file upl…
zowe · Jan 7, 2025 · bf77e64 · bf77e64
2 parents 7ac5ab7 + 9145334
commit bf77e64
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/packages/cli/CHANGELOG.md b/packages/cli/CHANGELOG.md
@@ -1,6 +1,10 @@
 # Change Log
 All notable changes to the Zowe CLI package will be documented in this file.
 
+## Recent Changes
+
+- BugFix: Fixed an issue where the `zowe files upload dir-to-uss` command was missing progress bar to track progress of file uploads. [#2344](https://github.com/zowe/zowe-cli/issues/2344)
+
 ## `8.10.3`
 
 - BugFix: The `zowe files copy data-set` command no longer copies all partitioned data set members if a member is specified. [#2402](https://github.com/zowe/zowe-cli/pull/2402)

diff --git a/packages/cli/src/zosfiles/upload/dtu/DirToUSSDir.handler.ts b/packages/cli/src/zosfiles/upload/dtu/DirToUSSDir.handler.ts
@@ -54,12 +54,19 @@ export default class DirToUSSDirHandler extends ZosFilesBaseHandler {
             uploadOptions.filesMap = this.buildFilesMap(commandParameters);
         }
 
-        const uploadApi = commandParameters.arguments.recursive ? Upload.dirToUSSDirRecursive : Upload.dirToUSSDir;
-        const response = await uploadApi.bind(Upload)(session, inputDir, commandParameters.arguments.USSDir, uploadOptions);
+        commandParameters.response.progress.startBar({ task: status });
 
-        const formatMessage = TextUtils.prettyJson(response.apiResponse);
-        commandParameters.response.console.log(formatMessage);
-        return response;
+        try {
+            const uploadApi = commandParameters.arguments.recursive ? Upload.dirToUSSDirRecursive : Upload.dirToUSSDir;
+            const response = await uploadApi.bind(Upload)(session, inputDir, commandParameters.arguments.USSDir, uploadOptions);
+
+            const formatMessage = TextUtils.prettyJson(response.apiResponse);
+            commandParameters.response.console.log(formatMessage);
+
+            return response;
+        } finally {
+            commandParameters.response.progress.endBar();
+        }
     }
 
     private buildFilesMap(commandParameters: IHandlerParameters) {