whole update logic working

app.js

@@ -21,12 +21,25 @@ const run = () => {
 
         mainWindow = createWindow();
         mainWindow.on('show', async () => {
+            await doUpdateCheck(mainWindow);
+        })
+        app.on('activate', function () {
+            if (BrowserWindow.getAllWindows().length === 0) mainWindow = createWindow();
+        })
+        app.on('window-all-closed', () => {
+            app.quit()
+        })
+        ipcMain.on('do-update-check', async () => {
+            await doUpdateCheck(mainWindow);
+        })
+        ipcMain.on('do-update', async () => {
             const osuPath = await config.get("osuPath", "");
             const isValid = await osuUtil.isValidOsuFolder(osuPath);
             if (osuPath.trim == "" || !isValid) {
                 mainWindow.webContents.send('status_update', {
-                    type: "missing-folder"
+                    type: "error",
-                })
+                    message: "Invalid osu! folder"
+                });
                 return;
             }
             if (fs.existsSync(osuPath)) {
@@ -41,23 +54,18 @@ const run = () => {
 
                 const releaseFiles = await osuUtil.getUpdateFiles(releaseStream);
                 const filesToDownload = await osuUtil.filesThatNeedUpdate(tempOsuPath, releaseFiles);
-                // const downloadTask = await osuUtil.downloadUpdateFiles(osuPath, filesToDownload);
+                const downloadTask = await osuUtil.downloadUpdateFiles(osuPath, filesToDownload);
-                // downloadTask.on('completed', () => {
+                downloadTask.on('completed', () => {
-                //     console.log("done!");
+                    mainWindow.webContents.send('status_update', {
-                // });
+                        type: "update-complete"
-                mainWindow.webContents.send('status_update', {
+                    })
-                    type: filesToDownload.length > 0 ? "update-available" : "up-to-date"
+                });
-                })
+
             } else
                 mainWindow.webContents.send('status_update', {
-                    type: "missing-folder"
+                    type: "error",
-                })
+                    message: "Invalid osu! folder"
-        })
+                });
-        app.on('activate', function () {
-            if (BrowserWindow.getAllWindows().length === 0) mainWindow = createWindow();
-        })
-        app.on('window-all-closed', () => {
-            app.quit()
         })
         ipcMain.handle('set-osu-dir', async (event) => {
             const yes = await dialog.showOpenDialog({
@@ -75,6 +83,37 @@ const run = () => {
     })
 }
 
+async function doUpdateCheck(window) {
+    const osuPath = await config.get("osuPath", "");
+    const isValid = await osuUtil.isValidOsuFolder(osuPath);
+    if (osuPath.trim == "" || !isValid) {
+        window.webContents.send('status_update', {
+            type: "missing-folder"
+        })
+        return;
+    }
+    if (fs.existsSync(osuPath)) {
+        tempOsuPath = osuPath;
+        const osuConfig = await osuUtil.getLatestConfig(tempOsuPath);
+        const lastVersion = await osuConfig.get("LastVersion");
+        let releaseStream = "stable40";
+        if (lastVersion.endsWith("cuttingedge"))
+            releaseStream = "cuttingedge"
+        else if (lastVersion.endsWith("beta"))
+            releaseStream = "beta";
+
+        const releaseFiles = await osuUtil.getUpdateFiles(releaseStream);
+        const filesToDownload = await osuUtil.filesThatNeedUpdate(tempOsuPath, releaseFiles);
+        console.log("sending update check " + (filesToDownload.length > 0 ? "update-available" : "up-to-date"))
+        window.webContents.send('status_update', {
+            type: filesToDownload.length > 0 ? "update-available" : "up-to-date"
+        })
+    } else
+        window.webContents.send('status_update', {
+            type: "missing-folder"
+        })
+}
+
 function createWindow() {
     // Create the browser window.
     const win = windowManager.createWindow(520, 420);

osuUtil.js

@@ -120,7 +120,9 @@ async function downloadUpdateFiles(osuPath, filesToUpdate) {
     let completedIndex = 0;
     filesToUpdate.forEach(async (fileToUpdate) => {
         const filePath = path.join(osuPath, fileToUpdate.fileName);
-        await fs.promises.rm(filePath);
+        if (await fu.existsAsync(filePath))
+            await fs.promises.rm(filePath);
+        
         const fileDownload = new DownloaderHelper(fileToUpdate.fileURL, osuPath, {
             fileName: fileToUpdate.fileName,
             override: true,

preload/preload.js

@@ -15,6 +15,21 @@ window.addEventListener('DOMContentLoaded', () => {
     const $ = require('jquery');
     const Swal = require('sweetalert2');
 
+    let currentState;
+
+    $("#launch-btn").on('click', async () => {
+        switch (currentState) {
+            case "up-to-date":
+                //TODO: launch client
+                break;
+            case "update-available":
+                $("#launch-btn").attr('disabled', true);
+                $('#launch-btn').html('Updating...');
+                ipcRenderer.send("do-update");
+                break;
+        }
+    });
+
     $("#folder-btn").on('click', async () => {
         const success = await ipcRenderer.invoke('set-osu-dir');
         if (success == undefined)
@@ -26,6 +41,7 @@ window.addEventListener('DOMContentLoaded', () => {
                 icon: 'success',
                 confirmButtonText: 'Cool'
             })
+            ipcRenderer.send("do-update-check");
         } else {
             Swal.fire({
                 title: 'Uh oh!',
@@ -37,6 +53,7 @@ window.addEventListener('DOMContentLoaded', () => {
     });
 
     ipcRenderer.on('status_update', (event, status) => {
+        currentState = status.type;
         switch (status.type) {
             case "up-to-date":
                 $("#launch-btn").attr('disabled', false);
@@ -47,7 +64,26 @@ window.addEventListener('DOMContentLoaded', () => {
                 $('#launch-btn').html('Update');
                 break;
             case "missing-folder":
-                $('#launch-btn').html('Please set your osu folder!');
+                $('#launch-btn').html('Please set your osu! folder');
+                break;
+            case "error":
+                Swal.fire({
+                    title: 'Uh oh!',
+                    text: status.message,
+                    icon: 'error',
+                    confirmButtonText: 'Okay'
+                });
+                ipcRenderer.send("do-update-check");
+                break;
+            case "update-complete":
+                Swal.fire({
+                    title: 'Yaaay!',
+                    text: "Your osu! client has been successfully updated!",
+                    icon: 'success',
+                    confirmButtonText: 'Thanks :3'
+                });
+                ipcRenderer.send("do-update-check");
+                break;
         }
     })
 

ui/windowManager.js

@@ -33,9 +33,9 @@ module.exports = {
 
         window.webContents.setUserAgent("EZPPLauncher");
         attachTitlebarToWindow(window);
-        window.webContents.openDevTools({
+        // window.webContents.openDevTools({
-            mode: "detach"
+        //     mode: "detach"
-        });
+        // });
 
         return window;
     },