From 23aefccbe98d71ad45a5576c24f5f45ab309ce30 Mon Sep 17 00:00:00 2001
From: "nicolas.dorier" <nicolas.dorier@gmail.com>
Date: Tue, 4 Aug 2020 11:21:16 +0900
Subject: [PATCH] Construct websocket uri for notification client side

---
 .../NotificationsNavItem.cshtml               | 20 ++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/BTCPayServer/Views/Shared/LayoutPartials/NotificationsNavItem.cshtml b/BTCPayServer/Views/Shared/LayoutPartials/NotificationsNavItem.cshtml
index 3d005c68b..b4c1a78ef 100644
--- a/BTCPayServer/Views/Shared/LayoutPartials/NotificationsNavItem.cshtml
+++ b/BTCPayServer/Views/Shared/LayoutPartials/NotificationsNavItem.cshtml
@@ -1,12 +1,22 @@
-<div></div>
+@inject LinkGenerator linkGenerator
+
+<div></div>
 @await Component.InvokeAsync("NotificationsDropdown")
-<script>
+<script type="text/javascript">
 
 var supportsWebSockets = 'WebSocket' in window && window.WebSocket.CLOSING === 2;
 
-if (supportsWebSockets) {
-    var ws_uri = "@Url.Action("SubscribeUpdates", "Notifications", new {}, Context.Request.Scheme)".replace("http", "ws");
-    var newDataEndpoint = "@Url.Action("GetNotificationDropdownUI", "Notifications", new {}, Context.Request.Scheme)";
+    if (supportsWebSockets) {
+
+    var loc = window.location, ws_uri;
+    if (loc.protocol === "https:") {
+        ws_uri = "wss:";
+    } else {
+        ws_uri = "ws:";
+    }
+    ws_uri += "//" + loc.host;
+    ws_uri += "@linkGenerator.GetPathByAction("SubscribeUpdates", "Notifications")";
+    var newDataEndpoint = "@linkGenerator.GetPathByAction("GetNotificationDropdownUI", "Notifications")";
 
     try {
         socket = new WebSocket(ws_uri);