From aba62bfaf4556c3b6285066b43a6063976a8d4f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9my=20Raes?= <raes.remy@gmail.com>
Date: Fri, 22 Nov 2024 15:19:07 +0100
Subject: [PATCH] Make MAD process cancellable (#865)

Mod counterpart to the launcher PR that makes a mod download via MAD cancellable.
---
 .github/nativefuncs.json                      |  6 ++++++
 .../vscripts/ui/menu_ns_moddownload.nut       | 19 +++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/.github/nativefuncs.json b/.github/nativefuncs.json
index abc213caf..10ceebd5d 100644
--- a/.github/nativefuncs.json
+++ b/.github/nativefuncs.json
@@ -414,6 +414,12 @@
 			"returnTypeString": "void",
 			"argTypes": "string name, string version"
 		},
+		{
+			"name": "NSCancelModDownload",
+			"helpText": "prevents installation of the mod currently being installed",
+			"returnTypeString": "void",
+			"argTypes": ""
+		},
 		{
 			"name": "NSGetModInstallState",
 			"helpText": "get status of the mod currently being installed",
diff --git a/Northstar.Client/mod/scripts/vscripts/ui/menu_ns_moddownload.nut b/Northstar.Client/mod/scripts/vscripts/ui/menu_ns_moddownload.nut
index 4968714c7..09001f57b 100644
--- a/Northstar.Client/mod/scripts/vscripts/ui/menu_ns_moddownload.nut
+++ b/Northstar.Client/mod/scripts/vscripts/ui/menu_ns_moddownload.nut
@@ -9,6 +9,7 @@ global enum eModInstallStatus
     CHECKSUMING,
     EXTRACTING,
     DONE,
+    ABORTED,
     FAILED,
     FAILED_READING_ARCHIVE,
     FAILED_WRITING_TO_DISK,
@@ -57,6 +58,11 @@ bool function DownloadMod( RequiredModInfo mod )
 	dialogData.message = Localize( "#DOWNLOADING_MOD_TEXT", mod.name, mod.version )
 	dialogData.showSpinner = true;
 
+	// Prevent download button
+	AddDialogButton( dialogData, "#CANCEL", void function() {
+		NSCancelModDownload()
+	})
+
 	// Prevent user from closing dialog
 	dialogData.forceChoice = true;
 	OpenDialog( dialogData )
@@ -77,6 +83,13 @@ bool function DownloadMod( RequiredModInfo mod )
 		WaitFrame()
 	}
 
+	// If download was aborted, don't close UI since it was closed by clicking cancel button
+	if ( state.status == eModInstallStatus.ABORTED )
+	{
+		print("Mod download was cancelled by the user.")
+		return false;
+	}
+
 	printt( "Mod status:", state.status )
 
 	// Close loading dialog
@@ -114,6 +127,12 @@ void function DisplayModDownloadErrorDialog( string modName )
 {
 	ModInstallState state = NSGetModInstallState()
 
+	// If user cancelled download, no need to display an error message
+	if ( state.status == eModInstallStatus.ABORTED )
+	{
+		return
+	}
+
 	DialogData dialogData
 	dialogData.header = Localize( "#FAILED_DOWNLOADING", modName )
 	dialogData.image = $"ui/menu/common/dialog_error"