[v2] Improve translation of URI to file to be loaded

v2/internal/frontend/desktop/common/uri_translate.go

@@ -0,0 +1,20 @@
+package common
+
+import "net/url"
+
+func TranslateUriToFile(uri string, expectedScheme string, expectedHost string) (file string, match bool, err error) {
+	url, err := url.Parse(uri)
+	if err != nil {
+		return "", false, err
+	}
+
+	if url.Scheme != expectedScheme || url.Host != expectedHost {
+		return "", false, nil
+	}
+
+	filePath := url.Path
+	if filePath == "" {
+		filePath = "/"
+	}
+	return filePath, true, nil
+}

v2/internal/frontend/desktop/darwin/frontend.go

@@ -19,12 +19,12 @@ import (
 	"html/template"
 	"log"
 	"strconv"
-	"strings"
 	"unsafe"
 
 	"github.com/wailsapp/wails/v2/internal/binding"
 	"github.com/wailsapp/wails/v2/internal/frontend"
 	"github.com/wailsapp/wails/v2/internal/frontend/assetserver"
+	"github.com/wailsapp/wails/v2/internal/frontend/desktop/common"
 	"github.com/wailsapp/wails/v2/internal/logger"
 	"github.com/wailsapp/wails/v2/pkg/options"
 )
@@ -270,12 +270,18 @@ func (f *Frontend) ExecJS(js string) {
 }
 
 func (f *Frontend) processRequest(r *request) {
-	url := C.GoString(r.url)
+	uri := C.GoString(r.url)
-	url = strings.TrimPrefix(url, "wails://wails")
+
-	if !strings.HasPrefix(url, "/") {
+	// Translate URI to file
+	file, match, err := common.TranslateUriToFile(uri, "wails", "wails")
+	if err != nil {
+		// TODO Handle errors
+		return
+	} else if !match {
 		return
 	}
-	_contents, _mimetype, err := f.assets.Load(url)
+
+	_contents, _mimetype, err := f.assets.Load(file)
 	if err != nil {
 		f.logger.Error(err.Error())
 		//TODO: Handle errors

v2/internal/frontend/desktop/linux/frontend.go

@@ -27,7 +27,6 @@ import (
 	"log"
 	"os"
 	"strconv"
-	"strings"
 	"sync"
 	"text/template"
 	"unsafe"
@@ -35,6 +34,7 @@ import (
 	"github.com/wailsapp/wails/v2/internal/binding"
 	"github.com/wailsapp/wails/v2/internal/frontend"
 	"github.com/wailsapp/wails/v2/internal/frontend/assetserver"
+	"github.com/wailsapp/wails/v2/internal/frontend/desktop/common"
 	"github.com/wailsapp/wails/v2/internal/logger"
 	"github.com/wailsapp/wails/v2/pkg/options"
 )
@@ -328,14 +328,16 @@ func (f *Frontend) processRequest(request unsafe.Pointer) {
 	uri := C.webkit_uri_scheme_request_get_uri(req)
 	goURI := C.GoString(uri)
 
-	// Translate URI
+	file, match, err := common.TranslateUriToFile(uri, "wails", "")
-	goURI = strings.TrimPrefix(goURI, "wails://")
+	if err != nil {
-	if !strings.HasPrefix(goURI, "/") {
+		// TODO Handle errors
+		return
+	} else if !match {
 		return
 	}
 
 	// Load file from asset store
-	content, mimeType, err := f.assets.Load(goURI)
+	content, mimeType, err := f.assets.Load(file)
 	if err != nil {
 		return
 	}

v2/internal/frontend/desktop/windows/frontend.go

@@ -19,6 +19,7 @@ import (
 	"github.com/wailsapp/wails/v2/internal/binding"
 	"github.com/wailsapp/wails/v2/internal/frontend"
 	"github.com/wailsapp/wails/v2/internal/frontend/assetserver"
+	"github.com/wailsapp/wails/v2/internal/frontend/desktop/common"
 	"github.com/wailsapp/wails/v2/internal/logger"
 	"github.com/wailsapp/wails/v2/pkg/options"
 )
@@ -335,14 +336,16 @@ func (f *Frontend) processRequest(req *edge.ICoreWebView2WebResourceRequest, arg
 	//Get the request
 	uri, _ := req.GetUri()
 
-	// Translate URI
+	file, match, err := common.TranslateUriToFile(uri, "file", "wails")
-	uri = strings.TrimPrefix(uri, "file://wails")
+	if err != nil {
-	if !strings.HasPrefix(uri, "/") {
+		// TODO Handle errors
+		return
+	} else if !match {
 		return
 	}
 
 	// Load file from asset store
-	content, mimeType, err := f.assets.Load(uri)
+	content, mimeType, err := f.assets.Load(file)
 	if err != nil {
 		return
 	}