From 07fb7654a10e31255d2986cc948585e980320c3b Mon Sep 17 00:00:00 2001
From: krymtkts <takatoshi.kuriyama@gmail.com>
Date: Sat, 29 Jul 2023 17:38:08 +0900
Subject: [PATCH 1/2] Replace raw JavaScript with Fable.Browser.WbSocket.

---
 js/live-reload.js |  5 -----
 src/App.fsproj    |  2 ++
 src/Dev.fs        | 10 ++++++++++
 src/Generator.fs  |  3 +--
 4 files changed, 13 insertions(+), 7 deletions(-)
 delete mode 100644 js/live-reload.js
 create mode 100644 src/Dev.fs

diff --git a/js/live-reload.js b/js/live-reload.js
deleted file mode 100644
index 76e5604..0000000
--- a/js/live-reload.js
+++ /dev/null
@@ -1,5 +0,0 @@
-const init = () =>{
-    websocket = new WebSocket("ws://"+window.location.host+"/websocket");
-    websocket.onmessage = (e) => location.reload();
-}
-window.addEventListener("load", init, false);
diff --git a/src/App.fsproj b/src/App.fsproj
index 1e436e5..908b963 100644
--- a/src/App.fsproj
+++ b/src/App.fsproj
@@ -8,11 +8,13 @@
     <Compile Include="bindings/Yaml.fs" />
     <Compile Include="Node.Extra.fs" />
     <Compile Include="Common.fs" />
+    <Compile Include="Dev.fs" />
     <Compile Include="Generator.fs" />
     <Compile Include="App.fs" />
   </ItemGroup>
   <ItemGroup>
     <PackageReference Include="Fable.Browser.Dom" Version="2.14.0" />
+    <PackageReference Include="Fable.Browser.WebSocket" Version="1.3.0" />
     <PackageReference Include="Fable.Core" Version="4.0.0" />
     <PackageReference Include="Fable.Node" Version="1.2.0" />
     <PackageReference Include="Fable.Promise" Version="3.2.0" />
diff --git a/src/Dev.fs b/src/Dev.fs
new file mode 100644
index 0000000..8ab1e89
--- /dev/null
+++ b/src/Dev.fs
@@ -0,0 +1,10 @@
+module Dev
+
+open Browser.Dom
+open Browser.WebSocket
+
+let private init (_) =
+    let ws = WebSocket.Create $"ws://{window.location.host}/websocket"
+    ws.onmessage <- fun (_) -> window.location.reload ()
+
+window.addEventListener ("load", init)
diff --git a/src/Generator.fs b/src/Generator.fs
index 884b3fa..1fb48d6 100644
--- a/src/Generator.fs
+++ b/src/Generator.fs
@@ -581,8 +581,7 @@ let render (opts: RnderOptions) =
 
         let devInjection, devScript =
             match opts.stage with
-            | Development ->
-                Some("/js/live-reload.js"), [ ("js/live-reload.js", $"{opts.dst}{opts.pathRoot}/js/live-reload.js") ]
+            | Development -> Some("/js/dev.js"), [ ("src/Dev.fs.js", $"{opts.dst}{opts.pathRoot}/js/dev.js") ]
             | Production -> None, []
 
         let site: FixedSiteContent =

From 202f88e5ebd2289caab16a3f5216e2f7981e36b3 Mon Sep 17 00:00:00 2001
From: krymtkts <takatoshi.kuriyama@gmail.com>
Date: Sat, 29 Jul 2023 18:20:29 +0900
Subject: [PATCH 2/2] Remove parens.

---
 src/Dev.fs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Dev.fs b/src/Dev.fs
index 8ab1e89..c2b120a 100644
--- a/src/Dev.fs
+++ b/src/Dev.fs
@@ -3,8 +3,8 @@ module Dev
 open Browser.Dom
 open Browser.WebSocket
 
-let private init (_) =
+let private init _ =
     let ws = WebSocket.Create $"ws://{window.location.host}/websocket"
-    ws.onmessage <- fun (_) -> window.location.reload ()
+    ws.onmessage <- fun _ -> window.location.reload ()
 
 window.addEventListener ("load", init)