Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: allow files in directories to be downloaded onto local machine #2199

Merged
merged 15 commits into from
May 21, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/file.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2288,12 +2288,13 @@ class File extends ServiceObject<File, FileMetadata> {

const fileStream = this.createReadStream(options);
let receivedData = false;

if (destination) {
fileStream
.on('error', callback)
.once('data', data => {
// We know that the file exists the server - now we can truncate/write to a file
receivedData = true;
// We know that the file exists the server - now we can truncate/write to a file
const writable = fs.createWriteStream(destination);
writable.write(data);
fileStream
Expand Down
20 changes: 17 additions & 3 deletions src/transfer-manager.ts
Original file line number Diff line number Diff line change
Expand Up @@ -526,7 +526,9 @@ export class TransferManager {
*
* @param {array | string} [filesOrFolder] An array of file name strings or file objects to be downloaded. If
* a string is provided this will be treated as a GCS prefix and all files with that prefix will be downloaded.
* @param {DownloadManyFilesOptions} [options] Configuration options.
* @param {DownloadManyFilesOptions} [options] Configuration options. Setting options.prefix or options.stripPrefix
* or options.passthroughOptions.destination will cause the downloaded files to be written to the file system
* instead of being returned as a buffer.
* @returns {Promise<DownloadResponse[]>}
*
* @example
Expand Down Expand Up @@ -587,7 +589,7 @@ export class TransferManager {
[GCCL_GCS_CMD_KEY]: GCCL_GCS_CMD_FEATURE.DOWNLOAD_MANY,
};

if (options.prefix) {
if (options.prefix || passThroughOptionsCopy.destination) {
vishwarajanand marked this conversation as resolved.
Show resolved Hide resolved
passThroughOptionsCopy.destination = path.join(
options.prefix || '',
passThroughOptionsCopy.destination || '',
Expand All @@ -598,7 +600,19 @@ export class TransferManager {
passThroughOptionsCopy.destination = file.name.replace(regex, '');
}

promises.push(limit(() => file.download(passThroughOptionsCopy)));
promises.push(
limit(async () => {
const destination = passThroughOptionsCopy.destination;
if (destination && destination.endsWith(path.sep)) {
await fsp.mkdir(destination, {recursive: true});
return Promise.resolve([
Buffer.alloc(0),
]) as Promise<DownloadResponse>;
}

return file.download(passThroughOptionsCopy);
})
);
}

return Promise.all(promises);
Expand Down
30 changes: 30 additions & 0 deletions test/file.ts
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ import assert from 'assert';
import * as crypto from 'crypto';
import duplexify from 'duplexify';
import * as fs from 'fs';
import * as path from 'path';
import proxyquire from 'proxyquire';
import * as resumableUpload from '../src/resumable-upload.js';
import * as sinon from 'sinon';
Expand Down Expand Up @@ -2571,6 +2572,12 @@ describe('File', () => {
});

describe('with destination', () => {
const sandbox = sinon.createSandbox();

afterEach(() => {
sandbox.restore();
});

it('should write the file to a destination if provided', done => {
tmp.setGracefulCleanup();
tmp.file((err, tmpFilePath) => {
Expand Down Expand Up @@ -2694,6 +2701,29 @@ describe('File', () => {
});
});
});

it('should fail if provided destination directory does not exist', done => {
tmp.setGracefulCleanup();
tmp.dir(async (err, tmpDirPath) => {
assert.ifError(err);

const fileContents = 'nested-abcdefghijklmnopqrstuvwxyz';

Object.assign(fileReadStream, {
_read(this: Readable) {
this.push(fileContents);
this.push(null);
},
});

const nestedPath = path.join(tmpDirPath, 'a', 'b', 'c', 'file.txt');

file.download({destination: nestedPath}, (err: Error) => {
assert.ok(err);
done();
});
});
});
});
});

Expand Down
73 changes: 73 additions & 0 deletions test/transfer-manager.ts
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ import {
Bucket,
File,
CRC32C,
DownloadCallback,
DownloadOptions,
IdempotencyStrategy,
MultiPartHelperGenerator,
Expand Down Expand Up @@ -233,6 +234,78 @@ describe('Transfer Manager', () => {

await transferManager.downloadManyFiles([file]);
});

it('sets the destination correctly when provided a passthroughOptions.destination', async () => {
const passthroughOptions = {
destination: 'test-destination',
};
const filename = 'first.txt';
const expectedDestination = path.normalize(
`${passthroughOptions.destination}/${filename}`
);
const download = (optionsOrCb?: DownloadOptions | DownloadCallback) => {
if (typeof optionsOrCb === 'function') {
optionsOrCb(null, Buffer.alloc(0));
} else if (optionsOrCb) {
assert.strictEqual(optionsOrCb.destination, expectedDestination);
}
return Promise.resolve([Buffer.alloc(0)]) as Promise<DownloadResponse>;
};

const file = new File(bucket, filename);
file.download = download;
await transferManager.downloadManyFiles([file], {passthroughOptions});
});

it('does not set the destination when prefix, strip prefix and passthroughOptions.destination are not provided', async () => {
const options = {};
const filename = 'first.txt';
const download = (optionsOrCb?: DownloadOptions | DownloadCallback) => {
if (typeof optionsOrCb === 'function') {
optionsOrCb(null, Buffer.alloc(0));
} else if (optionsOrCb) {
assert.strictEqual(optionsOrCb.destination, undefined);
}
return Promise.resolve([Buffer.alloc(0)]) as Promise<DownloadResponse>;
};

const file = new File(bucket, filename);
file.download = download;
await transferManager.downloadManyFiles([file], options);
});

it('should recursively create directory and write file contents if destination path is nested', async () => {
const prefix = 'text-prefix';
const folder = 'nestedFolder/';
const file = 'first.txt';
const filesOrFolder = [folder, path.join(folder, file)];
const expectedFilePath = path.join(prefix, folder, file);
const expectedDir = path.join(prefix, folder);
const mkdirSpy = sandbox.spy(fsp, 'mkdir');
const download = (optionsOrCb?: DownloadOptions | DownloadCallback) => {
if (typeof optionsOrCb === 'function') {
optionsOrCb(null, Buffer.alloc(0));
} else if (optionsOrCb) {
assert.strictEqual(optionsOrCb.destination, expectedFilePath);
}
return Promise.resolve([Buffer.alloc(0)]) as Promise<DownloadResponse>;
};

sandbox.stub(bucket, 'file').callsFake(filename => {
const file = new File(bucket, filename);
file.download = download;
return file;
});
await transferManager.downloadManyFiles(filesOrFolder, {
prefix: prefix,
});
assert.strictEqual(
mkdirSpy.calledOnceWith(expectedDir, {
recursive: true,
}),
true
);
});
});

describe('downloadFileInChunks', () => {
Expand Down