react-firebase-hooks
Version:
React Hooks for Firebase
186 lines (171 loc) • 8.71 kB
JavaScript
import { getDownloadURL, uploadBytesResumable } from 'firebase/storage';
import { useMemo, useReducer, useCallback, useRef, useEffect, useState } from 'react';
/*! *****************************************************************************
Copyright (c) Microsoft Corporation.
Permission to use, copy, modify, and/or distribute this software for any
purpose with or without fee is hereby granted.
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
***************************************************************************** */
var __assign = function() {
__assign = Object.assign || function __assign(t) {
for (var s, i = 1, n = arguments.length; i < n; i++) {
s = arguments[i];
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p];
}
return t;
};
return __assign.apply(this, arguments);
};
var defaultState = function (defaultValue) {
return {
loading: defaultValue === undefined || defaultValue === null,
value: defaultValue,
};
};
var reducer = function () { return function (state, action) {
switch (action.type) {
case 'error':
return __assign(__assign({}, state), { error: action.error, loading: false, value: undefined });
case 'reset':
return defaultState(action.defaultValue);
case 'value':
return __assign(__assign({}, state), { error: undefined, loading: false, value: action.value });
default:
return state;
}
}; };
var useLoadingValue = (function (getDefaultValue) {
var defaultValue = getDefaultValue ? getDefaultValue() : undefined;
var _a = useReducer(reducer(), defaultState(defaultValue)), state = _a[0], dispatch = _a[1];
var reset = useCallback(function () {
var defaultValue = getDefaultValue ? getDefaultValue() : undefined;
dispatch({ type: 'reset', defaultValue: defaultValue });
}, [getDefaultValue]);
var setError = useCallback(function (error) {
dispatch({ type: 'error', error: error });
}, []);
var setValue = useCallback(function (value) {
dispatch({ type: 'value', value: value });
}, []);
return useMemo(function () { return ({
error: state.error,
loading: state.loading,
reset: reset,
setError: setError,
setValue: setValue,
value: state.value,
}); }, [state.error, state.loading, reset, setError, setValue, state.value]);
});
var useComparatorRef = function (value, isEqual, onChange) {
var ref = useRef(value);
useEffect(function () {
if (!isEqual(value, ref.current)) {
ref.current = value;
if (onChange) {
onChange();
}
}
});
return ref;
};
var useDownloadURL = (function (storageRef) {
var _a = useLoadingValue(), error = _a.error, loading = _a.loading, reset = _a.reset, setError = _a.setError, setValue = _a.setValue, value = _a.value;
var ref = useComparatorRef(storageRef, isEqual, reset);
useEffect(function () {
if (!ref.current) {
setValue(undefined);
return;
}
getDownloadURL(ref.current).then(setValue).catch(setError);
}, [ref.current]);
return [value, loading, error];
});
var isEqual = function (v1, v2) {
var bothNull = !v1 && !v2;
var equal = !!v1 && !!v2 && v1.fullPath === v2.fullPath;
return bothNull || equal;
};
/*! *****************************************************************************
Copyright (c) Microsoft Corporation.
Permission to use, copy, modify, and/or distribute this software for any
purpose with or without fee is hereby granted.
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
***************************************************************************** */
function __awaiter(thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
}
function __generator(thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (_) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
}
var useUploadFile = (function () {
var _a = useState(), error = _a[0], setError = _a[1];
var _b = useState(false), uploading = _b[0], setUploading = _b[1];
var _c = useState(), snapshot = _c[0], setSnapshot = _c[1];
var uploadFile = useCallback(function (storageRef, data, metadata) { return __awaiter(void 0, void 0, void 0, function () {
return __generator(this, function (_a) {
return [2 /*return*/, new Promise(function (resolve, reject) {
setUploading(true);
setError(undefined);
var uploadTask = uploadBytesResumable(storageRef, data, metadata);
uploadTask.on('state_changed', function (snapshot) {
setSnapshot(snapshot);
}, function (error) {
setUploading(false);
setError(error);
resolve(undefined);
}, function () {
setUploading(false);
setSnapshot(undefined);
resolve({
metadata: uploadTask.snapshot.metadata,
ref: uploadTask.snapshot.ref,
});
});
})];
});
}); }, []);
return [uploadFile, uploading, snapshot, error];
});
export { useDownloadURL, useUploadFile };