From 3136d71032394cd715e19a7280cef21e12d31aa2 Mon Sep 17 00:00:00 2001
From: William Petit <wpetit@cadoles.com>
Date: Wed, 1 Mar 2023 14:39:36 +0100
Subject: [PATCH] feat(http): handle html5 routing

ref #5
---
 .../src/public/index.html                     | 10 ++--
 pkg/bundle/filesystem.go                      |  2 +
 pkg/http/handler.go                           |  2 +-
 pkg/http/html5_fileserver.go                  | 54 +++++++++++++++++++
 4 files changed, 62 insertions(+), 6 deletions(-)
 create mode 100644 pkg/http/html5_fileserver.go
diff --git a/misc/client-sdk-testsuite/src/public/index.html b/misc/client-sdk-testsuite/src/public/index.html
index 38034f9..256a29f 100644
--- a/misc/client-sdk-testsuite/src/public/index.html
+++ b/misc/client-sdk-testsuite/src/public/index.html
@@ -4,7 +4,7 @@
     <meta charset="utf-8" />
     <title>Client SDK Test suite</title>
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <link rel="stylesheet" href="vendor/mocha.css" />
+    <link rel="stylesheet" href="/vendor/mocha.css" />
     <style>
       body {
         background-color: white;
@@ -13,15 +13,15 @@
   </head>
   <body>
     <div id="mocha"></div>
-    <script src="vendor/chai.js"></script>
-    <script src="vendor/mocha.js"></script>
+    <script src="/vendor/chai.js"></script>
+    <script src="/vendor/mocha.js"></script>
     <script class="mocha-init">
       mocha.setup('bdd');
       mocha.checkLeaks();
     </script>
     <script src="/edge/sdk/client.js"></script>
-    <script src="test/client-sdk.js"></script>
-    <script src="test/auth-module.js"></script>
+    <script src="/test/client-sdk.js"></script>
+    <script src="/test/auth-module.js"></script>
     <script class="mocha-exec">
       mocha.run();
     </script>
diff --git a/pkg/bundle/filesystem.go b/pkg/bundle/filesystem.go
index 776996f..bb3d8d7 100644
--- a/pkg/bundle/filesystem.go
+++ b/pkg/bundle/filesystem.go
@@ -99,3 +99,5 @@ func (f *File) Readdir(count int) ([]os.FileInfo, error) {
 func (f *File) Stat() (os.FileInfo, error) {
 	return f.fi, nil
 }
+
+var _ http.FileSystem = &FileSystem{}
diff --git a/pkg/http/handler.go b/pkg/http/handler.go
index 0051d1f..7570d44 100644
--- a/pkg/http/handler.go
+++ b/pkg/http/handler.go
@@ -59,7 +59,7 @@ func (h *Handler) Load(bdle bundle.Bundle) error {
 	}
 
 	fs := bundle.NewFileSystem("public", bdle)
-	public := http.FileServer(fs)
+	public := HTML5Fileserver(fs)
 	sockjs := sockjs.NewHandler(sockJSPathPrefix, h.sockjsOpts, h.handleSockJSSession)
 
 	if h.server != nil {
diff --git a/pkg/http/html5_fileserver.go b/pkg/http/html5_fileserver.go
new file mode 100644
index 0000000..136cc98
--- /dev/null
+++ b/pkg/http/html5_fileserver.go
@@ -0,0 +1,54 @@
+package http
+
+import (
+	"net/http"
+	"os"
+	"path"
+	"strings"
+
+	"github.com/pkg/errors"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+func HTML5Fileserver(fs http.FileSystem) http.Handler {
+	handler := http.FileServer(fs)
+
+	fn := func(w http.ResponseWriter, r *http.Request) {
+		urlPath := r.URL.Path
+		if !strings.HasPrefix(urlPath, "/") {
+			urlPath = "/" + urlPath
+			r.URL.Path = urlPath
+		}
+		urlPath = path.Clean(urlPath)
+
+		file, err := fs.Open(urlPath)
+		if err != nil {
+			if errors.Is(err, os.ErrNotExist) {
+				r.URL.Path = "/"
+
+				handler.ServeHTTP(w, r)
+
+				return
+			}
+
+			logger.Error(r.Context(), "could not open bundle file", logger.E(err))
+
+			http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+			return
+		}
+
+		defer func() {
+			if err := file.Close(); err != nil {
+				logger.Error(r.Context(), "could not close file", logger.E(err))
+
+				http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+
+				return
+			}
+		}()
+
+		handler.ServeHTTP(w, r)
+	}
+
+	return http.HandlerFunc(fn)
+}
