From 6d9f145ada43e1ef3397dcf3e9221b7ce26fd1e4 Mon Sep 17 00:00:00 2001
From: Pauline <git@ethanlibs.co>
Date: Sun, 28 Jul 2024 21:05:31 +0200
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feature:=20add=20webview-based=20au?=
 =?UTF-8?q?th?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/desktop/build.rs                         |  3 +-
 .../permissions/onelauncher/default.toml      |  3 +-
 apps/desktop/src/api/commands.rs              | 55 +++++++++++++++----
 3 files changed, 47 insertions(+), 14 deletions(-)

diff --git a/apps/desktop/build.rs b/apps/desktop/build.rs
index 05964c47..f5b2cf71 100644
--- a/apps/desktop/build.rs
+++ b/apps/desktop/build.rs
@@ -6,8 +6,7 @@ fn main() {
 				"onelauncher",
 				tauri_build::InlinedPlugin::new().commands(&[
 					// User
-					"begin_msa",
-					"finish_msa",
+					"auth_login",
 					"get_users",
 					"get_user",
 					"remove_user",
diff --git a/apps/desktop/permissions/onelauncher/default.toml b/apps/desktop/permissions/onelauncher/default.toml
index c9fd51d7..a55b8d12 100644
--- a/apps/desktop/permissions/onelauncher/default.toml
+++ b/apps/desktop/permissions/onelauncher/default.toml
@@ -2,8 +2,7 @@
 description = "Default permissions for the plugin"
 permissions = [
   # User
-  "allow-begin-msa",
-  "allow-finish-msa",
+  "allow-auth-login",
   "allow-get-users",
   "allow-get-user",
   "allow-remove-user",
diff --git a/apps/desktop/src/api/commands.rs b/apps/desktop/src/api/commands.rs
index 1d5fb72d..98f196ff 100644
--- a/apps/desktop/src/api/commands.rs
+++ b/apps/desktop/src/api/commands.rs
@@ -4,10 +4,11 @@ use interpulse::api::minecraft::Version;
 use onelauncher::constants::{NATIVE_ARCH, TARGET_OS, VERSION};
 use onelauncher::data::{Loader, ManagedPackage, MinecraftCredentials, PackageData, Settings};
 use onelauncher::package::content;
-use onelauncher::store::{Cluster, ClusterPath, MinecraftLogin};
+use onelauncher::store::{Cluster, ClusterPath};
 use onelauncher::{cluster, minecraft, processor, settings};
 use serde::{Deserialize, Serialize};
 use specta::Type;
+use tauri::{AppHandle, Manager};
 use uuid::Uuid;
 
 #[macro_export]
@@ -21,8 +22,7 @@ macro_rules! collect_commands {
 			)
 			.commands(tauri_specta::collect_commands![
 				// User
-				begin_msa,
-				finish_msa,
+				auth_login,
 				get_users,
 				get_user,
 				remove_user,
@@ -214,14 +214,49 @@ pub async fn get_user(uuid: Uuid) -> Result<MinecraftCredentials, String> {
 
 #[specta::specta]
 #[tauri::command]
-pub async fn begin_msa() -> Result<MinecraftLogin, String> {
-	Ok(minecraft::begin().await?)
-}
+pub async fn auth_login(handle: AppHandle) -> Result<Option<MinecraftCredentials>, String> {
+	let flow = minecraft::begin().await?;
+	let now = chrono::Utc::now();
 
-#[specta::specta]
-#[tauri::command]
-pub async fn finish_msa(code: String, login: MinecraftLogin) -> Result<MinecraftCredentials, String> {
-	Ok(minecraft::finish(code.as_str(), login).await?)
+	if let Some(win) = handle.get_webview_window("login") {
+		win.close().map_err(|err| err.to_string())?;
+	}
+
+	let win = tauri::WebviewWindowBuilder::new(
+		&handle,
+		"login",
+		tauri::WebviewUrl::External(flow.redirect_uri.parse().map_err(|_|
+			anyhow::anyhow!("failed to parse auth redirect url")
+		).map_err(|err| err.to_string())?),
+	)
+	.title("Log into OneLauncher")
+	.always_on_top(true)
+	.center()
+	.build()
+	.map_err(|err| err.to_string())?;
+
+	win.request_user_attention(Some(tauri::UserAttentionType::Critical)).map_err(|err| err.to_string())?;
+
+	while (chrono::Utc::now() - now) < chrono::Duration::minutes(10) {
+		if win.title().is_err() {
+			return Ok(None);
+		}
+
+		if win.url().map_err(|err| err.to_string())?.as_str().starts_with("https://login.live.com/oauth20_desktop.srf") {
+			if let Some((_, code)) = win.url().map_err(|err| err.to_string())?.query_pairs().find(|x| x.0 == "code") {
+				win.close().map_err(|err| err.to_string())?;
+				let value = minecraft::finish(&code.clone(), flow).await?;
+
+				return Ok(Some(value));
+			}
+		}
+
+		tokio::time::sleep(std::time::Duration::from_millis(50)).await;
+	}
+
+	win.close().map_err(|err| err.to_string())?;
+
+	Ok(None)
 }
 
 #[specta::specta]