From 1ad5d0b726eced4230b25a192b057ef465f94191 Mon Sep 17 00:00:00 2001
From: rustdesk <info@rustdesk.com>
Date: Fri, 9 Jun 2023 21:36:12 +0800
Subject: [PATCH] simplify uri

---
 flutter/lib/common.dart | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/flutter/lib/common.dart b/flutter/lib/common.dart
index a78af2a94..102ee6e66 100644
--- a/flutter/lib/common.dart
+++ b/flutter/lib/common.dart
@@ -1598,18 +1598,25 @@ bool parseRustdeskUri(String uriPath) {
 bool callUniLinksUriHandler(Uri uri) {
   debugPrint("uni links called: $uri");
   // new connection
+  String peerId;
   if (uri.authority == "connection" && uri.path.startsWith("/new/")) {
-    final peerId = uri.path.substring("/new/".length);
-    var param = uri.queryParameters;
-    String? switch_uuid = param["switch_uuid"];
-    String? password = param["password"];
-    Future.delayed(Duration.zero, () {
-      rustDeskWinManager.newRemoteDesktop(peerId,
-          password: password, switch_uuid: switch_uuid);
-    });
-    return true;
+    peerId = uri.path.substring("/new/".length);
+  } else if (uri.authority == "connect") {
+    peerId = uri.path.substring(1);
+  } else if (uri.authority.length > 2 && uri.path.length <= 1) {
+    // "/" or ""
+    peerId = uri.authority;
+  } else {
+    return false;
   }
-  return false;
+  var param = uri.queryParameters;
+  String? switch_uuid = param["switch_uuid"];
+  String? password = param["password"];
+  Future.delayed(Duration.zero, () {
+    rustDeskWinManager.newRemoteDesktop(peerId,
+        password: password, switch_uuid: switch_uuid);
+  });
+  return true;
 }
 
 connectMainDesktop(String id,