99 lines
3.3 KiB
TypeScript
99 lines
3.3 KiB
TypeScript
/*
|
|
* Copyright (C) Contributors to the Suwayomi project
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
import React, { useEffect, useState } from 'react';
|
|
import IconButton from '@mui/material/IconButton';
|
|
import RefreshIcon from '@mui/icons-material/Refresh';
|
|
import CircularProgress from '@mui/material/CircularProgress';
|
|
import { Box } from '@mui/system';
|
|
import Typography from '@mui/material/Typography';
|
|
import makeToast from 'components/util/Toast';
|
|
import { IUpdateStatus } from 'typings';
|
|
import { useTranslation } from 'react-i18next';
|
|
import requestManager from 'lib/RequestManager';
|
|
|
|
interface IProgressProps {
|
|
progress: number;
|
|
}
|
|
|
|
function Progress({ progress }: IProgressProps) {
|
|
return (
|
|
<Box sx={{ display: 'grid', placeItems: 'center', position: 'relative' }}>
|
|
<CircularProgress variant="determinate" value={progress} />
|
|
<Box sx={{ position: 'absolute' }}>
|
|
<Typography fontSize="0.8rem">{`${Math.round(progress)}%`}</Typography>
|
|
</Box>
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
interface IUpdateCheckerProps {
|
|
handleFinishedUpdate: (time: number) => void;
|
|
}
|
|
|
|
function UpdateChecker({ handleFinishedUpdate }: IUpdateCheckerProps) {
|
|
const { t } = useTranslation();
|
|
|
|
const [loading, setLoading] = useState(false);
|
|
const [progress, setProgress] = useState(0);
|
|
|
|
const onClick = async () => {
|
|
try {
|
|
setLoading(true);
|
|
setProgress(0);
|
|
await requestManager.startGlobalUpdate().response;
|
|
} catch (e) {
|
|
makeToast(t('global.error.label.update_failed'), 'error');
|
|
setLoading(false);
|
|
}
|
|
};
|
|
|
|
useEffect(() => {
|
|
const wsc = requestManager.getUpdateWebSocket();
|
|
|
|
// "loading" can't be used since it will be outdated once the state gets changed
|
|
// it could be used by adding it as a dependency of "useEffect" but then the socket would
|
|
// get closed and connected again every time it changes
|
|
let updateStarted = false;
|
|
|
|
wsc.onmessage = (e) => {
|
|
const { running, statusMap } = JSON.parse(e.data) as IUpdateStatus;
|
|
const { COMPLETE = [], RUNNING = [], PENDING = [] } = statusMap;
|
|
|
|
const currentProgress = 100 * (COMPLETE.length / (COMPLETE.length + RUNNING.length + PENDING.length));
|
|
|
|
const isUpdateFinished = currentProgress === 100;
|
|
const ignoreFaultyMessage = !updateStarted && !running && isUpdateFinished;
|
|
|
|
// for some reason the server sends 100% completed manga updates when connecting to the
|
|
// socket while no update is running
|
|
if (ignoreFaultyMessage) {
|
|
return;
|
|
}
|
|
|
|
updateStarted = running;
|
|
setLoading(running);
|
|
setProgress(Number.isNaN(currentProgress) ? 0 : currentProgress);
|
|
|
|
if (isUpdateFinished) {
|
|
handleFinishedUpdate(Date.now());
|
|
}
|
|
};
|
|
|
|
return () => wsc.close();
|
|
}, []);
|
|
|
|
return (
|
|
<IconButton onClick={onClick} disabled={loading}>
|
|
{loading ? <Progress progress={progress} /> : <RefreshIcon />}
|
|
</IconButton>
|
|
);
|
|
}
|
|
|
|
export default UpdateChecker;
|