fix: live reload delay (#1822)

luckyframework · Aug 25, 2023 · 107dc61 · 107dc61
1 parent d60191e
commit 107dc61
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/lucky/tags/live_reload_tag.cr b/src/lucky/tags/live_reload_tag.cr
@@ -1,11 +1,15 @@
 module Lucky::LiveReloadTag
-  def live_reload_connect_tag : Nil
+  def live_reload_connect_tag(ms : Int32 = 1000) : Nil
     {% if flag?(:livereloadws) %}
       tag "script" do
         raw <<-JS
         (function() {
           var ws = new WebSocket("ws://#{Lucky::ServerSettings.host}:#{Lucky::ServerSettings.reload_port}");
-          ws.onmessage = function() { location.reload(); };
+          ws.onmessage = function() {
+            setTimeout(function() {
+              location.reload();
+            }, #{ms});
+          };
         })();
         JS
       end
@@ -14,7 +18,11 @@ module Lucky::LiveReloadTag
         raw <<-JS
         (function() {
           var stream = new EventSource("http://#{Lucky::ServerSettings.host}:#{Lucky::ServerSettings.reload_port}");
-          stream.onmessage = function() { location.reload(); };
+          stream.onmessage = function() {
+            setTimeout(function() {
+              location.reload();
+            }, #{ms});
+          };
         })();
         JS
       end