兼容window

douyu · May 5, 2023 · eeb322d · eeb322d
1 parent 660240f
commit eeb322d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/api/apiv1/static/filesystem.go b/api/apiv1/static/filesystem.go
@@ -28,6 +28,8 @@ func (d Dir) Exists(name string) bool {
 	}
 	fullName := filepath.Join(dir, filepath.FromSlash(path.Clean("/"+name)))
 	fullName = strings.TrimPrefix(fullName, strings.TrimPrefix(Prefix, "/"))
+	fullName = strings.TrimPrefix(fullName, string(filepath.Separator))
+	fullName = strings.ReplaceAll(fullName, "\\", "/")
 	_, err := statikFS.Open(fullName)
 	return err == nil
 }
@@ -38,7 +40,10 @@ func (d Dir) Open(name string) (http.File, error) {
 	if dir == "" {
 		dir = "."
 	}
+
 	fullName := filepath.Join(dir, filepath.FromSlash(path.Clean("/"+name)))
 	fullName = strings.TrimPrefix(fullName, strings.TrimPrefix(Prefix, "/"))
+	fullName = strings.TrimPrefix(fullName, string(filepath.Separator))
+	fullName = strings.ReplaceAll(fullName, "\\", "/")
 	return statikFS.Open(fullName)
 }
diff --git a/assets/assets.go b/assets/assets.go
@@ -5,10 +5,12 @@ package assets
 
 import (
 	"net/http"
+	"path/filepath"
 )
 
 var AssetsEmbedded = false
 
 func Assets() (http.FileSystem, error) {
-	return http.Dir("./assets/dist"), nil
+	pathDir := filepath.Join(".", "assets", "dist")
+	return http.Dir(pathDir), nil
 }