Converted /modals to typescript
This commit is contained in:
Nicola Thouliss
298
src/modals/ImportExportModal.tsx
Normal file
298
src/modals/ImportExportModal.tsx
Normal file
@@ -0,0 +1,298 @@
|
||||
import { useRef, useState, useEffect } from "react";
|
||||
import { Box, Label, Text, Button, Flex } from "theme-ui";
|
||||
import { saveAs } from "file-saver";
|
||||
import * as Comlink from "comlink";
|
||||
import shortid from "shortid";
|
||||
import { useToasts } from "react-toast-notifications";
|
||||
|
||||
import Modal from "../components/Modal";
|
||||
import LoadingOverlay from "../components/LoadingOverlay";
|
||||
import LoadingBar from "../components/LoadingBar";
|
||||
import ErrorBanner from "../components/banner/ErrorBanner";
|
||||
|
||||
import { useAuth } from "../contexts/AuthContext";
|
||||
import { useDatabase } from "../contexts/DatabaseContext";
|
||||
|
||||
import SelectDataModal from "./SelectDataModal";
|
||||
|
||||
import { getDatabase } from "../database";
|
||||
import { Map, MapState, TokenState } from "../components/map/Map";
|
||||
|
||||
const importDBName = "OwlbearRodeoImportDB";
|
||||
|
||||
function ImportExportModal({ isOpen, onRequestClose }: { isOpen: boolean, onRequestClose: () => void}) {
|
||||
const { worker } = useDatabase();
|
||||
const { userId } = useAuth();
|
||||
|
||||
const [isLoading, setIsLoading] = useState(false);
|
||||
const [error, setError] = useState<Error>();
|
||||
|
||||
const backgroundTaskRunningRef = useRef(false);
|
||||
const fileInputRef = useRef<any>();
|
||||
|
||||
const [showImportSelector, setShowImportSelector] = useState(false);
|
||||
const [showExportSelector, setShowExportSelector] = useState(false);
|
||||
|
||||
const { addToast } = useToasts();
|
||||
function addSuccessToast(message: string, maps: any, tokens: TokenState[]) {
|
||||
const mapText = `${maps.length} map${maps.length > 1 ? "s" : ""}`;
|
||||
const tokenText = `${tokens.length} token${tokens.length > 1 ? "s" : ""}`;
|
||||
if (maps.length > 0 && tokens.length > 0) {
|
||||
addToast(`${message} ${mapText} and ${tokenText}`);
|
||||
} else if (maps.length > 0) {
|
||||
addToast(`${message} ${mapText}`);
|
||||
} else if (tokens.length > 0) {
|
||||
addToast(`${message} ${tokenText}`);
|
||||
}
|
||||
}
|
||||
|
||||
function openFileDialog() {
|
||||
if (fileInputRef.current) {
|
||||
fileInputRef.current.click();
|
||||
}
|
||||
}
|
||||
|
||||
const loadingProgressRef = useRef(0);
|
||||
|
||||
function handleDBProgress({ completedRows, totalRows }: { completedRows: number, totalRows: number }) {
|
||||
loadingProgressRef.current = completedRows / totalRows;
|
||||
}
|
||||
|
||||
async function handleImportDatabase(file: File) {
|
||||
setIsLoading(true);
|
||||
backgroundTaskRunningRef.current = true;
|
||||
try {
|
||||
await worker.importData(
|
||||
file,
|
||||
importDBName,
|
||||
Comlink.proxy(handleDBProgress)
|
||||
);
|
||||
|
||||
setIsLoading(false);
|
||||
setShowImportSelector(true);
|
||||
backgroundTaskRunningRef.current = false;
|
||||
} catch (e) {
|
||||
setIsLoading(false);
|
||||
backgroundTaskRunningRef.current = false;
|
||||
if (e.message.startsWith("Max buffer length exceeded")) {
|
||||
setError(
|
||||
new Error(
|
||||
"Max image size exceeded ensure your database doesn't have an image over 100MB"
|
||||
)
|
||||
);
|
||||
} else {
|
||||
setError(e);
|
||||
}
|
||||
}
|
||||
// Set file input to null to allow adding the same data 2 times in a row
|
||||
if (fileInputRef.current) {
|
||||
fileInputRef.current.value = null;
|
||||
}
|
||||
}
|
||||
|
||||
function handleExportDatabase() {
|
||||
setShowExportSelector(true);
|
||||
}
|
||||
|
||||
useEffect(() => {
|
||||
function handleBeforeUnload(event: any) {
|
||||
if (backgroundTaskRunningRef.current) {
|
||||
event.returnValue =
|
||||
"Database is still processing, are you sure you want to leave?";
|
||||
}
|
||||
}
|
||||
|
||||
window.addEventListener("beforeunload", handleBeforeUnload);
|
||||
return () => {
|
||||
window.removeEventListener("beforeunload", handleBeforeUnload);
|
||||
};
|
||||
}, []);
|
||||
|
||||
function handleClose() {
|
||||
if (isLoading) {
|
||||
return;
|
||||
}
|
||||
onRequestClose();
|
||||
}
|
||||
|
||||
async function handleImportSelectorClose() {
|
||||
const importDB = getDatabase({ addons: [] }, importDBName);
|
||||
await importDB.delete();
|
||||
importDB.close();
|
||||
setShowImportSelector(false);
|
||||
}
|
||||
|
||||
async function handleImportSelectorConfirm(checkedMaps: any, checkedTokens: TokenState[]) {
|
||||
setIsLoading(true);
|
||||
backgroundTaskRunningRef.current = true;
|
||||
setShowImportSelector(false);
|
||||
loadingProgressRef.current = 0;
|
||||
|
||||
const importDB = getDatabase({ addons: [] }, importDBName);
|
||||
const db = getDatabase({});
|
||||
try {
|
||||
// Keep track of a mapping of old token ids to new ones to apply them to the map states
|
||||
let newTokenIds: {[id: string]: string} = {};
|
||||
if (checkedTokens.length > 0) {
|
||||
const tokenIds = checkedTokens.map((token) => token.id);
|
||||
const tokensToAdd: TokenState[] = await importDB.table("tokens").bulkGet(tokenIds);
|
||||
let newTokens: TokenState[] = [];
|
||||
for (let token of tokensToAdd) {
|
||||
const newId = shortid.generate();
|
||||
newTokenIds[token.id] = newId;
|
||||
// Generate new id and change owner
|
||||
newTokens.push({ ...token, id: newId, owner: userId });
|
||||
}
|
||||
await db.table("tokens").bulkAdd(newTokens);
|
||||
}
|
||||
|
||||
if (checkedMaps.length > 0) {
|
||||
const mapIds = checkedMaps.map((map: any) => map.id);
|
||||
const mapsToAdd = await importDB.table("maps").bulkGet(mapIds);
|
||||
let newMaps = [];
|
||||
let newStates = [];
|
||||
for (let map of mapsToAdd) {
|
||||
let state: MapState = await importDB.table("states").get(map.id);
|
||||
// Apply new token ids to imported state
|
||||
for (let tokenState of Object.values(state.tokens)) {
|
||||
if (tokenState.tokenId in newTokenIds) {
|
||||
state.tokens[tokenState.id].tokenId =
|
||||
newTokenIds[tokenState.tokenId];
|
||||
}
|
||||
}
|
||||
const newId = shortid.generate();
|
||||
// Generate new id and change owner
|
||||
newMaps.push({ ...map, id: newId, owner: userId });
|
||||
newStates.push({ ...state, mapId: newId });
|
||||
}
|
||||
await db.table("maps").bulkAdd(newMaps);
|
||||
await db.table("states").bulkAdd(newStates);
|
||||
}
|
||||
addSuccessToast("Imported", checkedMaps, checkedTokens);
|
||||
} catch (e) {
|
||||
console.error(e);
|
||||
setError(new Error("Unable to import data"));
|
||||
}
|
||||
await importDB.delete();
|
||||
importDB.close();
|
||||
db.close();
|
||||
setIsLoading(false);
|
||||
backgroundTaskRunningRef.current = false;
|
||||
}
|
||||
|
||||
function exportSelectorFilter(table: any, value: Map | TokenState) {
|
||||
// Only show owned maps and tokens
|
||||
if (table === "maps" || table === "tokens") {
|
||||
if (value.owner === userId) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
// Allow all states so tokens can be checked against maps
|
||||
if (table === "states") {
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
async function handleExportSelectorClose() {
|
||||
setShowExportSelector(false);
|
||||
}
|
||||
|
||||
async function handleExportSelectorConfirm(checkedMaps: Map[], checkedTokens: TokenState[]) {
|
||||
setShowExportSelector(false);
|
||||
setIsLoading(true);
|
||||
backgroundTaskRunningRef.current = true;
|
||||
|
||||
const mapIds = checkedMaps.map((map) => map.id);
|
||||
const tokenIds = checkedTokens.map((token) => token.id);
|
||||
|
||||
try {
|
||||
const buffer = await worker.exportData(
|
||||
Comlink.proxy(handleDBProgress),
|
||||
mapIds,
|
||||
tokenIds
|
||||
);
|
||||
const blob = new Blob([buffer]);
|
||||
saveAs(blob, `${shortid.generate()}.owlbear`);
|
||||
addSuccessToast("Exported", checkedMaps, checkedTokens);
|
||||
} catch (e) {
|
||||
setError(e);
|
||||
}
|
||||
setIsLoading(false);
|
||||
backgroundTaskRunningRef.current = false;
|
||||
}
|
||||
|
||||
return (
|
||||
<Modal isOpen={isOpen} onRequestClose={handleClose} allowClose={!isLoading}>
|
||||
<Flex
|
||||
sx={{
|
||||
flexDirection: "column",
|
||||
justifyContent: "center",
|
||||
maxWidth: "300px",
|
||||
flexGrow: 1,
|
||||
}}
|
||||
m={2}
|
||||
>
|
||||
<Box>
|
||||
<Label>Import / Export</Label>
|
||||
<Text as="p" mb={2} variant="caption">
|
||||
Select import or export then select the data you wish to use
|
||||
</Text>
|
||||
<input
|
||||
onChange={(event) => event.target.files && handleImportDatabase(event.target.files[0])}
|
||||
type="file"
|
||||
accept=".owlbear"
|
||||
style={{ display: "none" }}
|
||||
ref={fileInputRef}
|
||||
/>
|
||||
<Flex>
|
||||
<Button mr={1} sx={{ flexGrow: 1 }} onClick={openFileDialog}>
|
||||
Import
|
||||
</Button>
|
||||
<Button ml={1} sx={{ flexGrow: 1 }} onClick={handleExportDatabase}>
|
||||
Export
|
||||
</Button>
|
||||
</Flex>
|
||||
</Box>
|
||||
{isLoading && (
|
||||
<Box
|
||||
sx={{
|
||||
width: "100%",
|
||||
height: "100%",
|
||||
position: "absolute",
|
||||
top: 0,
|
||||
left: 0,
|
||||
}}
|
||||
>
|
||||
<LoadingOverlay bg="overlay" />
|
||||
<Box sx={{ zIndex: 3, position: "absolute", width: "100%" }}>
|
||||
<LoadingBar
|
||||
isLoading={isLoading}
|
||||
loadingProgressRef={loadingProgressRef}
|
||||
/>
|
||||
</Box>
|
||||
</Box>
|
||||
)}
|
||||
<ErrorBanner error={error} onRequestClose={() => setError(undefined)} />
|
||||
<SelectDataModal
|
||||
isOpen={showImportSelector}
|
||||
onRequestClose={handleImportSelectorClose}
|
||||
onConfirm={handleImportSelectorConfirm}
|
||||
databaseName={importDBName}
|
||||
confirmText="Import"
|
||||
label="Select data to import"
|
||||
/>
|
||||
<SelectDataModal
|
||||
isOpen={showExportSelector}
|
||||
onRequestClose={handleExportSelectorClose}
|
||||
onConfirm={handleExportSelectorConfirm}
|
||||
confirmText="Export"
|
||||
label="Select data to export"
|
||||
filter={exportSelectorFilter}
|
||||
/>
|
||||
</Flex>
|
||||
</Modal>
|
||||
);
|
||||
}
|
||||
|
||||
export default ImportExportModal;
|
||||
Reference in New Issue
Block a user