xofine/SiteProxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 proxy/rewriter.go

Browse Source
This commit is contained in:
XOF
2025-12-15 18:04:17 +08:00
parent 43506ab6e1
commit 3bc1d0a637
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
proxy/rewriter.go
View File

@@ -161,6 +161,8 @@ func (r *ContentRewriter) rewriteURL(urlStr string) string {
if !u.IsAbs() { if !u.IsAbs() {
resolved := r.baseURL.ResolveReference(u) resolved := r.baseURL.ResolveReference(u)
// 同域相对路径，只保留路径部分
if resolved.Host == r.baseURL.Host {
proxyPath := resolved.Path proxyPath := resolved.Path
if resolved.RawQuery != "" { if resolved.RawQuery != "" {
proxyPath += "?" + resolved.RawQuery proxyPath += "?" + resolved.RawQuery
@@ -170,7 +172,11 @@ func (r *ContentRewriter) rewriteURL(urlStr string) string {
} }
return "/p/" + r.token + proxyPath return "/p/" + r.token + proxyPath
} }
// 跨域相对路径（罕见），保留完整 URL
return "/p/" + r.token + "/" + resolved.String()
}
// 同域绝对 URL，只保留路径
if u.Host == r.baseURL.Host { if u.Host == r.baseURL.Host {
proxyPath := u.Path proxyPath := u.Path
if u.RawQuery != "" { if u.RawQuery != "" {
@@ -182,6 +188,7 @@ func (r *ContentRewriter) rewriteURL(urlStr string) string {
return "/p/" + r.token + proxyPath return "/p/" + r.token + proxyPath
} }
// 跨域绝对 URL，保留完整 URL
return "/p/" + r.token + "/" + u.String() return "/p/" + r.token + "/" + u.String()
} }