From 98e73b237218b68e9aceff050c904ab77c5497e6 Mon Sep 17 00:00:00 2001
From: "Namhyeon, Go" <gnh1201@gmail.com>
Date: Tue, 22 Apr 2025 21:26:42 +0900
Subject: [PATCH] Fix bug when use the devtools protocol with the remote allow
 origins (Fix WebSocket error)

---
 WelsonJS.Toolkit/WelsonJS.Launcher/Program.cs  | 6 ++++--
 WelsonJS.Toolkit/WelsonJS.Launcher/editor.html | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/WelsonJS.Toolkit/WelsonJS.Launcher/Program.cs b/WelsonJS.Toolkit/WelsonJS.Launcher/Program.cs
index cb51fca..59be1ba 100644
--- a/WelsonJS.Toolkit/WelsonJS.Launcher/Program.cs
+++ b/WelsonJS.Toolkit/WelsonJS.Launcher/Program.cs
@@ -150,9 +150,11 @@ namespace WelsonJS.Launcher
         }
         public static void OpenWebBrowser(string url)
         {
+            Uri resourceServerUri = new Uri(GetAppConfig("ResourceServerPrefix"));
+            Uri devToolsUri = new Uri(GetAppConfig("DevToolsPrefix"));
+
             string userDataDir = Path.Combine(GetAppDataPath(), "EdgeUserProfile");
-            string remoteAllowOrigins = GetAppConfig("ResourceServerPrefix");
-            int remoteDebuggingPort = new Uri(GetAppConfig("DevToolsPrefix")).Port;
+            string remoteAllowOrigins = $"{resourceServerUri.Scheme}://{resourceServerUri.Host}:{resourceServerUri.Port}";
             string[] arguments = {
                 $"\"{url}\"",
                 $"--remote-debugging-port={remoteDebuggingPort}",
diff --git a/WelsonJS.Toolkit/WelsonJS.Launcher/editor.html b/WelsonJS.Toolkit/WelsonJS.Launcher/editor.html
index 71b5b9e..54b97d2 100644
--- a/WelsonJS.Toolkit/WelsonJS.Launcher/editor.html
+++ b/WelsonJS.Toolkit/WelsonJS.Launcher/editor.html
@@ -395,7 +395,7 @@
                         if (response.id == 3) {
                             const responseContent = response.result.result.value;
 
-                            appendTextToEditor(responseContent);
+                            appendTextToEditor("/*\n" + responseContent + "\n*/");
                             pushPromptMessage("assistant", responseContent);
 
                             socket.close();