From 7b73aa2908f8d2ac04870ce901b982f0da2ea36d Mon Sep 17 00:00:00 2001
From: Josiah Glosson <soujournme@gmail.com>
Date: Tue, 8 Jul 2025 19:44:08 -0500
Subject: [PATCH] Implement the Update Now button

---
 .../src/components/ui/UpdateModal.vue         | 42 ++++++++++++++++---
 .../src/components/ui/modal/ModalWrapper.vue  |  8 +++-
 2 files changed, 44 insertions(+), 6 deletions(-)

diff --git a/apps/app-frontend/src/components/ui/UpdateModal.vue b/apps/app-frontend/src/components/ui/UpdateModal.vue
index fc831a777..806be7dbd 100644
--- a/apps/app-frontend/src/components/ui/UpdateModal.vue
+++ b/apps/app-frontend/src/components/ui/UpdateModal.vue
@@ -1,5 +1,10 @@
 <template>
-  <ModalWrapper ref="modal" :header="formatMessage(messages.header)" :on-hide="onHide">
+  <ModalWrapper
+    ref="modal"
+    :header="formatMessage(messages.header)"
+    :on-hide="onHide"
+    :closable="!updateInProgress"
+  >
     <div>{{ formatMessage(messages.bodyVersion, { version: update!.version }) }}</div>
     <div v-if="updateSize">
       {{ formatMessage(messages.bodySize, { size: formatBytes(updateSize) }) }}
@@ -9,20 +14,23 @@
         formatMessage(messages.bodyChangelog)
       }}</a>
     </div>
+    <ProgressBar class="mt-4" :progress="downloadProgress" />
     <div class="mt-4 flex flex-wrap gap-2">
       <ButtonStyled color="green">
-        <button>
+        <button :disabled="updateInProgress" @click="installUpdateNow">
           <RefreshCwIcon />
           {{ formatMessage(messages.restartNow) }}
         </button>
       </ButtonStyled>
       <ButtonStyled>
-        <button @click="hide">
+        <button :disabled="updateInProgress">
+          <RightArrowIcon />
           {{ formatMessage(messages.later) }}
         </button>
       </ButtonStyled>
       <ButtonStyled color="red">
-        <button @click="skipUpdate">
+        <button :disabled="updateInProgress" @click="skipUpdate">
+          <XIcon />
           {{ formatMessage(messages.skip) }}
         </button>
       </ButtonStyled>
@@ -35,10 +43,12 @@ import ModalWrapper from '@/components/ui/modal/ModalWrapper.vue'
 import { defineMessages, useVIntl } from '@vintl/vintl'
 import { useTemplateRef, ref } from 'vue'
 import { ButtonStyled } from '@modrinth/ui'
-import { RefreshCwIcon } from '@modrinth/assets'
+import { RefreshCwIcon, XIcon, RightArrowIcon } from '@modrinth/assets'
 import { getUpdateSize } from '@/helpers/utils'
 import { formatBytes } from '@modrinth/utils'
 import { handleError } from '@/store/notifications'
+import ProgressBar from '@/components/ui/ProgressBar.vue'
+import { Update } from '@tauri-apps/plugin-updater'
 
 const emit = defineEmits<{
   (e: 'updateSkipped', version: string): void
@@ -79,11 +89,17 @@ const messages = defineMessages({
 
 type UpdateData = {
   rid: number
+  currentVersion: string
   version: string
+  date?: string
+  body?: string
+  rawJson: Record<string, unknown>
 }
 
 const update = ref<UpdateData>()
 const updateSize = ref<number>()
+const updateInProgress = ref(false)
+const downloadProgress = ref(0)
 
 const modal = useTemplateRef('modal')
 const isOpen = ref(false)
@@ -105,6 +121,22 @@ function hide() {
 
 defineExpose({ show, hide, isOpen })
 
+function installUpdateNow() {
+  updateInProgress.value = true
+  let totalSize = 0
+  let totalDownloaded = 0
+  new Update(update.value!).downloadAndInstall((event) => {
+    if (event.event === 'Started') {
+      totalSize = event.data.contentLength!
+    } else if (event.event === 'Progress') {
+      totalDownloaded += event.data.chunkLength
+    } else if (event.event === 'Finished') {
+      totalDownloaded = totalSize
+    }
+    downloadProgress.value = (totalDownloaded / totalSize) * 100
+  })
+}
+
 function skipUpdate() {
   hide()
   emit('updateSkipped', update.value!.version)
diff --git a/apps/app-frontend/src/components/ui/modal/ModalWrapper.vue b/apps/app-frontend/src/components/ui/modal/ModalWrapper.vue
index 81f386f62..b48438d7c 100644
--- a/apps/app-frontend/src/components/ui/modal/ModalWrapper.vue
+++ b/apps/app-frontend/src/components/ui/modal/ModalWrapper.vue
@@ -48,7 +48,13 @@ function onModalHide() {
 </script>
 
 <template>
-  <Modal ref="modal" :header="header" :noblur="!themeStore.advancedRendering" @hide="onModalHide">
+  <Modal
+    ref="modal"
+    :header="header"
+    :noblur="!themeStore.advancedRendering"
+    @hide="onModalHide"
+    :closable="closable"
+  >
     <template #title>
       <slot name="title" />
     </template>