Rocket.Chat/packages/rocketchat-file-upload/client/lib/FileUploadGoogleStorage.js

/* globals FileUpload, FileUploadBase, Slingshot */

FileUpload.GoogleCloudStorage = class FileUploadGoogleCloudStorage extends FileUploadBase {
	constructor(meta, file) {
		super(meta, file);
		this.uploader = new Slingshot.Upload('rocketchat-uploads-gs', { rid: meta.rid });
	}

	start() {
		this.uploader.send(this.file, (error, downloadUrl) => {
			if (this.computation) {
				this.computation.stop();
			}

			if (error) {
				let uploading = Session.get('uploading');
				if (!Array.isArray(uploading)) {
					uploading = [];
				}

				const item = _.findWhere(uploading, { id: this.id });

				if (_.isObject(item)) {
					item.error = error.error;
					item.percentage = 0;
				} else {
					uploading.push({
						error: error.error,
						percentage: 0
					});
				}

				Session.set('uploading', uploading);
			} else {
				const file = _.pick(this.meta, 'type', 'size', 'name', 'identify', 'description');
				file._id = downloadUrl.substr(downloadUrl.lastIndexOf('/') + 1);
				file.url = downloadUrl;

				Meteor.call('sendFileMessage', this.meta.rid, 'googleCloudStorage', file, () => {
					Meteor.setTimeout(() => {
						const uploading = Session.get('uploading');
						if (uploading !== null) {
							const item = _.findWhere(uploading, {
								id: this.id
							});
							return Session.set('uploading', _.without(uploading, item));
						}
					}, 2000);
				});
			}
		});

		this.computation = Tracker.autorun(() => {
			this.onProgress(this.uploader.progress());
		});
	}

	onProgress() {}

	stop() {
		if (this.uploader && this.uploader.xhr) {
			this.uploader.xhr.abort();
		}
	}
};