Sets page error checking and download button fixes

willie-yao · Feb 10, 2023 · a4e0e32 · a4e0e32
1 parent 05637bd
commit a4e0e32
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 1 deletion.
diff --git a/src/main/main.ts b/src/main/main.ts
@@ -83,6 +83,8 @@ ipcMain.handle('retrieve-video-information', async (event, arg) => {
   }).then((output) => {
     // @ts-ignore
     return output.timestamp;
+  }).catch((error) => {
+    return error;
   });
 });
 

diff --git a/src/renderer/components/VideoSearch.tsx b/src/renderer/components/VideoSearch.tsx
@@ -6,7 +6,7 @@ import { useTheme } from '@mui/material/styles';
 import { PropagateLoader } from 'react-spinners';
 import { useNavigate } from 'react-router-dom';
 import HiddenTextField from 'renderer/common/HiddenTextField';
-// import validator from 'validator';
+import SnackbarPopup from 'renderer/common/SnackbarPopup';
 
 export interface VODMetadata {
   title: string;
@@ -37,6 +37,8 @@ const RetrieveSets = (
   }
 
   const [waiting, setWaiting] = useState(false);
+  const [errorMessage, setErrorMessage] = useState('');
+  const [errorOpen, setErrorOpen] = useState(false);
 
   const [getSets, { loading, error, data }] = useLazyQuery(
     GET_SETS_AT_STATION,
@@ -113,6 +115,10 @@ const RetrieveSets = (
               tournamentName: data.event.tournament.name,
             },
           });
+        }).catch((err) => {
+          setErrorMessage("Error retrieving sets: " + err.message);
+          setErrorOpen(true);
+          setWaiting(false);
         });
     }
   }, [data]);
@@ -127,6 +133,7 @@ const RetrieveSets = (
         justifyContent: 'center',
       }}
     >
+      {SnackbarPopup(errorMessage, 'error', errorOpen, setErrorOpen)}
       <Button
         variant="contained"
         onClick={() => {

diff --git a/src/renderer/pages/SetsView.tsx b/src/renderer/pages/SetsView.tsx
@@ -50,6 +50,15 @@ const SetsView = () => {
   const [errorMessage, setErrorMessage] = useState('');
   const [infoMessage, setInfoMessage] = useState('')
   const [downloaded, setDownloaded] = useState(false)
+  const [enableDownload, setEnableDownload] = useState(false)
+
+  useEffect(() => {
+    if (checked.length == location.state.sets.length + 1) {
+      setEnableDownload(false)
+    } else {
+      setEnableDownload(true)
+    }
+  }, [checked])
 
   const handleSelectAll = () => {
     let newChecked = [...checked];
@@ -319,13 +328,15 @@ const SetsView = () => {
       </List>
       <Box sx={{ display: 'flex', justifyContent: 'space-around', width: '100%' }}>
         <Button
+          disabled={!enableDownload}
           variant="contained"
           color="secondary"
           sx={{ width: '27vw', marginTop: '20px', color: 'white' }}
           onClick={() => {
             setDownloaded(true)
             setInfoMessage('Your VODs are being downloaded to ./downloadedVODs/' + location.state.tournamentName);
             setInfoOpen(true)
+            setEnableDownload(false)
             location.state.sets.map((set: VODMetadata) => {
               if (set.download) {
                 console.log('Downloading set: ', set);
@@ -340,10 +351,12 @@ const SetsView = () => {
                   .then((res: any) => {
                     setSuccessMessage('Download complete: ' + set.title);
                     setSuccessOpen(true);
+                    setEnableDownload(true);
                   })
                   .catch((err: any) => {
                     setErrorMessage('Download failed: ' + err);
                     setErrorOpen(true);
+                    setEnableDownload(true);
                   });
               }
             });
@@ -361,6 +374,7 @@ const SetsView = () => {
           Open VOD Folder
         </Button>
         <Button
+          disabled={!downloaded}
           variant="contained"
           color="secondary"
           sx={{ width: '27vw', marginTop: '20px', color: 'white' }}