vn-verdnaturachat/app/lib/methods/sendFileMessage.js

import RNFetchBlob from 'rn-fetch-blob';
import * as SDK from '@rocket.chat/sdk';

import reduxStore from '../createStore';
import database from '../realm';

const promises = {};

function _ufsCreate(fileInfo) {
	return SDK.driver.asyncCall('ufsCreate', fileInfo);
}

function _ufsComplete(fileId, store, token) {
	return SDK.driver.asyncCall('ufsComplete', fileId, store, token);
}

function _sendFileMessage(rid, data, msg = {}) {
	// RC 0.22.0
	return SDK.driver.asyncCall('sendFileMessage', rid, null, data, msg);
}

export function isUploadActive(path) {
	return !!promises[path];
}

export async function cancelUpload(path) {
	if (promises[path]) {
		await promises[path].cancel();
	}
}

export async function sendFileMessage(rid, fileInfo) {
	try {
		const data = await RNFetchBlob.wrap(fileInfo.path);
		if (!fileInfo.size) {
			const fileStat = await RNFetchBlob.fs.stat(fileInfo.path);
			fileInfo.size = fileStat.size;
			fileInfo.name = fileStat.filename;
		}

		const { FileUpload_MaxFileSize } = reduxStore.getState().settings;

		// -1 maxFileSize means there is no limit
		if (FileUpload_MaxFileSize > -1 && fileInfo.size > FileUpload_MaxFileSize) {
			return Promise.reject({ error: 'error-file-too-large' }); // eslint-disable-line
		}

		fileInfo.rid = rid;

		database.write(() => {
			database.create('uploads', fileInfo, true);
		});

		const result = await _ufsCreate.call(this, fileInfo);

		promises[fileInfo.path] = RNFetchBlob.fetch('POST', result.url, {
			'Content-Type': 'octet-stream'
		}, data);
		// Workaround for https://github.com/joltup/rn-fetch-blob/issues/96
		setTimeout(() => {
			promises[fileInfo.path].uploadProgress((loaded, total) => {
				database.write(() => {
					fileInfo.progress = Math.floor((loaded / total) * 100);
					database.create('uploads', fileInfo, true);
				});
			});
		});
		await promises[fileInfo.path];

		const completeResult = await _ufsComplete.call(this, result.fileId, fileInfo.store, result.token);

		await _sendFileMessage.call(this, completeResult.rid, {
			_id: completeResult._id,
			type: completeResult.type,
			size: completeResult.size,
			name: completeResult.name,
			description: completeResult.description,
			url: completeResult.path
		});

		database.write(() => {
			const upload = database.objects('uploads').filtered('path = $0', fileInfo.path);
			database.delete(upload);
		});
	} catch (e) {
		database.write(() => {
			fileInfo.error = true;
			database.create('uploads', fileInfo, true);
		});
	}
}