From 56f2cb84412252ae0a90076b861f3b1be32be01d Mon Sep 17 00:00:00 2001
From: Jan200101 <sentrycraft123@gmail.com>
Date: Sun, 4 Aug 2024 18:44:53 +0200
Subject: [PATCH] feat: support thunderstore ror2mm protocol for installing
 mods

---
 src/index.js            |  6 +++-
 src/modules/packages.js |  2 ++
 src/modules/protocol.js | 66 +++++++++++++++++++++++++++++++++++++++++
 3 files changed, 73 insertions(+), 1 deletion(-)
 create mode 100644 src/modules/protocol.js

diff --git a/src/index.js b/src/index.js
index 9321598..b26ef1e 100644
--- a/src/index.js
+++ b/src/index.js
@@ -15,6 +15,7 @@ const mods = require("./modules/mods");
 const update = require("./modules/update");
 const version = require("./modules/version");
 const settings = require("./modules/settings");
+const protocol = require("./modules/protocol");
 
 // loads `ipcMain` events that dont fit in any of the modules directly
 require("./modules/ipc");
@@ -88,6 +89,7 @@ function start() {
 
 	// load list of mods on initial load
 	win.webContents.on("dom-ready", () => {
+		protocol();
 		send("mods", mods.list());
 	})
 
@@ -103,6 +105,7 @@ function start() {
 	}
 }
 
+
 // starts the GUI or CLI
 if (cli.hasArgs()) {
 	if (cli.hasParam("update-viper")) {
@@ -112,9 +115,10 @@ if (cli.hasArgs()) {
 		cli.init();
 	}
 } else {
+	app.setAsDefaultProtocolClient("ror2mm");
+
 	// start the window/GUI
 	app.on("ready", () => {
-
 		start();
 	})
 }
diff --git a/src/modules/packages.js b/src/modules/packages.js
index 8df6665..f7e7d62 100644
--- a/src/modules/packages.js
+++ b/src/modules/packages.js
@@ -16,6 +16,7 @@ var packages = {};
 
 // lets renderer install packages
 ipcMain.on("install-from-url", (event, url, author, package_name, version) => {
+	console.log(url);
 	packages.install(url, author, package_name, version);
 })
 
@@ -194,6 +195,7 @@ packages.install = async (url, author, package_name, version) => {
 		return false;
 	}
 
+
 	let name = packages.format_name(author, package_name, version);
 
 	// removes zip's and folders
diff --git a/src/modules/protocol.js b/src/modules/protocol.js
new file mode 100644
index 0000000..8b6cfc6
--- /dev/null
+++ b/src/modules/protocol.js
@@ -0,0 +1,66 @@
+const { app, ipcMain } = require("electron");
+
+const requests = require("./requests");
+const version = require("./version");
+
+async function install_mod(domain, author, package_name, version) {
+	let package_data = JSON.parse(await requests.get(
+		domain, `/api/experimental/package/${author}/${package_name}/${version}/`
+	));
+
+	for (const dep of package_data.dependencies) {
+		let fragments = dep.split("-");
+		if (fragments.length != 3) {
+			console.error("bad dep")
+			return;
+		}
+
+		if (fragments[0] != "northstar")
+			await install_mod(domain, ...fragments);
+	}
+
+	let result = ipcMain.emit("install-from-url", null, package_data.download_url, author, package_name, version);
+	if (!result) {
+		console.error("no install-from-url handler")
+	}
+}
+
+module.exports = async () => {
+	if (version.northstar() == "unknown")
+		return;
+
+	const args = process.argv.slice(app.isPackaged ? 1 : 2);
+
+	for (const key of args) {
+		if (key.startsWith("ror2mm://")) {
+			let fragments = key.slice(9).split("/");
+		
+			if (fragments.length < 6)
+				return;
+
+			const ver = fragments[0];
+			const term = fragments[1];
+			const domain = fragments[2];
+			const author = fragments[3];
+			const mod = fragments[4];
+			const version = fragments[5];
+
+			// There is only v1
+			if (ver != "v1")
+				continue;
+
+			// No support for custom thunderstore instances
+			if (domain != "thunderstore.io")
+				continue;
+
+			try {
+				if (term == "install") {
+					await install_mod(domain, author, mod, version);
+				}
+			}catch(err) {
+				console.error(err);
+				continue;
+			}
+		}
+	}
+}