Spaces:
Running
Running
File size: 1,300 Bytes
d669ddb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
package web
import (
"adams549659584/go-proxy-bingai/common"
"embed"
"io/fs"
"net/http"
"net/http/httputil"
"path/filepath"
)
//go:embed *
var webFS embed.FS
var WEB_PATH_MAP = make(map[string]bool)
func init() {
var err error
if common.IS_DEBUG_MODE {
err = initWebPathMapByDir()
} else {
err = initWebPathMapByFS()
}
if err != nil {
panic(err)
}
}
func initWebPathMapByDir() error {
err := filepath.WalkDir("web", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if !d.IsDir() {
WEB_PATH_MAP["/"+path] = true
}
return nil
})
return err
}
func initWebPathMapByFS() error {
err := fs.WalkDir(webFS, ".", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if !d.IsDir() {
WEB_PATH_MAP["/web/"+path] = true
}
return nil
})
return err
}
func GetWebFS() http.FileSystem {
return http.FS(webFS)
}
func DebugWebHandler(w http.ResponseWriter, r *http.Request) {
reverseProxy := &httputil.ReverseProxy{
Director: func(req *http.Request) {
req.URL.Scheme = common.DEBUG_PROXY_WEB.Scheme
req.URL.Host = common.DEBUG_PROXY_WEB.Host
req.Host = common.DEBUG_PROXY_WEB.Host
req.Header.Set("Origin", common.DEBUG_PROXY_WEB.String())
},
}
reverseProxy.ServeHTTP(w, r)
}
|