feat: Data set and USS download commands

packages/zowe-explorer-api/__tests__/__unit__/profiles/ZoweExplorerZosmfApi.unit.test.ts

@@ -457,6 +457,11 @@ describe("ZosmfUssApi", () => {
             spy: jest.spyOn(zosfiles.Download, "ussFile"),
             args: ["ussPath", fakeProperties],
         },
+        {
+            name: "downloadDirectory",
+            spy: jest.spyOn(zosfiles.Download, "ussDir"),
+            args: ["localPath", fakeProperties],
+        },
         {
             name: "putContent",
             spy: jest.spyOn(zosfiles.Upload, "fileToUssFile"),
@@ -513,6 +518,11 @@ describe("ZosmfMvsApi", () => {
             spy: jest.spyOn(zosfiles.Download, "dataSet"),
             args: ["dsname", fakeProperties],
         },
+        {
+            name: "downloadAllMembers",
+            spy: jest.spyOn(zosfiles.Download, "allMembers"),
+            args: ["dsname", fakeProperties],
+        },
         {
             name: "putContents",
             spy: jest.spyOn(zosfiles.Upload, "pathToDataSet"),

packages/zowe-explorer-api/src/extend/MainframeInteraction.ts

@@ -119,9 +119,24 @@ export namespace MainframeInteraction {
          *
          * @param {string} ussFilePath
          * @param {zosfiles.IDownloadOptions} options
+         * @returns {Promise<zosfiles.IZosFilesResponse>}
          */
         getContents(ussFilePath: string, options: zosfiles.IDownloadSingleOptions): Promise<zosfiles.IZosFilesResponse>;
 
+        /**
+         * Download a USS directory to the local file system.
+         *
+         * @param {string} ussDirectoryPath The path of the USS directory to download
+         * @param {zosfiles.IDownloadOptions} fileOptions Download options including local directory path
+         * @param {zosfiles.IUSSListOptions} listOptions Options for listing files in USS
+         * @returns {Promise<zosfiles.IZosFilesResponse>}
+         */
+        downloadDirectory(
+            ussDirectoryPath: string,
+            fileOptions?: zosfiles.IDownloadOptions,
+            listOptions?: zosfiles.IUSSListOptions
+        ): Promise<zosfiles.IZosFilesResponse>;
+
         /**
          * Uploads a given buffer as the contents of a file on USS.
          * @param {Buffer} buffer
@@ -235,6 +250,16 @@ export namespace MainframeInteraction {
          */
         getContents(dataSetName: string, options?: zosfiles.IDownloadSingleOptions): Promise<zosfiles.IZosFilesResponse>;
 
+        /**
+         * Retrieve all members of a partitioned data set and save them to a directory.
+         *
+         * @param {string} dataSetName
+         * @param {string} directoryPath
+         * @param {zosfiles.IDownloadOptions} [options]
+         * @returns {Promise<zosfiles.IZosFilesResponse>}
+         */
+        downloadAllMembers(dataSetName: string, options?: zosfiles.IDownloadOptions): Promise<zosfiles.IZosFilesResponse>;
+
         /**
          * Uploads a given buffer as the contents of a file to a data set or member.
          *

packages/zowe-explorer-api/src/profiles/ZoweExplorerZosmfApi.ts

@@ -129,6 +129,18 @@ import { IDataSetCount } from "../dataset/IDataSetCount";
             });
         }
 
+        public downloadDirectory(
+            ussDirectoryPath: string,
+            fileOptions?: zosfiles.IDownloadOptions,
+            listOptions?: zosfiles.IUSSListOptions
+        ): Promise<zosfiles.IZosFilesResponse> {
+            return zosfiles.Download.ussDir(this.getSession(), ussDirectoryPath, {
+                responseTimeout: this.profile?.profile?.responseTimeout,
+                ...fileOptions,
+                ...listOptions,
+            });
+        }
+
         public copy(outputPath: string, options?: Omit<object, "request">): Promise<Buffer> {
             return zosfiles.Utilities.putUSSPayload(this.getSession(), outputPath, { ...options, request: "copy" });
         }
@@ -260,6 +272,13 @@ import { IDataSetCount } from "../dataset/IDataSetCount";
             });
         }
 
+        public downloadAllMembers(dataSetName: string, options?: zosfiles.IDownloadOptions): Promise<zosfiles.IZosFilesResponse> {
+            return zosfiles.Download.allMembers(this.getSession(), dataSetName, {
+                responseTimeout: this.profile?.profile?.responseTimeout,
+                ...options,
+            });
+        }
+
         public uploadFromBuffer(buffer: Buffer, dataSetName: string, options?: zosfiles.IUploadOptions): Promise<zosfiles.IZosFilesResponse> {
             return zosfiles.Upload.bufferToDataSet(this.getSession(), buffer, dataSetName, {
                 responseTimeout: this.profile?.profile?.responseTimeout,

packages/zowe-explorer-ftp-extension/src/ZoweExplorerFtpMvsApi.ts

@@ -371,6 +371,10 @@ export class FtpMvsApi extends AbstractFtpApi implements MainframeInteraction.IM
         }
     }
 
+    public downloadAllMembers(_dataSetName: string, _options?: zosfiles.IDownloadOptions): Promise<zosfiles.IZosFilesResponse> {
+        throw new ZoweFtpExtensionError("Download all members is not supported in ftp extension.");
+    }
+
     public hMigrateDataSet(_dataSetName: string): Promise<zosfiles.IZosFilesResponse> {
         throw new ZoweFtpExtensionError("Migrate dataset is not supported in ftp extension.");
     }

packages/zowe-explorer-ftp-extension/src/ZoweExplorerFtpUssApi.ts

@@ -101,6 +101,23 @@ export class FtpUssApi extends AbstractFtpApi implements MainframeInteraction.IU
         }
     }
 
+    /**
+     * Download a USS directory to the local file system.
+     * Currently not supported by FTP extension.
+     *
+     * @param ussDirectoryPath The path of the USS directory to download
+     * @param fileOptions Download options including local directory path
+     * @param listOptions Options for listing files in USS
+     * @returns A file response with the results of the download operation.
+     */
+    public downloadDirectory(
+        _ussDirectoryPath: string,
+        _fileOptions?: zosfiles.IDownloadOptions,
+        _listOptions?: zosfiles.IUSSListOptions
+    ): Promise<zosfiles.IZosFilesResponse> {
+        throw new ZoweFtpExtensionError("Download directory operation is not supported in FTP extension.");
+    }
+
     /**
      * Uploads a USS file from the given buffer.
      * @param buffer The buffer containing the contents of the USS file

packages/zowe-explorer/__tests__/__mocks__/@zowe/zos-files-for-zowe-sdk.ts

@@ -186,3 +186,8 @@ export class IZosFilesResponse {
      */
     public apiResponse?: any;
 }
+
+export class ZosFilesUtils {
+    public static getDirsFromDataSet = jest.fn().mockReturnValue("test/dataset/path");
+    public static DEFAULT_FILE_EXTENSION = "txt";
+}

packages/zowe-explorer/__tests__/__mocks__/vscode.ts

@@ -1651,7 +1651,9 @@ export interface TextDocument {
 export class Uri {
     private static _regexp = /^(([^:/?#]+?):)?(\/\/([^/?#]*))?([^?#]*)(\?([^#]*))?(#(.*))?/;
     public static file(path: string): Uri {
-        return Uri.parse(path);
+        const uri = Uri.parse(path);
+        uri.fsPath = path;
+        return uri;
     }
     public static parse(value: string, _strict?: boolean): Uri {
         const match = Uri._regexp.exec(value);

packages/zowe-explorer/__tests__/__unit__/extension.unit.test.ts

@@ -185,6 +185,9 @@ async function createGlobalMocks() {
             "zowe.ds.filteredDataSetsSearchFor",
             "zowe.ds.tableView",
             "zowe.ds.listDataSets",
+            "zowe.ds.downloadAllMembers",
+            "zowe.ds.downloadMember",
+            "zowe.ds.downloadDataSet",
             "zowe.uss.addSession",
             "zowe.uss.refreshAll",
             "zowe.uss.refresh",
@@ -201,6 +204,8 @@ async function createGlobalMocks() {
             "zowe.uss.uploadDialog",
             "zowe.uss.uploadDialogBinary",
             "zowe.uss.uploadDialogWithEncoding",
+            "zowe.uss.downloadFile",
+            "zowe.uss.downloadDirectory",
             "zowe.uss.copyPath",
             "zowe.uss.editFile",
             "zowe.uss.editAttributes",