@react-native-firebase/storage
Version:
React Native Firebase - React Native Firebase provides native integration with Cloud Storage, providing support to upload and download files directly from your device and from your Firebase Cloud Storage bucket.
243 lines (232 loc) • 7.74 kB
JavaScript
"use strict";
/*
* Copyright (c) 2016-present Invertase Limited & Contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this library except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
import { isFunction, isNull, isObject } from '@react-native-firebase/app/dist/module/common';
import { TaskEvent } from "./StorageStatics.js";
let TASK_ID = 0;
function createEmptyMetadata(ref) {
return {
bucket: ref.bucket,
fullPath: ref.fullPath,
generation: '',
metageneration: '',
name: ref.name,
size: 0,
timeCreated: '',
updated: '',
downloadTokens: undefined,
ref
};
}
function wrapErrorEventListener(listenerFn, unsubscribe) {
return snapshot => {
if (unsubscribe) {
setTimeout(() => unsubscribe(), 0);
} // 1 frame = 16ms, pushing to next frame
if (isFunction(listenerFn)) {
const errorEvent = snapshot;
if (errorEvent.error) {
listenerFn(errorEvent.error);
}
}
};
}
function wrapSnapshotEventListener(task, listenerFn, unsubscribe) {
if (!isFunction(listenerFn)) {
return null;
}
return snapshot => {
if (unsubscribe) {
setTimeout(() => unsubscribe(), 0);
} // 1 frame = 16ms, pushing to next frame
if (isFunction(listenerFn)) {
const taskSnapshot = Object.assign({}, snapshot);
taskSnapshot.task = task;
taskSnapshot.ref = task._ref;
if (!taskSnapshot.metadata) {
taskSnapshot.metadata = createEmptyMetadata(task._ref);
} else {
if (!taskSnapshot.metadata.generation) taskSnapshot.metadata.generation = '';
if (!taskSnapshot.metadata.bucket) taskSnapshot.metadata.bucket = task._ref.bucket;
if (!taskSnapshot.metadata.metageneration) taskSnapshot.metadata.metageneration = '';
if (!taskSnapshot.metadata.fullPath) taskSnapshot.metadata.fullPath = task._ref.fullPath;
if (!taskSnapshot.metadata.name) taskSnapshot.metadata.name = task._ref.name;
if (!taskSnapshot.metadata.ref) taskSnapshot.metadata.ref = task._ref;
}
Object.freeze(taskSnapshot);
task._snapshot = taskSnapshot;
listenerFn(taskSnapshot);
}
};
}
function addTaskEventListener(task, eventName, listener) {
let _eventName = eventName;
if (_eventName !== TaskEvent.STATE_CHANGED) {
_eventName = `${task._type}_${eventName}`;
}
return task._storage.emitter.addListener(task._storage.eventNameForApp(task._id, _eventName), listener);
}
function subscribeToEvents(task, nextOrObserver, error, complete) {
let _error;
let _errorSubscription;
let _next;
let _nextSubscription;
let _complete;
let _completeSubscription;
const unsubscribe = () => {
if (_nextSubscription) {
_nextSubscription.remove();
}
if (_errorSubscription) {
_errorSubscription.remove();
}
if (_completeSubscription) {
_completeSubscription.remove();
}
};
if (isFunction(nextOrObserver)) {
_error = wrapErrorEventListener(error, unsubscribe);
_next = wrapSnapshotEventListener(task, nextOrObserver, null);
_complete = wrapSnapshotEventListener(task, complete, unsubscribe);
} else if (isObject(nextOrObserver)) {
const observer = nextOrObserver;
_error = wrapErrorEventListener(observer.error, unsubscribe);
_next = wrapSnapshotEventListener(task, observer.next, null);
_complete = wrapSnapshotEventListener(task, observer.complete, unsubscribe);
} else if (isNull(nextOrObserver)) {
_error = wrapErrorEventListener(error, unsubscribe);
_complete = wrapSnapshotEventListener(task, complete, unsubscribe);
} else {
throw new Error("firebase.storage.StorageTask.on(*, _) 'nextOrObserver' must be a Function, an Object or Null.");
}
if (_next) {
_nextSubscription = addTaskEventListener(task, TaskEvent.STATE_CHANGED, _next);
}
if (_error) {
_errorSubscription = addTaskEventListener(task, 'failure', _error);
}
if (_complete) {
_completeSubscription = addTaskEventListener(task, 'success', _complete);
}
return unsubscribe;
}
export default class StorageTask {
constructor(type, storageRef, beginTaskFn) {
this._type = type;
this._id = TASK_ID++;
this._promise = null;
this._ref = storageRef;
this._beginTask = beginTaskFn;
this._storage = storageRef._storage;
this._snapshot = {
bytesTransferred: 0,
totalBytes: 0,
state: 'running',
metadata: createEmptyMetadata(storageRef),
task: this,
ref: storageRef
};
}
/**
* @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#then
*/
get then() {
if (!this._promise) {
this._promise = this._beginTask(this);
}
const promise = this._promise;
return (onFulfilled, onRejected) => {
return new Promise((resolve, reject) => {
promise.then(response => {
this._snapshot = {
...response,
ref: this._ref,
task: this,
state: response?.state ?? this._snapshot.state,
metadata: response?.metadata ? response.metadata : createEmptyMetadata(this._ref)
};
if (onFulfilled) {
resolve(onFulfilled(this._snapshot));
} else {
resolve(response);
}
}).catch(error => {
if (onRejected) {
resolve(onRejected(error));
} else {
reject(error);
}
});
});
};
}
/**
* @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#catch
*/
get catch() {
if (!this._promise) {
this._promise = this._beginTask(this);
}
return this._promise.catch.bind(this._promise);
}
get snapshot() {
return this._snapshot;
}
// // NOT on Web SDK
// /**
// * @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#finally
// */
// get finally() {
// if (!this._promise) this._promise = this._beginTask(this);
// return this._promise.finally.bind(this._promise);
// }
/**
* @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#on
*/
on(event, nextOrObserver, error, complete) {
if (event !== TaskEvent.STATE_CHANGED) {
throw new Error(`firebase.storage.StorageTask.on event argument must be a string with a value of '${TaskEvent.STATE_CHANGED}'`);
}
if (!this._promise) {
this._promise = this._beginTask(this);
}
// if only event provided return the subscriber function
if (!nextOrObserver && !error && !complete) {
return (...args) => subscribeToEvents(this, ...args);
}
return subscribeToEvents(this, nextOrObserver, error, complete);
}
/**
* @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#pause
*/
pause() {
return this._storage.native.setTaskStatus(this._id, 0);
}
/**
* @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#pause
*/
resume() {
return this._storage.native.setTaskStatus(this._id, 1);
}
/**
* @url https://firebase.google.com/docs/reference/js/firebase.storage.UploadTask#cancel
*/
cancel() {
return this._storage.native.setTaskStatus(this._id, 2);
}
}
//# sourceMappingURL=StorageTask.js.map